Increase IAST propagation to StringBuffer subSequence (#8038)

DataDog · Dec 12, 2024 · 9918d21 · 9918d21
1 parent 3d8130d
commit 9918d21
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/...ava-lang/src/main/java/datadog/trace/instrumentation/java/lang/StringBuilderCallSite.java b/...ava-lang/src/main/java/datadog/trace/instrumentation/java/lang/StringBuilderCallSite.java
@@ -162,6 +162,7 @@ public static String afterSubstring(
   }
 
   @CallSite.After("java.lang.CharSequence java.lang.StringBuilder.subSequence(int, int)")
+  @CallSite.After("java.lang.CharSequence java.lang.StringBuffer.subSequence(int, int)")
   public static CharSequence afterSubSequence(
       @CallSite.This final CharSequence self,
       @CallSite.Argument final int beginIndex,

diff --git a/.../src/test/groovy/datadog/trace/instrumentation/java/lang/StringBuilderCallSiteTest.groovy b/.../src/test/groovy/datadog/trace/instrumentation/java/lang/StringBuilderCallSiteTest.groovy
@@ -246,6 +246,7 @@ class StringBuilderCallSiteTest extends AgentTestRunner {
     where:
     type      | suite                        | param         | beginIndex | endIndex | expected
     "builder" | new TestStringBuilderSuite() | sb('012345')  | 1          | 5        | '1234'
+    "buffer"  | new TestStringBufferSuite()  | sbf('012345') | 1          | 5        | '1234'
   }
 
   private static class BrokenToString {