android_net_TrafficStats.cpp revision 8669a1669487d5974c77ffda833c6111f316861c
12b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor/* 22b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor * Copyright (C) 2010 The Android Open Source Project 32b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor * 42b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor * Licensed under the Apache License, Version 2.0 (the "License"); 52b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor * you may not use this file except in compliance with the License. 62b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor * You may obtain a copy of the License at 72b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor * 82b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor * http://www.apache.org/licenses/LICENSE-2.0 92b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor * 102b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor * Unless required by applicable law or agreed to in writing, software 112b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor * distributed under the License is distributed on an "AS IS" BASIS, 122b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 132b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor * See the License for the specific language governing permissions and 142b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor * limitations under the License. 152b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor */ 162b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor 172b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor#define LOG_TAG "TrafficStats" 182b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor 192b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor#include <dirent.h> 202b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor#include <errno.h> 212b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor#include <fcntl.h> 222b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor#include <sys/stat.h> 232b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor#include <sys/types.h> 242b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor 252b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor#include <android_runtime/AndroidRuntime.h> 262b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor#include <cutils/logger.h> 272b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor#include <jni.h> 28f17b971a75577a3a0e5ab574618e0756d167f5d4Elliott Hughes#include <ScopedUtfChars.h> 292b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor#include <utils/misc.h> 302b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor#include <utils/Log.h> 312b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor 322b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnornamespace android { 332b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor 344b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkeystatic const char* QTAGUID_IFACE_STATS = "/proc/net/xt_qtaguid/iface_stat_fmt"; 3592be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkeystatic const char* QTAGUID_UID_STATS = "/proc/net/xt_qtaguid/stats"; 36c39c1d4dee917560d174f6ba5402e4c6644edd47Ashish Sharma 37234766a36af6214644fa8205202287084ca9cf93Jeff Sharkey// NOTE: keep these in sync with TrafficStats.java 384b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkeystatic const uint64_t UNKNOWN = -1; 394b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey 404b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkeyenum StatsType { 41234766a36af6214644fa8205202287084ca9cf93Jeff Sharkey RX_BYTES = 0, 42234766a36af6214644fa8205202287084ca9cf93Jeff Sharkey RX_PACKETS = 1, 43234766a36af6214644fa8205202287084ca9cf93Jeff Sharkey TX_BYTES = 2, 444b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey TX_PACKETS = 3, 454b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey TCP_RX_PACKETS = 4, 464b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey TCP_TX_PACKETS = 5 47234766a36af6214644fa8205202287084ca9cf93Jeff Sharkey}; 48234766a36af6214644fa8205202287084ca9cf93Jeff Sharkey 494b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkeystruct Stats { 50234766a36af6214644fa8205202287084ca9cf93Jeff Sharkey uint64_t rxBytes; 51234766a36af6214644fa8205202287084ca9cf93Jeff Sharkey uint64_t rxPackets; 52234766a36af6214644fa8205202287084ca9cf93Jeff Sharkey uint64_t txBytes; 53234766a36af6214644fa8205202287084ca9cf93Jeff Sharkey uint64_t txPackets; 544b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey uint64_t tcpRxPackets; 554b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey uint64_t tcpTxPackets; 56234766a36af6214644fa8205202287084ca9cf93Jeff Sharkey}; 57234766a36af6214644fa8205202287084ca9cf93Jeff Sharkey 584b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkeystatic uint64_t getStatsType(struct Stats* stats, StatsType type) { 594b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey switch (type) { 604b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey case RX_BYTES: 614b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey return stats->rxBytes; 624b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey case RX_PACKETS: 634b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey return stats->rxPackets; 644b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey case TX_BYTES: 654b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey return stats->txBytes; 664b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey case TX_PACKETS: 674b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey return stats->txPackets; 684b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey case TCP_RX_PACKETS: 694b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey return stats->tcpRxPackets; 704b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey case TCP_TX_PACKETS: 714b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey return stats->tcpTxPackets; 724b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey default: 734b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey return UNKNOWN; 744b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey } 754b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey} 764b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey 774b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkeystatic int parseIfaceStats(const char* iface, struct Stats* stats) { 784b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey FILE *fp = fopen(QTAGUID_IFACE_STATS, "r"); 794b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey if (fp == NULL) { 804b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey return -1; 8164ba5eaa40614877c679b3fbaaa3c2efd17d50d2Kazuhiro Ondo } 8264ba5eaa40614877c679b3fbaaa3c2efd17d50d2Kazuhiro Ondo 838669a1669487d5974c77ffda833c6111f316861cJeff Sharkey char buffer[384]; 84234766a36af6214644fa8205202287084ca9cf93Jeff Sharkey char cur_iface[32]; 854b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey bool foundTcp = false; 864b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey uint64_t rxBytes, rxPackets, txBytes, txPackets, tcpRxPackets, tcpTxPackets; 874b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey 884b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey while (fgets(buffer, sizeof(buffer), fp) != NULL) { 894b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey int matched = sscanf(buffer, "%31s %llu %llu %llu %llu " 904b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey "%*llu %llu %*llu %*llu %*llu %*llu " 914b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey "%*llu %llu %*llu %*llu %*llu %*llu", cur_iface, &rxBytes, 924b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey &rxPackets, &txBytes, &txPackets, &tcpRxPackets, &tcpTxPackets); 934b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey if (matched >= 5) { 944b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey if (matched == 7) { 954b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey foundTcp = true; 964b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey } 974b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey if (!iface || !strcmp(iface, cur_iface)) { 984b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey stats->rxBytes += rxBytes; 994b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey stats->rxPackets += rxPackets; 1004b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey stats->txBytes += txBytes; 1014b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey stats->txPackets += txPackets; 1024b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey if (matched == 7) { 1034b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey stats->tcpRxPackets += tcpRxPackets; 1044b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey stats->tcpTxPackets += tcpTxPackets; 1054b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey } 106234766a36af6214644fa8205202287084ca9cf93Jeff Sharkey } 1072b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor } 1082b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor } 1092b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor 1104b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey if (!foundTcp) { 1114b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey stats->tcpRxPackets = UNKNOWN; 1124b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey stats->tcpTxPackets = UNKNOWN; 113f17b971a75577a3a0e5ab574618e0756d167f5d4Elliott Hughes } 114227bec49157bc496f7c9e8e8f63c12728a448922Irfan Sheriff 1154b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey if (fclose(fp) != 0) { 1164b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey return -1; 117234766a36af6214644fa8205202287084ca9cf93Jeff Sharkey } 1184b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey return 0; 1192b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor} 1202b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor 12192be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkeystatic int parseUidStats(const uint32_t uid, struct Stats* stats) { 12292be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey FILE *fp = fopen(QTAGUID_UID_STATS, "r"); 12392be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey if (fp == NULL) { 12492be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey return -1; 12592be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey } 12692be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey 1278669a1669487d5974c77ffda833c6111f316861cJeff Sharkey char buffer[384]; 12892be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey char iface[32]; 12992be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey uint32_t idx, cur_uid, set; 13092be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey uint64_t tag, rxBytes, rxPackets, txBytes, txPackets; 13192be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey 1328669a1669487d5974c77ffda833c6111f316861cJeff Sharkey while (fgets(buffer, sizeof(buffer), fp) != NULL) { 13392be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey if (sscanf(buffer, "%d %31s 0x%llx %u %u %llu %llu %llu %llu", &idx, 13492be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey iface, &tag, &cur_uid, &set, &rxBytes, &rxPackets, &txBytes, 13592be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey &txPackets) == 9) { 13692be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey if (uid == cur_uid && tag == 0L) { 13792be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey stats->rxBytes += rxBytes; 13892be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey stats->rxPackets += rxPackets; 13992be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey stats->txBytes += txBytes; 14092be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey stats->txPackets += txPackets; 14192be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey } 14292be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey } 14392be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey } 14492be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey 14592be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey if (fclose(fp) != 0) { 14692be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey return -1; 14792be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey } 14892be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey return 0; 14992be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey} 15092be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey 151234766a36af6214644fa8205202287084ca9cf93Jeff Sharkeystatic jlong getTotalStat(JNIEnv* env, jclass clazz, jint type) { 1524b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey struct Stats stats; 1534b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey memset(&stats, 0, sizeof(Stats)); 1544b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey if (parseIfaceStats(NULL, &stats) == 0) { 1554b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey return getStatsType(&stats, (StatsType) type); 1564b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey } else { 1574b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey return UNKNOWN; 1584b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey } 1592b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor} 1602b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor 161234766a36af6214644fa8205202287084ca9cf93Jeff Sharkeystatic jlong getIfaceStat(JNIEnv* env, jclass clazz, jstring iface, jint type) { 1624b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey ScopedUtfChars iface8(env, iface); 1634b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey if (iface8.c_str() == NULL) { 1644b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey return UNKNOWN; 1654b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey } 1664b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey 1674b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey struct Stats stats; 1684b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey memset(&stats, 0, sizeof(Stats)); 1694b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey if (parseIfaceStats(NULL, &stats) == 0) { 1704b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey return getStatsType(&stats, (StatsType) type); 171234766a36af6214644fa8205202287084ca9cf93Jeff Sharkey } else { 1724b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey return UNKNOWN; 173234766a36af6214644fa8205202287084ca9cf93Jeff Sharkey } 1742b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor} 1752b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor 17692be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkeystatic jlong getUidStat(JNIEnv* env, jclass clazz, jint uid, jint type) { 17792be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey struct Stats stats; 17892be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey memset(&stats, 0, sizeof(Stats)); 17992be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey if (parseUidStats(uid, &stats) == 0) { 18092be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey return getStatsType(&stats, (StatsType) type); 18192be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey } else { 18292be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey return UNKNOWN; 183c39c1d4dee917560d174f6ba5402e4c6644edd47Ashish Sharma } 1842b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor} 1852b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor 1862b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnorstatic JNINativeMethod gMethods[] = { 187234766a36af6214644fa8205202287084ca9cf93Jeff Sharkey {"nativeGetTotalStat", "(I)J", (void*) getTotalStat}, 188234766a36af6214644fa8205202287084ca9cf93Jeff Sharkey {"nativeGetIfaceStat", "(Ljava/lang/String;I)J", (void*) getIfaceStat}, 18992be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey {"nativeGetUidStat", "(II)J", (void*) getUidStat}, 1902b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor}; 1912b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor 1922b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnorint register_android_net_TrafficStats(JNIEnv* env) { 1932b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor return AndroidRuntime::registerNativeMethods(env, "android/net/TrafficStats", 1942b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor gMethods, NELEM(gMethods)); 1952b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor} 1962b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor 1972b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor} 198