108 lines
2.8 KiB
C++
108 lines
2.8 KiB
C++
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
|
|
#include <MESA/MESA_prof_load.h>
|
|
#include <MESA/MESA_handle_logger.h>
|
|
|
|
#include "tsg_rule.h"
|
|
#include "tsg_entry.h"
|
|
|
|
|
|
static int send_raw_packet(struct traffic_mirror *ttm,char * pkt_ptr,int pkt_len, struct mirrored_vlan *vlan,int vlan_num,int thread_seq)
|
|
{
|
|
int i=0,ret=0;
|
|
for(i=0; i<vlan_num; i++)
|
|
{
|
|
ret=tsg_traffic_mirror_send_burst(ttm, (char *)pkt_ptr, pkt_len, &(vlan[i].vlan_id), 1, thread_seq);
|
|
if(ret==TRAFFIC_MIRROR_SEND_SUCCESS)
|
|
{
|
|
FS_operate(g_tsg_para.fs2_handle, g_tsg_para.fs2_field_id[TSG_FS2_MIRRORED_PKT_SUCCESS], 0, FS_OP_ADD, 1);
|
|
FS_operate(g_tsg_para.fs2_handle, g_tsg_para.fs2_field_id[TSG_FS2_MIRRORED_BYTE_SUCCESS], 0, FS_OP_ADD, pkt_len);
|
|
}
|
|
else
|
|
{
|
|
FS_operate(g_tsg_para.fs2_handle, g_tsg_para.fs2_field_id[TSG_FS2_MIRRORED_PKT_FAILED], 0, FS_OP_ADD, 1);
|
|
FS_operate(g_tsg_para.fs2_handle, g_tsg_para.fs2_field_id[TSG_FS2_MIRRORED_BYTE_FAILED], 0, FS_OP_ADD, pkt_len);
|
|
}
|
|
|
|
vlan[i].mirrored_pkts+=1;
|
|
vlan[i].mirrored_bytes+=pkt_len;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int tsg_send_raw_packet(const struct streaminfo *a_stream, struct mirrored_vlan *vlan, int vlan_num, int thread_seq)
|
|
{
|
|
int ret=0;
|
|
void *p_eth_rawpkt=NULL;
|
|
int eth_rawpkt_len=0;
|
|
raw_ipfrag_list_t *frag_pkt=NULL;
|
|
|
|
if(g_tsg_para.mirror_switch==0 || vlan==NULL || vlan_num<=0)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
ret=get_rawpkt_opt_from_streaminfo(a_stream, RAW_PKT_GET_DATA, &p_eth_rawpkt);
|
|
switch(0)
|
|
{
|
|
case 0:
|
|
ret=get_rawpkt_opt_from_streaminfo(a_stream, RAW_PKT_GET_TOT_LEN, ð_rawpkt_len);
|
|
if(ret<0)
|
|
{
|
|
break;
|
|
}
|
|
|
|
send_raw_packet(g_tsg_para.mirror_handle, (char *)p_eth_rawpkt, eth_rawpkt_len, vlan, vlan_num, thread_seq);
|
|
break;
|
|
case 1:
|
|
frag_pkt=(raw_ipfrag_list_t *)p_eth_rawpkt;
|
|
while(frag_pkt)
|
|
{
|
|
send_raw_packet(g_tsg_para.mirror_handle, (char *)(frag_pkt->frag_packet), frag_pkt->pkt_len, vlan, vlan_num, thread_seq);
|
|
frag_pkt = frag_pkt->next;
|
|
}
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int tsg_notify_hited_monitor_result(const struct streaminfo *a_stream, struct Maat_rule_t *result, int result_num, int thread_seq)
|
|
{
|
|
if(g_tsg_para.mirror_switch==0)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
int vlan_num=0;
|
|
struct mirrored_vlan vlan[MAX_RESULT_NUM]={0};
|
|
struct tcpall_context *context=NULL;
|
|
|
|
vlan_num=tsg_get_vlan_id_by_monitor_rule(g_tsg_maat_feather, result, result_num, vlan, MAX_RESULT_NUM);
|
|
tsg_set_vlan_id_to_tcpall(a_stream, &context, vlan, vlan_num, thread_seq);
|
|
|
|
return 1;
|
|
}
|
|
|
|
int tsg_send_raw_packet_init(const char* conffile, void *logger)
|
|
{
|
|
MESA_load_profile_short_def(conffile, "TRAFFIC_MIRROR", "TRAFFIC_MIRROR_ENABLE", &g_tsg_para.mirror_switch, 0);
|
|
if(g_tsg_para.mirror_switch==0)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
g_tsg_para.mirror_handle=tsg_traffic_mirror_init(conffile, g_tsg_para.logger);
|
|
if(g_tsg_para.mirror_handle==NULL)
|
|
{
|
|
return -1;
|
|
}
|
|
|
|
return 0;
|
|
}
|