fix：hessian修复

dubbo使用hessian协议时，上游图无法串联，本次修复此问题
HXSecurity · Nov 20, 2023 · fa1da36 · fa1da36
1 parent f56b85d
commit fa1da36
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);