/*
 * Copyright (C) 2010 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#define LOG_TAG "TrafficStats"

#include <dirent.h>
#include <errno.h>
#include <fcntl.h>
#include <sys/stat.h>
#include <sys/types.h>

#include <android_runtime/AndroidRuntime.h>
#include <cutils/logger.h>
#include <jni.h>
#include <utils/misc.h>
#include <utils/Log.h>

namespace android {

// Returns an ASCII decimal number read from the specified file, -1 on error.
static jlong readNumber(char const* filename) {
#ifdef HAVE_ANDROID_OS
    char buf[80];
    int fd = open(filename, O_RDONLY);
    if (fd < 0) {
        if (errno != ENOENT) LOGE("Can't open %s: %s", filename, strerror(errno));
        return -1;
    }

    int len = read(fd, buf, sizeof(buf) - 1);
    if (len < 0) {
        LOGE("Can't read %s: %s", filename, strerror(errno));
        close(fd);
        return -1;
    }

    close(fd);
    buf[len] = '\0';
    return atoll(buf);
#else  // Simulator
    return -1;
#endif
}

// Returns the sum of numbers from the specified path under /sys/class/net/*,
// -1 if no such file exists.
static jlong readTotal(char const* suffix) {
#ifdef HAVE_ANDROID_OS
    char filename[PATH_MAX] = "/sys/class/net/";
    DIR *dir = opendir(filename);
    if (dir == NULL) {
        LOGE("Can't list %s: %s", filename, strerror(errno));
        return -1;
    }

    int len = strlen(filename);
    jlong total = -1;
    while (struct dirent *entry = readdir(dir)) {
        // Skip ., .., and localhost interfaces.
        if (entry->d_name[0] != '.' && strncmp(entry->d_name, "lo", 2) != 0) {
            strlcpy(filename + len, entry->d_name, sizeof(filename) - len);
            strlcat(filename, suffix, sizeof(filename));
            jlong num = readNumber(filename);
            if (num >= 0) total = total < 0 ? num : total + num;
        }
    }

    closedir(dir);
    return total;
#else  // Simulator
    return -1;
#endif
}

#define RMNET_PREFIX "rmnet"

// Returns the sum of numbers from all the rmnet interfaces
// -1 if no such file exists.
static jlong readRmNetStats(char const* suffix) {
#ifdef HAVE_ANDROID_OS
    char filename[PATH_MAX] = "/sys/class/net/";
    DIR *dir = opendir(filename);
    if (dir == NULL) {
        LOGE("Can't list %s: %s", filename, strerror(errno));
        return -1;
    }

    int len = strlen(filename);
    jlong total = -1;
    jlong num = 0;
    int ret = 0;
    struct stat fstat;
    while (struct dirent *entry = readdir(dir)) {
        // Skip ., .., and localhost interfaces but match RMNET_PREFIX
        if (entry->d_name[0] != '.' &&
                strncmp(entry->d_name, "lo", 2) != 0 &&
                strncmp(entry->d_name, RMNET_PREFIX, strlen(RMNET_PREFIX)) == 0) {
            strlcpy(filename + len, entry->d_name, sizeof(filename) - len);
            strlcat(filename, suffix, sizeof(filename));

            num = readNumber(filename);

            if (num >= 0) total = total < 0 ? num : total + num;
        }
    }

    closedir(dir);
    return total;
#else  // Simulator
    return -1;
#endif
}

// Mobile stats get accessed a lot more often than total stats.
// Note the individual files can come and go at runtime, so we check
// each file every time (rather than caching which ones exist).

static jlong getMobileTxPackets(JNIEnv* env, jobject clazz) {
    jlong stats = readRmNetStats("/statistics/tx_packets");
    return stats >= 0 ? stats : readNumber("/sys/class/net/ppp0/statistics/tx_packets");
}

static jlong getMobileRxPackets(JNIEnv* env, jobject clazz) {
    jlong stats = readRmNetStats("/statistics/rx_packets");
    return stats >= 0 ? stats : readNumber("/sys/class/net/ppp0/statistics/tx_packets");
}

static jlong getMobileTxBytes(JNIEnv* env, jobject clazz) {
    jlong stats = readRmNetStats("/statistics/tx_bytes");
    return stats >= 0 ? stats : readNumber("/sys/class/net/ppp0/statistics/tx_packets");
}

static jlong getMobileRxBytes(JNIEnv* env, jobject clazz) {
    jlong stats = readRmNetStats("/statistics/rx_bytes");
    return stats >= 0 ? stats : readNumber("/sys/class/net/ppp0/statistics/tx_packets");
}

// Total stats are read less often, so we're willing to put up
// with listing the directory and concatenating filenames.

static jlong getTotalTxPackets(JNIEnv* env, jobject clazz) {
    return readTotal("/statistics/tx_packets");
}

static jlong getTotalRxPackets(JNIEnv* env, jobject clazz) {
    return readTotal("/statistics/rx_packets");
}

static jlong getTotalTxBytes(JNIEnv* env, jobject clazz) {
    return readTotal("/statistics/tx_bytes");
}

static jlong getTotalRxBytes(JNIEnv* env, jobject clazz) {
    return readTotal("/statistics/rx_bytes");
}

// Per-UID stats require reading from a constructed filename.

static jlong getUidRxBytes(JNIEnv* env, jobject clazz, jint uid) {
    char filename[80];
    sprintf(filename, "/proc/uid_stat/%d/tcp_rcv", uid);
    return readNumber(filename);
}

static jlong getUidTxBytes(JNIEnv* env, jobject clazz, jint uid) {
    char filename[80];
    sprintf(filename, "/proc/uid_stat/%d/tcp_snd", uid);
    return readNumber(filename);
}

static JNINativeMethod gMethods[] = {
    {"getMobileTxPackets", "()J", (void*) getMobileTxPackets},
    {"getMobileRxPackets", "()J", (void*) getMobileRxPackets},
    {"getMobileTxBytes", "()J", (void*) getMobileTxBytes},
    {"getMobileRxBytes", "()J", (void*) getMobileRxBytes},
    {"getTotalTxPackets", "()J", (void*) getTotalTxPackets},
    {"getTotalRxPackets", "()J", (void*) getTotalRxPackets},
    {"getTotalTxBytes", "()J", (void*) getTotalTxBytes},
    {"getTotalRxBytes", "()J", (void*) getTotalRxBytes},
    {"getUidTxBytes", "(I)J", (void*) getUidTxBytes},
    {"getUidRxBytes", "(I)J", (void*) getUidRxBytes},
};

int register_android_net_TrafficStats(JNIEnv* env) {
    return AndroidRuntime::registerNativeMethods(env, "android/net/TrafficStats",
            gMethods, NELEM(gMethods));
}

}