diff --git a/dd-java-agent/instrumentation/commons-fileupload/src/main/java/datadog/trace/instrumentation/commons/fileupload/CommonsFileuploadInstrumenter.java b/dd-java-agent/instrumentation/commons-fileupload/src/main/java/datadog/trace/instrumentation/commons/fileupload/CommonsFileuploadInstrumenter.java index 5291962d58d..5487a380572 100644 --- a/dd-java-agent/instrumentation/commons-fileupload/src/main/java/datadog/trace/instrumentation/commons/fileupload/CommonsFileuploadInstrumenter.java +++ b/dd-java-agent/instrumentation/commons-fileupload/src/main/java/datadog/trace/instrumentation/commons/fileupload/CommonsFileuploadInstrumenter.java @@ -11,11 +11,12 @@ import datadog.trace.api.iast.InstrumentationBridge; import datadog.trace.api.iast.Source; import datadog.trace.api.iast.SourceTypes; -import datadog.trace.api.iast.source.WebModule; import java.util.Arrays; import java.util.Collection; import java.util.Collections; import java.util.Map; + +import datadog.trace.api.iast.propagation.PropagationModule; import net.bytebuddy.asm.Advice; @AutoService(Instrumenter.class) @@ -50,10 +51,10 @@ public static class ParseAdvice { @Advice.OnMethodExit(suppress = Throwable.class) @Source(SourceTypes.REQUEST_MULTIPART_PARAMETER) public static Map onExit(@Advice.Return final Map map) { - final WebModule module = InstrumentationBridge.WEB; + final PropagationModule module = InstrumentationBridge.PROPAGATION; if (module != null) { for (Map.Entry entry : map.entrySet()) { - module.onMultipartValues(entry.getKey(), Collections.singleton(entry.getValue())); + module.taint(entry.getValue(), SourceTypes.REQUEST_MULTIPART_PARAMETER, entry.getKey()); } } return map;