Example #1
0
static int qmqpd_next_line(VSTRING *message, char **start, int *len,
			           char **next)
{
    char   *beyond = STR(message) + LEN(message);
    char   *enough = *next + var_line_limit;
    char   *cp;

    /*
     * Stop at newline or at some limit. Don't look beyond the end of the
     * buffer.
     */
#define UCHARPTR(x) ((unsigned char *) (x))

    for (cp = *start = *next; /* void */ ; cp++) {
	if (cp >= beyond)
	    return ((*len = (*next = cp) - *start) > 0 ? UCHARPTR(cp)[-1] : -1);
	if (*cp == '\n')
	    return ((*len = cp - *start), (*next = cp + 1), '\n');
	if (cp >= enough)
	    return ((*len = cp - *start), (*next = cp), UCHARPTR(cp)[-1]);
    }
}
Example #2
0
int CspPackRefrom(struct audit_pack_info_head * p_info_hd,struct audit_pack_info * p_info,unsigned char * payload){
	char * data;
	char cspFilePathTmp[AUDIT_PATH_LEN] = {0},cspFilePath[AUDIT_PATH_LEN] = {0};
	char times[AUDIT_TIME_LEN] = {0},userip[AUDIT_IPADDR_LEN] = {0},desip[AUDIT_IPADDR_LEN]={0},src_mac[AUDIT_MAC_LEN] = {0},des_mac[AUDIT_MAC_LEN]={0};
	char conStr[AUDIT_PACK_LEN]={0};
	char requestUrl[AUDIT_URL_LEN];
	int cspFd;
	int type;
	int flag;
	int dir = 0;
	unsigned int policy_time;
	unsigned char * pos = NULL;
	int str_len = 0;
	int len = p_info_hd->payload_len;
#if CSP_DEBUG
	CspDebug( p_info_hd,p_info,payload);
#endif

//	printf("seq:%u\n",p_info_hd->seq);
         if(p_info_hd->seq > 4294967295) return 0;
	dir = p_info_hd->hash%10;

	if(p_info->audit_type == AUDIT_CACHE_CSP_HIS){
		type = AUDIT_CSP_TYPE_HIS;
	}
	else if(p_info->audit_type == AUDIT_CACHE_CSP_PORTAL){
		type = AUDIT_CSP_TYPE_PORTAL;
	}
	
	if(p_info_hd->isin){
		if( p_info->flag ==1){
			if(memcmp(payload,"HTTP/1.",7)==0)
			{
			
			char sip[100],dip[100];
			
			sprintf(dip,"%d.%d.%d.%d",(int)UCHARPTR(&p_info_hd->ser_ip)[0],(int)UCHARPTR(&p_info_hd->ser_ip)[1],(int)UCHARPTR(&p_info_hd->ser_ip)[2],(int)UCHARPTR(&p_info_hd->ser_ip)[3]);
	//	if(p_info->audit_type == AUDIT_CACHE_CSP_PORTAL)
	//	printf("HTTP  %d hash:%lu serpt:%u dip:%s seq:%u\n",dir,p_info_hd->hash,p_info_hd->ser_port,dip,p_info_hd->seq);
							
				p_info->in_seq = p_info_hd->seq;
				p_info->flag =2;
				memset(cspFilePathTmp,0,sizeof(cspFilePathTmp));
//				CspGetPath(cspFilePathTmp,AUDIT_TMP_CSP_PATH,p_info->id);
				sprintf(cspFilePathTmp,"%s%d/P_%lu",AUDIT_TMP_CSP_PATH,dir,p_info->id);
				cspFd = open(cspFilePathTmp, O_RDWR | O_CREAT, 0666);
				if(cspFd < 2) return 0;
				p_info->in_offset = AuditWrite(cspFd,payload,len);
				if(p_info->in_offset < 0){
					p_info->in_offset=0;
					close(cspFd);
					return 0;
				}
				close(cspFd);

			
				return 0;
			}
			return 0;
		}
		else if( p_info->flag == 2){
			if(memcmp(payload,"HTTP/1.",7)==0){
				return 0;
			}
			
		//	printf("intseq:%u,seq:%u,len :%u\n",p_info->in_seq,p_info_hd->seq,p_info_hd->seq-p_info->in_seq);
			memset(cspFilePathTmp,0,sizeof(cspFilePathTmp));
			sprintf(cspFilePathTmp,"%s%d/P_%lu",AUDIT_TMP_CSP_PATH,dir,p_info->id);
			cspFd = open(cspFilePathTmp, O_RDWR | O_CREAT, 0666);
			if(cspFd < 2) return 0;
	//		lseek(cspFd,p_info->in_offset, SEEK_SET);
			lseek(cspFd,p_info_hd->seq - p_info->in_seq, SEEK_SET);
			AuditWrite(cspFd,payload,len);
			close(cspFd);

			return 0;
		}	
		if(p_info_hd->rst == 1 || p_info_hd->fin == 1){
				if(p_info->id !=0){
					AuditFileRename(AUDIT_TMP_CSP_PATH,AUDIT_CSP_CONTENT_PATH,p_info->id, HTTP_REQUEST,dir);
					AuditFileRename(AUDIT_TMP_CSP_PATH,AUDIT_CSP_CONTENT_PATH,p_info->id,HTTP_RESPONSE,dir);
					p_info->in_offset = 0;
					p_info->id = 0;
				}
				p_info->flag = 0;
				//memset(p_info,0,sizeof(*p_info));
				return 0;
		}
		return 0;

	}
	else{
		
		if(p_info_hd->fin ==1||p_info_hd->rst == 1  ){
			if(p_info->id !=0){
				AuditFileRename(AUDIT_TMP_CSP_PATH,AUDIT_CSP_CONTENT_PATH,p_info->id, HTTP_REQUEST,dir);
				AuditFileRename(AUDIT_TMP_CSP_PATH,AUDIT_CSP_CONTENT_PATH,p_info->id,HTTP_RESPONSE,dir);
				p_info->in_offset = 0;
				p_info->id = 0;
			}
			p_info->flag = 0;
	//		memset(p_info,0,sizeof(*p_info));
			return 0;
		}
		
		if(memcmp(payload,"POST ",5) == 0){		
			
			if(p_info->flag == 2 ) {
				if(p_info->id !=0){
					AuditFileRename(AUDIT_TMP_CSP_PATH,AUDIT_CSP_CONTENT_PATH,p_info->id, HTTP_REQUEST,dir);
					AuditFileRename(AUDIT_TMP_CSP_PATH,AUDIT_CSP_CONTENT_PATH,p_info->id,HTTP_RESPONSE,dir);
					p_info->in_offset = 0;
					p_info->id = 0;
				}
				p_info->flag = 0;
			}
			memset(requestUrl,0,sizeof(requestUrl));
			pos = strstr(payload," HTTP/1.");
			if(!pos){
				return 0;
			}
			str_len  =  pos - payload ;
			str_len-=4;
			if(str_len < 0) return 0;
			if(str_len >= AUDIT_URL_LEN){
				str_len = AUDIT_URL_LEN-1;
			}
			memcpy(requestUrl,payload+5,str_len);
			
			pos = strstr(requestUrl,"?");
			if(pos){
				*pos = '\0';
			}


			if(!strstr(requestUrl,"dhc.logon.csp") && !strstr(requestUrl,"UtilHome.csp")){
				return 0;
			}
			if(pos){
				*pos = '?';
			}		//url match
			if(type == AUDIT_CSP_TYPE_HIS){
				pthread_rwlock_rdlock(&(p_info_hd->policy->audit_lock));
				flag = audit_cache_policy_match(type,p_info_hd->cli_ip,p_info_hd->timenow,requestUrl,p_info_hd->policy,p_info_hd->redisConn);
				pthread_rwlock_unlock(&(p_info_hd->policy->audit_lock));
					if(flag == 0){
						return 0;
					}
			}
	//		printf("POST %s\n",requestUrl);
	//		if(strncmp(payload+5,AUDIT_CSP_REQUEST_LOGON,strlen(AUDIT_CSP_REQUEST_LOGON)) != 0 &&strncmp(payload+5,AUDIT_CSP_REQUEST_LOGON_PORTAL,strlen(AUDIT_CSP_REQUEST_LOGON_PORTAL)) != 0  ) return 0;
			if(p_info->out_seq != p_info_hd->seq){
				pthread_mutex_lock(&csp_id_mutex);
				p_info->id=++csp_id;
		//		csp_id_hash[csp_id%AUDIT_INDEX_NUM] = p_info_hd->hash;
				SetId(AUDIT_CSP_ID_PATH,csp_id);
				pthread_mutex_unlock(&csp_id_mutex);
				p_info->out_seq = p_info_hd->seq;
			}
			
			#if REL_HBASE
				sprintf(times,"%lu",p_info_hd->audit_time);
			#else
				get_audit_time(times);
			#endif
				
	//		CspGetPath(cspFilePathTmp,AUDIT_TMP_CSP_PATH,p_info->id);
			sprintf(cspFilePathTmp,"%s%d/Q_%lu",AUDIT_TMP_CSP_PATH,dir,p_info->id);
			
			CspGetIp(userip,p_info_hd->cli_ip);

			CspGetIp(desip,p_info_hd->ser_ip);
			
			CspGetMac(src_mac,p_info_hd->cli_mac);

			CspGetMac(des_mac,p_info_hd->ser_mac);

		//	policy_time = get_min();
			
			pthread_rwlock_rdlock(&tablesTime_lock);
			sprintf(conStr,"%llu|%d|%s|%s|%s|%s|%s|%u|%u|%u|%d|%s|&\r\n",p_info->id,p_info_hd->user_id,times,userip,src_mac,desip,des_mac,p_info_hd->cli_port,p_info_hd->ser_port,p_info_hd->timenow,type,tablesTime);
			pthread_rwlock_unlock(&tablesTime_lock);
			cspFd=open(cspFilePathTmp, O_RDWR | O_CREAT, 0666);
			if(cspFd <= 2) return 0;
			p_info->out_offset = AuditWrite(cspFd,conStr,strlen(conStr));
			if(p_info->out_offset < 0){
				p_info->out_offset = 0;
				close(cspFd);
				return 0;
			}
			
			close(cspFd);
			p_info->flag=1;
		}
		else if(memcmp(payload,"GET ",4)==0){
			if(p_info->flag == 2 ) {
				if(p_info->id !=0){
					AuditFileRename(AUDIT_TMP_CSP_PATH,AUDIT_CSP_CONTENT_PATH,p_info->id, HTTP_REQUEST,dir);
					AuditFileRename(AUDIT_TMP_CSP_PATH,AUDIT_CSP_CONTENT_PATH,p_info->id,HTTP_RESPONSE,dir);
					p_info->in_offset = 0;
					p_info->id = 0;
				}
				p_info->flag = 0;
			}
			memset(requestUrl,0,sizeof(requestUrl));
			pos = strstr(payload," HTTP/1.");
			if(!pos){
				return 0;
			}
			str_len  =  pos - payload ;
			str_len-=4;
			if(str_len < 0) return 0;
			if(str_len >= AUDIT_URL_LEN){
				str_len = AUDIT_URL_LEN-1;
			}
			memcpy(requestUrl,payload+4,str_len);


			pos = strstr(requestUrl,"?");
			if(pos){
				*pos = '\0';
			}

			if(strstr(requestUrl,"dhcvisanopwait")){
				return 0;
			}
			if(strstr(requestUrl,"dhcvispatwait")){
				return 0;
			}
	//		dhcrisappbill  dhcnuripexeclist

			if(strstr(requestUrl,"dhcrisappbill")){
				return 0;
			}

			if(strstr(requestUrl,"dhcnuripexeclist")){
				return 0;
			}
			if(strstr(requestUrl,"dhc.epr.messagetab.csp")){
				return 0;
			}
			if(strstr(requestUrl,"dhc.bdp.ext.sys.csp")){
				return 0;
			}

			

			
			if(!strstr(requestUrl,".csp") &&!strstr(requestUrl,".cls"))
				return 0;

			if(pos){
				*pos = '?';
			}
			if(type == AUDIT_CSP_TYPE_HIS){
		//		printf("url:%s\n",requestUrl);
				pthread_rwlock_rdlock(&(p_info_hd->policy->audit_lock));
				flag = audit_cache_policy_match(type,p_info_hd->cli_ip,p_info_hd->timenow,requestUrl,p_info_hd->policy,p_info_hd->redisConn);
				pthread_rwlock_unlock(&(p_info_hd->policy->audit_lock));
			//	printf("url:%s,flag = %d\n",requestUrl,flag);
				if(flag == 0){
					return 0;
				}
			}


			char sip[100],dip[100];
			
			sprintf(dip,"%d.%d.%d.%d",(int)UCHARPTR(&p_info_hd->ser_ip)[0],(int)UCHARPTR(&p_info_hd->ser_ip)[1],(int)UCHARPTR(&p_info_hd->ser_ip)[2],(int)UCHARPTR(&p_info_hd->ser_ip)[3]);

			
			//if(p_info->audit_type == AUDIT_CACHE_CSP_PORTAL)
		//	printf("GET %s %d hash:%lu serpt:%u dip:%s seq:%u",requestUrl,dir,p_info_hd->hash,p_info_hd->ser_port,dip,p_info_hd->seq);
/*
			if(strcmp(&requestUrl[str_len-4],".gif")==0||strcmp(&requestUrl[str_len-4],".css")==0||strcmp(&requestUrl[str_len-4],".js")==0||strcmp(&requestUrl[str_len-3],".png")==0||strcmp(&requestUrl[str_len-3],".jpg")==0 ){
				return 0
			}
			
			if(strncmp(payload+4,AUDIT_CSP_REQUEST_CSP,strlen(AUDIT_CSP_REQUEST_CSP)) != 0 && strncmp(payload+4,AUDIT_CSP_GLOBLE_VIEW_CSP,strlen(AUDIT_CSP_GLOBLE_VIEW_CSP))!=0 && strncmp(payload+4,AUDIT_CSP_SQL_VIEW_CSP,strlen(AUDIT_CSP_SQL_VIEW_CSP))!=0) {
				return 0;
			}
*/			
	//		printf("0 outseq:%u,seq:%u\n",p_info->out_seq,p_info_hd->seq);

			if(p_info->out_seq != p_info_hd->seq){
				pthread_mutex_lock(&csp_id_mutex);
				p_info->id=++csp_id;
			//	csp_id_hash[csp_id%AUDIT_INDEX_NUM] = p_info_hd->hash;
				SetId(AUDIT_CSP_ID_PATH,csp_id);
				pthread_mutex_unlock(&csp_id_mutex);
				p_info->out_seq = p_info_hd->seq;
			}
		//	if(p_info->audit_type == AUDIT_CACHE_CSP_PORTAL)
	//		printf(" id:%lu\n",p_info->id);	
			
			
			#if REL_HBASE
				sprintf(times,"%lu",p_info_hd->audit_time);
			#else
				get_audit_time(times);
			#endif
//			CspGetPath(cspFilePathTmp,AUDIT_TMP_CSP_PATH,p_info->id);
			sprintf(cspFilePathTmp,"%s%d/Q_%lu",AUDIT_TMP_CSP_PATH,dir,p_info->id);

			CspGetIp(userip,p_info_hd->cli_ip);

			CspGetIp(desip,p_info_hd->ser_ip);
			
			CspGetMac(src_mac,p_info_hd->cli_mac);

			CspGetMac(des_mac,p_info_hd->ser_mac);

	//		policy_time = get_min();
			pthread_rwlock_rdlock(&tablesTime_lock);
			sprintf(conStr,"%llu|%d|%s|%s|%s|%s|%s|%u|%u|%u|%d|%s|&\r\n",p_info->id,p_info_hd->user_id,times,userip,src_mac,desip,des_mac,p_info_hd->cli_port,p_info_hd->ser_port,p_info_hd->timenow,type,tablesTime);
			pthread_rwlock_unlock(&tablesTime_lock);		
			cspFd=open(cspFilePathTmp, O_RDWR | O_CREAT, 0666);
			if(cspFd <= 2) return 0;
			p_info->out_offset = AuditWrite(cspFd,conStr,strlen(conStr));
			if(p_info->out_offset < 0){
				p_info->out_offset = 0;
				close(cspFd);
				return 0;
			}
			close(cspFd);
			p_info->flag=1;
		}
		if( p_info->flag==0)
			return 0;
		if( p_info->flag==1){
			
	///		printf("1 outseq:%u,seq:%u\n",p_info->out_seq,p_info_hd->seq);
			memset(cspFilePathTmp,0,sizeof(cspFilePathTmp));
			//CspGetPath(cspFilePathTmp,AUDIT_TMP_CSP_PATH,p_info->id);
			sprintf(cspFilePathTmp,"%s%d/Q_%lu",AUDIT_TMP_CSP_PATH,dir,p_info->id);
			cspFd = open(cspFilePathTmp, O_RDWR | O_CREAT, 0666);
			if(cspFd <= 2) return 0;
			lseek(cspFd,p_info->out_offset, SEEK_SET);
			lseek(cspFd,p_info_hd->seq - p_info->out_seq, SEEK_CUR);
			
			AuditWrite(cspFd,payload,len);
			// p_info->writtenbyes += len;
			AuditWrite(cspFd,"\r\n",2);
		//	 p_info->writtenbyes += 2;
			 
			close(cspFd);
			return 0;
		}
	}
}
Example #3
0
static void CspGetIp(OUT char * ip, IN u_int32_t int_ip){
	sprintf(ip,"%d.%d.%d.%d",(int) UCHARPTR(&int_ip)[0],(int) UCHARPTR(&int_ip)[1],(int)UCHARPTR(&int_ip)[2],(int)UCHARPTR(&int_ip)[3]);
}