Home | History | Annotate | Download | only in jni 
      1 /*
      2  * Copyright (C) 2010 The Android Open Source Project
      3  *
      4  * Licensed under the Apache License, Version 2.0 (the "License");
      5  * you may not use this file except in compliance with the License.
      6  * You may obtain a copy of the License at
      7  *
      8  *      http://www.apache.org/licenses/LICENSE-2.0
      9  *
     10  * Unless required by applicable law or agreed to in writing, software
     11  * distributed under the License is distributed on an "AS IS" BASIS,
     12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     13  * See the License for the specific language governing permissions and
     14  * limitations under the License.
     15  */
     16 
     17 #define LOG_TAG "NetworkStatsNative"
     18 
     19 #include <dirent.h>
     20 #include <errno.h>
     21 #include <fcntl.h>
     22 #include <inttypes.h>
     23 #include <sys/stat.h>
     24 #include <sys/types.h>
     25 
     26 #include "core_jni_helpers.h"
     27 #include <jni.h>
     28 #include <nativehelper/ScopedUtfChars.h>
     29 #include <utils/misc.h>
     30 #include <utils/Log.h>
     31 
     32 #include "android-base/unique_fd.h"
     33 #include "bpf/BpfNetworkStats.h"
     34 #include "bpf/BpfUtils.h"
     35 
     36 using android::bpf::Stats;
     37 using android::bpf::hasBpfSupport;
     38 using android::bpf::bpfGetUidStats;
     39 using android::bpf::bpfGetIfaceStats;
     40 
     41 namespace android {
     42 
     43 static const char* QTAGUID_IFACE_STATS = "/proc/net/xt_qtaguid/iface_stat_fmt";
     44 static const char* QTAGUID_UID_STATS = "/proc/net/xt_qtaguid/stats";
     45 
     46 // NOTE: keep these in sync with TrafficStats.java
     47 static const uint64_t UNKNOWN = -1;
     48 
     49 enum StatsType {
     50     RX_BYTES = 0,
     51     RX_PACKETS = 1,
     52     TX_BYTES = 2,
     53     TX_PACKETS = 3,
     54     TCP_RX_PACKETS = 4,
     55     TCP_TX_PACKETS = 5
     56 };
     57 
     58 static uint64_t getStatsType(struct Stats* stats, StatsType type) {
     59     switch (type) {
     60         case RX_BYTES:
     61             return stats->rxBytes;
     62         case RX_PACKETS:
     63             return stats->rxPackets;
     64         case TX_BYTES:
     65             return stats->txBytes;
     66         case TX_PACKETS:
     67             return stats->txPackets;
     68         case TCP_RX_PACKETS:
     69             return stats->tcpRxPackets;
     70         case TCP_TX_PACKETS:
     71             return stats->tcpTxPackets;
     72         default:
     73             return UNKNOWN;
     74     }
     75 }
     76 
     77 static int parseIfaceStats(const char* iface, struct Stats* stats) {
     78     FILE *fp = fopen(QTAGUID_IFACE_STATS, "r");
     79     if (fp == NULL) {
     80         return -1;
     81     }
     82 
     83     char buffer[384];
     84     char cur_iface[32];
     85     bool foundTcp = false;
     86     uint64_t rxBytes, rxPackets, txBytes, txPackets, tcpRxPackets, tcpTxPackets;
     87 
     88     while (fgets(buffer, sizeof(buffer), fp) != NULL) {
     89         int matched = sscanf(buffer, "%31s %" SCNu64 " %" SCNu64 " %" SCNu64
     90                 " %" SCNu64 " " "%*u %" SCNu64 " %*u %*u %*u %*u "
     91                 "%*u %" SCNu64 " %*u %*u %*u %*u", cur_iface, &rxBytes,
     92                 &rxPackets, &txBytes, &txPackets, &tcpRxPackets, &tcpTxPackets);
     93         if (matched >= 5) {
     94             if (matched == 7) {
     95                 foundTcp = true;
     96             }
     97             if (!iface || !strcmp(iface, cur_iface)) {
     98                 stats->rxBytes += rxBytes;
     99                 stats->rxPackets += rxPackets;
    100                 stats->txBytes += txBytes;
    101                 stats->txPackets += txPackets;
    102                 if (matched == 7) {
    103                     stats->tcpRxPackets += tcpRxPackets;
    104                     stats->tcpTxPackets += tcpTxPackets;
    105                 }
    106             }
    107         }
    108     }
    109 
    110     if (!foundTcp) {
    111         stats->tcpRxPackets = UNKNOWN;
    112         stats->tcpTxPackets = UNKNOWN;
    113     }
    114 
    115     if (fclose(fp) != 0) {
    116         return -1;
    117     }
    118     return 0;
    119 }
    120 
    121 static int parseUidStats(const uint32_t uid, struct Stats* stats) {
    122     FILE *fp = fopen(QTAGUID_UID_STATS, "r");
    123     if (fp == NULL) {
    124         return -1;
    125     }
    126 
    127     char buffer[384];
    128     char iface[32];
    129     uint32_t idx, cur_uid, set;
    130     uint64_t tag, rxBytes, rxPackets, txBytes, txPackets;
    131 
    132     while (fgets(buffer, sizeof(buffer), fp) != NULL) {
    133         if (sscanf(buffer,
    134                 "%" SCNu32 " %31s 0x%" SCNx64 " %u %u %" SCNu64 " %" SCNu64
    135                 " %" SCNu64 " %" SCNu64 "",
    136                 &idx, iface, &tag, &cur_uid, &set, &rxBytes, &rxPackets,
    137                 &txBytes, &txPackets) == 9) {
    138             if (uid == cur_uid && tag == 0L) {
    139                 stats->rxBytes += rxBytes;
    140                 stats->rxPackets += rxPackets;
    141                 stats->txBytes += txBytes;
    142                 stats->txPackets += txPackets;
    143             }
    144         }
    145     }
    146 
    147     if (fclose(fp) != 0) {
    148         return -1;
    149     }
    150     return 0;
    151 }
    152 
    153 static jlong getTotalStat(JNIEnv* env, jclass clazz, jint type, jboolean useBpfStats) {
    154     struct Stats stats;
    155     memset(&stats, 0, sizeof(Stats));
    156 
    157     if (useBpfStats) {
    158         if (bpfGetIfaceStats(NULL, &stats) == 0) {
    159             return getStatsType(&stats, (StatsType) type);
    160         } else {
    161             return UNKNOWN;
    162         }
    163     }
    164 
    165     if (parseIfaceStats(NULL, &stats) == 0) {
    166         return getStatsType(&stats, (StatsType) type);
    167     } else {
    168         return UNKNOWN;
    169     }
    170 }
    171 
    172 static jlong getIfaceStat(JNIEnv* env, jclass clazz, jstring iface, jint type,
    173                           jboolean useBpfStats) {
    174     ScopedUtfChars iface8(env, iface);
    175     if (iface8.c_str() == NULL) {
    176         return UNKNOWN;
    177     }
    178 
    179     struct Stats stats;
    180     memset(&stats, 0, sizeof(Stats));
    181 
    182     if (useBpfStats) {
    183         if (bpfGetIfaceStats(iface8.c_str(), &stats) == 0) {
    184             return getStatsType(&stats, (StatsType) type);
    185         } else {
    186             return UNKNOWN;
    187         }
    188     }
    189 
    190     if (parseIfaceStats(iface8.c_str(), &stats) == 0) {
    191         return getStatsType(&stats, (StatsType) type);
    192     } else {
    193         return UNKNOWN;
    194     }
    195 }
    196 
    197 static jlong getUidStat(JNIEnv* env, jclass clazz, jint uid, jint type, jboolean useBpfStats) {
    198     struct Stats stats;
    199     memset(&stats, 0, sizeof(Stats));
    200 
    201     if (useBpfStats) {
    202         if (bpfGetUidStats(uid, &stats) == 0) {
    203             return getStatsType(&stats, (StatsType) type);
    204         } else {
    205             return UNKNOWN;
    206         }
    207     }
    208 
    209     if (parseUidStats(uid, &stats) == 0) {
    210         return getStatsType(&stats, (StatsType) type);
    211     } else {
    212         return UNKNOWN;
    213     }
    214 }
    215 
    216 static const JNINativeMethod gMethods[] = {
    217     {"nativeGetTotalStat", "(IZ)J", (void*) getTotalStat},
    218     {"nativeGetIfaceStat", "(Ljava/lang/String;IZ)J", (void*) getIfaceStat},
    219     {"nativeGetUidStat", "(IIZ)J", (void*) getUidStat},
    220 };
    221 
    222 int register_android_server_net_NetworkStatsService(JNIEnv* env) {
    223     jclass netStatsService = env->FindClass("com/android/server/net/NetworkStatsService");
    224     jfieldID rxBytesId = env->GetStaticFieldID(netStatsService, "TYPE_RX_BYTES", "I");
    225     jfieldID rxPacketsId = env->GetStaticFieldID(netStatsService, "TYPE_RX_PACKETS", "I");
    226     jfieldID txBytesId = env->GetStaticFieldID(netStatsService, "TYPE_TX_BYTES", "I");
    227     jfieldID txPacketsId = env->GetStaticFieldID(netStatsService, "TYPE_TX_PACKETS", "I");
    228     jfieldID tcpRxPacketsId = env->GetStaticFieldID(netStatsService, "TYPE_TCP_RX_PACKETS", "I");
    229     jfieldID tcpTxPacketsId = env->GetStaticFieldID(netStatsService, "TYPE_TCP_TX_PACKETS", "I");
    230 
    231     env->SetStaticIntField(netStatsService, rxBytesId, RX_BYTES);
    232     env->SetStaticIntField(netStatsService, rxPacketsId, RX_PACKETS);
    233     env->SetStaticIntField(netStatsService, txBytesId, TX_BYTES);
    234     env->SetStaticIntField(netStatsService, txPacketsId, TX_PACKETS);
    235     env->SetStaticIntField(netStatsService, tcpRxPacketsId, TCP_RX_PACKETS);
    236     env->SetStaticIntField(netStatsService, tcpTxPacketsId, TCP_TX_PACKETS);
    237 
    238     return jniRegisterNativeMethods(env, "com/android/server/net/NetworkStatsService", gMethods,
    239                                     NELEM(gMethods));
    240 }
    241 
    242 }
    243