From fa1da368b58a6c9559b618b9e6b80dbf87d745fa Mon Sep 17 00:00:00 2001
From: 15911075183ma <141103706+15911075183ma@users.noreply.github.com>
Date: Mon, 20 Nov 2023 18:39:20 +0800
Subject: [PATCH] =?UTF-8?q?fix=EF=BC=9Ahessian=E4=BF=AE=E5=A4=8D=20dubbo?=
 =?UTF-8?q?=E4=BD=BF=E7=94=A8hessian=E5=8D=8F=E8=AE=AE=E6=97=B6=EF=BC=8C?=
 =?UTF-8?q?=E4=B8=8A=E6=B8=B8=E5=9B=BE=E6=97=A0=E6=B3=95=E4=B8=B2=E8=81=94?=
 =?UTF-8?q?=EF=BC=8C=E6=9C=AC=E6=AC=A1=E4=BF=AE=E5=A4=8D=E6=AD=A4=E9=97=AE?=
 =?UTF-8?q?=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../core/handler/hookpoint/service/trace/DubboService.java     | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/dongtai-core/src/main/java/io/dongtai/iast/core/handler/hookpoint/service/trace/DubboService.java b/dongtai-core/src/main/java/io/dongtai/iast/core/handler/hookpoint/service/trace/DubboService.java
index e127198f5..752aeeb22 100644
--- a/dongtai-core/src/main/java/io/dongtai/iast/core/handler/hookpoint/service/trace/DubboService.java
+++ b/dongtai-core/src/main/java/io/dongtai/iast/core/handler/hookpoint/service/trace/DubboService.java
@@ -24,6 +24,9 @@ public static void solveSyncInvoke(MethodEvent event, Object invocation, String
                 Method method = invocation.getClass().getMethod("addHeader", String.class, String.class);
                 method.setAccessible(true);
                 method.invoke(invocation, ContextManager.getHeaderKey(), traceId);
+                //因为dubbo已经添加事件，我取出上次事件并对traceId进行修改
+                MethodEvent methodEvent = EngineManager.TRACK_MAP.get().get(invokeIdSequencer.get() - 1);
+                methodEvent.traceId = traceId;
                 return;
             }
             TaintPoolUtils.trackObject(event, null, event.parameterInstances, 0, false);