From ca0a1024c4492e91ec1ce2a8d429206f2d7b0983 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=B4=94=E4=B8=80=E9=B8=A3?= Date: Thu, 12 Sep 2019 16:13:35 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9kni=E6=8B=A6=E6=88=AA?= =?UTF-8?q?=E5=AD=97=E8=8A=82=E6=95=B0=E7=BB=9F=E8=AE=A1=E7=AC=94=E8=AF=AF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- entry/src/kni_entry.cpp | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/entry/src/kni_entry.cpp b/entry/src/kni_entry.cpp index c850601..4176a06 100644 --- a/entry/src/kni_entry.cpp +++ b/entry/src/kni_entry.cpp @@ -726,8 +726,6 @@ static int send_to_tfe_normal_mode(char *raw_data, uint16_t raw_len, int thread_ } static int send_to_tfe_tun_mode(char *raw_data, uint16_t raw_len, addr_type_t addr_type){ - //intercept traffic stat - FS_operate(g_kni_fs_handle->handle, g_kni_fs_handle->fields[KNI_FIELD_KNI_INTCP_BYTES], 0, FS_OP_ADD, raw_len); struct kni_tun_handle *handle = g_kni_handle->tun_handle; char *dst_data = ALLOC(char, KNI_MTU); add_ether_header(dst_data, raw_data, raw_len, addr_type); @@ -748,6 +746,10 @@ static int send_to_tfe(char *raw_data, uint16_t raw_len, int thread_seq, int tfe else{ ret = send_to_tfe_normal_mode(raw_data, raw_len, thread_seq, tfe_id, addr_type); } + if(ret >= 0){ + //intercept traffic stat + FS_operate(g_kni_fs_handle->handle, g_kni_fs_handle->fields[KNI_FIELD_KNI_INTCP_BYTES], 0, FS_OP_ADD, raw_len); + } return ret; }