From 8baaf454b485fafa17ee25d1be0e1ddc58c41a4f Mon Sep 17 00:00:00 2001
From: Mikael Magnusson <mikma@users.sourceforge.net>
Date: Sat, 11 Mar 2023 23:10:42 +0100
Subject: WIP: Add TrafficStats

---
 tunnel/src/main/java/com/wireguard/util/Resolver.java | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'tunnel/src/main/java/com/wireguard/util')

diff --git a/tunnel/src/main/java/com/wireguard/util/Resolver.java b/tunnel/src/main/java/com/wireguard/util/Resolver.java
index 62b11286..24f5ab88 100644
--- a/tunnel/src/main/java/com/wireguard/util/Resolver.java
+++ b/tunnel/src/main/java/com/wireguard/util/Resolver.java
@@ -17,6 +17,7 @@ import java.net.UnknownHostException;
 import android.net.IpPrefix;
 import android.net.LinkProperties;
 import android.net.Network;
+import android.net.TrafficStats;
 import android.util.Log;
 
 import androidx.annotation.Nullable;
@@ -24,6 +25,7 @@ import androidx.annotation.Nullable;
 @NonNullForAll
 public class Resolver {
     private static final String TAG = "WireGuard/Resolver";
+    private static final int STATS_TAG = 3; // FIXME
     @Nullable private final Network network;
     @Nullable private final LinkProperties linkProps;
     @Nullable private IpPrefix nat64Prefix;
@@ -82,6 +84,7 @@ public class Resolver {
     }
 
     public InetAddress resolve(String host) throws UnknownHostException {
+        TrafficStats.setThreadStatsTag(STATS_TAG);
         final InetAddress[] candidates = network != null ? network.getAllByName(host) : InetAddress.getAllByName(host);
         InetAddress address = candidates[0];
         for (final InetAddress candidate : candidates) {
@@ -89,6 +92,7 @@ public class Resolver {
 
             try {
                 sock = new DatagramSocket();
+                TrafficStats.tagDatagramSocket(sock);
                 if (network != null) {
                     network.bindSocket(sock);
                 }
-- 
cgit v1.2.3