diff --git a/src/gquic_process.cpp b/src/gquic_process.cpp index 630c467..3d1183d 100644 --- a/src/gquic_process.cpp +++ b/src/gquic_process.cpp @@ -1,1400 +1,1399 @@ -/* - * quic_process.c - * - * Created on: 2019年4月2日 - * Author: root - */ - -#include -#include -#include -#include -#include -#include - -#include "gquic_process.h" -#include "quic_analysis.h" -#include "parser_quic.h" - -#ifndef PRINTADDR -#define PRINTADDR(a, b) ((b)addr), a->threadnum) : "") -#endif - -const unsigned char PCAP_FILE_HEAD[24] = {0xD4, 0xC3, 0xB2, 0xA1, 0x02, 0x00, 0x04, 0x00, - 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, - 0xFF, 0xFF, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00}; - -struct pcap_hdr -{ - unsigned int tv_sec; - unsigned int tv_usec; - unsigned int len; - unsigned int caplen; -}; - -int dump_packet(struct streaminfo *pstream) -{ - int ret=0; - char buff[2048]={0}; - char filename[512]={0}; - void *p_eth_rawpkt=NULL; - int eth_rawpkt_len=0; - struct pcap_hdr pcap_hdr; - struct timeval current_time; - - if(g_quic_param.dump_packet_switch==0) - { - return 0; - } - - ret=get_rawpkt_opt_from_streaminfo(pstream, RAW_PKT_GET_DATA, &p_eth_rawpkt); - if(ret==0) - { - ret=get_rawpkt_opt_from_streaminfo(pstream, RAW_PKT_GET_TOT_LEN, ð_rawpkt_len); - if(ret<0) - { - return -1; - } - snprintf(filename, sizeof(filename), "%s-%s.pcap", g_quic_param.log_path, printaddr(&(pstream->addr), pstream->threadnum)); - FILE *fp=fopen(filename, "a+"); - if(fp) - { - gettimeofday(¤t_time, NULL); - pcap_hdr.tv_sec = current_time.tv_sec; - pcap_hdr.tv_usec = current_time.tv_usec; - pcap_hdr.caplen = eth_rawpkt_len; - pcap_hdr.len = pcap_hdr.caplen; - - memcpy(buff, PCAP_FILE_HEAD, 24); - memcpy(buff+24, &pcap_hdr, sizeof(pcap_hdr)); - memcpy(buff+24+sizeof(pcap_hdr), p_eth_rawpkt, eth_rawpkt_len); - fwrite(buff, eth_rawpkt_len+24+sizeof(pcap_hdr), 1, fp); - - fclose(fp); - fp=NULL; - } - } - - return 0; -} - -int is_iquic(enum _QUIC_VERSION quic_version) -{ - switch(quic_version) - { - case IQUIC_VERSION_I001: - case IQUIC_VERSION_I002: - case IQUIC_VERSION_I003: - case IQUIC_VERSION_I004: - case IQUIC_VERSION_I005: - case IQUIC_VERSION_I006: - case IQUIC_VERSION_I007: - case IQUIC_VERSION_I008: - case IQUIC_VERSION_I009: - case IQUIC_VERSION_I010: - case IQUIC_VERSION_I011: - case IQUIC_VERSION_I012: - case IQUIC_VERSION_I013: - case IQUIC_VERSION_I014: - case IQUIC_VERSION_I015: - case IQUIC_VERSION_I016: - case IQUIC_VERSION_I017: - case IQUIC_VERSION_I018: - case IQUIC_VERSION_I019: - case IQUIC_VERSION_I020: - case IQUIC_VERSION_I021: - case IQUIC_VERSION_I022: - case IQUIC_VERSION_I023: - case IQUIC_VERSION_I024: - case IQUIC_VERSION_I025: - case IQUIC_VERSION_I026: - case IQUIC_VERSION_I027: - case IQUIC_VERSION_I028: - case IQUIC_VERSION_I029: - case IQUIC_VERSION_I030: - case IQUIC_VERSION_I031: - case IQUIC_VERSION_I032: - return TRUE; - break; - default: - break; - } - - return FALSE; -} - -int is_quic_port(struct streaminfo *pstream) -{ - switch(pstream->addr.addrtype) - { - case ADDR_TYPE_IPV4: - case __ADDR_TYPE_IP_PAIR_V4: - if(ntohs(pstream->addr.ipv4->source)!=443 && ntohs(pstream->addr.ipv4->dest)!=443) - { - return 0; - } - break; - case ADDR_TYPE_IPV6: - case __ADDR_TYPE_IP_PAIR_V6: - if(ntohs(pstream->addr.ipv6->source)!=443 && ntohs(pstream->addr.ipv6->dest)!=443) - { - return 0; - } - break; - default: - return 0; - break; - } - - return 1; -} - -static int get_value(unsigned char *payload, int *offset, int len) -{ - switch(len) - { - case 1: - return (int)(payload[(*offset)++]); - break; - case 2: - (*offset)+=len; - return (int)ntohs(*(unsigned short *)(payload+*offset-len)); - break; - case 3: - (*offset)+=len; - return ((int)*(payload-2+*offset)<<16| - (int)*(payload-1+*offset)<<8| - (int)*(payload+*offset)<<0); - break; - case 4: - (*offset)+=len; - return (int)ntohl(*(unsigned int *)(payload+*offset-len)); - break; - case 32: - (*offset)+=len; - return 0; - break; - default: - break; - } - - return 0; -} - - -static int check_length(int last_len, int field_len) -{ - return ((last_len-field_len>0) ? 1 : 0); -} - -int quic_getLinkState(struct _quic_context *_context) -{ - UCHAR state = 0; - - if(0==_context->link_state) - { - state=SESSION_STATE_PENDING|SESSION_STATE_DATA; - _context->link_state=1; - } - else - { - state=SESSION_STATE_DATA; - } - - return state; -} - -char quic_callPlugins(struct streaminfo *pstream, struct _quic_context *_context, void *buff, int buff_len, enum quic_interested_region region_mask, void *a_packet) -{ - char state=PROT_STATE_GIVEME; - char app_state=APP_STATE_GIVEME; - stSessionInfo session_info={0}; - - if(region_mask==QUIC_INTEREST_KEY_MASK) - { - session_info.plugid=g_quic_param.quic_plugid; - session_info.prot_flag=0; - session_info.session_state=SESSION_STATE_CLOSE; - session_info.app_info=NULL; - session_info.buf=NULL; - session_info.buflen=0; - } - else - { - session_info.plugid=g_quic_param.quic_plugid; - session_info.prot_flag=(((unsigned long long)1)<quic_info); - session_info.buf=buff; - session_info.buflen=buff_len; - } - state=PROT_PROCESS(&session_info, &(_context->business_pme), pstream->threadnum, pstream, a_packet); - - if(state&PROT_STATE_DROPPKT) - { - app_state=APP_STATE_DROPPKT; - } - - return app_state; -} - - -unsigned long long get_variable_length(char *p, int offset, int v_len) -{ - switch(v_len) - { - case 1: - return (unsigned long long)(p[offset]); - break; - case 2: - return (unsigned long long)ntohs(*(unsigned short *)((char *)p+offset)); - break; - case 3: - return (unsigned long long)*(p+0+offset)<<16| - (unsigned long long)*(p+1+offset)<<8| - (unsigned long long)*(p+2+offset)<<0; - break; - case 4: - return (unsigned long long)ntohl(*(unsigned int *)(p+offset)); - break; - case 5: - return (unsigned long long)*((unsigned char *)(p)+0+offset)<<32| - (unsigned long long)*((unsigned char *)(p)+1+offset)<<24| - (unsigned long long)*((unsigned char *)(p)+2+offset)<<16| - (unsigned long long)*((unsigned char *)(p)+3+offset)<<8| - (unsigned long long)*((unsigned char *)(p)+4+offset)<<0; - break; - case 6: - return (unsigned long long)*((unsigned char *)(p)+0+offset)<<40| - (unsigned long long)*((unsigned char *)(p)+1+offset)<<32| - (unsigned long long)*((unsigned char *)(p)+2+offset)<<24| - (unsigned long long)*((unsigned char *)(p)+3+offset)<<16| - (unsigned long long)*((unsigned char *)(p)+4+offset)<<8| - (unsigned long long)*((unsigned char *)(p)+5+offset)<<0; - break; - case 7: - return (unsigned long long)*((unsigned char *)(p)+0+offset)<<56| - (unsigned long long)*((unsigned char *)(p)+1+offset)<<40| - (unsigned long long)*((unsigned char *)(p)+2+offset)<<32| - (unsigned long long)*((unsigned char *)(p)+3+offset)<<24| - (unsigned long long)*((unsigned char *)(p)+4+offset)<<16| - (unsigned long long)*((unsigned char *)(p)+5+offset)<<8| - (unsigned long long)*((unsigned char *)(p)+6+offset)<<0; - break; - case 8: - return (unsigned long long)*((unsigned char *)(p)+0+offset)<<56| - (unsigned long long)*((unsigned char *)(p)+1+offset)<<48| - (unsigned long long)*((unsigned char *)(p)+2+offset)<<40| - (unsigned long long)*((unsigned char *)(p)+3+offset)<<32| - (unsigned long long)*((unsigned char *)(p)+4+offset)<<24| - (unsigned long long)*((unsigned char *)(p)+5+offset)<<16| - (unsigned long long)*((unsigned char *)(p)+6+offset)<<8| - (unsigned long long)*((unsigned char *)(p)+7+offset)<<0; - break; - default: - break; - } - - return 0; -} - -long long bit_to_value(char *payload, int payload_len, unsigned char flags, unsigned long long *out_value, int *used_len) -{ - switch(flags&0x3) // packet number - { - case 0x3: // 6 bytes - if(!check_length(payload_len-*used_len, 6)) - { - return -1; - } - *out_value=get_variable_length(payload, *used_len, 6); - *used_len+=6; - break; - case 0x2: // 4 bytes - if(!check_length(payload_len-*used_len, sizeof(unsigned int))) - { - return -1; - } - *out_value=(unsigned long long)ntohl(*(unsigned int *)(payload+*used_len)); - *used_len+=4; - break; - case 0x1: // 2bytes - if(!check_length(payload_len-*used_len, sizeof(unsigned short))) - { - return -1; - } - *out_value=(unsigned long long)ntohs(*(unsigned short *)(payload+*used_len)); - *used_len+=2; - break; - default: // 1 byte - if(!check_length(payload_len-*used_len, sizeof(unsigned char))) - { - return -1; - } - *out_value=payload[*used_len]; - *used_len+=1; - break; - } - - return 0; -} - -int get_quic_tlv(char *start_pos, quic_tlv_t *tlv, int len, int type, int thread_seq) -{ - if(tlv->value==NULL && len>0) - { - tlv->value=(char *)dictator_malloc(thread_seq, len+1); - memset(tlv->value, 0, len+1); - tlv->length=len; - tlv->type=type; - memcpy(tlv->value, start_pos, tlv->length); - } - - return 0; -} - -int get_stream_id(struct streaminfo *pstream, struct _quic_context* _context, char* payload, int payload_len, unsigned char frame_type, int *used_len) -{ - int stream_len=0,offset_len=0; - - _context->quic_info.frame_hdr.frame_type=frame_type; - - stream_len=(frame_type&GQUIC_SPECIAL_FRAME_STREAM_ID)+1; - if(!check_length(payload_len-*used_len, stream_len)) - { - return -1; - } - - _context->quic_info.frame_hdr.stream_id=(unsigned int)get_variable_length(payload, *used_len, stream_len); - *used_len+=stream_len; // stream ID length - - if(frame_type&GQUIC_SPECIAL_FRAME_STREAM_DLEN) - { - if(!check_length(payload_len-*used_len, 2)) - { - return -1; - } - _context->quic_info.frame_hdr.data_len=ntohs(*(unsigned short *)(payload+*used_len)); - *used_len+=2; //data length - } - - offset_len=(frame_type&GQUIC_SPECIAL_FRAME_STREAM_OFFSET) ? (((frame_type&GQUIC_SPECIAL_FRAME_STREAM_OFFSET))>>2)+1 : 0; - if(!check_length(payload_len-*used_len, offset_len)) - { - return -1; - } - - _context->quic_info.frame_hdr.offset=get_variable_length(payload, *used_len, offset_len); - *used_len+=offset_len; - - MESA_handle_runtime_log(g_quic_param.logger, - RLOG_LV_DEBUG, - "QUIC_FRAME", - "frame_type: 0X%02X stream_id: %u data length: %u offset length: %u addr: %s", - frame_type, - _context->quic_info.frame_hdr.stream_id, - _context->quic_info.frame_hdr.data_len, - offset_len, - PRINTADDR(pstream, g_quic_param.level) - ); - - return _context->quic_info.frame_hdr.stream_id; -} - -unsigned long long get_packet_number(char* data, int offset, char pkn_len) -{ - switch(pkn_len) - { - case 1: - return (unsigned long long)data[offset]; - break; - case 2: - return (unsigned long long)ntohs(*(unsigned short *)(data+offset)); - break; - case 4: - return (unsigned long long)ntohl(*(unsigned int *)(data+offset)); - break; - case 8: - return get_variable_length(data, offset, 8);; - break; - } - - return 0; -} - -// GQUIC version from 0 to 43 -static enum _QUIC_VERSION parse_q0to43_header(struct streaminfo *pstream, struct _quic_context* _context, char *payload, int payload_len, int *used_len) -{ - int ret=0; - char public_flags=0; - - struct _quic_public_header *gquic_hdr=&(_context->quic_info.quic_hdr); - - public_flags=payload[*used_len]; - *used_len+=1; - gquic_hdr->public_flags=public_flags; - - if((public_flags&GQUIC_PUBLIC_FLAG_RST) && _context->is_quic==TRUE) - { - gquic_hdr->is_reset=TRUE; //Public Reset Packet - return QUIC_VERSION_UNKNOWN; - - } - - if(pstream->curdir==DIR_S2C && gquic_hdr->public_flags&GQUIC_PUBLIC_FLAG_VERSION) - { - return QUIC_VERSION_UNKNOWN; - } - - //For Public Reset and Version Negotiation Packets (sent by the server) which don't have a packet number - if(!public_flags&GQUIC_PUBLIC_FLAG_PKT_NUM) - { - if(public_flags&GQUIC_PUBLIC_FLAG_VERSION) //Public Reset Packet - { - return QUIC_VERSION_UNKNOWN;// todo - } - else // Version Negotiation Packet - { - return QUIC_VERSION_UNKNOWN; - } - } - - if(gquic_hdr->public_flags&GQUIC_PUBLIC_FLAG_CID) - { - *(unsigned long long *)gquic_hdr->server_CID=get_variable_length(payload, *used_len, sizeof(gquic_hdr->server_CID)); - *used_len+=sizeof(unsigned long long); // CID length - - _context->is_quic=TRUE; - } - - if(gquic_hdr->public_flags&GQUIC_PUBLIC_FLAG_VERSION && (*(unsigned char *)(payload+*used_len)==0x51)) - { - gquic_hdr->quic_version=(unsigned int)ntohl(*(unsigned int *)(payload+*used_len)); - *used_len+=sizeof(int); // skip version - - _context->is_quic=TRUE; - } - - if(_context->is_quic==FALSE || gquic_hdr->quic_versionquic_version>GQUIC_VERSION_Q043) - { - _context->is_quic=FALSE; - return QUIC_VERSION_UNKNOWN; - } - - ret=bit_to_value(payload, payload_len, gquic_hdr->public_flags>>4, &gquic_hdr->packet_number, used_len); - if(ret<0) - { - return (enum _QUIC_VERSION)gquic_hdr->quic_version; - } - - if(gquic_hdr->public_flags==GQUIC_PUBLIC_FLAG_NONCE) - { - *used_len+=32; //diversification nonce - } - - // Version 11 reduced the length of null encryption authentication tag from 16 to 12 bytes - if(gquic_hdr->quic_version > GQUIC_VERSION_Q010) - { - *used_len+=12; - } - else - { - *used_len+=16; - } - - // Version 34 removed entropy bits from packets and ACK frames, - // removed private flag from packet header and changed the ACK format to specify ranges of packets acknowledged rather than missing ranges. - if(gquic_hdr->quic_version < GQUIC_VERSION_Q034) - { - *used_len+=1; //private flags - } - - _context->is_quic=TRUE; - - MESA_handle_runtime_log(g_quic_param.logger, - RLOG_LV_DEBUG, - "QUIC_IDETIFY", - "pub_flags: 0X%02X conection ID:[ destination: %llu ] version: Q%03u packet number: %llu dir(1: C2S;2: S2C): %d addr: %s", - gquic_hdr->public_flags, - *(unsigned long long *)gquic_hdr->server_CID, - (((gquic_hdr->quic_version>>8)&0x0000000F)*10) + ((gquic_hdr->quic_version)&0x0000000F), - gquic_hdr->packet_number, - pstream->curdir, - PRINTADDR(pstream, g_quic_param.level) - ); - - return (enum _QUIC_VERSION)gquic_hdr->quic_version; -} - -enum _QUIC_VERSION parse_quic_header(struct streaminfo *pstream, struct _quic_context* _context, char *payload, int payload_len, int *used_len) -{ - int i=0,len=0; - char client_CID[MAX_CONNECT_ID_LEN*2]={0}; - char server_CID[MAX_CONNECT_ID_LEN*2]={0}; - - struct _quic_public_header *long_hdr=&(_context->quic_info.quic_hdr); - - long_hdr->public_flags=payload[*used_len]; - *used_len+=1; //skip public flags - - if(long_hdr->public_flags&0x80) - { - long_hdr->quic_version=(unsigned int)ntohl(*(unsigned int *)(payload+*used_len)); - *used_len+=sizeof(int); // skip version - - long_hdr->client_CID_len=(payload[*used_len]&0xF) ? (payload[*used_len]&0xF)+3 : 0; - long_hdr->server_CID_len=((payload[*used_len]>>4)&0xF) ? ((payload[*used_len]>>4)&0xF)+3 : 0; - *used_len+=sizeof(char); // both connection_id length - - memcpy(long_hdr->server_CID, (void *)(payload+*used_len), long_hdr->server_CID_len); - *used_len+=long_hdr->server_CID_len; // Destination connection_id length - memcpy(long_hdr->client_CID, (void *)(payload+*used_len), long_hdr->client_CID_len); - *used_len+=long_hdr->client_CID_len; // source connection_id length - } - else - { - if(pstream->curdir==DIR_C2S)// short header only destination connection ID - { - *used_len+=long_hdr->server_CID_len; // every packet destination connection ID is same - } - } - - len=(long_hdr->public_flags&0x03)+1; - long_hdr->packet_number=get_packet_number(payload, *used_len, len); - *used_len+=len; - - *used_len+=12; //message authentication hash - - _context->is_quic=TRUE; - - for(i=0,len=0;iserver_CID_len; i++) - { - len+=snprintf(server_CID+len, sizeof(server_CID)-len, "%02X", long_hdr->server_CID[i]); - } - - for(i=0,len=0;iclient_CID_len; i++) - { - len+=snprintf(client_CID+len, sizeof(client_CID)-len, "%02X", long_hdr->client_CID[i]); - } - - MESA_handle_runtime_log(g_quic_param.logger, - RLOG_LV_DEBUG, - "QUIC_IDETIFY", - "pub_flags: 0X%02X conection ID:[ destination: %s source: %s ] version: Q%03u packet number: %llu dir(1: C2S;2: S2C): %d addr: %s", - long_hdr->public_flags, - server_CID, - client_CID, - (((long_hdr->quic_version>>8)&0x0000000F)*10) + ((long_hdr->quic_version)&0x0000000F), - long_hdr->packet_number, - pstream->curdir, - PRINTADDR(pstream, g_quic_param.level) - ); - - return (enum _QUIC_VERSION)long_hdr->quic_version; -} - -enum _QUIC_VERSION is_quic_protocol(struct streaminfo *pstream, struct _quic_context* _context, char *payload, int payload_len, int *used_len) -{ - enum _QUIC_VERSION quic_version=QUIC_VERSION_UNKNOWN; - - if(_context->quic_info.quic_hdr.quic_version!=QUIC_VERSION_UNKNOWN) - { - if(is_iquic((enum _QUIC_VERSION)(_context->quic_info.quic_hdr.quic_version))) - { - return (enum _QUIC_VERSION)(_context->quic_info.quic_hdr.quic_version); - } - - quic_version=(enum _QUIC_VERSION)(_context->quic_info.quic_hdr.quic_version); - } - else - { - // The most significant bit (0x80) of byte 0 (the first byte) is set to 1 for long headers - (payload[*used_len]&0x80) ? (quic_version=(enum _QUIC_VERSION)ntohl(*(unsigned int *)(payload+(*used_len+1)))) : QUIC_VERSION_UNKNOWN; - } - - switch(quic_version) // +1 meaning: skip public flags - { - case GQUIC_VERSION_Q044: - case GQUIC_VERSION_Q045: - case GQUIC_VERSION_Q046: - case GQUIC_VERSION_Q047: - case GQUIC_VERSION_Q048: - quic_version=parse_quic_header(pstream, _context, payload, payload_len, used_len); - return quic_version; - break; - default: - if( - (quic_version==GQUIC_VERSION_Q099) || - (quic_version==PICOQUIC_VERSION_30) || - (quic_version==PQUIC_VERSION_PROX) || - (quic_version==GQUIC_VERSION_T099) || - (quic_version>=GQUIC_VERSION_Q049 && quic_version<=GQUIC_VERSION_Q050) || - (quic_version>=GQUIC_VERSION_Q051 && quic_version<=GQUIC_VERSION_Q059) || - (quic_version>=GQUIC_VERSION_T048 && quic_version<=GQUIC_VERSION_T049) || - (quic_version>=GQUIC_VERSION_T050 && quic_version<=GQUIC_VERSION_T059) || - (quic_version>=QUANT_VERSION_00 && quic_version<=QUANT_VERSION_FF) || - (quic_version>=QUIC_GO_VERSION_00 && quic_version<=QUIC_GO_VERSION_FF) || - (quic_version>=QUICLY_VERSION_00 && quic_version<=QUICLY_VERSION_FF) || - (quic_version>=MSQUIC_VERSION_00 && quic_version<=MSQUIC_VERSION_0F) || - (quic_version>=MOZQUIC_VERSION_00 && quic_version<=MOZQUIC_VERSION_0F) || - (quic_version>=MVFST_VERSION_00 && quic_version<=MVFST_VERSION_0F) || - (quic_version>=IQUIC_VERSION_I001 && quic_version<=IQUIC_VERSION_I032) || - (quic_version==IQUIC_VERSION_RFC9000) - ) - { - MESA_handle_runtime_log(g_quic_param.logger, - RLOG_LV_DEBUG, - "QUIC", - "version: 0x%x addr: %s", - quic_version, PRINTADDR(pstream, g_quic_param.level) - ); - - _context->is_quic=TRUE; - _context->quic_info.quic_hdr.quic_version=quic_version; - return quic_version; - } - break; - } - - // Q001~Q043: 0x80 is currently unused, and must be set to 0 - if(payload[*used_len]>0x80) - { - return QUIC_VERSION_UNKNOWN; - } - - return parse_q0to43_header(pstream, _context, payload, payload_len, used_len); -} - -int parse_extension_tag(struct streaminfo *pstream, struct _quic_stream **quic_stream, void *a_packet, char *payload, int payload_len, int *used_len, int tag_num) -{ - int tag_used_num=0; - int tag_type=0; - int total_tag_len=0,tag_len=0; - int tag_offset_end=0,pre_tag_offset_end=0; - - if(tag_num>64 || tag_num<0) - { - (*used_len)=payload_len; - - dump_packet(pstream); - MESA_handle_runtime_log(g_quic_param.logger, RLOG_LV_FATAL, "QUIC_TAG_NUM", "QUIC_TAG_NUM:%d addr: %s", tag_num, printaddr(&pstream->addr, pstream->threadnum)); - return -1; - } - - struct _quic_stream *stream=*quic_stream; - int tag_value_start=tag_num*4*2+(*used_len); // skip length of type and offset, type(offset)=szieof(int) - - if(stream==NULL) - { - stream=(struct _quic_stream *)dictator_malloc(pstream->threadnum, sizeof(struct _quic_stream)); - memset(stream, 0, sizeof(struct _quic_stream)); - stream->ext_tags=(quic_tlv_t *)dictator_malloc(pstream->threadnum, tag_num*sizeof(quic_tlv_t)); - memset(stream->ext_tags, 0, tag_num*sizeof(quic_tlv_t)); - *quic_stream=stream; - stream->sni_idx=0xFF; - stream->ver_idx=0xFF; - stream->ua_idx=0xFF; - } - else - { - quic_release_exts(pstream->threadnum, stream->ext_tags, stream->ext_tag_num); - stream->ext_tags=(quic_tlv_t *)dictator_malloc(pstream->threadnum, tag_num*sizeof(quic_tlv_t)); - memset(stream->ext_tags, 0, tag_num*sizeof(quic_tlv_t)); - *quic_stream=stream; - stream->ext_tag_num=0; - stream->count++; - stream->sni_idx=0xFF; - stream->ver_idx=0xFF; - stream->ua_idx=0xFF; - } - - while(tag_num>tag_used_num) - { - tag_type=ntohl(*(unsigned int *)(payload+*used_len)); - *used_len+=sizeof(int); - - tag_offset_end=*(unsigned int *)(payload+*used_len); - *used_len+=sizeof(int); - - tag_len=tag_offset_end-pre_tag_offset_end; - if(tag_len<0 || (tag_offset_end>=payload_len) || (tag_len>payload_len-tag_value_start)) - { - return -1; - } - - switch(tag_type) - { - case TAG_PAD: - break; - case TAG_VER: - stream->ver_idx=stream->ext_tag_num; - get_quic_tlv(payload+tag_value_start, &stream->ext_tags[stream->ext_tag_num], tag_len, tag_type, pstream->threadnum); - *(unsigned int *)(stream->ext_tags[stream->ext_tag_num].value)=(unsigned int)ntohl(*(unsigned int *)(stream->ext_tags[stream->ext_tag_num].value)); - MESA_handle_runtime_log(g_quic_param.logger, - RLOG_LV_DEBUG, - "QUIC_VERSION", - "Quic version: 0X%X addr: %s", - *(unsigned int *)(stream->ext_tags[stream->ext_tag_num].value), - PRINTADDR(pstream, g_quic_param.level) - ); - break; - case TAG_UAID: - stream->ua_idx=stream->ext_tag_num; - get_quic_tlv(payload+tag_value_start, &stream->ext_tags[stream->ext_tag_num], tag_len, tag_type, pstream->threadnum); - MESA_handle_runtime_log(g_quic_param.logger, - RLOG_LV_DEBUG, - "QUIC_UA", - "User Agent: %s addr: %s", - stream->ext_tags[stream->ext_tag_num].value, - PRINTADDR(pstream, g_quic_param.level) - ); - stream->ext_tag_num++; - break; - case TAG_SNI: - stream->sni_idx=stream->ext_tag_num; - get_quic_tlv(payload+tag_value_start, &stream->ext_tags[stream->ext_tag_num], tag_len, tag_type, pstream->threadnum); - MESA_handle_runtime_log(g_quic_param.logger, - RLOG_LV_DEBUG, - "QUIC_SNI", - "SNI: %s addr: %s", - stream->ext_tags[stream->ext_tag_num].value, - PRINTADDR(pstream, g_quic_param.level) - ); - stream->ext_tag_num++; - break; - default: - get_quic_tlv(payload+tag_value_start, &stream->ext_tags[stream->ext_tag_num], tag_len, tag_type, pstream->threadnum); - stream->ext_tag_num++; - break; - } - - tag_used_num++; - tag_value_start+=tag_len; - total_tag_len+=tag_len; - pre_tag_offset_end=tag_offset_end; - } - - *used_len += total_tag_len; - - return 0; -} - -int gquic_frame_type_stream(struct streaminfo *pstream, struct _quic_context* _context, char *payload, int payload_len, int *used_len, void *a_packet) -{ - int ret=0; - char state=APP_STATE_GIVEME; - int tag_num = 0; - unsigned int message_tag; - - if(!check_length(payload_len-*used_len, 8)) - { - return state; - } - - switch(_context->quic_info.quic_hdr.quic_version) - { - case GQUIC_VERSION_Q041: - *used_len+=1; // unknown - case GQUIC_VERSION_Q044: - message_tag=(unsigned int)ntohl(*(unsigned int *)(payload+*used_len)); - *used_len+=4; - - tag_num=*(int *)(payload+*used_len); - *used_len+=4; //tag_num - break; - default: - message_tag=(unsigned int)ntohl(*(unsigned int *)(payload+*used_len)); - *used_len+=4; - - tag_num=*(unsigned short *)(payload+*used_len); - *used_len+=2; //tag_num - *used_len+=2; //padding - break; - } - - switch(message_tag) - { - case CHLO: //MTAG_CHLO; - ret=parse_extension_tag(pstream, &(_context->quic_info.client_hello), a_packet, payload, payload_len, used_len, tag_num); - if(ret>=0 && _context->call_business) - { - state=quic_callPlugins(pstream, _context, (void *)(_context->quic_info.client_hello), sizeof(void *), QUIC_CLIENT_HELLO_MASK, a_packet); - } - break; - case SHLO: //MTAG_SHLO; - ret=parse_extension_tag(pstream, &(_context->quic_info.server_hello), a_packet, payload, payload_len, used_len, tag_num); - if(ret>=0 && _context->call_business) - { - state=quic_callPlugins(pstream, _context, (void *)(_context->quic_info.server_hello), sizeof(void *), QUIC_SERVER_HELLO_MASK, a_packet); - } - break; - case REJ: //MTAG_REJ; - ret=parse_extension_tag(pstream, &(_context->quic_info.rejection), a_packet, payload, payload_len, used_len, tag_num); - if(ret>=0 && _context->call_business) - { - state=quic_callPlugins(pstream, _context, (void *)(_context->quic_info.rejection), sizeof(void *), QUIC_REJECTION_MASK, a_packet); - } - break; - default: - break; - } - - return state; -} - -//frame type->stream->offset->data length -int gquic_proc_unencrypt(struct streaminfo *pstream, struct _quic_context* _context, void *a_packet, char * payload, int payload_len, int *used_len) -{ - unsigned int ret=0; - unsigned char frame_type=0; - unsigned int stream_id=0; - unsigned int error_code=0; - unsigned short reason_phrase_length=0; - unsigned long long byte_offset=0; - unsigned long long least_unacked_delta=0; - - while(*used_lenaddr, pstream->threadnum)); - - return quic_callPlugins(pstream, _context, NULL, 0, QUIC_INTEREST_KEY_MASK, a_packet); - break; - case GQUIC_REGULAR_FRAME_CONNECTION_CLOSE: - if(!check_length(payload_len-*used_len, sizeof(unsigned int)+sizeof(unsigned short))) - { - return APP_STATE_GIVEME; - } - error_code=(unsigned int)get_variable_length(payload, *used_len, sizeof(unsigned int)); - *used_len+=sizeof(unsigned int); - - reason_phrase_length=(unsigned short)get_variable_length(payload, *used_len, sizeof(unsigned short)); - *used_len+=sizeof(unsigned short); - - if(!check_length(payload_len-*used_len, reason_phrase_length)) - { - return APP_STATE_GIVEME; - } - - *used_len+=reason_phrase_length; // skip Reason Phrase - - MESA_handle_runtime_log(g_quic_param.logger, - RLOG_LV_DEBUG, - "QUIC_CONNECTION_CLOSE", - "error_code: %u reason_phrase_length: %d addr: %s", - error_code, - reason_phrase_length, - printaddr(&pstream->addr, pstream->threadnum)); - - return quic_callPlugins(pstream, _context, NULL, 0, QUIC_INTEREST_KEY_MASK, a_packet); - break; - case GQUIC_REGULAR_FRAME_GOAWAY: - if(!check_length(payload_len-*used_len, sizeof(unsigned int)+sizeof(unsigned int)+sizeof(unsigned short))) - { - return APP_STATE_GIVEME; - } - error_code=(unsigned int)get_variable_length(payload, *used_len, sizeof(unsigned int)); - *used_len+=sizeof(unsigned int); - - //Last Good Stream ID - stream_id=(unsigned int)get_variable_length(payload, *used_len, sizeof(unsigned int)); - *used_len+=sizeof(unsigned int); - - reason_phrase_length=(unsigned short)get_variable_length(payload, *used_len, sizeof(unsigned short)); - *used_len+=sizeof(unsigned short); - - if(!check_length(payload_len-*used_len, reason_phrase_length)) - { - return APP_STATE_GIVEME; - } - *used_len+=reason_phrase_length; // skip Reason Phrase - - MESA_handle_runtime_log(g_quic_param.logger, - RLOG_LV_DEBUG, - "QUIC_GOAWAY", - "error_code: %u Last Good Stream ID: %u reason_phrase_length: %d addr: %s", - error_code, - stream_id, - reason_phrase_length, - printaddr(&pstream->addr, pstream->threadnum)); - break; - case GQUIC_REGULAR_FRAME_WINDOW_UPDATE: - if(!check_length(payload_len-*used_len, sizeof(unsigned int)+sizeof(unsigned long long))) - { - return APP_STATE_GIVEME; - } - stream_id=(unsigned int)get_variable_length(payload, *used_len, sizeof(unsigned int)); - *used_len+=sizeof(unsigned int); - - byte_offset=get_variable_length(payload, *used_len, sizeof(unsigned long long)); - *used_len+=sizeof(unsigned long long); - - MESA_handle_runtime_log(g_quic_param.logger, - RLOG_LV_DEBUG, - "QUIC_WINDOW_UPDATE", - "stream_id: %u byte_offset: %llu addr: %s", - stream_id, - byte_offset, - printaddr(&pstream->addr, pstream->threadnum)); - break; - case GQUIC_REGULAR_FRAME_BLOCKED: - if(!check_length(payload_len-*used_len, sizeof(unsigned int))) - { - return APP_STATE_GIVEME; - } - stream_id=(unsigned int)get_variable_length(payload, *used_len, sizeof(unsigned int)); - *used_len+=sizeof(unsigned int); - - MESA_handle_runtime_log(g_quic_param.logger, - RLOG_LV_DEBUG, - "QUIC_STOP_WAITING", - "stream_id: %u addr: %s", - stream_id, - printaddr(&pstream->addr, pstream->threadnum)); - break; - case GQUIC_REGULAR_FRAME_STOP_WAITING: - ret=bit_to_value(payload, payload_len, _context->quic_info.quic_hdr.public_flags>>4, &least_unacked_delta, used_len); - if(ret<0) - { - return APP_STATE_GIVEME; - } - - MESA_handle_runtime_log(g_quic_param.logger, - RLOG_LV_DEBUG, - "QUIC_STOP_WAITING", - "least_unacked_delta: %llu addr: %s", - least_unacked_delta, - printaddr(&pstream->addr, pstream->threadnum)); - break; - case GQUIC_REGULAR_FRAME_PING: - //The PING frame contains no payload. - //The receiver of a PING frame simply needs to ACK the packet containing this frame - break; - default: //Regular Frame Types - if(frame_type&GQUIC_SPECIAL_FRAME_STREAM || (frame_type&0xC0)==GQUIC_SPECIAL_FRAME_ACK) - { - stream_id=get_stream_id(pstream, _context, payload, payload_len, frame_type, used_len); - if(stream_id<0) - { - return APP_STATE_GIVEME; - } - ret=gquic_frame_type_stream(pstream, _context, payload, payload_len, used_len, a_packet); - } - else if((frame_type&0xE0)==GQUIC_SPECIAL_FRAME_CONGEST_FB) // high two bits set 0; (frame_type: 01nullmmB) - { - //not used - } - else - { - return APP_STATE_GIVEME; - } - break; - } - - if(ret&APP_STATE_DROPME || ret&APP_STATE_DROPPKT) - { - return ret; - } - } - - return APP_STATE_GIVEME; -} - - -//QUIC_DATA:is quic data pcap;QUIC_TRUE:is handshake pcap;QUIC_RETURN_DROPME:not quic protocol; -int parse_gquic_Q046(struct streaminfo *pstream, struct _quic_context* _context, void *a_packet, char * payload, int payload_len, int *used_len) -{ - int stream_id=0; - int ret=APP_STATE_GIVEME; - unsigned char frame_type; - - while(*used_len < payload_len) - { - frame_type=payload[*used_len]; - *used_len+=1; //skip frame_type - - if(frame_type&IQUIC_FRAME_STREAM_HEX08) //0x08=Q048 - { - stream_id=get_stream_id(pstream, _context, payload, payload_len, frame_type, used_len); - if(stream_id<0) - { - return APP_STATE_GIVEME; - } - ret=gquic_frame_type_stream(pstream, _context, payload, payload_len, used_len, a_packet); - } - else - { - return APP_STATE_GIVEME; //todo - } - - if(ret&APP_STATE_DROPME || ret&APP_STATE_DROPPKT) - { - return ret; - } - } - - return APP_STATE_GIVEME; -} - -int parse_encrypt_parameter(struct _quic_stream *quic_stream, unsigned char *payload, int payload_len, int thread_seq) -{ - int used_len=0,length=0; - while(payload_len>used_len) - { - if(payload[used_len]> 0x00 && payload[used_len]<=0x20) - { - get_value(payload, &used_len, 1); //type=1 - length=get_value(payload, &used_len, 1); // length=1 - used_len+=length; - - continue; - } - - if((*(unsigned short *)(payload+used_len)) == htons(EXT_QUIC_PARAM_USER_AGENT)) - { - quic_stream->ua_idx=quic_stream->ext_tag_num++; - get_value(payload, &used_len, 2); //type=2 - length=get_value(payload, &used_len, 1); // length=1 - get_quic_tlv((char *)payload+used_len, &(quic_stream->ext_tags[quic_stream->ua_idx]), length, EXT_QUIC_PARAM_USER_AGENT, thread_seq); - used_len+=length; - - continue; - } - - if(*(unsigned int *)(payload+used_len) == htonl(EXT_QUIC_PARAM_QUIC_VERSION)) - { - quic_stream->ver_idx=quic_stream->ext_tag_num++; - get_value(payload, &used_len, 4); //type=4 - length=get_value(payload, &used_len, 1); // length=1 - get_quic_tlv((char *)payload+used_len, &(quic_stream->ext_tags[quic_stream->ver_idx]), length, EXT_QUIC_PARAM_QUIC_VERSION, thread_seq); - *(unsigned int *)quic_stream->ext_tags[quic_stream->ver_idx].value=(unsigned int)htonl(*(unsigned int *)quic_stream->ext_tags[quic_stream->ver_idx].value); - used_len+=length; - - continue; - } - - if((*(unsigned int *)(payload+used_len))== htonl(EXT_QUIC_PARAM_GREASE_HIGH4) && (*(unsigned int *)(payload+used_len+4))== htonl(EXT_QUIC_PARAM_GREASE_LOW4)) - { - used_len+=8; //type=8 - length=get_value(payload, &used_len, 1); // length=1 - used_len+=length; - - continue; - } - - break; - } - - return 0; -} -int parse_encrypt_server_name(struct _quic_stream *quic_stream, unsigned char *payload, int payload_len, int thread_seq) -{ - int ext_len=0,used_len=0; - - quic_stream->sni_idx=quic_stream->ext_tag_num++; - get_value(payload, &used_len, 2); //Server Name List length - if(get_value(payload, &used_len, 1)==0) //Server Name type - { - ext_len=get_value(payload, &used_len, 2); //Server Name length - get_quic_tlv((char *)payload+used_len, &(quic_stream->ext_tags[quic_stream->sni_idx]), ext_len, EXTENSION_SERVER_NAME, thread_seq); - } - - return 1; -} - -int parse_encrypt_client_hello(struct streaminfo *pstream, struct _quic_stream *quic_stream, void *a_packet, unsigned char *payload, int payload_len) -{ - int skip_len=0; - int used_len=0; - int flags=0; - int ext_type=0, extension_total_len=0; - - get_value(payload, &used_len, 1); //handshake type - get_value(payload, &used_len, 3); //client hello length - get_value(payload, &used_len, 2); //ssl_version - - get_value(payload, &used_len, 32); //Random - - skip_len=(int)get_value(payload, &used_len, 1); //Session ID length - if(!check_length(payload_len-used_len, skip_len)) - { - return flags; - } - used_len+=skip_len; - - skip_len=(int)get_value(payload, &used_len, 2); //Ciper Suites length - if(!check_length(payload_len-used_len, skip_len)) - { - return flags; - } - used_len+=skip_len; - - skip_len=(int)get_value(payload, &used_len, 1); //Compression Methods - if(!check_length(payload_len-used_len, skip_len)) - { - return flags; - } - used_len+=skip_len; - - extension_total_len=(int)get_value(payload, &used_len, 2); //Extension length - if(!check_length(payload_len-used_len, extension_total_len)) - { - return flags; - } - - quic_stream->ext_tags=(quic_tlv_t *)dictator_malloc(pstream->threadnum, sizeof(quic_tlv_t)*3); - memset(quic_stream->ext_tags, 0, sizeof(quic_tlv_t)*3); - quic_stream->sni_idx=0xFF; - quic_stream->ua_idx=0xFF; - quic_stream->ver_idx=0xFF; - - while(extension_total_len>used_len) - { - ext_type=get_value(payload, &used_len, 2); //Extension type - skip_len=get_value(payload, &used_len, 2); //length - if(!check_length(payload_len-used_len, skip_len)) - { - return flags; - } - - switch(ext_type) - { - case EXTENSION_SERVER_NAME: - parse_encrypt_server_name(quic_stream, payload+used_len, skip_len, pstream->threadnum); - flags=1; - break; - case EXTENSION_QUIC_PARAM: - parse_encrypt_parameter(quic_stream, payload+used_len, skip_len, pstream->threadnum); - break; - case EXTENSION_SUPPORT_GROUP: - case EXTENSION_APP_PROT_NEGO: - case EXTENSION_SIG_ALGORITHM: - case EXTENSION_KEY_SHARE: - case EXTENSION_PSK_EXCHANGE: - case EXTENSION_SUPP_SSL_VER: - case EXTENSION_COMPRESS_CERT: - break; - default: - break; - } - - used_len+=skip_len; - } - - return flags; -} - -int parse_decrypt_quic(struct streaminfo *pstream, struct _quic_context* _context, void *a_packet, unsigned char * payload, int payload_len, int *used_len) -{ - int ret=0,state=APP_STATE_GIVEME; - unsigned int quic_version=_context->quic_info.quic_hdr.quic_version; - - get_value(payload, used_len, 4); //Frame Type=1, offset=1, length=2 - - if( (quic_version>=MVFST_VERSION_00 && quic_version<=MVFST_VERSION_0F) || - (quic_version>=GQUIC_VERSION_T050 && quic_version<=GQUIC_VERSION_T059) || - (quic_version>=IQUIC_VERSION_I022 && quic_version<=IQUIC_VERSION_I029) || - (quic_version==IQUIC_VERSION_RFC9000) - ) - { - if(payload[*used_len] == 0x01) - { - if(_context->quic_info.client_hello==NULL) - { - _context->quic_info.client_hello=(struct _quic_stream *)dictator_malloc(pstream->threadnum, sizeof(struct _quic_stream)); - memset(_context->quic_info.client_hello, 0, sizeof(struct _quic_stream)); - } - ret=parse_encrypt_client_hello(pstream, _context->quic_info.client_hello, a_packet, payload+*used_len, payload_len-*used_len); //Frame Type=1, offset=1, length=2 - if(ret>0 && _context->call_business) - { - state=quic_callPlugins(pstream, _context, (void *)(_context->quic_info.client_hello), sizeof(void *), QUIC_CLIENT_HELLO_MASK, a_packet); - } - } - } - else if( (quic_version>=GQUIC_VERSION_Q047 && quic_version<=GQUIC_VERSION_Q059)) - { - state=gquic_frame_type_stream(pstream, _context, (char *)payload, payload_len, used_len, a_packet); - } - else - { - state=APP_STATE_DROPME; - } - - return state; -} - -//cid->version->nounce->pkt num->ahn hash(12) -int quic_process(struct streaminfo *pstream, struct _quic_context* _context, int thread_seq, void* a_packet) -{ - int used_len=0; - int ret=APP_STATE_GIVEME; - unsigned char decrypt_payload[1500]={0}; - unsigned int decrypt_payload_len=sizeof(decrypt_payload); - - enum _QUIC_VERSION is_gquic=QUIC_VERSION_UNKNOWN; - struct udpdetail *udp_detail=pstream->pudpdetail; - - if(udp_detail->pdata==NULL || udp_detail->datalen<=0) - { - return APP_STATE_GIVEME; - } - - is_gquic=is_quic_protocol(pstream, _context, (char *)udp_detail->pdata, udp_detail->datalen, &used_len); - if(!_context->call_business) - { - return APP_STATE_GIVEME; - } - - if(is_gquic!=QUIC_VERSION_UNKNOWN) - { - if(_context->cb_version==0) - { - _context->cb_version=1; - ret=quic_callPlugins(pstream, _context, &(_context->quic_info.quic_hdr.quic_version), sizeof(_context->quic_info.quic_hdr.quic_version), QUIC_USEING_VERSION_MASK, a_packet); - if((ret&APP_STATE_DROPME) || (ret&APP_STATE_DROPPKT)) - { - return ret; - } - } - - switch(is_gquic) - { - case GQUIC_VERSION_Q044: - case GQUIC_VERSION_Q045: - case GQUIC_VERSION_Q046: - case GQUIC_VERSION_Q047: - case GQUIC_VERSION_Q048: - ret=parse_gquic_Q046(pstream, _context, a_packet, (char *)udp_detail->pdata, udp_detail->datalen, &used_len); - break; - default: - if(is_gquic>=GQUIC_VERSION_Q001 && is_gquic<=GQUIC_VERSION_Q046) - { - ret=gquic_proc_unencrypt(pstream, _context, a_packet, (char *)udp_detail->pdata, udp_detail->datalen, &used_len); - } - - if( ((is_gquic>=MVFST_VERSION_00 && is_gquic<=MVFST_VERSION_0F) || - (is_gquic>=GQUIC_VERSION_Q049 && is_gquic<=GQUIC_VERSION_Q059) || - (is_gquic>=GQUIC_VERSION_T050 && is_gquic<=GQUIC_VERSION_T059) || - (is_gquic>=GQUIC_VERSION_T050 && is_gquic<=GQUIC_VERSION_T059) || - (is_gquic>=IQUIC_VERSION_I022 && is_gquic<=IQUIC_VERSION_I029) || - (is_gquic==IQUIC_VERSION_RFC9000) - ) - && _context->is_decrypt==0 - ) - { - _context->is_decrypt=1; - ret=dissect_quic((char *)udp_detail->pdata, udp_detail->datalen, decrypt_payload, &decrypt_payload_len); - if(ret!=1) - { - return APP_STATE_DROPME; - } - ret=parse_decrypt_quic(pstream, _context, a_packet, decrypt_payload, decrypt_payload_len, &used_len); - break; - } - - ret=quic_callPlugins(pstream, _context, (char *)udp_detail->pdata, udp_detail->datalen, QUIC_APPLICATION_DATA_MASK, a_packet); - if((ret&APP_STATE_DROPME) || (ret&APP_STATE_DROPPKT)) - { - return ret; - } - break; - } - - if((ret&APP_STATE_DROPME) || (ret&APP_STATE_DROPPKT)) - { - return ret; - } - } - - if(_context->is_quic==TRUE) - { - if(_context->quic_info.quic_hdr.is_reset) - { - return quic_callPlugins(pstream, _context, NULL, 0, QUIC_INTEREST_KEY_MASK, a_packet); - } - - if(_context->quic_info.quic_hdr.is_version_negotiation) - { - return quic_callPlugins(pstream, _context, NULL, 0, QUIC_NEGOTIATION_VERSION_MASK, a_packet); - } - return APP_STATE_GIVEME; - } - - return APP_STATE_DROPME;; -} - - -int quic_protocol_identify(struct streaminfo *a_stream, void *a_packet, char *out_sni, int out_sni_len) -{ - int ret=APP_STATE_GIVEME; - int sni_len=0,len=-1; - void *pme=NULL; - char *sni=NULL; - struct _quic_context *_context=NULL; - - if(!is_quic_port(a_stream)) - { - return len; - } - - quic_init_stream(&pme, a_stream->threadnum); - _context=(struct _quic_context *)pme; - - ret=quic_process(a_stream, _context, a_stream->threadnum, a_packet); - if(ret!=PROT_STATE_DROPME && _context->is_quic!=QUIC_VERSION_UNKNOWN) - { - if(_context->quic_info.client_hello!=NULL) - { - if(_context->quic_info.client_hello->sni_idx!=0xFF) - { - sni=(char *)(_context->quic_info.client_hello->ext_tags[_context->quic_info.client_hello->sni_idx].value); - sni_len=_context->quic_info.client_hello->ext_tags[_context->quic_info.client_hello->sni_idx].length; - len= sni_len>(out_sni_len-1) ? (out_sni_len-1) : sni_len; - memcpy(out_sni, sni, len); - } - else - { - len=0; - } - } - else - { - if(_context->is_quic==TRUE) - { - len=0; - } - } - } - - quic_release_stream(&pme, a_stream->threadnum); - - return len; -} +/* + * quic_process.c + * + * Created on: 2019年4月2日 + * Author: root + */ + +#include +#include +#include +#include +#include +#include + +#include "gquic_process.h" +#include "quic_analysis.h" +#include "parser_quic.h" + +#ifndef PRINTADDR +#define PRINTADDR(a, b) ((b)addr), a->threadnum) : "") +#endif + +const unsigned char PCAP_FILE_HEAD[24] = {0xD4, 0xC3, 0xB2, 0xA1, 0x02, 0x00, 0x04, 0x00, + 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, + 0xFF, 0xFF, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00}; + +struct pcap_hdr +{ + unsigned int tv_sec; + unsigned int tv_usec; + unsigned int len; + unsigned int caplen; +}; + +int dump_packet(struct streaminfo *pstream) +{ + int ret=0; + char buff[2048]={0}; + char filename[512]={0}; + void *p_eth_rawpkt=NULL; + int eth_rawpkt_len=0; + struct pcap_hdr pcap_hdr; + struct timeval current_time; + + if(g_quic_param.dump_packet_switch==0) + { + return 0; + } + + ret=get_rawpkt_opt_from_streaminfo(pstream, RAW_PKT_GET_DATA, &p_eth_rawpkt); + if(ret==0) + { + ret=get_rawpkt_opt_from_streaminfo(pstream, RAW_PKT_GET_TOT_LEN, ð_rawpkt_len); + if(ret<0) + { + return -1; + } + snprintf(filename, sizeof(filename), "%s-%s.pcap", g_quic_param.log_path, printaddr(&(pstream->addr), pstream->threadnum)); + FILE *fp=fopen(filename, "a+"); + if(fp) + { + gettimeofday(¤t_time, NULL); + pcap_hdr.tv_sec = current_time.tv_sec; + pcap_hdr.tv_usec = current_time.tv_usec; + pcap_hdr.caplen = eth_rawpkt_len; + pcap_hdr.len = pcap_hdr.caplen; + + memcpy(buff, PCAP_FILE_HEAD, 24); + memcpy(buff+24, &pcap_hdr, sizeof(pcap_hdr)); + memcpy(buff+24+sizeof(pcap_hdr), p_eth_rawpkt, eth_rawpkt_len); + fwrite(buff, eth_rawpkt_len+24+sizeof(pcap_hdr), 1, fp); + + fclose(fp); + fp=NULL; + } + } + + return 0; +} + +int is_iquic(enum _QUIC_VERSION quic_version) +{ + switch(quic_version) + { + case IQUIC_VERSION_I001: + case IQUIC_VERSION_I002: + case IQUIC_VERSION_I003: + case IQUIC_VERSION_I004: + case IQUIC_VERSION_I005: + case IQUIC_VERSION_I006: + case IQUIC_VERSION_I007: + case IQUIC_VERSION_I008: + case IQUIC_VERSION_I009: + case IQUIC_VERSION_I010: + case IQUIC_VERSION_I011: + case IQUIC_VERSION_I012: + case IQUIC_VERSION_I013: + case IQUIC_VERSION_I014: + case IQUIC_VERSION_I015: + case IQUIC_VERSION_I016: + case IQUIC_VERSION_I017: + case IQUIC_VERSION_I018: + case IQUIC_VERSION_I019: + case IQUIC_VERSION_I020: + case IQUIC_VERSION_I021: + case IQUIC_VERSION_I022: + case IQUIC_VERSION_I023: + case IQUIC_VERSION_I024: + case IQUIC_VERSION_I025: + case IQUIC_VERSION_I026: + case IQUIC_VERSION_I027: + case IQUIC_VERSION_I028: + case IQUIC_VERSION_I029: + case IQUIC_VERSION_I030: + case IQUIC_VERSION_I031: + case IQUIC_VERSION_I032: + return TRUE; + break; + default: + break; + } + + return FALSE; +} + +int is_quic_port(struct streaminfo *pstream) +{ + switch(pstream->addr.addrtype) + { + case ADDR_TYPE_IPV4: + case __ADDR_TYPE_IP_PAIR_V4: + if(ntohs(pstream->addr.ipv4->source)!=443 && ntohs(pstream->addr.ipv4->dest)!=443) + { + return 0; + } + break; + case ADDR_TYPE_IPV6: + case __ADDR_TYPE_IP_PAIR_V6: + if(ntohs(pstream->addr.ipv6->source)!=443 && ntohs(pstream->addr.ipv6->dest)!=443) + { + return 0; + } + break; + default: + return 0; + break; + } + + return 1; +} + +static int get_value(unsigned char *payload, int *offset, int len) +{ + switch(len) + { + case 1: + return (int)(payload[(*offset)++]); + break; + case 2: + (*offset)+=len; + return (int)ntohs(*(unsigned short *)(payload+*offset-len)); + break; + case 3: + (*offset)+=len; + return ((int)*(payload-2+*offset)<<16| + (int)*(payload-1+*offset)<<8| + (int)*(payload+*offset)<<0); + break; + case 4: + (*offset)+=len; + return (int)ntohl(*(unsigned int *)(payload+*offset-len)); + break; + case 32: + (*offset)+=len; + return 0; + break; + default: + break; + } + + return 0; +} + + +static int check_length(int last_len, int field_len) +{ + return ((last_len-field_len>0) ? 1 : 0); +} + +int quic_getLinkState(struct _quic_context *_context) +{ + UCHAR state = 0; + + if(0==_context->link_state) + { + state=SESSION_STATE_PENDING|SESSION_STATE_DATA; + _context->link_state=1; + } + else + { + state=SESSION_STATE_DATA; + } + + return state; +} + +char quic_callPlugins(struct streaminfo *pstream, struct _quic_context *_context, void *buff, int buff_len, enum quic_interested_region region_mask, void *a_packet) +{ + char state=PROT_STATE_GIVEME; + char app_state=APP_STATE_GIVEME; + stSessionInfo session_info={0}; + + if(region_mask==QUIC_INTEREST_KEY_MASK) + { + session_info.plugid=g_quic_param.quic_plugid; + session_info.prot_flag=0; + session_info.session_state=SESSION_STATE_CLOSE; + session_info.app_info=NULL; + session_info.buf=NULL; + session_info.buflen=0; + } + else + { + session_info.plugid=g_quic_param.quic_plugid; + session_info.prot_flag=(((unsigned long long)1)<quic_info); + session_info.buf=buff; + session_info.buflen=buff_len; + } + state=PROT_PROCESS(&session_info, &(_context->business_pme), pstream->threadnum, pstream, a_packet); + + if(state&PROT_STATE_DROPPKT) + { + app_state=APP_STATE_DROPPKT; + } + + return app_state; +} + + +unsigned long long get_variable_length(char *p, int offset, int v_len) +{ + switch(v_len) + { + case 1: + return (unsigned long long)(p[offset]); + break; + case 2: + return (unsigned long long)ntohs(*(unsigned short *)((char *)p+offset)); + break; + case 3: + return (unsigned long long)*(p+0+offset)<<16| + (unsigned long long)*(p+1+offset)<<8| + (unsigned long long)*(p+2+offset)<<0; + break; + case 4: + return (unsigned long long)ntohl(*(unsigned int *)(p+offset)); + break; + case 5: + return (unsigned long long)*((unsigned char *)(p)+0+offset)<<32| + (unsigned long long)*((unsigned char *)(p)+1+offset)<<24| + (unsigned long long)*((unsigned char *)(p)+2+offset)<<16| + (unsigned long long)*((unsigned char *)(p)+3+offset)<<8| + (unsigned long long)*((unsigned char *)(p)+4+offset)<<0; + break; + case 6: + return (unsigned long long)*((unsigned char *)(p)+0+offset)<<40| + (unsigned long long)*((unsigned char *)(p)+1+offset)<<32| + (unsigned long long)*((unsigned char *)(p)+2+offset)<<24| + (unsigned long long)*((unsigned char *)(p)+3+offset)<<16| + (unsigned long long)*((unsigned char *)(p)+4+offset)<<8| + (unsigned long long)*((unsigned char *)(p)+5+offset)<<0; + break; + case 7: + return (unsigned long long)*((unsigned char *)(p)+0+offset)<<56| + (unsigned long long)*((unsigned char *)(p)+1+offset)<<40| + (unsigned long long)*((unsigned char *)(p)+2+offset)<<32| + (unsigned long long)*((unsigned char *)(p)+3+offset)<<24| + (unsigned long long)*((unsigned char *)(p)+4+offset)<<16| + (unsigned long long)*((unsigned char *)(p)+5+offset)<<8| + (unsigned long long)*((unsigned char *)(p)+6+offset)<<0; + break; + case 8: + return (unsigned long long)*((unsigned char *)(p)+0+offset)<<56| + (unsigned long long)*((unsigned char *)(p)+1+offset)<<48| + (unsigned long long)*((unsigned char *)(p)+2+offset)<<40| + (unsigned long long)*((unsigned char *)(p)+3+offset)<<32| + (unsigned long long)*((unsigned char *)(p)+4+offset)<<24| + (unsigned long long)*((unsigned char *)(p)+5+offset)<<16| + (unsigned long long)*((unsigned char *)(p)+6+offset)<<8| + (unsigned long long)*((unsigned char *)(p)+7+offset)<<0; + break; + default: + break; + } + + return 0; +} + +long long bit_to_value(char *payload, int payload_len, unsigned char flags, unsigned long long *out_value, int *used_len) +{ + switch(flags&0x3) // packet number + { + case 0x3: // 6 bytes + if(!check_length(payload_len-*used_len, 6)) + { + return -1; + } + *out_value=get_variable_length(payload, *used_len, 6); + *used_len+=6; + break; + case 0x2: // 4 bytes + if(!check_length(payload_len-*used_len, sizeof(unsigned int))) + { + return -1; + } + *out_value=(unsigned long long)ntohl(*(unsigned int *)(payload+*used_len)); + *used_len+=4; + break; + case 0x1: // 2bytes + if(!check_length(payload_len-*used_len, sizeof(unsigned short))) + { + return -1; + } + *out_value=(unsigned long long)ntohs(*(unsigned short *)(payload+*used_len)); + *used_len+=2; + break; + default: // 1 byte + if(!check_length(payload_len-*used_len, sizeof(unsigned char))) + { + return -1; + } + *out_value=payload[*used_len]; + *used_len+=1; + break; + } + + return 0; +} + +int get_quic_tlv(char *start_pos, quic_tlv_t *tlv, int len, int type, int thread_seq) +{ + if(tlv->value==NULL && len>0) + { + tlv->value=(char *)dictator_malloc(thread_seq, len+1); + memset(tlv->value, 0, len+1); + tlv->length=len; + tlv->type=type; + memcpy(tlv->value, start_pos, tlv->length); + } + + return 0; +} + +int get_stream_id(struct streaminfo *pstream, struct _quic_context* _context, char* payload, int payload_len, unsigned char frame_type, int *used_len) +{ + int stream_len=0,offset_len=0; + + _context->quic_info.frame_hdr.frame_type=frame_type; + + stream_len=(frame_type&GQUIC_SPECIAL_FRAME_STREAM_ID)+1; + if(!check_length(payload_len-*used_len, stream_len)) + { + return -1; + } + + _context->quic_info.frame_hdr.stream_id=(unsigned int)get_variable_length(payload, *used_len, stream_len); + *used_len+=stream_len; // stream ID length + + if(frame_type&GQUIC_SPECIAL_FRAME_STREAM_DLEN) + { + if(!check_length(payload_len-*used_len, 2)) + { + return -1; + } + _context->quic_info.frame_hdr.data_len=ntohs(*(unsigned short *)(payload+*used_len)); + *used_len+=2; //data length + } + + offset_len=(frame_type&GQUIC_SPECIAL_FRAME_STREAM_OFFSET) ? (((frame_type&GQUIC_SPECIAL_FRAME_STREAM_OFFSET))>>2)+1 : 0; + if(!check_length(payload_len-*used_len, offset_len)) + { + return -1; + } + + _context->quic_info.frame_hdr.offset=get_variable_length(payload, *used_len, offset_len); + *used_len+=offset_len; + + MESA_handle_runtime_log(g_quic_param.logger, + RLOG_LV_DEBUG, + "QUIC_FRAME", + "frame_type: 0X%02X stream_id: %u data length: %u offset length: %u addr: %s", + frame_type, + _context->quic_info.frame_hdr.stream_id, + _context->quic_info.frame_hdr.data_len, + offset_len, + PRINTADDR(pstream, g_quic_param.level) + ); + + return _context->quic_info.frame_hdr.stream_id; +} + +unsigned long long get_packet_number(char* data, int offset, char pkn_len) +{ + switch(pkn_len) + { + case 1: + return (unsigned long long)data[offset]; + break; + case 2: + return (unsigned long long)ntohs(*(unsigned short *)(data+offset)); + break; + case 4: + return (unsigned long long)ntohl(*(unsigned int *)(data+offset)); + break; + case 8: + return get_variable_length(data, offset, 8);; + break; + } + + return 0; +} + +// GQUIC version from 0 to 43 +static enum _QUIC_VERSION parse_q0to43_header(struct streaminfo *pstream, struct _quic_context* _context, char *payload, int payload_len, int *used_len) +{ + int ret=0; + char public_flags=0; + + struct _quic_public_header *gquic_hdr=&(_context->quic_info.quic_hdr); + + public_flags=payload[*used_len]; + *used_len+=1; + gquic_hdr->public_flags=public_flags; + + if((public_flags&GQUIC_PUBLIC_FLAG_RST) && _context->is_quic==TRUE) + { + gquic_hdr->is_reset=TRUE; //Public Reset Packet + return QUIC_VERSION_UNKNOWN; + + } + + if(pstream->curdir==DIR_S2C && gquic_hdr->public_flags&GQUIC_PUBLIC_FLAG_VERSION) + { + return QUIC_VERSION_UNKNOWN; + } + + //For Public Reset and Version Negotiation Packets (sent by the server) which don't have a packet number + if(!public_flags&GQUIC_PUBLIC_FLAG_PKT_NUM) + { + if(public_flags&GQUIC_PUBLIC_FLAG_VERSION) //Public Reset Packet + { + return QUIC_VERSION_UNKNOWN;// todo + } + else // Version Negotiation Packet + { + return QUIC_VERSION_UNKNOWN; + } + } + + if(gquic_hdr->public_flags&GQUIC_PUBLIC_FLAG_CID) + { + *(unsigned long long *)gquic_hdr->server_CID=get_variable_length(payload, *used_len, sizeof(gquic_hdr->server_CID)); + *used_len+=sizeof(unsigned long long); // CID length + + _context->is_quic=TRUE; + } + + if(gquic_hdr->public_flags&GQUIC_PUBLIC_FLAG_VERSION && (*(unsigned char *)(payload+*used_len)==0x51)) + { + gquic_hdr->quic_version=(unsigned int)ntohl(*(unsigned int *)(payload+*used_len)); + *used_len+=sizeof(int); // skip version + + _context->is_quic=TRUE; + } + + if(_context->is_quic==FALSE || gquic_hdr->quic_versionquic_version>GQUIC_VERSION_Q043) + { + _context->is_quic=FALSE; + return QUIC_VERSION_UNKNOWN; + } + + ret=bit_to_value(payload, payload_len, gquic_hdr->public_flags>>4, &gquic_hdr->packet_number, used_len); + if(ret<0) + { + return (enum _QUIC_VERSION)gquic_hdr->quic_version; + } + + if(gquic_hdr->public_flags==GQUIC_PUBLIC_FLAG_NONCE) + { + *used_len+=32; //diversification nonce + } + + // Version 11 reduced the length of null encryption authentication tag from 16 to 12 bytes + if(gquic_hdr->quic_version > GQUIC_VERSION_Q010) + { + *used_len+=12; + } + else + { + *used_len+=16; + } + + // Version 34 removed entropy bits from packets and ACK frames, + // removed private flag from packet header and changed the ACK format to specify ranges of packets acknowledged rather than missing ranges. + if(gquic_hdr->quic_version < GQUIC_VERSION_Q034) + { + *used_len+=1; //private flags + } + + _context->is_quic=TRUE; + + MESA_handle_runtime_log(g_quic_param.logger, + RLOG_LV_DEBUG, + "QUIC_IDETIFY", + "pub_flags: 0X%02X conection ID:[ destination: %llu ] version: Q%03u packet number: %llu dir(1: C2S;2: S2C): %d addr: %s", + gquic_hdr->public_flags, + *(unsigned long long *)gquic_hdr->server_CID, + (((gquic_hdr->quic_version>>8)&0x0000000F)*10) + ((gquic_hdr->quic_version)&0x0000000F), + gquic_hdr->packet_number, + pstream->curdir, + PRINTADDR(pstream, g_quic_param.level) + ); + + return (enum _QUIC_VERSION)gquic_hdr->quic_version; +} + +enum _QUIC_VERSION parse_quic_header(struct streaminfo *pstream, struct _quic_context* _context, char *payload, int payload_len, int *used_len) +{ + int i=0,len=0; + char client_CID[MAX_CONNECT_ID_LEN*2]={0}; + char server_CID[MAX_CONNECT_ID_LEN*2]={0}; + + struct _quic_public_header *long_hdr=&(_context->quic_info.quic_hdr); + + long_hdr->public_flags=payload[*used_len]; + *used_len+=1; //skip public flags + + if(long_hdr->public_flags&0x80) + { + long_hdr->quic_version=(unsigned int)ntohl(*(unsigned int *)(payload+*used_len)); + *used_len+=sizeof(int); // skip version + + long_hdr->client_CID_len=(payload[*used_len]&0xF) ? (payload[*used_len]&0xF)+3 : 0; + long_hdr->server_CID_len=((payload[*used_len]>>4)&0xF) ? ((payload[*used_len]>>4)&0xF)+3 : 0; + *used_len+=sizeof(char); // both connection_id length + + memcpy(long_hdr->server_CID, (void *)(payload+*used_len), long_hdr->server_CID_len); + *used_len+=long_hdr->server_CID_len; // Destination connection_id length + memcpy(long_hdr->client_CID, (void *)(payload+*used_len), long_hdr->client_CID_len); + *used_len+=long_hdr->client_CID_len; // source connection_id length + } + else + { + if(pstream->curdir==DIR_C2S)// short header only destination connection ID + { + *used_len+=long_hdr->server_CID_len; // every packet destination connection ID is same + } + } + + len=(long_hdr->public_flags&0x03)+1; + long_hdr->packet_number=get_packet_number(payload, *used_len, len); + *used_len+=len; + + *used_len+=12; //message authentication hash + + _context->is_quic=TRUE; + + for(i=0,len=0;iserver_CID_len; i++) + { + len+=snprintf(server_CID+len, sizeof(server_CID)-len, "%02X", long_hdr->server_CID[i]); + } + + for(i=0,len=0;iclient_CID_len; i++) + { + len+=snprintf(client_CID+len, sizeof(client_CID)-len, "%02X", long_hdr->client_CID[i]); + } + + MESA_handle_runtime_log(g_quic_param.logger, + RLOG_LV_DEBUG, + "QUIC_IDETIFY", + "pub_flags: 0X%02X conection ID:[ destination: %s source: %s ] version: Q%03u packet number: %llu dir(1: C2S;2: S2C): %d addr: %s", + long_hdr->public_flags, + server_CID, + client_CID, + (((long_hdr->quic_version>>8)&0x0000000F)*10) + ((long_hdr->quic_version)&0x0000000F), + long_hdr->packet_number, + pstream->curdir, + PRINTADDR(pstream, g_quic_param.level) + ); + + return (enum _QUIC_VERSION)long_hdr->quic_version; +} + +enum _QUIC_VERSION is_quic_protocol(struct streaminfo *pstream, struct _quic_context* _context, char *payload, int payload_len, int *used_len) +{ + enum _QUIC_VERSION quic_version=QUIC_VERSION_UNKNOWN; + + if(_context->quic_info.quic_hdr.quic_version!=QUIC_VERSION_UNKNOWN) + { + if(is_iquic((enum _QUIC_VERSION)(_context->quic_info.quic_hdr.quic_version))) + { + return (enum _QUIC_VERSION)(_context->quic_info.quic_hdr.quic_version); + } + + quic_version=(enum _QUIC_VERSION)(_context->quic_info.quic_hdr.quic_version); + } + else + { + // The most significant bit (0x80) of byte 0 (the first byte) is set to 1 for long headers + (payload[*used_len]&0x80) ? (quic_version=(enum _QUIC_VERSION)ntohl(*(unsigned int *)(payload+(*used_len+1)))) : QUIC_VERSION_UNKNOWN; + } + + switch(quic_version) // +1 meaning: skip public flags + { + case GQUIC_VERSION_Q044: + case GQUIC_VERSION_Q045: + case GQUIC_VERSION_Q046: + case GQUIC_VERSION_Q047: + case GQUIC_VERSION_Q048: + quic_version=parse_quic_header(pstream, _context, payload, payload_len, used_len); + return quic_version; + break; + default: + if( + (quic_version==GQUIC_VERSION_Q099) || + (quic_version==PICOQUIC_VERSION_30) || + (quic_version==PQUIC_VERSION_PROX) || + (quic_version==GQUIC_VERSION_T099) || + (quic_version>=GQUIC_VERSION_Q049 && quic_version<=GQUIC_VERSION_Q050) || + (quic_version>=GQUIC_VERSION_Q051 && quic_version<=GQUIC_VERSION_Q059) || + (quic_version>=GQUIC_VERSION_T048 && quic_version<=GQUIC_VERSION_T049) || + (quic_version>=GQUIC_VERSION_T050 && quic_version<=GQUIC_VERSION_T059) || + (quic_version>=QUANT_VERSION_00 && quic_version<=QUANT_VERSION_FF) || + (quic_version>=QUIC_GO_VERSION_00 && quic_version<=QUIC_GO_VERSION_FF) || + (quic_version>=QUICLY_VERSION_00 && quic_version<=QUICLY_VERSION_FF) || + (quic_version>=MSQUIC_VERSION_00 && quic_version<=MSQUIC_VERSION_0F) || + (quic_version>=MOZQUIC_VERSION_00 && quic_version<=MOZQUIC_VERSION_0F) || + (quic_version>=MVFST_VERSION_00 && quic_version<=MVFST_VERSION_0F) || + (quic_version>=IQUIC_VERSION_I001 && quic_version<=IQUIC_VERSION_I032) || + (quic_version==IQUIC_VERSION_RFC9000) + ) + { + MESA_handle_runtime_log(g_quic_param.logger, + RLOG_LV_DEBUG, + "QUIC", + "version: 0x%x addr: %s", + quic_version, PRINTADDR(pstream, g_quic_param.level) + ); + + _context->is_quic=TRUE; + _context->quic_info.quic_hdr.quic_version=quic_version; + return quic_version; + } + break; + } + + // Q001~Q043: 0x80 is currently unused, and must be set to 0 + if(payload[*used_len]>0x80) + { + return QUIC_VERSION_UNKNOWN; + } + + return parse_q0to43_header(pstream, _context, payload, payload_len, used_len); +} + +int parse_extension_tag(struct streaminfo *pstream, struct _quic_stream **quic_stream, void *a_packet, char *payload, int payload_len, int *used_len, int tag_num) +{ + int tag_used_num=0; + int tag_type=0; + int total_tag_len=0,tag_len=0; + int tag_offset_end=0,pre_tag_offset_end=0; + + if(tag_num>64 || tag_num<0) + { + (*used_len)=payload_len; + + dump_packet(pstream); + MESA_handle_runtime_log(g_quic_param.logger, RLOG_LV_FATAL, "QUIC_TAG_NUM", "QUIC_TAG_NUM:%d addr: %s", tag_num, printaddr(&pstream->addr, pstream->threadnum)); + return -1; + } + + struct _quic_stream *stream=*quic_stream; + int tag_value_start=tag_num*4*2+(*used_len); // skip length of type and offset, type(offset)=szieof(int) + + if(stream==NULL) + { + stream=(struct _quic_stream *)dictator_malloc(pstream->threadnum, sizeof(struct _quic_stream)); + memset(stream, 0, sizeof(struct _quic_stream)); + stream->ext_tags=(quic_tlv_t *)dictator_malloc(pstream->threadnum, tag_num*sizeof(quic_tlv_t)); + memset(stream->ext_tags, 0, tag_num*sizeof(quic_tlv_t)); + *quic_stream=stream; + stream->sni_idx=0xFF; + stream->ver_idx=0xFF; + stream->ua_idx=0xFF; + } + else + { + quic_release_exts(pstream->threadnum, stream->ext_tags, stream->ext_tag_num); + stream->ext_tags=(quic_tlv_t *)dictator_malloc(pstream->threadnum, tag_num*sizeof(quic_tlv_t)); + memset(stream->ext_tags, 0, tag_num*sizeof(quic_tlv_t)); + *quic_stream=stream; + stream->ext_tag_num=0; + stream->count++; + stream->sni_idx=0xFF; + stream->ver_idx=0xFF; + stream->ua_idx=0xFF; + } + + while(tag_num>tag_used_num) + { + tag_type=ntohl(*(unsigned int *)(payload+*used_len)); + *used_len+=sizeof(int); + + tag_offset_end=*(unsigned int *)(payload+*used_len); + *used_len+=sizeof(int); + + tag_len=tag_offset_end-pre_tag_offset_end; + if(tag_len<0 || (tag_offset_end>=payload_len) || (tag_len>payload_len-tag_value_start)) + { + return -1; + } + + switch(tag_type) + { + case TAG_PAD: + break; + case TAG_VER: + stream->ver_idx=stream->ext_tag_num; + get_quic_tlv(payload+tag_value_start, &stream->ext_tags[stream->ext_tag_num], tag_len, tag_type, pstream->threadnum); + *(unsigned int *)(stream->ext_tags[stream->ext_tag_num].value)=(unsigned int)ntohl(*(unsigned int *)(stream->ext_tags[stream->ext_tag_num].value)); + MESA_handle_runtime_log(g_quic_param.logger, + RLOG_LV_DEBUG, + "QUIC_VERSION", + "Quic version: 0X%X addr: %s", + *(unsigned int *)(stream->ext_tags[stream->ext_tag_num].value), + PRINTADDR(pstream, g_quic_param.level) + ); + break; + case TAG_UAID: + stream->ua_idx=stream->ext_tag_num; + get_quic_tlv(payload+tag_value_start, &stream->ext_tags[stream->ext_tag_num], tag_len, tag_type, pstream->threadnum); + MESA_handle_runtime_log(g_quic_param.logger, + RLOG_LV_DEBUG, + "QUIC_UA", + "User Agent: %s addr: %s", + stream->ext_tags[stream->ext_tag_num].value, + PRINTADDR(pstream, g_quic_param.level) + ); + stream->ext_tag_num++; + break; + case TAG_SNI: + stream->sni_idx=stream->ext_tag_num; + get_quic_tlv(payload+tag_value_start, &stream->ext_tags[stream->ext_tag_num], tag_len, tag_type, pstream->threadnum); + MESA_handle_runtime_log(g_quic_param.logger, + RLOG_LV_DEBUG, + "QUIC_SNI", + "SNI: %s addr: %s", + stream->ext_tags[stream->ext_tag_num].value, + PRINTADDR(pstream, g_quic_param.level) + ); + stream->ext_tag_num++; + break; + default: + get_quic_tlv(payload+tag_value_start, &stream->ext_tags[stream->ext_tag_num], tag_len, tag_type, pstream->threadnum); + stream->ext_tag_num++; + break; + } + + tag_used_num++; + tag_value_start+=tag_len; + total_tag_len+=tag_len; + pre_tag_offset_end=tag_offset_end; + } + + *used_len += total_tag_len; + + return 0; +} + +int gquic_frame_type_stream(struct streaminfo *pstream, struct _quic_context* _context, char *payload, int payload_len, int *used_len, void *a_packet) +{ + int ret=0; + char state=APP_STATE_GIVEME; + int tag_num = 0; + unsigned int message_tag; + + if(!check_length(payload_len-*used_len, 8)) + { + return state; + } + + switch(_context->quic_info.quic_hdr.quic_version) + { + case GQUIC_VERSION_Q041: + *used_len+=1; // unknown + case GQUIC_VERSION_Q044: + message_tag=(unsigned int)ntohl(*(unsigned int *)(payload+*used_len)); + *used_len+=4; + + tag_num=*(int *)(payload+*used_len); + *used_len+=4; //tag_num + break; + default: + message_tag=(unsigned int)ntohl(*(unsigned int *)(payload+*used_len)); + *used_len+=4; + + tag_num=*(unsigned short *)(payload+*used_len); + *used_len+=2; //tag_num + *used_len+=2; //padding + break; + } + + switch(message_tag) + { + case CHLO: //MTAG_CHLO; + ret=parse_extension_tag(pstream, &(_context->quic_info.client_hello), a_packet, payload, payload_len, used_len, tag_num); + if(ret>=0 && _context->call_business) + { + state=quic_callPlugins(pstream, _context, (void *)(_context->quic_info.client_hello), sizeof(void *), QUIC_CLIENT_HELLO_MASK, a_packet); + } + break; + case SHLO: //MTAG_SHLO; + ret=parse_extension_tag(pstream, &(_context->quic_info.server_hello), a_packet, payload, payload_len, used_len, tag_num); + if(ret>=0 && _context->call_business) + { + state=quic_callPlugins(pstream, _context, (void *)(_context->quic_info.server_hello), sizeof(void *), QUIC_SERVER_HELLO_MASK, a_packet); + } + break; + case REJ: //MTAG_REJ; + ret=parse_extension_tag(pstream, &(_context->quic_info.rejection), a_packet, payload, payload_len, used_len, tag_num); + if(ret>=0 && _context->call_business) + { + state=quic_callPlugins(pstream, _context, (void *)(_context->quic_info.rejection), sizeof(void *), QUIC_REJECTION_MASK, a_packet); + } + break; + default: + break; + } + + return state; +} + +//frame type->stream->offset->data length +int gquic_proc_unencrypt(struct streaminfo *pstream, struct _quic_context* _context, void *a_packet, char * payload, int payload_len, int *used_len) +{ + unsigned int ret=0; + unsigned char frame_type=0; + unsigned int stream_id=0; + unsigned int error_code=0; + unsigned short reason_phrase_length=0; + unsigned long long byte_offset=0; + unsigned long long least_unacked_delta=0; + + while(*used_lenaddr, pstream->threadnum)); + + return quic_callPlugins(pstream, _context, NULL, 0, QUIC_INTEREST_KEY_MASK, a_packet); + break; + case GQUIC_REGULAR_FRAME_CONNECTION_CLOSE: + if(!check_length(payload_len-*used_len, sizeof(unsigned int)+sizeof(unsigned short))) + { + return APP_STATE_GIVEME; + } + error_code=(unsigned int)get_variable_length(payload, *used_len, sizeof(unsigned int)); + *used_len+=sizeof(unsigned int); + + reason_phrase_length=(unsigned short)get_variable_length(payload, *used_len, sizeof(unsigned short)); + *used_len+=sizeof(unsigned short); + + if(!check_length(payload_len-*used_len, reason_phrase_length)) + { + return APP_STATE_GIVEME; + } + + *used_len+=reason_phrase_length; // skip Reason Phrase + + MESA_handle_runtime_log(g_quic_param.logger, + RLOG_LV_DEBUG, + "QUIC_CONNECTION_CLOSE", + "error_code: %u reason_phrase_length: %d addr: %s", + error_code, + reason_phrase_length, + printaddr(&pstream->addr, pstream->threadnum)); + + return quic_callPlugins(pstream, _context, NULL, 0, QUIC_INTEREST_KEY_MASK, a_packet); + break; + case GQUIC_REGULAR_FRAME_GOAWAY: + if(!check_length(payload_len-*used_len, sizeof(unsigned int)+sizeof(unsigned int)+sizeof(unsigned short))) + { + return APP_STATE_GIVEME; + } + error_code=(unsigned int)get_variable_length(payload, *used_len, sizeof(unsigned int)); + *used_len+=sizeof(unsigned int); + + //Last Good Stream ID + stream_id=(unsigned int)get_variable_length(payload, *used_len, sizeof(unsigned int)); + *used_len+=sizeof(unsigned int); + + reason_phrase_length=(unsigned short)get_variable_length(payload, *used_len, sizeof(unsigned short)); + *used_len+=sizeof(unsigned short); + + if(!check_length(payload_len-*used_len, reason_phrase_length)) + { + return APP_STATE_GIVEME; + } + *used_len+=reason_phrase_length; // skip Reason Phrase + + MESA_handle_runtime_log(g_quic_param.logger, + RLOG_LV_DEBUG, + "QUIC_GOAWAY", + "error_code: %u Last Good Stream ID: %u reason_phrase_length: %d addr: %s", + error_code, + stream_id, + reason_phrase_length, + printaddr(&pstream->addr, pstream->threadnum)); + break; + case GQUIC_REGULAR_FRAME_WINDOW_UPDATE: + if(!check_length(payload_len-*used_len, sizeof(unsigned int)+sizeof(unsigned long long))) + { + return APP_STATE_GIVEME; + } + stream_id=(unsigned int)get_variable_length(payload, *used_len, sizeof(unsigned int)); + *used_len+=sizeof(unsigned int); + + byte_offset=get_variable_length(payload, *used_len, sizeof(unsigned long long)); + *used_len+=sizeof(unsigned long long); + + MESA_handle_runtime_log(g_quic_param.logger, + RLOG_LV_DEBUG, + "QUIC_WINDOW_UPDATE", + "stream_id: %u byte_offset: %llu addr: %s", + stream_id, + byte_offset, + printaddr(&pstream->addr, pstream->threadnum)); + break; + case GQUIC_REGULAR_FRAME_BLOCKED: + if(!check_length(payload_len-*used_len, sizeof(unsigned int))) + { + return APP_STATE_GIVEME; + } + stream_id=(unsigned int)get_variable_length(payload, *used_len, sizeof(unsigned int)); + *used_len+=sizeof(unsigned int); + + MESA_handle_runtime_log(g_quic_param.logger, + RLOG_LV_DEBUG, + "QUIC_STOP_WAITING", + "stream_id: %u addr: %s", + stream_id, + printaddr(&pstream->addr, pstream->threadnum)); + break; + case GQUIC_REGULAR_FRAME_STOP_WAITING: + ret=bit_to_value(payload, payload_len, _context->quic_info.quic_hdr.public_flags>>4, &least_unacked_delta, used_len); + if(ret<0) + { + return APP_STATE_GIVEME; + } + + MESA_handle_runtime_log(g_quic_param.logger, + RLOG_LV_DEBUG, + "QUIC_STOP_WAITING", + "least_unacked_delta: %llu addr: %s", + least_unacked_delta, + printaddr(&pstream->addr, pstream->threadnum)); + break; + case GQUIC_REGULAR_FRAME_PING: + //The PING frame contains no payload. + //The receiver of a PING frame simply needs to ACK the packet containing this frame + break; + default: //Regular Frame Types + if(frame_type&GQUIC_SPECIAL_FRAME_STREAM || (frame_type&0xC0)==GQUIC_SPECIAL_FRAME_ACK) + { + stream_id=get_stream_id(pstream, _context, payload, payload_len, frame_type, used_len); + if(stream_id<0) + { + return APP_STATE_GIVEME; + } + ret=gquic_frame_type_stream(pstream, _context, payload, payload_len, used_len, a_packet); + } + else if((frame_type&0xE0)==GQUIC_SPECIAL_FRAME_CONGEST_FB) // high two bits set 0; (frame_type: 01nullmmB) + { + //not used + } + else + { + return APP_STATE_GIVEME; + } + break; + } + + if(ret&APP_STATE_DROPME || ret&APP_STATE_DROPPKT) + { + return ret; + } + } + + return APP_STATE_GIVEME; +} + + +//QUIC_DATA:is quic data pcap;QUIC_TRUE:is handshake pcap;QUIC_RETURN_DROPME:not quic protocol; +int parse_gquic_Q046(struct streaminfo *pstream, struct _quic_context* _context, void *a_packet, char * payload, int payload_len, int *used_len) +{ + int stream_id=0; + int ret=APP_STATE_GIVEME; + unsigned char frame_type; + + while(*used_len < payload_len) + { + frame_type=payload[*used_len]; + *used_len+=1; //skip frame_type + + if(frame_type&IQUIC_FRAME_STREAM_HEX08) //0x08=Q048 + { + stream_id=get_stream_id(pstream, _context, payload, payload_len, frame_type, used_len); + if(stream_id<0) + { + return APP_STATE_GIVEME; + } + ret=gquic_frame_type_stream(pstream, _context, payload, payload_len, used_len, a_packet); + } + else + { + return APP_STATE_GIVEME; //todo + } + + if(ret&APP_STATE_DROPME || ret&APP_STATE_DROPPKT) + { + return ret; + } + } + + return APP_STATE_GIVEME; +} + +int parse_encrypt_parameter(struct _quic_stream *quic_stream, unsigned char *payload, int payload_len, int thread_seq) +{ + int used_len=0,length=0; + while(payload_len>used_len) + { + if(payload[used_len]> 0x00 && payload[used_len]<=0x20) + { + get_value(payload, &used_len, 1); //type=1 + length=get_value(payload, &used_len, 1); // length=1 + used_len+=length; + + continue; + } + + if((*(unsigned short *)(payload+used_len)) == htons(EXT_QUIC_PARAM_USER_AGENT)) + { + quic_stream->ua_idx=quic_stream->ext_tag_num++; + get_value(payload, &used_len, 2); //type=2 + length=get_value(payload, &used_len, 1); // length=1 + get_quic_tlv((char *)payload+used_len, &(quic_stream->ext_tags[quic_stream->ua_idx]), length, EXT_QUIC_PARAM_USER_AGENT, thread_seq); + used_len+=length; + + continue; + } + + if(*(unsigned int *)(payload+used_len) == htonl(EXT_QUIC_PARAM_QUIC_VERSION)) + { + quic_stream->ver_idx=quic_stream->ext_tag_num++; + get_value(payload, &used_len, 4); //type=4 + length=get_value(payload, &used_len, 1); // length=1 + get_quic_tlv((char *)payload+used_len, &(quic_stream->ext_tags[quic_stream->ver_idx]), length, EXT_QUIC_PARAM_QUIC_VERSION, thread_seq); + *(unsigned int *)quic_stream->ext_tags[quic_stream->ver_idx].value=(unsigned int)htonl(*(unsigned int *)quic_stream->ext_tags[quic_stream->ver_idx].value); + used_len+=length; + + continue; + } + + if((*(unsigned int *)(payload+used_len))== htonl(EXT_QUIC_PARAM_GREASE_HIGH4) && (*(unsigned int *)(payload+used_len+4))== htonl(EXT_QUIC_PARAM_GREASE_LOW4)) + { + used_len+=8; //type=8 + length=get_value(payload, &used_len, 1); // length=1 + used_len+=length; + + continue; + } + + break; + } + + return 0; +} +int parse_encrypt_server_name(struct _quic_stream *quic_stream, unsigned char *payload, int payload_len, int thread_seq) +{ + int ext_len=0,used_len=0; + + quic_stream->sni_idx=quic_stream->ext_tag_num++; + get_value(payload, &used_len, 2); //Server Name List length + if(get_value(payload, &used_len, 1)==0) //Server Name type + { + ext_len=get_value(payload, &used_len, 2); //Server Name length + get_quic_tlv((char *)payload+used_len, &(quic_stream->ext_tags[quic_stream->sni_idx]), ext_len, EXTENSION_SERVER_NAME, thread_seq); + } + + return 1; +} + +int parse_encrypt_client_hello(struct streaminfo *pstream, struct _quic_stream *quic_stream, void *a_packet, unsigned char *payload, int payload_len) +{ + int skip_len=0; + int used_len=0; + int flags=0; + int ext_type=0, extension_total_len=0; + + get_value(payload, &used_len, 1); //handshake type + get_value(payload, &used_len, 3); //client hello length + get_value(payload, &used_len, 2); //ssl_version + + get_value(payload, &used_len, 32); //Random + + skip_len=(int)get_value(payload, &used_len, 1); //Session ID length + if(!check_length(payload_len-used_len, skip_len)) + { + return flags; + } + used_len+=skip_len; + + skip_len=(int)get_value(payload, &used_len, 2); //Ciper Suites length + if(!check_length(payload_len-used_len, skip_len)) + { + return flags; + } + used_len+=skip_len; + + skip_len=(int)get_value(payload, &used_len, 1); //Compression Methods + if(!check_length(payload_len-used_len, skip_len)) + { + return flags; + } + used_len+=skip_len; + + extension_total_len=(int)get_value(payload, &used_len, 2); //Extension length + if(!check_length(payload_len-used_len, extension_total_len)) + { + return flags; + } + + quic_stream->ext_tags=(quic_tlv_t *)dictator_malloc(pstream->threadnum, sizeof(quic_tlv_t)*3); + memset(quic_stream->ext_tags, 0, sizeof(quic_tlv_t)*3); + quic_stream->sni_idx=0xFF; + quic_stream->ua_idx=0xFF; + quic_stream->ver_idx=0xFF; + + while(extension_total_len>used_len) + { + ext_type=get_value(payload, &used_len, 2); //Extension type + skip_len=get_value(payload, &used_len, 2); //length + if(!check_length(payload_len-used_len, skip_len)) + { + return flags; + } + + switch(ext_type) + { + case EXTENSION_SERVER_NAME: + parse_encrypt_server_name(quic_stream, payload+used_len, skip_len, pstream->threadnum); + flags=1; + break; + case EXTENSION_QUIC_PARAM: + parse_encrypt_parameter(quic_stream, payload+used_len, skip_len, pstream->threadnum); + break; + case EXTENSION_SUPPORT_GROUP: + case EXTENSION_APP_PROT_NEGO: + case EXTENSION_SIG_ALGORITHM: + case EXTENSION_KEY_SHARE: + case EXTENSION_PSK_EXCHANGE: + case EXTENSION_SUPP_SSL_VER: + case EXTENSION_COMPRESS_CERT: + break; + default: + break; + } + + used_len+=skip_len; + } + + return flags; +} + +int parse_decrypt_quic(struct streaminfo *pstream, struct _quic_context* _context, void *a_packet, unsigned char * payload, int payload_len, int *used_len) +{ + int ret=0,state=APP_STATE_GIVEME; + unsigned int quic_version=_context->quic_info.quic_hdr.quic_version; + + get_value(payload, used_len, 4); //Frame Type=1, offset=1, length=2 + + if( (quic_version>=MVFST_VERSION_00 && quic_version<=MVFST_VERSION_0F) || + (quic_version>=GQUIC_VERSION_T050 && quic_version<=GQUIC_VERSION_T059) || + (quic_version>=IQUIC_VERSION_I022 && quic_version<=IQUIC_VERSION_I029) || + (quic_version==IQUIC_VERSION_RFC9000) + ) + { + if(payload[*used_len] == 0x01) + { + if(_context->quic_info.client_hello==NULL) + { + _context->quic_info.client_hello=(struct _quic_stream *)dictator_malloc(pstream->threadnum, sizeof(struct _quic_stream)); + memset(_context->quic_info.client_hello, 0, sizeof(struct _quic_stream)); + } + ret=parse_encrypt_client_hello(pstream, _context->quic_info.client_hello, a_packet, payload+*used_len, payload_len-*used_len); //Frame Type=1, offset=1, length=2 + if(ret>0 && _context->call_business) + { + state=quic_callPlugins(pstream, _context, (void *)(_context->quic_info.client_hello), sizeof(void *), QUIC_CLIENT_HELLO_MASK, a_packet); + } + } + } + else if( (quic_version>=GQUIC_VERSION_Q047 && quic_version<=GQUIC_VERSION_Q059)) + { + state=gquic_frame_type_stream(pstream, _context, (char *)payload, payload_len, used_len, a_packet); + } + else + { + state=APP_STATE_DROPME; + } + + return state; +} + +//cid->version->nounce->pkt num->ahn hash(12) +int quic_process(struct streaminfo *pstream, struct _quic_context* _context, int thread_seq, void* a_packet) +{ + int used_len=0; + int ret=APP_STATE_GIVEME; + unsigned char decrypt_payload[1500]={0}; + unsigned int decrypt_payload_len=sizeof(decrypt_payload); + + enum _QUIC_VERSION is_gquic=QUIC_VERSION_UNKNOWN; + struct udpdetail *udp_detail=pstream->pudpdetail; + + if(udp_detail->pdata==NULL || udp_detail->datalen<=0) + { + return APP_STATE_GIVEME; + } + + is_gquic=is_quic_protocol(pstream, _context, (char *)udp_detail->pdata, udp_detail->datalen, &used_len); + if(is_gquic!=QUIC_VERSION_UNKNOWN) + { + if(_context->cb_version==0 && _context->call_business) + { + _context->cb_version=1; + ret=quic_callPlugins(pstream, _context, &(_context->quic_info.quic_hdr.quic_version), sizeof(_context->quic_info.quic_hdr.quic_version), QUIC_USEING_VERSION_MASK, a_packet); + if((ret&APP_STATE_DROPME) || (ret&APP_STATE_DROPPKT)) + { + return ret; + } + } + + switch(is_gquic) + { + case GQUIC_VERSION_Q044: + case GQUIC_VERSION_Q045: + case GQUIC_VERSION_Q046: + case GQUIC_VERSION_Q047: + case GQUIC_VERSION_Q048: + ret=parse_gquic_Q046(pstream, _context, a_packet, (char *)udp_detail->pdata, udp_detail->datalen, &used_len); + break; + default: + if(is_gquic>=GQUIC_VERSION_Q001 && is_gquic<=GQUIC_VERSION_Q046) + { + ret=gquic_proc_unencrypt(pstream, _context, a_packet, (char *)udp_detail->pdata, udp_detail->datalen, &used_len); + break; + } + + if( ((is_gquic>=MVFST_VERSION_00 && is_gquic<=MVFST_VERSION_0F) || + (is_gquic>=GQUIC_VERSION_Q049 && is_gquic<=GQUIC_VERSION_Q059) || + (is_gquic>=GQUIC_VERSION_T050 && is_gquic<=GQUIC_VERSION_T059) || + (is_gquic>=GQUIC_VERSION_T050 && is_gquic<=GQUIC_VERSION_T059) || + (is_gquic>=IQUIC_VERSION_I022 && is_gquic<=IQUIC_VERSION_I029) || + (is_gquic==IQUIC_VERSION_RFC9000) + ) + && _context->is_decrypt==0 + ) + { + _context->is_decrypt=1; + ret=dissect_quic((char *)udp_detail->pdata, udp_detail->datalen, decrypt_payload, &decrypt_payload_len); + if(ret!=1) + { + return APP_STATE_DROPME; + } + ret=parse_decrypt_quic(pstream, _context, a_packet, decrypt_payload, decrypt_payload_len, &used_len); + break; + } + + if(_context->call_business) + { + ret=quic_callPlugins(pstream, _context, (char *)udp_detail->pdata, udp_detail->datalen, QUIC_APPLICATION_DATA_MASK, a_packet); + if((ret&APP_STATE_DROPME) || (ret&APP_STATE_DROPPKT)) + { + return ret; + } + } + break; + } + + if((ret&APP_STATE_DROPME) || (ret&APP_STATE_DROPPKT)) + { + return ret; + } + } + + if(_context->is_quic==TRUE) + { + if(_context->quic_info.quic_hdr.is_reset && _context->call_business) + { + return quic_callPlugins(pstream, _context, NULL, 0, QUIC_INTEREST_KEY_MASK, a_packet); + } + + if(_context->quic_info.quic_hdr.is_version_negotiation && _context->call_business) + { + return quic_callPlugins(pstream, _context, NULL, 0, QUIC_NEGOTIATION_VERSION_MASK, a_packet); + } + return APP_STATE_GIVEME; + } + + return APP_STATE_DROPME;; +} + + +int quic_protocol_identify(struct streaminfo *a_stream, void *a_packet, char *out_sni, int out_sni_len) +{ + int ret=APP_STATE_GIVEME; + int sni_len=0,len=-1; + void *pme=NULL; + char *sni=NULL; + struct _quic_context *_context=NULL; + + if(!is_quic_port(a_stream)) + { + return len; + } + + quic_init_stream(&pme, a_stream->threadnum); + _context=(struct _quic_context *)pme; + + ret=quic_process(a_stream, _context, a_stream->threadnum, a_packet); + if(ret!=PROT_STATE_DROPME && _context->is_quic!=QUIC_VERSION_UNKNOWN) + { + if(_context->quic_info.client_hello!=NULL) + { + if(_context->quic_info.client_hello->sni_idx!=0xFF) + { + sni=(char *)(_context->quic_info.client_hello->ext_tags[_context->quic_info.client_hello->sni_idx].value); + sni_len=_context->quic_info.client_hello->ext_tags[_context->quic_info.client_hello->sni_idx].length; + len= sni_len>(out_sni_len-1) ? (out_sni_len-1) : sni_len; + memcpy(out_sni, sni, len); + } + else + { + len=0; + } + } + else + { + if(_context->is_quic==TRUE) + { + len=0; + } + } + } + + quic_release_stream(&pme, a_stream->threadnum); + + return len; +}