Merge "API to adjust network stats. DO NOT MERGE." into jb-mr1-dev

2 files changed, 31 insertions, 0 deletions

diff --git a/core/java/android/net/INetworkStatsService.aidl b/core/java/android/net/INetworkStatsService.aidl
index b7b8731..4d46334 100644
--- a/core/java/android/net/INetworkStatsService.aidl
+++ b/core/java/android/net/INetworkStatsService.aidl
@@ -45,4 +45,7 @@ interface INetworkStatsService {
     /** Advise persistance threshold; may be overridden internally. */
     void advisePersistThreshold(long thresholdBytes);
 
+    /** Adjust recorded network stats. */
+    void adjustNetworkStats(int uid, int tag, long rxBytes, long rxPackets, long txBytes, long txPackets, long operationCount);
+
 }
diff --git a/core/java/android/net/TrafficStats.java b/core/java/android/net/TrafficStats.java
index e437d2e..b66c2fe 100644
--- a/core/java/android/net/TrafficStats.java
+++ b/core/java/android/net/TrafficStats.java
@@ -88,6 +88,13 @@ public class TrafficStats {
      */
     public static final int TAG_SYSTEM_BACKUP = 0xFFFFFF03;
 
+    /**
+     * Default tag value for cloud messaging traffic.
+     *
+     * @hide
+     */
+    public static final int TAG_SYSTEM_CLOUD_MESSAGING = 0xFFFFFF04;
+
     private static INetworkStatsService sStatsService;
 
     private synchronized static INetworkStatsService getStatsService() {
@@ -246,6 +253,27 @@ public class TrafficStats {
         }
     }
 
+    /**
+     * Adjust network statistics for the given UID and tag by the requested
+     * amount. This can be used to correctly account network usage performed on
+     * behalf of another application. Values can be negative.
+     * <p>
+     * Requires that caller holds
+     * {@link android.Manifest.permission#MODIFY_NETWORK_ACCOUNTING} permission.
+     *
+     * @see #setThreadStatsUid(int)
+     * @hide
+     */
+    public static void adjustNetworkStats(int uid, int tag, long rxBytes, long rxPackets,
+            long txBytes, long txPackets, long operationCount) {
+        try {
+            getStatsService().adjustNetworkStats(
+                    uid, tag, rxBytes, rxPackets, txBytes, txPackets, operationCount);
+        } catch (RemoteException e) {
+            throw new RuntimeException(e);
+        }
+    }
+
     /** {@hide} */
     public static void closeQuietly(INetworkStatsSession session) {
         // TODO: move to NetworkStatsService once it exists