Tag sockets traffic originating from Sentry's HttpConnection

Author: romtsn

sentry-android-core/api/sentry-android-core.api

@@ -111,6 +111,12 @@ public class io/sentry/android/core/AndroidProfiler$ProfileStartData {
 	public fun <init> (JJLjava/util/Date;)V
 }
 
+public final class io/sentry/android/core/AndroidSocketTagger : io/sentry/ISocketTagger {
+	public static fun getInstance ()Lio/sentry/android/core/AndroidSocketTagger;
+	public fun tagSockets ()V
+	public fun untagSockets ()V
+}
+
 public final class io/sentry/android/core/AnrIntegration : io/sentry/Integration, java/io/Closeable {
 	public fun <init> (Landroid/content/Context;)V
 	public fun close ()V

sentry-android-core/src/main/java/io/sentry/android/core/AndroidOptionsInitializer.java

@@ -15,6 +15,7 @@
 import io.sentry.NoOpCompositePerformanceCollector;
 import io.sentry.NoOpConnectionStatusProvider;
 import io.sentry.NoOpContinuousProfiler;
+import io.sentry.NoOpSocketTagger;
 import io.sentry.NoOpTransactionProfiler;
 import io.sentry.NoopVersionDetector;
 import io.sentry.ScopeType;
@@ -238,6 +239,9 @@ static void initializeIntegrationsAndProcessors(
     if (options.getThreadChecker() instanceof NoOpThreadChecker) {
       options.setThreadChecker(AndroidThreadChecker.getInstance());
     }
+    if (options.getSocketTagger() instanceof NoOpSocketTagger) {
+      options.setSocketTagger(AndroidSocketTagger.getInstance());
+    }
     if (options.getPerformanceCollectors().isEmpty()) {
       options.addPerformanceCollector(new AndroidMemoryCollector());
       options.addPerformanceCollector(new AndroidCpuCollector(options.getLogger()));

sentry-android-core/src/main/java/io/sentry/android/core/AndroidSocketTagger.java

@@ -0,0 +1,27 @@
+package io.sentry.android.core;
+
+import android.net.TrafficStats;
+import io.sentry.ISocketTagger;
+import org.jetbrains.annotations.ApiStatus;
+
+@ApiStatus.Internal
+public final class AndroidSocketTagger implements ISocketTagger {
+
+  private static final AndroidSocketTagger instance = new AndroidSocketTagger();
+
+  private AndroidSocketTagger() {}
+
+  public static AndroidSocketTagger getInstance() {
+    return instance;
+  }
+
+  @Override
+  public void tagSockets() {
+    TrafficStats.setThreadStatsTag(1);
+  }
+
+  @Override
+  public void untagSockets() {
+    TrafficStats.clearThreadStatsTag();
+  }
+}

sentry-android-core/src/test/java/io/sentry/android/core/AndroidOptionsInitializerTest.kt

@@ -11,6 +11,7 @@ import io.sentry.DefaultCompositePerformanceCollector
 import io.sentry.IConnectionStatusProvider
 import io.sentry.IContinuousProfiler
 import io.sentry.ILogger
+import io.sentry.ISocketTagger
 import io.sentry.ITransactionProfiler
 import io.sentry.MainEventProcessor
 import io.sentry.NoOpContinuousProfiler
@@ -746,6 +747,13 @@ class AndroidOptionsInitializerTest {
         assertTrue { fixture.sentryOptions.threadChecker is AndroidThreadChecker }
     }
 
+    @Test
+    fun `AndroidSocketTagger is set to options`() {
+        fixture.initSut()
+
+        assertTrue { fixture.sentryOptions.socketTagger is AndroidSocketTagger }
+    }
+
     @Test
     fun `does not install ComposeGestureTargetLocator, if sentry-compose is not available`() {
         fixture.initSutWithClassLoader()
@@ -859,6 +867,7 @@ class AndroidOptionsInitializerTest {
             setModulesLoader(mock<IModulesLoader>())
             setDebugMetaLoader(mock<IDebugMetaLoader>())
             threadChecker = mock<IThreadChecker>()
+            setSocketTagger(mock<ISocketTagger>())
             compositePerformanceCollector = mock<CompositePerformanceCollector>()
         })
 
@@ -868,6 +877,7 @@ class AndroidOptionsInitializerTest {
         assertFalse { fixture.sentryOptions.modulesLoader is AssetsModulesLoader }
         assertFalse { fixture.sentryOptions.debugMetaLoader is AssetsDebugMetaLoader }
         assertFalse { fixture.sentryOptions.threadChecker is AndroidThreadChecker }
+        assertFalse { fixture.sentryOptions.socketTagger is AndroidSocketTagger }
         assertFalse { fixture.sentryOptions.compositePerformanceCollector is DefaultCompositePerformanceCollector }
     }
 }

sentry/api/sentry.api

@@ -1028,6 +1028,11 @@ public abstract interface class io/sentry/ISerializer {
 	public abstract fun serialize (Ljava/util/Map;)Ljava/lang/String;
 }
 
+public abstract interface class io/sentry/ISocketTagger {
+	public abstract fun tagSockets ()V
+	public abstract fun untagSockets ()V
+}
+
 public abstract interface class io/sentry/ISpan {
 	public abstract fun finish ()V
 	public abstract fun finish (Lio/sentry/SpanStatus;)V
@@ -1696,6 +1701,12 @@ public final class io/sentry/NoOpScopesStorage : io/sentry/IScopesStorage {
 	public fun set (Lio/sentry/IScopes;)Lio/sentry/ISentryLifecycleToken;
 }
 
+public final class io/sentry/NoOpSocketTagger : io/sentry/ISocketTagger {
+	public static fun getInstance ()Lio/sentry/ISocketTagger;
+	public fun tagSockets ()V
+	public fun untagSockets ()V
+}
+
 public final class io/sentry/NoOpSpan : io/sentry/ISpan {
 	public fun finish ()V
 	public fun finish (Lio/sentry/SpanStatus;)V
@@ -3105,6 +3116,7 @@ public class io/sentry/SentryOptions {
 	public fun getSessionReplay ()Lio/sentry/SentryReplayOptions;
 	public fun getSessionTrackingIntervalMillis ()J
 	public fun getShutdownTimeoutMillis ()J
+	public fun getSocketTagger ()Lio/sentry/ISocketTagger;
 	public fun getSpanFactory ()Lio/sentry/ISpanFactory;
 	public fun getSpotlightConnectionUrl ()Ljava/lang/String;
 	public fun getSslSocketFactory ()Ljavax/net/ssl/SSLSocketFactory;
@@ -3241,6 +3253,7 @@ public class io/sentry/SentryOptions {
 	public fun setSessionReplay (Lio/sentry/SentryReplayOptions;)V
 	public fun setSessionTrackingIntervalMillis (J)V
 	public fun setShutdownTimeoutMillis (J)V
+	public fun setSocketTagger (Lio/sentry/ISocketTagger;)V
 	public fun setSpanFactory (Lio/sentry/ISpanFactory;)V
 	public fun setSpotlightConnectionUrl (Ljava/lang/String;)V
 	public fun setSslSocketFactory (Ljavax/net/ssl/SSLSocketFactory;)V

sentry/src/jmh/java/io/sentry/benchmark/StringAfterDotBenchmark.java

@@ -0,0 +1,60 @@
+package io.sentry.benchmark;
+
+import java.util.concurrent.TimeUnit;
+import org.jetbrains.annotations.NotNull;
+import org.openjdk.jmh.annotations.*;
+
+@BenchmarkMode(Mode.Throughput)
+@OutputTimeUnit(TimeUnit.MILLISECONDS)
+@State(Scope.Thread)
+public class StringAfterDotBenchmark {
+
+  @NotNull private String testString = "";
+
+  private char[] buf = new char[64];
+
+  @Setup
+  public void setup() {
+    testString = "com.example.deep.package.ClassName";
+  }
+  //
+  // @Benchmark
+  // public String usingSubstring() {
+  //    int idx = testString.lastIndexOf('.');
+  //    return (idx >= 0 && idx + 1 < testString.length())
+  //            ? testString.substring(idx + 1)
+  //            : testString;
+  // }
+  //
+  // @Benchmark
+  // public String usingManualCharCopy() {
+  //    int len = testString.length();
+  //    for (int i = len - 1; i >= 0; i--) {
+  //        if (testString.charAt(i) == '.') {
+  //            int newLen = len - i - 1;
+  //            char[] buf = new char[newLen];
+  //            testString.getChars(i + 1, len, buf, 0);
+  //            return new String(buf);
+  //        }
+  //    }
+  //    return testString;
+  // }
+
+  @Benchmark
+  public String usingThreadLocalBuffer() {
+    int len = testString.length();
+    int bufIndex = buf.length;
+
+    for (int i = len - 1; i >= 0; i--) {
+      char c = testString.charAt(i);
+      if (c == '.') {
+        int suffixLen = buf.length - bufIndex;
+        return new String(buf, bufIndex, suffixLen);
+      }
+      buf[--bufIndex] = c;
+    }
+
+    // No dot found — return original
+    return testString;
+  }
+}

sentry/src/main/java/io/sentry/ISocketTagger.java

@@ -0,0 +1,9 @@
+package io.sentry;
+
+public interface ISocketTagger {
+  /** Tags the sockets traffic originating from the Sentry HttpConnection thread. */
+  void tagSockets();
+
+  /** Untags the sockets traffic originating from the Sentry HttpConnection thread. */
+  void untagSockets();
+}

sentry/src/main/java/io/sentry/NoOpSocketTagger.java

@@ -0,0 +1,24 @@
+package io.sentry;
+
+import org.jetbrains.annotations.NotNull;
+
+public final class NoOpSocketTagger implements ISocketTagger {
+
+  private static final NoOpSocketTagger instance = new NoOpSocketTagger();
+
+  private NoOpSocketTagger() {}
+
+  public static @NotNull ISocketTagger getInstance() {
+    return instance;
+  }
+
+  @Override
+  public void tagSockets() {
+    // No operation
+  }
+
+  @Override
+  public void untagSockets() {
+    // No operation
+  }
+}

sentry/src/main/java/io/sentry/SentryOptions.java

@@ -566,6 +566,8 @@ public class SentryOptions {
    */
   private boolean startProfilerOnAppStart = false;
 
+  private @NotNull ISocketTagger socketTagger = NoOpSocketTagger.getInstance();
+
   /**
    * Adds an event processor
    *
@@ -2827,6 +2829,24 @@ public boolean isCaptureOpenTelemetryEvents() {
     return captureOpenTelemetryEvents;
   }
 
+  /**
+   * Returns the SocketTagger
+   *
+   * @return the socket tagger
+   */
+  public @NotNull ISocketTagger getSocketTagger() {
+    return socketTagger;
+  }
+
+  /**
+   * Sets the SocketTagger
+   *
+   * @param socketTagger the socket tagger
+   */
+  public void setSocketTagger(final @Nullable ISocketTagger socketTagger) {
+    this.socketTagger = socketTagger != null ? socketTagger : NoOpSocketTagger.getInstance();
+  }
+
   /**
    * Load the lazy fields. Useful to load in the background, so that results are already cached. DO
    * NOT CALL THIS METHOD ON THE MAIN THREAD.

sentry/src/main/java/io/sentry/transport/HttpConnection.java

@@ -146,6 +146,7 @@ HttpURLConnection open() throws IOException {
   }
 
   public @NotNull TransportResult send(final @NotNull SentryEnvelope envelope) throws IOException {
+    options.getSocketTagger().tagSockets();
     final HttpURLConnection connection = createConnection();
     TransportResult result;
 
@@ -161,6 +162,7 @@ HttpURLConnection open() throws IOException {
               "An exception occurred while submitting the envelope to the Sentry server.");
     } finally {
       result = readAndLog(connection);
+      options.getSocketTagger().untagSockets();
     }
     return result;
   }