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> 22853940331ff2be48ed3e63f459845e5e43d0a131Mark Salyzyn#include <inttypes.h> 232b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor#include <sys/stat.h> 242b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor#include <sys/types.h> 252b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor 26ed6b9dff563c5e22f040ff37e12c0d771e0478aeAndreas Gampe#include "core_jni_helpers.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) { 89853940331ff2be48ed3e63f459845e5e43d0a131Mark Salyzyn int matched = sscanf(buffer, "%31s %" SCNu64 " %" SCNu64 " %" SCNu64 90853940331ff2be48ed3e63f459845e5e43d0a131Mark Salyzyn " %" SCNu64 " " "%*u %" SCNu64 " %*u %*u %*u %*u " 91853940331ff2be48ed3e63f459845e5e43d0a131Mark Salyzyn "%*u %" SCNu64 " %*u %*u %*u %*u", 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) { 133853940331ff2be48ed3e63f459845e5e43d0a131Mark Salyzyn if (sscanf(buffer, 134853940331ff2be48ed3e63f459845e5e43d0a131Mark Salyzyn "%" SCNu32 " %31s 0x%" SCNx64 " %u %u %" SCNu64 " %" SCNu64 135853940331ff2be48ed3e63f459845e5e43d0a131Mark Salyzyn " %" SCNu64 " %" SCNu64 "", 136853940331ff2be48ed3e63f459845e5e43d0a131Mark Salyzyn &idx, iface, &tag, &cur_uid, &set, &rxBytes, &rxPackets, 137853940331ff2be48ed3e63f459845e5e43d0a131Mark Salyzyn &txBytes, &txPackets) == 9) { 13892be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey if (uid == cur_uid && tag == 0L) { 13992be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey stats->rxBytes += rxBytes; 14092be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey stats->rxPackets += rxPackets; 14192be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey stats->txBytes += txBytes; 14292be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey stats->txPackets += txPackets; 14392be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey } 14492be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey } 14592be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey } 14692be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey 14792be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey if (fclose(fp) != 0) { 14892be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey return -1; 14992be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey } 15092be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey return 0; 15192be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey} 15292be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey 153234766a36af6214644fa8205202287084ca9cf93Jeff Sharkeystatic jlong getTotalStat(JNIEnv* env, jclass clazz, jint type) { 1544b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey struct Stats stats; 1554b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey memset(&stats, 0, sizeof(Stats)); 1564b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey if (parseIfaceStats(NULL, &stats) == 0) { 1574b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey return getStatsType(&stats, (StatsType) type); 1584b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey } else { 1594b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey return UNKNOWN; 1604b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey } 1612b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor} 1622b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor 163234766a36af6214644fa8205202287084ca9cf93Jeff Sharkeystatic jlong getIfaceStat(JNIEnv* env, jclass clazz, jstring iface, jint type) { 1644b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey ScopedUtfChars iface8(env, iface); 1654b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey if (iface8.c_str() == NULL) { 1664b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey return UNKNOWN; 1674b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey } 1684b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey 1694b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey struct Stats stats; 1704b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey memset(&stats, 0, sizeof(Stats)); 17162f16bfc5e90f0320ab68b66b5b5b125cada1867JP Abgrall if (parseIfaceStats(iface8.c_str(), &stats) == 0) { 1724b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey return getStatsType(&stats, (StatsType) type); 173234766a36af6214644fa8205202287084ca9cf93Jeff Sharkey } else { 1744b17a1321db24b1a59c29b580aed7482a43febeeJeff Sharkey return UNKNOWN; 175234766a36af6214644fa8205202287084ca9cf93Jeff Sharkey } 1762b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor} 1772b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor 17892be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkeystatic jlong getUidStat(JNIEnv* env, jclass clazz, jint uid, jint type) { 17992be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey struct Stats stats; 18092be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey memset(&stats, 0, sizeof(Stats)); 18192be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey if (parseUidStats(uid, &stats) == 0) { 18292be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey return getStatsType(&stats, (StatsType) type); 18392be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey } else { 18492be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey return UNKNOWN; 185c39c1d4dee917560d174f6ba5402e4c6644edd47Ashish Sharma } 1862b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor} 1872b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor 18876f6a86de25e1bf74717e047e55fd44b089673f3Daniel Micaystatic const JNINativeMethod gMethods[] = { 189234766a36af6214644fa8205202287084ca9cf93Jeff Sharkey {"nativeGetTotalStat", "(I)J", (void*) getTotalStat}, 190234766a36af6214644fa8205202287084ca9cf93Jeff Sharkey {"nativeGetIfaceStat", "(Ljava/lang/String;I)J", (void*) getIfaceStat}, 19192be93a94edafb5906e8bc48e6fee9dd07f5049eJeff Sharkey {"nativeGetUidStat", "(II)J", (void*) getUidStat}, 1922b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor}; 1932b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor 1942b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnorint register_android_net_TrafficStats(JNIEnv* env) { 195ed6b9dff563c5e22f040ff37e12c0d771e0478aeAndreas Gampe return RegisterMethodsOrDie(env, "android/net/TrafficStats", gMethods, NELEM(gMethods)); 1962b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor} 1972b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor 1982b4abcd0c7c4361af8ab6d5d7b073fb75ac6d219Dan Egnor} 199