Merge pull request #606 from 15911075183ma/fix-dubbo-hessian-upstream…

…Diagram Fix dubbo hessian upstream diagram
HXSecurity · Nov 21, 2023 · d4627d4 · d4627d4
2 parents 650d3a9 + fa1da36
commit d4627d4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/...core/src/main/java/io/dongtai/iast/core/handler/hookpoint/service/trace/DubboService.java b/...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);