创建
This commit is contained in:
linuxrc@163.com
23
bin/conf/project_list.conf
Normal file
23
bin/conf/project_list.conf
Normal file
@@ -0,0 +1,23 @@
|
|||||||
|
#project_requirement_name value_type
|
||||||
|
#current requirement definitions:
|
||||||
|
# terminal_tag,
|
||||||
|
# ipv4_frag_list,
|
||||||
|
# ipv6_frag_list,
|
||||||
|
# stream_id,
|
||||||
|
# value type definitions
|
||||||
|
# char,
|
||||||
|
# short,
|
||||||
|
# int,
|
||||||
|
# long,
|
||||||
|
# struct
|
||||||
|
|
||||||
|
#terminal_tag struct
|
||||||
|
#ipv4_frag_list struct
|
||||||
|
#ipv6_frag_list struct
|
||||||
|
#stream_id struct
|
||||||
|
#test_project struct
|
||||||
|
#tcp_flow_stat struct
|
||||||
|
#udp_flow_stat struct
|
||||||
|
|
||||||
|
JT_IP_PKT struct
|
||||||
|
STREAMID long
|
||||||
33
bin/conf/qq_file_trans.conf
Normal file
33
bin/conf/qq_file_trans.conf
Normal file
@@ -0,0 +1,33 @@
|
|||||||
|
[MODULE]
|
||||||
|
#日志
|
||||||
|
RUN_LOG_DIR=./log/qq_file_trans/
|
||||||
|
RUN_LOG_LEVEL=40
|
||||||
|
|
||||||
|
#存本地文件开关
|
||||||
|
DEBUG_SWITCH=0
|
||||||
|
|
||||||
|
#统计信息
|
||||||
|
STAT_FIELD_CYCLE=300
|
||||||
|
STAT_FIELD_TRIG=0
|
||||||
|
|
||||||
|
|
||||||
|
[FEATURE]
|
||||||
|
#FEATURE_QQ_NUMBER_TCP=0:27,00,00,00,00,14;0:04,37,21
|
||||||
|
#FEATURE_QQ_TCPFILENAME=0:0c,00,00,00,00,2b;31:0c,00,00,00,00,2b
|
||||||
|
#FEATURE_QQ_TCPFILE=0:04,36,43,00&39:65,65,00,00
|
||||||
|
|
||||||
|
FEATURE_QQ_NUMBER_TCP=0:04&13:00,01,00,00,00,00,00
|
||||||
|
FEATURE_QQ_TCPFILENAME=0:04&13:00,01,00,00,00,00,00&31:0c,00,00,00,00,2b
|
||||||
|
FEATURE_QQ_TCPFILE=0:04&13:00,01,00,00,00,00,00&39:65,65,00,00
|
||||||
|
FEATURE_QQ_TCPFILE_END=0:04&13:00,01,00,00,00,00,00&39:56,56,00,00
|
||||||
|
|
||||||
|
#文件应该也有特征13:00,01,00,00,00,00,00
|
||||||
|
FEATURE_QQ_NUMBER_UDP=0:03,00,01,00,00,00,00,00;13:00,01,00,00,00,00,00
|
||||||
|
FEATURE_QQ_UDPFILENAME=3:00,2b,52,00,00;34:00,2b,52,00,00
|
||||||
|
FEATURE_QQ_UDPFILE_START=3:00,2d,63,00;34:00,2d,63,00
|
||||||
|
FEATURE_QQ_UDPFILE_END=3:00,2b,56,00,00;34:00,2b,56,00,00
|
||||||
|
|
||||||
|
#其他QQ协议特征，上面的也会作为协议特征
|
||||||
|
FEATURE_QQTCP_PROTOCOL=
|
||||||
|
FEATURE_QQUDP_PROTOCOL=
|
||||||
|
|
||||||
24
bin/plug/business/conflist_business.inf
Normal file
24
bin/plug/business/conflist_business.inf
Normal file
@@ -0,0 +1,24 @@
|
|||||||
|
##./plug/business/test_app/test_app.inf
|
||||||
|
##./plug/business/test_http/test_http.inf
|
||||||
|
##./plug/business/ipsec_biz_sample/ipsec_biz_sample.inf
|
||||||
|
##./plug/business/l2tp_biz_sample/l2tp_biz_sample.inf
|
||||||
|
##./plug/business/pptp_biz_sample/pptp_biz_sample.inf
|
||||||
|
##./plug/business/terminal_tag/terminal_tag.inf
|
||||||
|
##./plug/business/g_device_plug/g_device_plug.inf
|
||||||
|
#
|
||||||
|
#./plug/business/ycl_ftp/ycl_ftp.inf
|
||||||
|
#./plug/business/ycl_mail/ycl_mail.inf
|
||||||
|
#./plug/business/ycl_http_service/ycl_http_service.inf
|
||||||
|
##./plug/business/ycl_http_netdisk/ycl_http_netdisk.inf
|
||||||
|
#./plug/business/ycl_http_webmail/ycl_http_webmail.inf
|
||||||
|
##./plug/business/ycl_qq_biz/ycl_qq_biz.inf
|
||||||
|
##./plug/business/ycl_qq/ycl_qq_biz.inf
|
||||||
|
#
|
||||||
|
#./plug/business/http_jt_audit/http_jt_audit.inf
|
||||||
|
#./plug/business/dns_audit/dns_jt_audit.inf
|
||||||
|
#./plug/business/mail_audit/mail_jt_audit.inf
|
||||||
|
#./plug/business/ssl_audit/ssl_audit.inf
|
||||||
|
#./plug/business/ftp_audit/ftp_audit.inf
|
||||||
|
#./plug/business/ip_jt/ip_jt.inf
|
||||||
|
##./plug/business/MESA_capfile/MESA_capfile.inf
|
||||||
|
./plug/business/qq_file_trans/qq_file_trans.inf
|
||||||
13
bin/plug/business/qq_file_trans/qq_file_trans.inf
Normal file
13
bin/plug/business/qq_file_trans/qq_file_trans.inf
Normal file
@@ -0,0 +1,13 @@
|
|||||||
|
[PLUGINFO]
|
||||||
|
PLUGNAME=QQ_FILE_TRANS
|
||||||
|
SO_PATH=./plug/business/qq_file_trans/qq_file_trans.so
|
||||||
|
INIT_FUNC=QQ_PROT_INIT
|
||||||
|
DESTROY_FUNC=QQ_PROT_DESTROY
|
||||||
|
|
||||||
|
[UDP]
|
||||||
|
FUNC_FLAG=ALL
|
||||||
|
FUNC_NAME=QQ_PROT_UDP_ENTRY
|
||||||
|
|
||||||
|
[TCP]
|
||||||
|
FUNC_FLAG=ALL
|
||||||
|
FUNC_NAME=QQ_PROT_TCP_ENTRY
|
||||||
BIN
bin/plug/business/qq_file_trans/qq_file_trans.so
Normal file
BIN
bin/plug/business/qq_file_trans/qq_file_trans.so
Normal file
Binary file not shown.
21
src/Makefile
Normal file
21
src/Makefile
Normal file
@@ -0,0 +1,21 @@
|
|||||||
|
CC=g++
|
||||||
|
CCC=g++
|
||||||
|
|
||||||
|
INC_PATH = -I./include
|
||||||
|
CFLAGS = -Wall -g -fPIC $(INC_PATH)
|
||||||
|
|
||||||
|
LIBS = -lMESA_prof_load -lMESA_handle_logger -ldocumentanalyze -lMESA_field_stat -lpthread
|
||||||
|
|
||||||
|
OBJS = qq_file_global.o qq_file_entry.o
|
||||||
|
|
||||||
|
TARGET = qq_file_trans.so
|
||||||
|
|
||||||
|
$(TARGET):$(OBJS)
|
||||||
|
$(CCC) -fPIC -shared $^ -o $@ $(LIBS)
|
||||||
|
cp $@ /home/zhangcw/sapp/plug/business/qq_file_trans/
|
||||||
|
|
||||||
|
.c.o:
|
||||||
|
$(CC) -c $(CFLAGS) $<
|
||||||
|
|
||||||
|
clean:
|
||||||
|
rm -f $(TARGET) $(OBJS)
|
||||||
86
src/include/connector.h
Normal file
86
src/include/connector.h
Normal file
@@ -0,0 +1,86 @@
|
|||||||
|
#ifndef CONNECTOR_H_
|
||||||
|
#define CONNECTOR_H_
|
||||||
|
|
||||||
|
extern int CONNECTOR_VERSION_1_20170704;
|
||||||
|
|
||||||
|
typedef unsigned long ulong64;
|
||||||
|
typedef unsigned int uint32;
|
||||||
|
|
||||||
|
typedef enum
|
||||||
|
{
|
||||||
|
APPD_SENDBACK=0, //ǰ<><C7B0><EFBFBD><EFBFBD>Ӧ<EFBFBD>ò<EFBFBD><C3B2><EFBFBD><EFBFBD>ݻش<DDBB><D8B4><EFBFBD><EFBFBD><EFBFBD>
|
||||||
|
APPD_FILE=1, //<2F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ļ<EFBFBD><C4BC><EFBFBD><EFBFBD><EFBFBD>ģ<EFBFBD><C4A3><EFBFBD><EFBFBD>Ӧ<EFBFBD>ò<EFBFBD><C3B2><EFBFBD><EFBFBD>ݷַ<DDB7>
|
||||||
|
}APPD_HEADER_TYPE;
|
||||||
|
|
||||||
|
typedef enum
|
||||||
|
{
|
||||||
|
DEST_APPD=0, //<2F><><EFBFBD>ݷַ<DDB7><D6B7><EFBFBD><EFBFBD><EFBFBD>appd
|
||||||
|
DEST_WM=1, //webmail<69><6C><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>WM
|
||||||
|
DEST_MLFA=2, //<2F>ļ<EFBFBD><C4BC><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>mlfa
|
||||||
|
}APPD_DESTINATION;
|
||||||
|
|
||||||
|
typedef enum
|
||||||
|
{
|
||||||
|
COMPLETE=0,
|
||||||
|
FRAG=1,
|
||||||
|
FRAG_END=2,
|
||||||
|
}APPD_MSG_TYPE;
|
||||||
|
|
||||||
|
typedef struct _appd_basic_t
|
||||||
|
{
|
||||||
|
APPD_HEADER_TYPE appd_type; //Ӧ<>ò<EFBFBD><C3B2><EFBFBD><EFBFBD>ݻ<EFBFBD><DDBB><EFBFBD><EFBFBD>ļ<EFBFBD><C4BC><EFBFBD><EFBFBD><EFBFBD>
|
||||||
|
const char* service_name; //<2F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>,'\0'<27><><EFBFBD><EFBFBD>
|
||||||
|
APPD_DESTINATION destination; //<2F><><EFBFBD>ݽ<EFBFBD><DDBD>ճ<EFBFBD><D5B3><EFBFBD>
|
||||||
|
ulong64* streamid; //<2F><>̬<EFBFBD><CCAC><EFBFBD><EFBFBD>ָ<EFBFBD>룬<EFBFBD><EBA3AC>id
|
||||||
|
uint32 streamid_num;
|
||||||
|
struct ipaddr* addr; //<2F><>̬<EFBFBD><CCAC><EFBFBD><EFBFBD>ָ<EFBFBD>룬 <20><><EFBFBD><EFBFBD>Ԫ<EFBFBD><D4AA>
|
||||||
|
uint32 addr_num;
|
||||||
|
}appd_basic_t;
|
||||||
|
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
extern "C" {
|
||||||
|
#endif
|
||||||
|
|
||||||
|
/*<2A><>ʼ<EFBFBD><CABC><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>connector<6F><72><EFBFBD><EFBFBD><EFBFBD><EFBFBD> ÿһ<C3BF><D2BB><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>һ<EFBFBD><D2BB>*/
|
||||||
|
//void* connector_start(appd_basic_t* baisc_info);
|
||||||
|
void* connector_start(appd_basic_t* baisc_info, const struct streaminfo *a_tcp);
|
||||||
|
/* msg_type:<3A><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> COMPLETE FRAG FRAG_END*/
|
||||||
|
void connector(void* handler, APPD_MSG_TYPE msg_type, void* simple_opt_handle, void* nested_opt_handle);
|
||||||
|
/*ÿһ<C3BF><D2BB><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>*/
|
||||||
|
void connector_finish(void* handler);
|
||||||
|
|
||||||
|
|
||||||
|
/*<2A><><EFBFBD><EFBFBD>ѡ<EFBFBD><D1A1><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>*/
|
||||||
|
/*1. <20><><EFBFBD>ü<EFBFBD><C3BC><EFBFBD>ѡ<EFBFBD><D1A1><EFBFBD>ĸ<EFBFBD><C4B8><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>simple_opt_handle*/
|
||||||
|
void* connector_create_simple_opt(uint32 max_size);
|
||||||
|
/*2. <20><><EFBFBD><EFBFBD>ѡ<EFBFBD><D1A1>*/
|
||||||
|
int connector_append_simple_opt(void* simple_opt_handle, const char* opt_name, char* opt_value, uint32 opt_len);
|
||||||
|
/*3. <20>ͷ<EFBFBD>*/
|
||||||
|
void connector_free_simple_opt(void** simple_opt_handle);
|
||||||
|
|
||||||
|
|
||||||
|
/*Ƕ<><C7B6>ѡ<EFBFBD><D1A1><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>*/
|
||||||
|
/*1. <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ƕ<EFBFBD><C7B6>ѡ<EFBFBD><D1A1><EFBFBD><EFBFBD><EFBFBD>ܸ<EFBFBD><DCB8><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>nested_opt_handle*/
|
||||||
|
void* connector_create_nested_opt(uint32 max_size);
|
||||||
|
/*2. <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ij<EFBFBD><C4B3>Ƕ<EFBFBD><C7B6>ѡ<EFBFBD><D1A1><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Լ<EFBFBD><D4BC>ڲ<EFBFBD>ѡ<EFBFBD><D1A1><EFBFBD>ĸ<EFBFBD><C4B8><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>nested_unit_handle*/
|
||||||
|
void* connector_create_nested_opt_unit(const char* nested_unit_name, uint32 max_size);
|
||||||
|
/*3.<2E><><EFBFBD>䵥<EFBFBD><E4B5A5>Ƕ<EFBFBD><C7B6>ѡ<EFBFBD><D1A1><EFBFBD><EFBFBD><EFBFBD>ڲ<EFBFBD>ѡ<EFBFBD><D1A1>*/
|
||||||
|
/*opt_unit_handle <20><>connector_create_nested_opt_unit<69>ķ<EFBFBD><C4B7><EFBFBD>ֵ*/
|
||||||
|
/*<2A><><EFBFBD><EFBFBD>ֵ 0 OK -1 ERROR*/
|
||||||
|
int connector_append_nested_opt_unit(void* nested_unit_handle, const char* subopt_name, char* subopt_value, uint32 subopt_len);
|
||||||
|
/*4. <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ƕ<EFBFBD><C7B6>ѡ<EFBFBD><D1A1><EFBFBD><D7B7><EFBFBD><EFBFBD>Ƕ<EFBFBD><C7B6>ѡ<EFBFBD><D1A1><EFBFBD><EFBFBD><EFBFBD><EFBFBD>nested_opt_handle<6C><65>*/
|
||||||
|
/*<2A><><EFBFBD><EFBFBD>ֵ 0 OK -1 ERROR*/
|
||||||
|
int connector_append_nested_opt(void* nested_opt_handle, void* nested_unit_handle);
|
||||||
|
/*5. <20>ͷŵ<CDB7><C5B5><EFBFBD>Ƕ<EFBFBD><C7B6>ѡ<EFBFBD><D1A1>*/
|
||||||
|
void connector_free_nested_opt_unit(void** nested_unit_handle);
|
||||||
|
/*6. <20>ͷ<EFBFBD>*/
|
||||||
|
void connector_free_nested_opt(void** nested_opt_handle);
|
||||||
|
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
#endif
|
||||||
|
|
||||||
|
|
||||||
868
src/qq_file_entry.c
Normal file
868
src/qq_file_entry.c
Normal file
@@ -0,0 +1,868 @@
|
|||||||
|
#include <stdio.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
#include <sys/stat.h>
|
||||||
|
#include <sys/types.h>
|
||||||
|
#include <errno.h>
|
||||||
|
#include <assert.h>
|
||||||
|
|
||||||
|
#include <MESA/stream.h>
|
||||||
|
#include <MESA/MESA_prof_load.h>
|
||||||
|
#include <MESA/MESA_handle_logger.h>
|
||||||
|
#include <MESA/field_stat.h>
|
||||||
|
|
||||||
|
#include "qq_file_entry.h"
|
||||||
|
#include "qq_file_global.h"
|
||||||
|
#include "connector.h"
|
||||||
|
|
||||||
|
qq_global_info_t g_qq_global_info;
|
||||||
|
|
||||||
|
int data_hit_feature(const char *data, unsigned int datalen, unsigned int min_len, FEATURE_INDEX_T index)
|
||||||
|
{
|
||||||
|
unsigned int i, j;
|
||||||
|
|
||||||
|
if(datalen > min_len)
|
||||||
|
{
|
||||||
|
for(i=0; i<g_qq_global_info.fnum[index]; i++)
|
||||||
|
{
|
||||||
|
for(j=0; j<g_qq_global_info.rulenum[index][i]; j++)
|
||||||
|
{
|
||||||
|
//<2F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ƥ<EFBFBD><C6A5>
|
||||||
|
if(datalen <= (g_qq_global_info.ftoff[index][i][j] + g_qq_global_info.ftlen[index][i][j]) || \
|
||||||
|
memcmp(data+g_qq_global_info.ftoff[index][i][j], g_qq_global_info.feature[index][i][j], g_qq_global_info.ftlen[index][i][j]))
|
||||||
|
{
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
//һ<><D2BB><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ڵ<EFBFBD><DAB5><EFBFBD><EFBFBD>й<EFBFBD><D0B9><EFBFBD><EFBFBD><EFBFBD>ƥ<EFBFBD>䣬<EFBFBD><E4A3AC><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
||||||
|
if(j == g_qq_global_info.rulenum[index][i])
|
||||||
|
{
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
QQ_PACKET_TYPE_t is_qq_protocol_tcp(const char *data, unsigned int datalen)
|
||||||
|
{
|
||||||
|
if(data_hit_feature(data, datalen, 0, FINDEX_TCPFILE))
|
||||||
|
return QQ_PACKET_TCPFILE;
|
||||||
|
if(data_hit_feature(data, datalen, PACKET_MIN_LEN_TCPNAME, FINDEX_TCPNAME))
|
||||||
|
return QQ_PACKET_TCPFILENAME;
|
||||||
|
//<2F>ŵ<EFBFBD>FINDEX_TCPFILE<4C><45>FINDEX_TCPNAME֮<45><D6AE><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ϊ<EFBFBD><CEAA><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ҳ<EFBFBD><D2B2><EFBFBD><EFBFBD><EFBFBD><EFBFBD>FINDEX_TCPNUM<55><4D><EFBFBD>Ͷ<EFBFBD><CDB6><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ݣ<EFBFBD>UDPһ<50><D2BB><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
||||||
|
if(data_hit_feature(data, datalen, PACKET_MIN_LEN_QQNUM, FINDEX_TCPNUM))
|
||||||
|
return QQ_PACKET_QQNUM;
|
||||||
|
if(data_hit_feature(data, datalen, 0, FINDEX_TCPPROT))
|
||||||
|
return QQ_PACKET_OTHER;
|
||||||
|
|
||||||
|
return QQ_PACKET_NOT;
|
||||||
|
}
|
||||||
|
|
||||||
|
QQ_PACKET_TYPE_t is_qq_protocol_udp(const char *data, unsigned int datalen)
|
||||||
|
{
|
||||||
|
if(data_hit_feature(data, datalen, 0, FINDEX_UDPFILE))
|
||||||
|
return QQ_PACKET_UDPFILE;
|
||||||
|
if(data_hit_feature(data, datalen, PACKET_MIN_LEN_UDPNAME, FINDEX_UDPNAME))
|
||||||
|
return QQ_PACKET_UDPFILENAME;
|
||||||
|
if(data_hit_feature(data, datalen, 0, FINDEX_UDPFILE_END))
|
||||||
|
return QQ_PACKET_UDPFILE_END;
|
||||||
|
if(data_hit_feature(data, datalen, PACKET_MIN_LEN_QQNUM, FINDEX_UDPNUM))
|
||||||
|
return QQ_PACKET_QQNUM;
|
||||||
|
if(data_hit_feature(data, datalen, 0, FINDEX_UDPPROT))
|
||||||
|
return QQ_PACKET_OTHER;
|
||||||
|
|
||||||
|
return QQ_PACKET_NOT;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
QQ_STATE_t assign_qq_state(QQ_PACKET_TYPE_t type)
|
||||||
|
{
|
||||||
|
switch(type)
|
||||||
|
{
|
||||||
|
case QQ_PACKET_QQNUM: return QQ_STATE_QQNUM;
|
||||||
|
|
||||||
|
case QQ_PACKET_TCPFILE:
|
||||||
|
case QQ_PACKET_UDPFILE: return QQ_STATE_FCONT_START;
|
||||||
|
|
||||||
|
case QQ_PACKET_UDPFILE_END: return QQ_STATE_FCONT_END;
|
||||||
|
|
||||||
|
case QQ_PACKET_TCPFILENAME:
|
||||||
|
case QQ_PACKET_UDPFILENAME: return QQ_STATE_FNAME;
|
||||||
|
|
||||||
|
case QQ_PACKET_NOT:
|
||||||
|
case QQ_PACKET_OTHER: return QQ_STATE_INIT;
|
||||||
|
default:
|
||||||
|
assert(0);
|
||||||
|
return QQ_STATE_INIT;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static int init_qq_pme_info(qq_proto_pme_t **pme, int thread_id, char tcp_flag)
|
||||||
|
{
|
||||||
|
qq_proto_pme_t *qq_pme=NULL;
|
||||||
|
|
||||||
|
qq_pme= (qq_proto_pme_t *)dictator_malloc(thread_id, sizeof(qq_proto_pme_t));
|
||||||
|
memset(qq_pme, 0, sizeof(qq_proto_pme_t));
|
||||||
|
qq_pme->thread_id = thread_id;
|
||||||
|
qq_pme->qq_state = QQ_STATE_INIT;
|
||||||
|
qq_pme->pending_flag = 1;
|
||||||
|
qq_pme->fileinfo.block_seq = -1; //UDP<44><50>ʼ<EFBFBD><CABC><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>-1<><31><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ʶ<EFBFBD><CAB6><EFBFBD>ش<EFBFBD><D8B4><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
||||||
|
qq_pme->tcp_flag = tcp_flag;
|
||||||
|
|
||||||
|
*pme = qq_pme;
|
||||||
|
|
||||||
|
if(g_qq_global_info.stat_trig)
|
||||||
|
{
|
||||||
|
stat_field_operation(g_qq_global_info.stat_handle, g_qq_global_info.stat_id[STAT_ID_SESSION_NUM], FS_OP_TYPE_ADD, 1);
|
||||||
|
stat_field_operation(g_qq_global_info.stat_handle, g_qq_global_info.stat_id[STAT_ID_NEW_SESSION], FS_OP_TYPE_ADD, 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void clear_qq_pme_info(qq_proto_pme_t *qq_pme, struct streaminfo *a_udp, const void *raw_pkt)
|
||||||
|
{
|
||||||
|
if(qq_pme->connector_hdl != NULL)
|
||||||
|
{
|
||||||
|
connector_finish(qq_pme->connector_hdl);
|
||||||
|
}
|
||||||
|
if(qq_pme->fileinfo.buf)
|
||||||
|
{
|
||||||
|
dictator_free(qq_pme->thread_id, qq_pme->fileinfo.buf);
|
||||||
|
}
|
||||||
|
|
||||||
|
dictator_free(qq_pme->thread_id, qq_pme);
|
||||||
|
|
||||||
|
if(g_qq_global_info.stat_trig)
|
||||||
|
{
|
||||||
|
stat_field_operation(g_qq_global_info.stat_handle, g_qq_global_info.stat_id[STAT_ID_SESSION_NUM], FS_OP_TYPE_ADD, -1);
|
||||||
|
stat_field_operation(g_qq_global_info.stat_handle, g_qq_global_info.stat_id[STAT_ID_DEL_SESSION], FS_OP_TYPE_ADD, 1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
QQ_STATE_t analyse_udp_pdata_state(QQ_STATE_t state, const char *data, int datalen)
|
||||||
|
{
|
||||||
|
if(state != QQ_STATE_INIT)
|
||||||
|
return state;
|
||||||
|
|
||||||
|
return assign_qq_state(is_qq_protocol_udp(data, datalen));
|
||||||
|
}
|
||||||
|
|
||||||
|
QQ_STATE_t analyse_tcp_pdata_state(QQ_STATE_t state, const char *data, int datalen)
|
||||||
|
{
|
||||||
|
if(state != QQ_STATE_INIT)
|
||||||
|
return state;
|
||||||
|
|
||||||
|
return assign_qq_state(is_qq_protocol_tcp(data, datalen));
|
||||||
|
}
|
||||||
|
|
||||||
|
int process_udp_data_state(struct streaminfo *a_udp, qq_proto_pme_t *qq_pme, int thread_seq,const void *raw_pkt)
|
||||||
|
{
|
||||||
|
char *data = (char *)a_udp->pudpdetail->pdata;
|
||||||
|
int datalen = a_udp->pudpdetail->datalen;
|
||||||
|
char filename[512];
|
||||||
|
int tmp, i, j;
|
||||||
|
|
||||||
|
switch(qq_pme->qq_state)
|
||||||
|
{
|
||||||
|
case QQ_STATE_QQNUM:
|
||||||
|
qq_pme->qq_state = QQ_STATE_INIT;
|
||||||
|
if(qq_pme->qqnum[a_udp->curdir-1] || qq_pme->qqnum_from)
|
||||||
|
break;
|
||||||
|
|
||||||
|
if(*data == 0x03 && datalen==27)
|
||||||
|
{
|
||||||
|
qq_pme->qqnum[a_udp->curdir-1] = ntohl(*((unsigned int *)&data[23]));
|
||||||
|
}
|
||||||
|
else if(*data == 0x04) //TODO
|
||||||
|
{
|
||||||
|
qq_pme->qqnum_from = ntohl(*((unsigned int *)&data[9]));
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
|
||||||
|
case QQ_STATE_FNAME:
|
||||||
|
qq_pme->qq_state = QQ_STATE_INIT;
|
||||||
|
qq_pme->file_dir = a_udp->curdir - 1;
|
||||||
|
|
||||||
|
if(*data == 0x05 && datalen < 800) //800---<2D><><EFBFBD><EFBFBD>д<EFBFBD><D0B4>
|
||||||
|
{
|
||||||
|
qq_pme->total_file_len = ntohl(*((unsigned int *)&data[11]));
|
||||||
|
for (i = 23, j = 0; i < datalen - 1; i+=2, j+=2)
|
||||||
|
{
|
||||||
|
if ((data[i] == 0x00) && (data[i + 1] == 0x00))
|
||||||
|
break;
|
||||||
|
filename[j] = data[i];
|
||||||
|
filename[j+1] = data[i+1];
|
||||||
|
}
|
||||||
|
tmp = j;
|
||||||
|
}
|
||||||
|
else if(*data == 0x04 && datalen < 800)
|
||||||
|
{
|
||||||
|
if(qq_pme->qqnum_from == 0)
|
||||||
|
{
|
||||||
|
qq_pme->qqnum_from = ntohl(*((unsigned int *)&data[9]));
|
||||||
|
}
|
||||||
|
qq_pme->total_file_len = ntohl(*((unsigned int *)&data[42]));
|
||||||
|
for (i = 54, j = 0; i < datalen - 1; i+=2, j+=2)
|
||||||
|
{
|
||||||
|
if ((data[i] == 0x00) && (data[i + 1] == 0x00))
|
||||||
|
break;
|
||||||
|
filename[j] = data[i];
|
||||||
|
filename[j+1] = data[i+1];
|
||||||
|
}
|
||||||
|
tmp = j;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
return APP_STATE_GIVEME;
|
||||||
|
}
|
||||||
|
return protocol_process(qq_pme, QQ_FILENAME, filename, tmp, a_udp, raw_pkt);
|
||||||
|
|
||||||
|
case QQ_STATE_FCONT_START:
|
||||||
|
qq_pme->qq_state = QQ_STATE_INIT; //ÿ<><C3BF><EFBFBD><EFBFBD>ǰ<EFBFBD>涼<EFBFBD><E6B6BC><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
||||||
|
if(*data == 0x05 && datalen>18)
|
||||||
|
{
|
||||||
|
tmp = ntohl(*((int *)&data[10])); //TODO<44><4F>֪<EFBFBD><D6AA>int<6E>ͻ<EFBFBD><CDBB><EFBFBD>short
|
||||||
|
if(qq_pme->fileinfo.block_seq >= tmp)
|
||||||
|
return APP_STATE_GIVEME;
|
||||||
|
|
||||||
|
qq_pme->fileinfo.block_seq = tmp;
|
||||||
|
qq_pme->compress_flag = data[17];
|
||||||
|
return protocol_process(qq_pme, QQ_FILECONT, data+18, datalen-18, a_udp, raw_pkt);
|
||||||
|
}
|
||||||
|
else if(*data==0x04 && datalen>50)
|
||||||
|
{
|
||||||
|
tmp = ntohl(*((int *)&data[41])); //TODO<44><4F>֪<EFBFBD><D6AA>int<6E>ͻ<EFBFBD><CDBB><EFBFBD>short
|
||||||
|
if(qq_pme->fileinfo.block_seq >= tmp)
|
||||||
|
return APP_STATE_GIVEME;
|
||||||
|
|
||||||
|
qq_pme->fileinfo.block_seq = tmp;
|
||||||
|
qq_pme->compress_flag = data[48];
|
||||||
|
return protocol_process(qq_pme, QQ_FILECONT, data+49, datalen-50, a_udp, raw_pkt);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
|
||||||
|
case QQ_STATE_FCONT_END:
|
||||||
|
qq_pme->qq_state = QQ_STATE_INIT;
|
||||||
|
if(*data == 0x05 && datalen>18)
|
||||||
|
{
|
||||||
|
qq_pme->compress_flag = data[17];
|
||||||
|
return protocol_process(qq_pme, QQ_FILECONT_END, data+18, datalen-18, a_udp, raw_pkt);
|
||||||
|
}
|
||||||
|
else if(*data == 0x04 && datalen>50)
|
||||||
|
{
|
||||||
|
qq_pme->compress_flag = data[48];
|
||||||
|
return protocol_process(qq_pme, QQ_FILECONT_END, data+49, datalen-50, a_udp, raw_pkt);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
|
||||||
|
case QQ_STATE_FCONT:
|
||||||
|
case QQ_STATE_INIT:
|
||||||
|
default:
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
return APP_STATE_GIVEME;
|
||||||
|
}
|
||||||
|
|
||||||
|
extern "C" char QQ_PROT_UDP_ENTRY(struct streaminfo *a_udp, void **pme, int thread_seq,const void *raw_pkt)
|
||||||
|
{
|
||||||
|
QQ_PACKET_TYPE_t type;
|
||||||
|
qq_proto_pme_t *qq_pme=(qq_proto_pme_t*)(*pme);
|
||||||
|
int rec = APP_STATE_GIVEME;
|
||||||
|
|
||||||
|
switch(a_udp->opstate)
|
||||||
|
{
|
||||||
|
case OP_STATE_PENDING:
|
||||||
|
type = is_qq_protocol_udp((char *)a_udp->pudpdetail->pdata, a_udp->pudpdetail->datalen);
|
||||||
|
if(type)
|
||||||
|
{
|
||||||
|
if(init_qq_pme_info(&qq_pme, a_udp->threadnum, 0))
|
||||||
|
return APP_STATE_DROPME;
|
||||||
|
qq_pme->qq_state = assign_qq_state(type);
|
||||||
|
*pme = qq_pme;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
return APP_STATE_DROPME;
|
||||||
|
}
|
||||||
|
//no break here
|
||||||
|
|
||||||
|
case OP_STATE_DATA:
|
||||||
|
qq_pme->qq_state = analyse_udp_pdata_state(qq_pme->qq_state, (char *)a_udp->pudpdetail->pdata, a_udp->pudpdetail->datalen);
|
||||||
|
|
||||||
|
rec = process_udp_data_state(a_udp, qq_pme, thread_seq, raw_pkt);
|
||||||
|
break;
|
||||||
|
|
||||||
|
case OP_STATE_CLOSE:
|
||||||
|
clear_qq_pme_info(qq_pme, a_udp, raw_pkt);
|
||||||
|
break;
|
||||||
|
|
||||||
|
default:
|
||||||
|
return APP_STATE_DROPME;
|
||||||
|
}
|
||||||
|
|
||||||
|
if(rec & APP_STATE_DROPME)
|
||||||
|
{
|
||||||
|
clear_qq_pme_info(qq_pme, a_udp, raw_pkt);
|
||||||
|
}
|
||||||
|
|
||||||
|
return rec;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
int process_tcp_data_state(struct streaminfo *a_tcp, qq_proto_pme_t *qq_pme, int thread_seq,const void *raw_pkt)
|
||||||
|
{
|
||||||
|
char *data = (char *)a_tcp->pudpdetail->pdata;
|
||||||
|
int datalen = a_tcp->pudpdetail->datalen;
|
||||||
|
char filename[512];
|
||||||
|
int cur_len=0, i, j, rec = APP_STATE_GIVEME, need_len;
|
||||||
|
|
||||||
|
switch(qq_pme->qq_state)
|
||||||
|
{
|
||||||
|
case QQ_STATE_QQNUM:
|
||||||
|
qq_pme->qq_state = QQ_STATE_INIT;
|
||||||
|
if(qq_pme->qqnum_from)
|
||||||
|
break;
|
||||||
|
|
||||||
|
if(*data == 0x04) //TODO
|
||||||
|
{
|
||||||
|
qq_pme->qqnum_from = ntohl(*((unsigned int *)&data[9]));
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
|
||||||
|
case QQ_STATE_FNAME:
|
||||||
|
qq_pme->qq_state = QQ_STATE_INIT;
|
||||||
|
qq_pme->file_dir = a_tcp->curdir - 1;
|
||||||
|
|
||||||
|
//ȡ<><C8A1><EFBFBD>ļ<EFBFBD><C4BC><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ļ<EFBFBD><C4BC><EFBFBD><EFBFBD><EFBFBD>UNICODE<44><45><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
||||||
|
if(*data == TCPFILE_START_CHAR0c && datalen < 800) //800---<2D><><EFBFBD><EFBFBD>д<EFBFBD><D0B4>
|
||||||
|
{
|
||||||
|
qq_pme->total_file_len = ntohl(*((unsigned int *)&data[27]));
|
||||||
|
for (i = 39, j = 0; i < datalen - 1; i+=2, j+=2)
|
||||||
|
{
|
||||||
|
if ((data[i] == 0x00) && (data[i + 1] == 0x00))
|
||||||
|
break;
|
||||||
|
filename[j] = data[i];
|
||||||
|
filename[j+1] = data[i+1];
|
||||||
|
}
|
||||||
|
cur_len = j;
|
||||||
|
}
|
||||||
|
else if(*data == TCPFILE_START_CHAR04 && datalen < 800)
|
||||||
|
{
|
||||||
|
qq_pme->total_file_len = ntohl(*((unsigned int *)&data[90]));
|
||||||
|
for (i = 102, j = 0; i < datalen - 1; i+=2, j+=2)
|
||||||
|
{
|
||||||
|
if ((data[i] == 0x00) && (data[i + 1] == 0x00))
|
||||||
|
break;
|
||||||
|
filename[j] = data[i];
|
||||||
|
filename[j+1] = data[i+1];
|
||||||
|
}
|
||||||
|
cur_len = j;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
return APP_STATE_GIVEME;
|
||||||
|
}
|
||||||
|
return protocol_process(qq_pme, QQ_FILENAME, filename, cur_len, a_tcp, raw_pkt);
|
||||||
|
|
||||||
|
case QQ_STATE_FCONT_START:
|
||||||
|
//<2F><>UDP<44><50>ͬ<EFBFBD><CDAC><EFBFBD>ǣ<EFBFBD>TCP<43><50><EFBFBD><EFBFBD>ͷ<EFBFBD><CDB7><EFBFBD><EFBFBD>һ<EFBFBD><D2BB><EFBFBD>ǰ<EFBFBD><C7B0><EFBFBD>ƥ<EFBFBD><C6A5><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ʽ<EFBFBD><CABD><EFBFBD>䣻
|
||||||
|
if(a_tcp->curdir != qq_pme->file_dir+1)
|
||||||
|
break;
|
||||||
|
|
||||||
|
if(a_tcp->ptcpdetail->lostlen) //<2F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ѱ<EFBFBD>Ұ<EFBFBD><D2B0><EFBFBD>ͷ
|
||||||
|
{
|
||||||
|
qq_pme->fileinfo.len = 0;
|
||||||
|
qq_pme->qq_state = QQ_STATE_INIT;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
if(qq_pme->fileinfo.len > 0)
|
||||||
|
{
|
||||||
|
if(qq_pme->fileinfo.len < TCPFILE_HDR_LEN04+1) //<2F>ж<EFBFBD><D0B6><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ʱ<EFBFBD><CAB1>Ҫ<EFBFBD><D2AA><EFBFBD><EFBFBD><EFBFBD>ݳ<EFBFBD><DDB3>ȴ<EFBFBD><C8B4><EFBFBD>ͷ<EFBFBD><CDB7><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
||||||
|
{
|
||||||
|
need_len = TCPFILE_HDR_LEN04+1 - qq_pme->fileinfo.len;
|
||||||
|
if(datalen < need_len)
|
||||||
|
{
|
||||||
|
if(save_initial_tcp_data(&qq_pme->fileinfo, data, datalen, qq_pme->thread_id) < 0)
|
||||||
|
{
|
||||||
|
return APP_STATE_DROPME;
|
||||||
|
}
|
||||||
|
return APP_STATE_GIVEME;
|
||||||
|
}
|
||||||
|
memcpy(qq_pme->fileinfo.buf+qq_pme->fileinfo.len, data, need_len);
|
||||||
|
qq_pme->fileinfo.len += need_len;
|
||||||
|
data += need_len;
|
||||||
|
datalen -= need_len;
|
||||||
|
if(data_hit_feature(qq_pme->fileinfo.buf, qq_pme->fileinfo.len, 0, FINDEX_TCPFILE))
|
||||||
|
{
|
||||||
|
qq_pme->fileinfo.prot_flag = QQ_FILECONT;
|
||||||
|
}
|
||||||
|
else if(data_hit_feature(qq_pme->fileinfo.buf, qq_pme->fileinfo.len, 0, FINDEX_TCPFILE_END))
|
||||||
|
{
|
||||||
|
qq_pme->fileinfo.prot_flag = QQ_FILECONT_END;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
return APP_STATE_DROPME;
|
||||||
|
}
|
||||||
|
if(*(qq_pme->fileinfo.buf) == TCPFILE_START_CHAR04)
|
||||||
|
{
|
||||||
|
qq_pme->fileinfo.block_len = ntohs(*((unsigned short *)&qq_pme->fileinfo.buf[3]));
|
||||||
|
qq_pme->compress_flag = qq_pme->fileinfo.buf[TCPFILE_HDR_LEN04-1];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//<2F><><EFBFBD><EFBFBD>һ<EFBFBD><D2BB><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ij<EFBFBD><C4B3><EFBFBD>
|
||||||
|
need_len = qq_pme->fileinfo.block_len - qq_pme->fileinfo.len;
|
||||||
|
if(need_len > datalen)
|
||||||
|
{
|
||||||
|
if(save_initial_tcp_data(&qq_pme->fileinfo, data, datalen, qq_pme->thread_id) < 0)
|
||||||
|
{
|
||||||
|
return APP_STATE_DROPME;
|
||||||
|
}
|
||||||
|
return APP_STATE_GIVEME;
|
||||||
|
}
|
||||||
|
if(save_initial_tcp_data(&qq_pme->fileinfo, data, need_len, qq_pme->thread_id) < 0)
|
||||||
|
{
|
||||||
|
return APP_STATE_DROPME;
|
||||||
|
}
|
||||||
|
qq_pme->fileinfo.len = 0;
|
||||||
|
data += need_len;
|
||||||
|
datalen -= need_len;
|
||||||
|
|
||||||
|
if(*(qq_pme->fileinfo.buf) == TCPFILE_START_CHAR04 && qq_pme->fileinfo.block_len>TCPFILE_HDR_LEN04)
|
||||||
|
{
|
||||||
|
if(*(qq_pme->fileinfo.buf + qq_pme->fileinfo.block_len - 1) != TCPFILE_END_CHAR04)
|
||||||
|
return APP_STATE_DROPME;
|
||||||
|
rec = protocol_process(qq_pme, qq_pme->fileinfo.prot_flag, qq_pme->fileinfo.buf+TCPFILE_HDR_LEN04, qq_pme->fileinfo.block_len-TCPFILE_HDR_LEN04-1, a_tcp, raw_pkt);
|
||||||
|
}
|
||||||
|
if(rec & APP_STATE_DROPME)
|
||||||
|
{
|
||||||
|
return rec;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
while(datalen>0)
|
||||||
|
{
|
||||||
|
//ͷ<><CDB7><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ȳ<EFBFBD><C8B2><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
||||||
|
if(datalen < TCPFILE_HDR_LEN04+1)
|
||||||
|
{
|
||||||
|
if(save_initial_tcp_data(&qq_pme->fileinfo, data, datalen, qq_pme->thread_id) < 0)
|
||||||
|
{
|
||||||
|
return APP_STATE_DROPME;
|
||||||
|
}
|
||||||
|
return APP_STATE_GIVEME;
|
||||||
|
}
|
||||||
|
|
||||||
|
if(data_hit_feature(data, datalen, 0, FINDEX_TCPFILE))
|
||||||
|
{
|
||||||
|
qq_pme->fileinfo.prot_flag = QQ_FILECONT;
|
||||||
|
}
|
||||||
|
else if(data_hit_feature(data, datalen, 0, FINDEX_TCPFILE_END))
|
||||||
|
{
|
||||||
|
qq_pme->fileinfo.prot_flag = QQ_FILECONT_END;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
//<2F>е<EFBFBD>ʱ<EFBFBD><EFBFBD><F2BFB4B5>м<EFBFBD><D0BC><EFBFBD><EFBFBD><EFBFBD>С<EFBFBD>飬<EFBFBD>轫<EFBFBD><E8BDAB><EFBFBD>߳<EFBFBD>
|
||||||
|
unsigned short block_len = ntohs(*((unsigned short *)&data[3]));
|
||||||
|
if(*data == TCPFILE_START_CHAR04 && block_len < TCPFILE_HDR_LEN04)
|
||||||
|
{
|
||||||
|
data += block_len;
|
||||||
|
datalen -= block_len;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
return APP_STATE_DROPME;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//ͷ<><CDB7><EFBFBD><EFBFBD><EFBFBD>ȹ<EFBFBD><C8B9><EFBFBD><EFBFBD><EFBFBD>ͷ<EFBFBD><CDB7>ȡ<EFBFBD><C8A1><EFBFBD>鳤<EFBFBD>ȣ<EFBFBD><C8A3>鲻<EFBFBD><E9B2BB>Ҳ<EFBFBD><D2B2><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
||||||
|
if(*data == TCPFILE_START_CHAR04 && datalen>TCPFILE_HDR_LEN04)
|
||||||
|
{
|
||||||
|
qq_pme->fileinfo.block_len = ntohs(*((unsigned short *)&data[3]));
|
||||||
|
qq_pme->compress_flag = data[TCPFILE_HDR_LEN04-1];
|
||||||
|
if(datalen < qq_pme->fileinfo.block_len)
|
||||||
|
{
|
||||||
|
if(save_initial_tcp_data(&qq_pme->fileinfo, data, datalen, qq_pme->thread_id) < 0)
|
||||||
|
{
|
||||||
|
return APP_STATE_DROPME;
|
||||||
|
}
|
||||||
|
return APP_STATE_GIVEME;
|
||||||
|
}
|
||||||
|
|
||||||
|
rec = protocol_process(qq_pme, qq_pme->fileinfo.prot_flag, data+TCPFILE_HDR_LEN04, qq_pme->fileinfo.block_len-TCPFILE_HDR_LEN04-1, a_tcp, raw_pkt);
|
||||||
|
if(rec & APP_STATE_DROPME)
|
||||||
|
{
|
||||||
|
return rec;
|
||||||
|
}
|
||||||
|
data += qq_pme->fileinfo.block_len;
|
||||||
|
datalen -= qq_pme->fileinfo.block_len;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
|
||||||
|
case QQ_STATE_FCONT_END:
|
||||||
|
case QQ_STATE_FCONT:
|
||||||
|
case QQ_STATE_INIT:
|
||||||
|
default:
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
return APP_STATE_GIVEME;
|
||||||
|
}
|
||||||
|
|
||||||
|
extern "C" char QQ_PROT_TCP_ENTRY(struct streaminfo *a_tcp, void **pme, int thread_seq,const void *raw_pkt)
|
||||||
|
{
|
||||||
|
QQ_PACKET_TYPE_t type;
|
||||||
|
qq_proto_pme_t *qq_pme=(qq_proto_pme_t*)(*pme);
|
||||||
|
int rec = APP_STATE_GIVEME;
|
||||||
|
|
||||||
|
switch(a_tcp->opstate)
|
||||||
|
{
|
||||||
|
case OP_STATE_PENDING:
|
||||||
|
type = is_qq_protocol_tcp((char *)a_tcp->ptcpdetail->pdata, a_tcp->ptcpdetail->datalen);
|
||||||
|
if(type)
|
||||||
|
{
|
||||||
|
if(init_qq_pme_info(&qq_pme, a_tcp->threadnum, 1))
|
||||||
|
return APP_STATE_DROPME;
|
||||||
|
qq_pme->qq_state = assign_qq_state(type);
|
||||||
|
*pme = qq_pme;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
return APP_STATE_DROPME;
|
||||||
|
}
|
||||||
|
//no break here
|
||||||
|
|
||||||
|
case OP_STATE_DATA:
|
||||||
|
qq_pme->qq_state = analyse_tcp_pdata_state(qq_pme->qq_state, (char *)a_tcp->ptcpdetail->pdata, a_tcp->ptcpdetail->datalen);
|
||||||
|
|
||||||
|
rec = process_tcp_data_state(a_tcp, qq_pme, thread_seq, raw_pkt);
|
||||||
|
break;
|
||||||
|
|
||||||
|
case OP_STATE_CLOSE:
|
||||||
|
clear_qq_pme_info(qq_pme, a_tcp, raw_pkt);
|
||||||
|
break;
|
||||||
|
|
||||||
|
default:
|
||||||
|
return APP_STATE_DROPME;
|
||||||
|
}
|
||||||
|
|
||||||
|
if(rec & APP_STATE_DROPME)
|
||||||
|
{
|
||||||
|
clear_qq_pme_info(qq_pme, a_tcp, raw_pkt);
|
||||||
|
}
|
||||||
|
|
||||||
|
return rec;
|
||||||
|
}
|
||||||
|
|
||||||
|
int mkdir_according_path(const char * path)
|
||||||
|
{
|
||||||
|
char buffer[256];
|
||||||
|
const char *ps=path, *pc;
|
||||||
|
|
||||||
|
if(*ps == '/')
|
||||||
|
ps += 1;
|
||||||
|
|
||||||
|
while((pc = strchr(ps, '/')) != NULL)
|
||||||
|
{
|
||||||
|
while(*(pc+1) == '/')
|
||||||
|
pc++;
|
||||||
|
|
||||||
|
memcpy(buffer, path, pc - path);
|
||||||
|
buffer[pc-path] = '\0';
|
||||||
|
|
||||||
|
if(access(buffer, F_OK))
|
||||||
|
{
|
||||||
|
if(mkdir(buffer, 0777))
|
||||||
|
{
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
ps = pc + 1;
|
||||||
|
}
|
||||||
|
if(access(path, F_OK))
|
||||||
|
{
|
||||||
|
if(mkdir(path, 0777))
|
||||||
|
{
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int register_field_stat(const char *logpath)
|
||||||
|
{
|
||||||
|
unsigned int stat_cycle;
|
||||||
|
int ret, i;
|
||||||
|
char stat_log[128];
|
||||||
|
const char *region_name[STAT_ID_NUM]={"NEW_SESSION", "DEL_SESSION", "SESSION_NUM"};
|
||||||
|
|
||||||
|
snprintf(stat_log, 128, "%s/qq_stat.log", logpath);
|
||||||
|
|
||||||
|
ret = MESA_load_profile_uint_def(CONF_FILENAME, "MODULE", "STAT_FIELD_CYCLE",&stat_cycle, 5);
|
||||||
|
if(ret < 0)
|
||||||
|
{
|
||||||
|
MESA_HANDLE_RUNTIME_LOG(g_qq_global_info.runtime_log, RLOG_LV_INFO, MODULE_NAME,"get [MODULE] STAT_FIELD_CYCLE failed, using default 5.");
|
||||||
|
}
|
||||||
|
ret = MESA_load_profile_int_def(CONF_FILENAME, "MODULE", "STAT_FIELD_TRIG",&g_qq_global_info.stat_trig, 0);
|
||||||
|
if(ret < 0)
|
||||||
|
{
|
||||||
|
MESA_HANDLE_RUNTIME_LOG(g_qq_global_info.runtime_log, RLOG_LV_INFO, MODULE_NAME,"get [MODULE] STAT_FIELD_TRIG failed, using default 0.");
|
||||||
|
}
|
||||||
|
|
||||||
|
if(g_qq_global_info.stat_trig==0)
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
FILE* stat_fd = fopen(stat_log, "w");
|
||||||
|
if(stat_fd == NULL)
|
||||||
|
{
|
||||||
|
MESA_HANDLE_RUNTIME_LOG(g_qq_global_info.runtime_log, RLOG_LV_INFO, MODULE_NAME,"fopen %s failed.", stat_log);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
g_qq_global_info.stat_handle = init_screen_stat(stat_fd, stat_cycle, g_qq_global_info.stat_trig);
|
||||||
|
if(g_qq_global_info.stat_handle==NULL)
|
||||||
|
{
|
||||||
|
MESA_HANDLE_RUNTIME_LOG(g_qq_global_info.runtime_log, RLOG_LV_INFO, MODULE_NAME,"init_screen_stat failed.");
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
for(i=0;i<STAT_ID_NUM;i++)
|
||||||
|
{
|
||||||
|
g_qq_global_info.stat_id[i] = stat_field_register(g_qq_global_info.stat_handle, region_name[i]);
|
||||||
|
if(g_qq_global_info.stat_id[i] < 0)
|
||||||
|
{
|
||||||
|
MESA_HANDLE_RUNTIME_LOG(g_qq_global_info.runtime_log, RLOG_LV_INFO, MODULE_NAME,"stat_field_register %s failed.", region_name[i]);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
/*<2A><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ʽ(<28><>ʾ<EFBFBD><CABE>)<29><>
|
||||||
|
*FEATURE_QQ_TCPFILENAME=0:04&13:00,01,00,00,00,00,00&31:0c,00,00,00,00,2b;3:00,2b,52,00,00&34:00,2b,52,00,00
|
||||||
|
*1.<2E>Էֺŷָ<C5B7>һ<EFBFBD><D2BB><EFBFBD><EFBFBD><EFBFBD>ã<EFBFBD><C3A3><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ж<EFBFBD><D0B6><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ã<EFBFBD><C3A3><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ǻ<EFBFBD><C7BB>Ĺ<EFBFBD>ϵ<EFBFBD><CFB5>
|
||||||
|
*2.<2E><>&<26>ָ<EFBFBD>һ<EFBFBD><D2BB><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ڵĶ<DAB5><C4B6><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ĺ<EFBFBD>ϵ<EFBFBD><CFB5>
|
||||||
|
*3.ÿ<><C3BF><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ǰ<EFBFBD><C7B0>һ<EFBFBD><D2BB><EFBFBD><EFBFBD><EFBFBD>ֺ<EFBFBD>ð<EFBFBD>ű<EFBFBD>ʾ<EFBFBD><CABE>ʮ<EFBFBD><CAAE><EFBFBD><EFBFBD><EFBFBD>ƴ<EFBFBD><C6B4><EFBFBD>һ<EFBFBD><D2BB><EFBFBD><EFBFBD><EFBFBD>ڵ<EFBFBD>ƫ<EFBFBD><C6AB><EFBFBD><EFBFBD>(<28>ֽ<EFBFBD>)<29><>ÿ<EFBFBD><C3BF>ʮ<EFBFBD><CAAE><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ռ<EFBFBD><D5BC>λ<EFBFBD><CEBB><EFBFBD>ö<EFBFBD><C3B6>ŷָ
|
||||||
|
*/
|
||||||
|
int decode_feature_to_binary(char *feature, const char *feature_name,
|
||||||
|
char *feature_bin[MAX_QQ_FEATURE_NUM][MAX_AND_RULE],
|
||||||
|
unsigned int feature_len[MAX_QQ_FEATURE_NUM][MAX_AND_RULE],
|
||||||
|
unsigned int feature_off[MAX_QQ_FEATURE_NUM][MAX_AND_RULE],
|
||||||
|
unsigned int rulenum[MAX_QQ_FEATURE_NUM],
|
||||||
|
unsigned int *feature_num)
|
||||||
|
{
|
||||||
|
char *ptmp0, *save_ptr0=NULL, *ptmp1, *save_ptr1=NULL, *ptmp2, *save_ptr2=NULL;
|
||||||
|
int fnum=0, rnum, i, val;
|
||||||
|
|
||||||
|
for(ptmp0 = strtok_r(feature, ";", &save_ptr0); ptmp0 != NULL; ptmp0 = strtok_r(NULL, ";", &save_ptr0))
|
||||||
|
{
|
||||||
|
rnum = 0;
|
||||||
|
for(ptmp1 = strtok_r(ptmp0, "&", &save_ptr1); ptmp1 != NULL; ptmp1 = strtok_r(NULL, "&", &save_ptr1))
|
||||||
|
{
|
||||||
|
feature_bin[fnum][rnum] = (char *)calloc(1, strlen(ptmp1)/2+1);
|
||||||
|
i = 0;
|
||||||
|
if(sscanf(ptmp1, "%u:", &feature_off[fnum][rnum]) != 1)
|
||||||
|
{
|
||||||
|
MESA_HANDLE_RUNTIME_LOGV2(g_qq_global_info.runtime_log,RLOG_LV_FATAL, MODULE_NAME, "Decode [FEATURE] %s error, format invalid.", feature_name);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
while(*ptmp1 != ':')
|
||||||
|
ptmp1++;
|
||||||
|
for(ptmp2 = strtok_r(ptmp1+1, ",", &save_ptr2); ptmp2 != NULL; ptmp2 = strtok_r(NULL, ",", &save_ptr2))
|
||||||
|
{
|
||||||
|
if(strlen(ptmp2)!=2 || sscanf(ptmp2, "%02x", &val)!=1)
|
||||||
|
{
|
||||||
|
MESA_HANDLE_RUNTIME_LOGV2(g_qq_global_info.runtime_log,RLOG_LV_FATAL, MODULE_NAME, "Decode [FEATURE] %s error, format invalid.", feature_name);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
feature_bin[fnum][rnum][i++] = val;
|
||||||
|
}
|
||||||
|
feature_len[fnum][rnum] = i;
|
||||||
|
rnum++;
|
||||||
|
}
|
||||||
|
rulenum[fnum] = rnum;
|
||||||
|
fnum++;
|
||||||
|
}
|
||||||
|
|
||||||
|
*feature_num = fnum;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
extern "C" int QQ_PROT_INIT(void)
|
||||||
|
{
|
||||||
|
int log_level;
|
||||||
|
char tmp_buffer[512]={0}, tmp_buffer2[512]={0};
|
||||||
|
|
||||||
|
memset(&g_qq_global_info, 0, sizeof(qq_global_info_t));
|
||||||
|
|
||||||
|
MESA_load_profile_string_def(CONF_FILENAME, "MODULE", "RUN_LOG_DIR", tmp_buffer, sizeof(tmp_buffer), "./log/qq_file_trans");
|
||||||
|
MESA_load_profile_int_def(CONF_FILENAME, "MODULE", "RUN_LOG_LEVEL", &log_level,10);
|
||||||
|
if(mkdir_according_path(tmp_buffer))
|
||||||
|
{
|
||||||
|
printf("mkdir %s failed: %s\n", tmp_buffer, strerror(errno));
|
||||||
|
assert(0);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
snprintf(tmp_buffer2, sizeof(tmp_buffer2), "%s/qq_file_trans.log", tmp_buffer);
|
||||||
|
g_qq_global_info.runtime_log = MESA_create_runtime_log_handle(tmp_buffer2, log_level);
|
||||||
|
if(g_qq_global_info.runtime_log==NULL)
|
||||||
|
{
|
||||||
|
assert(0);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
if(register_field_stat(tmp_buffer) < 0)
|
||||||
|
{
|
||||||
|
assert(0);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
memset(tmp_buffer, 0 , sizeof(tmp_buffer));
|
||||||
|
if(MESA_load_profile_string_nodef(CONF_FILENAME, "FEATURE", "FEATURE_QQTCP_PROTOCOL", tmp_buffer, sizeof(tmp_buffer)) < 0)
|
||||||
|
{
|
||||||
|
MESA_HANDLE_RUNTIME_LOGV2(g_qq_global_info.runtime_log,RLOG_LV_FATAL, MODULE_NAME, "Get [FEATURE] FEATURE_QQTCP_PROTOCOL error.");
|
||||||
|
assert(0);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
if(decode_feature_to_binary(tmp_buffer, "FEATURE_QQTCP_PROTOCOL", g_qq_global_info.feature[FINDEX_TCPPROT], g_qq_global_info.ftlen[FINDEX_TCPPROT], g_qq_global_info.ftoff[FINDEX_TCPPROT], g_qq_global_info.rulenum[FINDEX_TCPPROT], &g_qq_global_info.fnum[FINDEX_TCPPROT]))
|
||||||
|
{
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
memset(tmp_buffer, 0 , sizeof(tmp_buffer));
|
||||||
|
if(MESA_load_profile_string_nodef(CONF_FILENAME, "FEATURE", "FEATURE_QQUDP_PROTOCOL", tmp_buffer, sizeof(tmp_buffer)) < 0)
|
||||||
|
{
|
||||||
|
MESA_HANDLE_RUNTIME_LOGV2(g_qq_global_info.runtime_log,RLOG_LV_FATAL, MODULE_NAME, "Get [FEATURE] FEATURE_QQUDP_PROTOCOL error.");
|
||||||
|
assert(0);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
if(decode_feature_to_binary(tmp_buffer, "FEATURE_QQUDP_PROTOCOL", g_qq_global_info.feature[FINDEX_UDPPROT], g_qq_global_info.ftlen[FINDEX_UDPPROT], g_qq_global_info.ftoff[FINDEX_UDPPROT], g_qq_global_info.rulenum[FINDEX_UDPPROT], &g_qq_global_info.fnum[FINDEX_UDPPROT]))
|
||||||
|
{
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
memset(tmp_buffer, 0 , sizeof(tmp_buffer));
|
||||||
|
if(MESA_load_profile_string_nodef(CONF_FILENAME, "FEATURE", "FEATURE_QQ_NUMBER_TCP", tmp_buffer, sizeof(tmp_buffer)) < 0)
|
||||||
|
{
|
||||||
|
MESA_HANDLE_RUNTIME_LOGV2(g_qq_global_info.runtime_log,RLOG_LV_FATAL, MODULE_NAME, "Get [FEATURE] FEATURE_QQ_NUMBER_TCP error.");
|
||||||
|
assert(0);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
if(decode_feature_to_binary(tmp_buffer, "FEATURE_QQ_NUMBER_TCP", g_qq_global_info.feature[FINDEX_TCPNUM], g_qq_global_info.ftlen[FINDEX_TCPNUM], g_qq_global_info.ftoff[FINDEX_TCPNUM], g_qq_global_info.rulenum[FINDEX_TCPNUM], &g_qq_global_info.fnum[FINDEX_TCPNUM]))
|
||||||
|
{
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
memset(tmp_buffer, 0 , sizeof(tmp_buffer));
|
||||||
|
if(MESA_load_profile_string_nodef(CONF_FILENAME, "FEATURE", "FEATURE_QQ_NUMBER_UDP", tmp_buffer, sizeof(tmp_buffer)) < 0)
|
||||||
|
{
|
||||||
|
MESA_HANDLE_RUNTIME_LOGV2(g_qq_global_info.runtime_log,RLOG_LV_FATAL, MODULE_NAME, "Get [FEATURE] FEATURE_QQ_NUMBER_UDP error.");
|
||||||
|
assert(0);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
if(decode_feature_to_binary(tmp_buffer, "FEATURE_QQ_NUMBER_UDP", g_qq_global_info.feature[FINDEX_UDPNUM], g_qq_global_info.ftlen[FINDEX_UDPNUM], g_qq_global_info.ftoff[FINDEX_UDPNUM], g_qq_global_info.rulenum[FINDEX_UDPNUM], &g_qq_global_info.fnum[FINDEX_UDPNUM]))
|
||||||
|
{
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
memset(tmp_buffer, 0 , sizeof(tmp_buffer));
|
||||||
|
if(MESA_load_profile_string_nodef(CONF_FILENAME, "FEATURE", "FEATURE_QQ_TCPFILE", tmp_buffer, sizeof(tmp_buffer)) < 0)
|
||||||
|
{
|
||||||
|
MESA_HANDLE_RUNTIME_LOGV2(g_qq_global_info.runtime_log,RLOG_LV_FATAL, MODULE_NAME, "Get [FEATURE] FEATURE_QQ_TCPFILE error.");
|
||||||
|
assert(0);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
if(decode_feature_to_binary(tmp_buffer, "FEATURE_QQ_TCPFILE", g_qq_global_info.feature[FINDEX_TCPFILE], g_qq_global_info.ftlen[FINDEX_TCPFILE], g_qq_global_info.ftoff[FINDEX_TCPFILE], g_qq_global_info.rulenum[FINDEX_TCPFILE], &g_qq_global_info.fnum[FINDEX_TCPFILE]))
|
||||||
|
{
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
memset(tmp_buffer, 0 , sizeof(tmp_buffer));
|
||||||
|
if(MESA_load_profile_string_nodef(CONF_FILENAME, "FEATURE", "FEATURE_QQ_TCPFILE_END", tmp_buffer, sizeof(tmp_buffer)) < 0)
|
||||||
|
{
|
||||||
|
MESA_HANDLE_RUNTIME_LOGV2(g_qq_global_info.runtime_log,RLOG_LV_FATAL, MODULE_NAME, "Get [FEATURE] FEATURE_QQ_TCPFILE_END error.");
|
||||||
|
assert(0);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
if(decode_feature_to_binary(tmp_buffer, "FEATURE_QQ_TCPFILE_END", g_qq_global_info.feature[FINDEX_TCPFILE_END], g_qq_global_info.ftlen[FINDEX_TCPFILE_END], g_qq_global_info.ftoff[FINDEX_TCPFILE_END], g_qq_global_info.rulenum[FINDEX_TCPFILE_END], &g_qq_global_info.fnum[FINDEX_TCPFILE_END]))
|
||||||
|
{
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
memset(tmp_buffer, 0 , sizeof(tmp_buffer));
|
||||||
|
if(MESA_load_profile_string_nodef(CONF_FILENAME, "FEATURE", "FEATURE_QQ_TCPFILENAME", tmp_buffer, sizeof(tmp_buffer)) < 0)
|
||||||
|
{
|
||||||
|
MESA_HANDLE_RUNTIME_LOGV2(g_qq_global_info.runtime_log,RLOG_LV_FATAL, MODULE_NAME, "Get [FEATURE] FEATURE_QQ_TCPFILENAME error.");
|
||||||
|
assert(0);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
if(decode_feature_to_binary(tmp_buffer, "FEATURE_QQ_TCPFILENAME", g_qq_global_info.feature[FINDEX_TCPNAME], g_qq_global_info.ftlen[FINDEX_TCPNAME], g_qq_global_info.ftoff[FINDEX_TCPNAME], g_qq_global_info.rulenum[FINDEX_TCPNAME], &g_qq_global_info.fnum[FINDEX_TCPNAME]))
|
||||||
|
{
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
memset(tmp_buffer, 0 , sizeof(tmp_buffer));
|
||||||
|
if(MESA_load_profile_string_nodef(CONF_FILENAME, "FEATURE", "FEATURE_QQ_UDPFILE_START", tmp_buffer, sizeof(tmp_buffer)) < 0)
|
||||||
|
{
|
||||||
|
MESA_HANDLE_RUNTIME_LOGV2(g_qq_global_info.runtime_log,RLOG_LV_FATAL, MODULE_NAME, "Get [FEATURE] FEATURE_QQ_UDPFILE_START error.");
|
||||||
|
assert(0);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
if(decode_feature_to_binary(tmp_buffer, "FEATURE_QQ_UDPFILE_START", g_qq_global_info.feature[FINDEX_UDPFILE], g_qq_global_info.ftlen[FINDEX_UDPFILE], g_qq_global_info.ftoff[FINDEX_UDPFILE], g_qq_global_info.rulenum[FINDEX_UDPFILE], &g_qq_global_info.fnum[FINDEX_UDPFILE]))
|
||||||
|
{
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
memset(tmp_buffer, 0 , sizeof(tmp_buffer));
|
||||||
|
if(MESA_load_profile_string_nodef(CONF_FILENAME, "FEATURE", "FEATURE_QQ_UDPFILE_END", tmp_buffer, sizeof(tmp_buffer)) < 0)
|
||||||
|
{
|
||||||
|
MESA_HANDLE_RUNTIME_LOGV2(g_qq_global_info.runtime_log,RLOG_LV_FATAL, MODULE_NAME, "Get [FEATURE] FEATURE_QQ_UDPFILE_END error.");
|
||||||
|
assert(0);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
if(decode_feature_to_binary(tmp_buffer, "FEATURE_QQ_UDPFILE_END", g_qq_global_info.feature[FINDEX_UDPFILE_END], g_qq_global_info.ftlen[FINDEX_UDPFILE_END], g_qq_global_info.ftoff[FINDEX_UDPFILE_END], g_qq_global_info.rulenum[FINDEX_UDPFILE_END], &g_qq_global_info.fnum[FINDEX_UDPFILE_END]))
|
||||||
|
{
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
memset(tmp_buffer, 0 , sizeof(tmp_buffer));
|
||||||
|
if(MESA_load_profile_string_nodef(CONF_FILENAME, "FEATURE", "FEATURE_QQ_UDPFILENAME", tmp_buffer, sizeof(tmp_buffer)) < 0)
|
||||||
|
{
|
||||||
|
MESA_HANDLE_RUNTIME_LOGV2(g_qq_global_info.runtime_log,RLOG_LV_FATAL, MODULE_NAME, "Get [FEATURE] FEATURE_QQ_UDPFILENAME error.");
|
||||||
|
assert(0);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
if(decode_feature_to_binary(tmp_buffer, "FEATURE_QQ_UDPFILENAME", g_qq_global_info.feature[FINDEX_UDPNAME], g_qq_global_info.ftlen[FINDEX_UDPNAME], g_qq_global_info.ftoff[FINDEX_UDPNAME], g_qq_global_info.rulenum[FINDEX_UDPNAME], &g_qq_global_info.fnum[FINDEX_UDPNAME]))
|
||||||
|
{
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
g_qq_global_info.instanse = docanalyze_initialize(get_thread_count());
|
||||||
|
if(g_qq_global_info.instanse==NULL)
|
||||||
|
{
|
||||||
|
MESA_HANDLE_RUNTIME_LOGV2(g_qq_global_info.runtime_log,RLOG_LV_FATAL, MODULE_NAME, "docanalyze_initialize() error.");
|
||||||
|
assert(0);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
MESA_load_profile_int_def(CONF_FILENAME, "MODULE", "DEBUG_SWITCH", &g_qq_global_info.debug_sw, 0);
|
||||||
|
if(g_qq_global_info.debug_sw)
|
||||||
|
{
|
||||||
|
if(mkdir_according_path(QQ_DEBUG_FILEPATH))
|
||||||
|
{
|
||||||
|
printf("mkdir %s failed: %s\n", tmp_buffer, strerror(errno));
|
||||||
|
assert(0);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
g_qq_global_info.project_id = project_customer_register("STREAMID", "long");
|
||||||
|
assert(g_qq_global_info.project_id >= 0);
|
||||||
|
|
||||||
|
srand(time(NULL));
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
extern "C" void QQ_PROT_DESTROY(void)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
125
src/qq_file_entry.h
Normal file
125
src/qq_file_entry.h
Normal file
@@ -0,0 +1,125 @@
|
|||||||
|
#ifndef __QQ_FILE_ENTRY_H__
|
||||||
|
#define __QQ_FILE_ENTRY_H__
|
||||||
|
|
||||||
|
#include <pthread.h>
|
||||||
|
|
||||||
|
#include <MESA/DocumentAnalyze.h>
|
||||||
|
|
||||||
|
#define CONF_FILENAME "./conf/qq_file_trans.conf"
|
||||||
|
#define QQ_DEBUG_FILEPATH "./qq_files"
|
||||||
|
#define MODULE_NAME "QQ_FILE"
|
||||||
|
|
||||||
|
#define MAX_QQ_FEATURE_NUM 16
|
||||||
|
#define MAX_AND_RULE 32
|
||||||
|
|
||||||
|
#define PACKET_MIN_LEN_QQNUM 16
|
||||||
|
#define PACKET_MIN_LEN_TCPNAME 551
|
||||||
|
#define PACKET_MIN_LEN_UDPNAME 535
|
||||||
|
|
||||||
|
#define TCPFILE_START_CHAR04 0x04
|
||||||
|
#define TCPFILE_END_CHAR04 0x03
|
||||||
|
#define TCPFILE_HDR_LEN04 53
|
||||||
|
|
||||||
|
#define TCPFILE_START_CHAR0c 0x0c
|
||||||
|
#define TCPFILE_HDR_LEN0c 22
|
||||||
|
|
||||||
|
typedef enum __QQ_PACKET_TYPE
|
||||||
|
{
|
||||||
|
QQ_PACKET_NOT=0,
|
||||||
|
QQ_PACKET_QQNUM,
|
||||||
|
QQ_PACKET_TCPFILE,
|
||||||
|
QQ_PACKET_TCPFILENAME,
|
||||||
|
QQ_PACKET_UDPFILE,
|
||||||
|
QQ_PACKET_UDPFILE_END,
|
||||||
|
QQ_PACKET_UDPFILENAME,
|
||||||
|
QQ_PACKET_OTHER,
|
||||||
|
}QQ_PACKET_TYPE_t;
|
||||||
|
|
||||||
|
typedef enum __QQ_STATE
|
||||||
|
{
|
||||||
|
QQ_STATE_INIT=0,
|
||||||
|
QQ_STATE_QQNUM,
|
||||||
|
QQ_STATE_FNAME,
|
||||||
|
QQ_STATE_FCONT_START,
|
||||||
|
QQ_STATE_FCONT,
|
||||||
|
QQ_STATE_FCONT_END,
|
||||||
|
}QQ_STATE_t;
|
||||||
|
|
||||||
|
typedef enum __FEATURE_INDEX
|
||||||
|
{
|
||||||
|
FINDEX_TCPPROT=0,
|
||||||
|
FINDEX_TCPNUM,
|
||||||
|
FINDEX_TCPNAME,
|
||||||
|
FINDEX_TCPFILE,
|
||||||
|
FINDEX_TCPFILE_END,
|
||||||
|
FINDEX_UDPPROT,
|
||||||
|
FINDEX_UDPNUM,
|
||||||
|
FINDEX_UDPNAME,
|
||||||
|
FINDEX_UDPFILE,
|
||||||
|
FINDEX_UDPFILE_END,
|
||||||
|
|
||||||
|
FINDEX_NUM,
|
||||||
|
}FEATURE_INDEX_T;
|
||||||
|
|
||||||
|
enum __FIELD_STAT_ID
|
||||||
|
{
|
||||||
|
STAT_ID_NEW_SESSION,
|
||||||
|
STAT_ID_DEL_SESSION,
|
||||||
|
STAT_ID_SESSION_NUM,
|
||||||
|
STAT_ID_NUM,
|
||||||
|
};
|
||||||
|
|
||||||
|
typedef struct __trans_file_info
|
||||||
|
{
|
||||||
|
char *buf;
|
||||||
|
unsigned int len;
|
||||||
|
unsigned int max_len;
|
||||||
|
long long prot_flag;
|
||||||
|
|
||||||
|
int block_seq; //<2F><><EFBFBD>ݰ<EFBFBD><DDB0><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ţ<EFBFBD><C5A3><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ʱ<EFBFBD><CAB1><EFBFBD>ش<EFBFBD><D8B4><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>UDP
|
||||||
|
unsigned short block_len; //һ<><D2BB>С<EFBFBD><D0A1><EFBFBD>Ĵ<EFBFBD>С<EFBFBD><D0A1><EFBFBD><EFBFBD><EFBFBD><EFBFBD>һ<EFBFBD><D2BB><EFBFBD><EFBFBD>һ<EFBFBD><D2BB><EFBFBD><EFBFBD><EFBFBD>Ĵ<EFBFBD>С<EFBFBD><D0A1><EFBFBD><EFBFBD><EFBFBD><EFBFBD>TCP
|
||||||
|
unsigned short file_seq; //һ<><D2BB>С<EFBFBD><D0A1><EFBFBD>Ĵ<EFBFBD>С<EFBFBD><D0A1><EFBFBD><EFBFBD><EFBFBD><EFBFBD>һ<EFBFBD><D2BB><EFBFBD><EFBFBD>һ<EFBFBD><D2BB><EFBFBD><EFBFBD><EFBFBD>Ĵ<EFBFBD>С<EFBFBD><D0A1><EFBFBD><EFBFBD><EFBFBD><EFBFBD>TCP
|
||||||
|
}trans_file_info_t;
|
||||||
|
|
||||||
|
typedef struct __qq_proto_pme
|
||||||
|
{
|
||||||
|
void *connector_hdl;
|
||||||
|
char filename[512];
|
||||||
|
int filename_len;
|
||||||
|
QQ_STATE_t qq_state;
|
||||||
|
unsigned int qqnum[2];
|
||||||
|
unsigned int qqnum_from;
|
||||||
|
unsigned int qqnum_to;
|
||||||
|
|
||||||
|
trans_file_info_t fileinfo;
|
||||||
|
|
||||||
|
int thread_id;
|
||||||
|
unsigned int file_dir;
|
||||||
|
unsigned int total_file_len;
|
||||||
|
unsigned int curr_file_len;
|
||||||
|
char pending_flag;
|
||||||
|
char end_flag;
|
||||||
|
char compress_flag;
|
||||||
|
char tcp_flag;
|
||||||
|
unsigned int random;
|
||||||
|
}qq_proto_pme_t;
|
||||||
|
|
||||||
|
typedef struct __qq_global_info
|
||||||
|
{
|
||||||
|
void *runtime_log;
|
||||||
|
void *stat_handle;
|
||||||
|
docanalyze_instance_t instanse;
|
||||||
|
char *feature[FINDEX_NUM][MAX_QQ_FEATURE_NUM][MAX_AND_RULE];
|
||||||
|
unsigned int ftlen[FINDEX_NUM][MAX_QQ_FEATURE_NUM][MAX_AND_RULE];
|
||||||
|
unsigned int ftoff[FINDEX_NUM][MAX_QQ_FEATURE_NUM][MAX_AND_RULE];
|
||||||
|
unsigned int rulenum[FINDEX_NUM][MAX_QQ_FEATURE_NUM]; //ÿһ<C3BF><D2BB><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ڲ<EFBFBD><DAB2><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
||||||
|
unsigned int fnum[FINDEX_NUM];
|
||||||
|
|
||||||
|
int project_id;
|
||||||
|
int stat_trig;
|
||||||
|
int stat_id[STAT_ID_NUM];
|
||||||
|
int debug_sw;
|
||||||
|
}qq_global_info_t;
|
||||||
|
|
||||||
|
#endif
|
||||||
|
|
||||||
308
src/qq_file_global.c
Normal file
308
src/qq_file_global.c
Normal file
@@ -0,0 +1,308 @@
|
|||||||
|
#include <stdio.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
#include <sys/stat.h>
|
||||||
|
#include <sys/types.h>
|
||||||
|
#include <errno.h>
|
||||||
|
#include <assert.h>
|
||||||
|
#include <iconv.h>
|
||||||
|
#include <pthread.h>
|
||||||
|
|
||||||
|
#include <MESA/stream.h>
|
||||||
|
#include <MESA/DocumentAnalyze.h>
|
||||||
|
#include <MESA/MESA_handle_logger.h>
|
||||||
|
|
||||||
|
#include "qq_file_entry.h"
|
||||||
|
#include "qq_file_global.h"
|
||||||
|
#include "connector.h"
|
||||||
|
|
||||||
|
#define QQ_OPT_SERVICE "SERVICE_IMQQ"
|
||||||
|
#define QQ_OPT_FILESET "opt_file_set"
|
||||||
|
#define QQ_OPT_FILENAME "opt_file_name"
|
||||||
|
#define QQ_OPT_FILECONT "opt_file_content"
|
||||||
|
#define QQ_OPT_NUMFROM "opt_im_from"
|
||||||
|
#define QQ_OPT_NUMTO "opt_im_to"
|
||||||
|
#define QQ_OPT_ACCOUNT "opt_im_account"
|
||||||
|
|
||||||
|
extern qq_global_info_t g_qq_global_info;
|
||||||
|
const unsigned char gzip_feature[4]={0x1f, 0x8b, 0x08, 0x00};
|
||||||
|
|
||||||
|
int write_file_cont2disk(char *buffer, int filelen, char *filename, int seq, int random)
|
||||||
|
{
|
||||||
|
FILE *fp;
|
||||||
|
int writelen, offset=0;
|
||||||
|
char filenamestr[512];
|
||||||
|
|
||||||
|
sprintf(filenamestr, "%s/%d_%u_%s", QQ_DEBUG_FILEPATH, seq, random, filename);
|
||||||
|
|
||||||
|
fp = fopen(filenamestr, "ab+");
|
||||||
|
if(fp == NULL)
|
||||||
|
{
|
||||||
|
printf("fopen file %s failed.\n", filenamestr);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
while(offset < filelen)
|
||||||
|
{
|
||||||
|
writelen = fwrite(buffer + offset, 1, filelen - offset, fp);
|
||||||
|
if(writelen < 0)
|
||||||
|
{
|
||||||
|
printf("read error: %s\n", strerror(errno));
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
offset += writelen;
|
||||||
|
}
|
||||||
|
fclose(fp);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int buf_cache_check_size(trans_file_info_t *pdstBuf, int wantedlen, int thread_seq)
|
||||||
|
{
|
||||||
|
int ret=0, needlen, block_num=0;
|
||||||
|
|
||||||
|
if(pdstBuf->max_len > wantedlen + pdstBuf->len)
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
needlen = wantedlen + pdstBuf->len - pdstBuf->max_len;
|
||||||
|
block_num = needlen/REALLOC_BLOCK_SIZE;
|
||||||
|
if(needlen%REALLOC_BLOCK_SIZE)
|
||||||
|
block_num += 1;
|
||||||
|
|
||||||
|
pdstBuf->max_len += block_num*REALLOC_BLOCK_SIZE;
|
||||||
|
|
||||||
|
if(pdstBuf->max_len > MAX_MALLOC_SIZE)
|
||||||
|
{
|
||||||
|
pdstBuf->max_len -= block_num*REALLOC_BLOCK_SIZE;
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
if(pdstBuf->buf == NULL)
|
||||||
|
{
|
||||||
|
pdstBuf->buf = (char *)dictator_malloc(thread_seq, pdstBuf->max_len);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
pdstBuf->buf = (char *)dictator_realloc(thread_seq, pdstBuf->buf, pdstBuf->max_len);
|
||||||
|
ret = 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
int save_initial_tcp_data(trans_file_info_t *pdstBuf, char *data, int datalen, int thread_seq)
|
||||||
|
{
|
||||||
|
if(buf_cache_check_size(pdstBuf, datalen, thread_seq) < 0)
|
||||||
|
{
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
memcpy(pdstBuf->buf+pdstBuf->len, data, datalen);
|
||||||
|
pdstBuf->len += datalen;
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
int convert_from_unicode2utf8(char *in, int in_len, char *out, int *out_len)
|
||||||
|
{
|
||||||
|
iconv_t cd;
|
||||||
|
size_t inlen = in_len, outlen=*out_len, ret;
|
||||||
|
|
||||||
|
cd = iconv_open("utf-8", "unicode");
|
||||||
|
if(cd == (iconv_t)-1)
|
||||||
|
{
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
ret = iconv(cd, &in, &inlen, &out, &outlen);
|
||||||
|
if(ret == (size_t)-1)
|
||||||
|
{
|
||||||
|
iconv_close(cd);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
iconv_close(cd);
|
||||||
|
*out_len = outlen;
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int sendback_qq_data(qq_proto_pme_t *qq_pme, struct streaminfo *a_udp, char *filecont, int contlen)
|
||||||
|
{
|
||||||
|
int nest_opt_num = 1;
|
||||||
|
void *nested_opt_handle, *nested_opt_unit_handle, *simple_opt=NULL;
|
||||||
|
|
||||||
|
if(qq_pme->connector_hdl == NULL)
|
||||||
|
{
|
||||||
|
appd_basic_t basic;
|
||||||
|
ulong64 streamid[1];
|
||||||
|
char qqfrom[32], qq_to[32];
|
||||||
|
|
||||||
|
streamid[0] = project_req_get_ulong(a_udp, g_qq_global_info.project_id);
|
||||||
|
|
||||||
|
basic.addr = (struct ipaddr*)&a_udp->addr;
|
||||||
|
basic.addr_num = 1;
|
||||||
|
basic.appd_type = APPD_SENDBACK;
|
||||||
|
basic.destination = DEST_APPD;
|
||||||
|
basic.service_name = QQ_OPT_SERVICE;
|
||||||
|
basic.streamid = streamid;
|
||||||
|
basic.streamid_num = 1;
|
||||||
|
|
||||||
|
qq_pme->connector_hdl = connector_start(&basic, a_udp);
|
||||||
|
if(qq_pme->connector_hdl == NULL)
|
||||||
|
return APP_STATE_DROPME;
|
||||||
|
|
||||||
|
if(qq_pme->filename_len > 0)
|
||||||
|
{
|
||||||
|
nest_opt_num = 2;
|
||||||
|
}
|
||||||
|
sprintf(qqfrom, "%u", qq_pme->qqnum_from);
|
||||||
|
sprintf(qq_to, "%u", qq_pme->qqnum_to);
|
||||||
|
|
||||||
|
simple_opt = connector_create_simple_opt(3);
|
||||||
|
connector_append_simple_opt(simple_opt, QQ_OPT_NUMFROM,qqfrom, strlen(qqfrom));
|
||||||
|
connector_append_simple_opt(simple_opt, QQ_OPT_ACCOUNT,qqfrom, strlen(qqfrom));
|
||||||
|
connector_append_simple_opt(simple_opt, QQ_OPT_NUMTO,qq_to, strlen(qq_to));
|
||||||
|
connector(qq_pme->connector_hdl, COMPLETE, simple_opt, NULL);
|
||||||
|
connector_free_simple_opt(&simple_opt);
|
||||||
|
}
|
||||||
|
|
||||||
|
nested_opt_handle = connector_create_nested_opt(1);
|
||||||
|
nested_opt_unit_handle = connector_create_nested_opt_unit(QQ_OPT_FILESET, nest_opt_num);
|
||||||
|
if(nest_opt_num == 2)
|
||||||
|
{
|
||||||
|
connector_append_nested_opt_unit(nested_opt_unit_handle, QQ_OPT_FILENAME,qq_pme->filename, qq_pme->filename_len);
|
||||||
|
}
|
||||||
|
connector_append_nested_opt_unit(nested_opt_unit_handle, QQ_OPT_FILECONT,filecont, contlen);
|
||||||
|
connector_append_nested_opt(nested_opt_handle,nested_opt_unit_handle);
|
||||||
|
|
||||||
|
if(qq_pme->end_flag)
|
||||||
|
{
|
||||||
|
connector(qq_pme->connector_hdl, FRAG_END, NULL, nested_opt_handle);
|
||||||
|
connector_finish(qq_pme->connector_hdl);
|
||||||
|
qq_pme->connector_hdl = NULL;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
connector(qq_pme->connector_hdl, FRAG, NULL, nested_opt_handle);
|
||||||
|
}
|
||||||
|
connector_free_nested_opt_unit(&nested_opt_unit_handle);
|
||||||
|
connector_free_nested_opt(&nested_opt_handle);
|
||||||
|
|
||||||
|
return APP_STATE_GIVEME;
|
||||||
|
}
|
||||||
|
|
||||||
|
int protocol_process(qq_proto_pme_t *qq_pme, long long prot_flag, char *buf_in, int inlen, struct streaminfo *a_udp, const void *raw_pkt)
|
||||||
|
{
|
||||||
|
int rec = APP_STATE_GIVEME;
|
||||||
|
char buffer[512]={0}, *buf=buf_in;
|
||||||
|
int outlen=512, ret, buflen=inlen, k;
|
||||||
|
result_array_t result_array;
|
||||||
|
|
||||||
|
switch(prot_flag)
|
||||||
|
{
|
||||||
|
case QQ_FILENAME:
|
||||||
|
if(qq_pme->pending_flag==0)
|
||||||
|
break;
|
||||||
|
else
|
||||||
|
{
|
||||||
|
qq_pme->fileinfo.block_seq = -1; //<2F><><EFBFBD>ݿ<EFBFBD><DDBF><EFBFBD><EFBFBD>¼<EFBFBD><C2BC><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
||||||
|
qq_pme->end_flag = 0;
|
||||||
|
qq_pme->pending_flag = 0;
|
||||||
|
qq_pme->fileinfo.file_seq += 1;
|
||||||
|
|
||||||
|
if(qq_pme->qqnum_from == 0)
|
||||||
|
{
|
||||||
|
qq_pme->qqnum_to = qq_pme->qqnum[qq_pme->file_dir];
|
||||||
|
if(qq_pme->qqnum_to == qq_pme->qqnum[0])
|
||||||
|
qq_pme->qqnum_from = qq_pme->qqnum[1];
|
||||||
|
else
|
||||||
|
qq_pme->qqnum_from = qq_pme->qqnum[0];
|
||||||
|
}
|
||||||
|
|
||||||
|
if(g_qq_global_info.debug_sw)
|
||||||
|
{
|
||||||
|
qq_pme->random = rand();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if(!convert_from_unicode2utf8(buf_in, inlen, buffer, &outlen))
|
||||||
|
{
|
||||||
|
snprintf(qq_pme->filename, sizeof(qq_pme->filename), "%s", buffer);
|
||||||
|
qq_pme->filename_len = strlen(qq_pme->filename);
|
||||||
|
MESA_HANDLE_RUNTIME_LOGV2(g_qq_global_info.runtime_log,RLOG_LV_DEBUG, MODULE_NAME, "DETECT_QQ_FILE %s, QQ_FROM: %u, QQ_TO: %u, FILE_LEN: %u, FILENAME: %s", \
|
||||||
|
qq_pme->tcp_flag?"TCP":"UDP", qq_pme->qqnum_from, qq_pme->qqnum_to, qq_pme->total_file_len,qq_pme->filename);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
MESA_HANDLE_RUNTIME_LOGV2(g_qq_global_info.runtime_log,RLOG_LV_INFO, MODULE_NAME, "convert_from_unicode2utf8 error: %s\n", strerror(errno));
|
||||||
|
return APP_STATE_DROPME;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
|
||||||
|
case QQ_FILECONT_END:
|
||||||
|
case QQ_FILECONT:
|
||||||
|
if(qq_pme->filename_len == 0)
|
||||||
|
return APP_STATE_DROPME;
|
||||||
|
|
||||||
|
if(prot_flag==QQ_FILECONT_END)
|
||||||
|
{
|
||||||
|
if(qq_pme->end_flag || qq_pme->pending_flag)
|
||||||
|
break;
|
||||||
|
else
|
||||||
|
{
|
||||||
|
qq_pme->end_flag = 1;
|
||||||
|
qq_pme->pending_flag = 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if(qq_pme->pending_flag)
|
||||||
|
{
|
||||||
|
qq_pme->end_flag = 0;
|
||||||
|
qq_pme->pending_flag = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
if(inlen >= 4 && !memcmp(buf_in, gzip_feature, 4) && qq_pme->compress_flag)
|
||||||
|
{
|
||||||
|
docanalyze_streamparam_t doc_hanlde = docanalyze_startstream(DOC_GZIP_TYPE, g_qq_global_info.instanse, qq_pme->thread_id);
|
||||||
|
if(doc_hanlde ==NULL)
|
||||||
|
{
|
||||||
|
MESA_HANDLE_RUNTIME_LOGV2(g_qq_global_info.runtime_log,RLOG_LV_DEBUG, MODULE_NAME, "docanalyze_startstream() error.");
|
||||||
|
return APP_STATE_GIVEME;
|
||||||
|
}
|
||||||
|
memset(&result_array, 0, sizeof(result_array_t));
|
||||||
|
ret = docanalyze_parsestream(doc_hanlde, buf_in, inlen, &result_array);
|
||||||
|
if(ret == DOC_PRO_OK)
|
||||||
|
{
|
||||||
|
for (k=0; k < result_array.result_num; k++)
|
||||||
|
{
|
||||||
|
buf = result_array.result_buff[k].presult;
|
||||||
|
buflen = result_array.result_buff[k].size;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
qq_pme->curr_file_len += buflen;
|
||||||
|
|
||||||
|
if(g_qq_global_info.debug_sw)
|
||||||
|
{
|
||||||
|
write_file_cont2disk(buf, buflen, qq_pme->filename, qq_pme->fileinfo.file_seq, qq_pme->random);
|
||||||
|
}
|
||||||
|
rec = sendback_qq_data(qq_pme, a_udp, buf, buflen);
|
||||||
|
|
||||||
|
docanalyze_freeresult(&result_array);
|
||||||
|
docanalyze_endstream(doc_hanlde);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
qq_pme->curr_file_len += buflen;
|
||||||
|
|
||||||
|
if(g_qq_global_info.debug_sw)
|
||||||
|
{
|
||||||
|
write_file_cont2disk(buf, buflen, qq_pme->filename, qq_pme->fileinfo.file_seq, qq_pme->random);
|
||||||
|
}
|
||||||
|
rec = sendback_qq_data(qq_pme, a_udp, buf, buflen);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
|
||||||
|
default: break;
|
||||||
|
}
|
||||||
|
|
||||||
|
return rec;
|
||||||
|
}
|
||||||
|
|
||||||
21
src/qq_file_global.h
Normal file
21
src/qq_file_global.h
Normal file
@@ -0,0 +1,21 @@
|
|||||||
|
#ifndef __QQ_FILE_GLOBAL_H__
|
||||||
|
#define __QQ_FILE_GLOBAL_H__
|
||||||
|
|
||||||
|
#define QQ_FILENAME 1
|
||||||
|
#define QQ_FILECONT 2
|
||||||
|
#define QQ_FILECONT_END 4
|
||||||
|
|
||||||
|
#define MAX_MALLOC_SIZE (16*1024)
|
||||||
|
#define REALLOC_BLOCK_SIZE 1024
|
||||||
|
|
||||||
|
#define MESA_HANDLE_RUNTIME_LOGV2(handle, lv, mod, fmt, args...) \
|
||||||
|
MESA_handle_runtime_log((handle), (lv), (mod), "%s:%d, " fmt, \
|
||||||
|
__FILE__, __LINE__, ##args)
|
||||||
|
|
||||||
|
int buf_cache_check_size(trans_file_info_t *pdstBuf, int wantedlen, int thread_seq);
|
||||||
|
int save_initial_tcp_data(trans_file_info_t *pdstBuf, char *data, int datalen, int thread_seq);
|
||||||
|
|
||||||
|
int protocol_process(qq_proto_pme_t *qq_pme, long long prot_flag, char *buf, int buflen, struct streaminfo *a_udp, const void *raw_pkt);
|
||||||
|
|
||||||
|
#endif
|
||||||
|
|
||||||
Reference in New Issue
Block a user