From 9207c4bb850844d13bfd622653f3c02e0b6969b8 Mon Sep 17 00:00:00 2001
From: Javier Santos <javier.santos@datadoghq.com>
Date: Thu, 2 Nov 2023 12:17:54 +0100
Subject: [PATCH] rebase to master

---
 .../commons/fileupload/CommonsFileuploadInstrumenter.java  | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

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<String, String> onExit(@Advice.Return final Map<String, String> map) {
-      final WebModule module = InstrumentationBridge.WEB;
+      final PropagationModule module = InstrumentationBridge.PROPAGATION;
       if (module != null) {
         for (Map.Entry<String, String> entry : map.entrySet()) {
-          module.onMultipartValues(entry.getKey(), Collections.singleton(entry.getValue()));
+          module.taint(entry.getValue(), SourceTypes.REQUEST_MULTIPART_PARAMETER, entry.getKey());
         }
       }
       return map;