From 4d7e51edf61e46668df7de428dc628a1b1430f3a Mon Sep 17 00:00:00 2001
From: huanchao <huanchao@yunshan.net>
Date: Thu, 12 Oct 2023 18:13:39 +0800
Subject: [PATCH] [Agent] Enhance the direction judgment of eBPF data

---
 agent/src/common/meta_packet.rs      | 2 ++
 agent/src/flow_generator/flow_map.rs | 5 +++++
 2 files changed, 7 insertions(+)

diff --git a/agent/src/common/meta_packet.rs b/agent/src/common/meta_packet.rs
index 73e35267105..70ad42d9d51 100644
--- a/agent/src/common/meta_packet.rs
+++ b/agent/src/common/meta_packet.rs
@@ -169,6 +169,7 @@ pub struct MetaPacket<'a> {
     pub process_kname: [u8; PACKET_KNAME_MAX_PADDING], // kernel process name
     // for PcapAssembler
     pub flow_id: u64,
+    pub socket_role: u8,
 
     /********** for GPID **********/
     pub gpid_0: u32,
@@ -932,6 +933,7 @@ impl<'a> MetaPacket<'a> {
         packet.thread_id = data.thread_id;
         packet.coroutine_id = data.coroutine_id;
         packet.syscall_trace_id = data.syscall_trace_id_call;
+        packet.socket_role = data.socket_role;
         #[cfg(target_arch = "aarch64")]
         ptr::copy(
             data.process_kname.as_ptr() as *const u8,
diff --git a/agent/src/flow_generator/flow_map.rs b/agent/src/flow_generator/flow_map.rs
index 444ae0e0030..cd5a8b0f982 100644
--- a/agent/src/flow_generator/flow_map.rs
+++ b/agent/src/flow_generator/flow_map.rs
@@ -1089,6 +1089,11 @@ impl FlowMap {
             ],
             signal_source: meta_packet.signal_source,
             is_active_service,
+            direction_score: if meta_packet.socket_role > 0 {
+                ServiceTable::MAX_SCORE
+            } else {
+                0
+            },
             ..Default::default()
         };
         tagged_flow.flow = flow;