/*
 * Copyright (c) 2022-2023 Huawei Device Co., Ltd.
 * 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.
 */

#include <net/if.h>
#include <vector>

#include "bpf_path.h"
#include "bpf_def.h"
#include "bpf_stats.h"
#include "securec.h"
#include "netnative_log_wrapper.h"
#include "net_stats_constants.h"

namespace OHOS::NetManagerStandard {
namespace {
constexpr const char *CELLULAR_IFACE = "rmnet0";
constexpr const char *WIFI_IFACE = "wlan0";
const std::string LOOPBACK_IFACE = "lo";
}
int32_t NetsysBpfStats::GetNumberFromStatsValue(uint64_t &stats, StatsType statsType, const stats_value &value)
{
    switch (statsType) {
        case StatsType::STATS_TYPE_RX_BYTES:
            stats = value.rxBytes;
            break;
        case StatsType::STATS_TYPE_RX_PACKETS:
            stats = value.rxPackets;
            break;
        case StatsType::STATS_TYPE_TX_BYTES:
            stats = value.txBytes;
            break;
        case StatsType::STATS_TYPE_TX_PACKETS:
            stats = value.txPackets;
            break;
        default:
            NETNATIVE_LOGE("invalid StatsType type %{public}d", statsType);
            return STATS_ERR_READ_BPF_FAIL;
    }
    return NETSYS_SUCCESS;
}

int32_t NetsysBpfStats::GetTotalStats(uint64_t &stats, StatsType statsType)
{
    stats = 0;
    BpfMapper<iface_stats_key, iface_stats_value> ifaceStatsMap(IFACE_STATS_MAP_PATH, BPF_F_RDONLY);
    if (!ifaceStatsMap.IsValid()) {
        NETNATIVE_LOGE("ifaceStatsMap IsValid");
        return STATS_ERR_INVALID_IFACE_NAME_MAP;
    }

    iface_stats_value totalStats = {0};
    auto keys = ifaceStatsMap.GetAllKeys();
    for (const auto &k : keys) {
        iface_stats_value v = {0};
        if (ifaceStatsMap.Read(k, v) < NETSYS_SUCCESS) {
            NETNATIVE_LOGE("Read ifaceStatsMap err");
            return STATS_ERR_READ_BPF_FAIL;
        }
        totalStats.rxPackets += v.rxPackets;
        totalStats.rxBytes += v.rxBytes;
        totalStats.txPackets += v.txPackets;
        totalStats.txBytes += v.txBytes;
    }

    return GetNumberFromStatsValue(stats, statsType, totalStats);
}

int32_t NetsysBpfStats::GetUidStats(uint64_t &stats, StatsType statsType, uint32_t uid)
{
    stats = 0;
    BpfMapper<app_uid_stats_key, app_uid_stats_value> appUidStatsMap(APP_UID_STATS_MAP_PATH, BPF_F_RDONLY);
    if (!appUidStatsMap.IsValid()) {
        return STATS_ERR_INVALID_IFACE_NAME_MAP;
    }

    app_uid_stats_value uidStats = {0};
    if (appUidStatsMap.Read(uid, uidStats) < 0) {
        return STATS_ERR_READ_BPF_FAIL;
    }
    return GetNumberFromStatsValue(stats, statsType, uidStats);
}

int32_t NetsysBpfStats::GetAllSimStatsInfo(std::vector<OHOS::NetManagerStandard::NetStatsInfo> &stats)
{
    BpfMapper<stats_key, stats_value> uidSimStatsMap(APP_UID_SIM_STATS_MAP_PATH, BPF_F_RDONLY);
    if (!uidSimStatsMap.IsValid()) {
        return STATS_ERR_INVALID_IFACE_NAME_MAP;
    }

    stats.clear();
    char if_name[IFNAME_SIZE] = {0};
    auto keys = uidSimStatsMap.GetAllKeys();
    for (const auto &k : keys) {
        stats_value v = {};
        if (uidSimStatsMap.Read(k, v) < 0) {
            NETNATIVE_LOGE("Read uid_sim_map err");
            return STATS_ERR_READ_BPF_FAIL;
        }

        NetStatsInfo tempStats;
        tempStats.uid_ = k.uId;
        if (memset_s(if_name, sizeof(if_name), 0, sizeof(if_name)) != EOK) {
            return STATS_ERR_READ_BPF_FAIL;
        }

        char *pName = if_indextoname(k.ifIndex, if_name);
        if (pName != nullptr) {
            tempStats.iface_ = pName;
        }
        if (k.ifType == IFACE_TYPE_WIFI && std::string(pName) != LOOPBACK_IFACE) {
            tempStats.iface_ = WIFI_IFACE;
        } else if (k.ifType == IFACE_TYPE_CELLULAR) {
            tempStats.iface_ = CELLULAR_IFACE;
        }
        tempStats.rxBytes_ = v.rxBytes;
        tempStats.txBytes_ = v.txBytes;
        tempStats.rxPackets_ = v.rxPackets;
        tempStats.txPackets_ = v.txPackets;
        auto findRet = std::find_if(stats.begin(), stats.end(),
                                    [&tempStats](const NetStatsInfo &info) { return info.Equals(tempStats); });
        if (findRet == stats.end()) {
            stats.push_back(std::move(tempStats));
        } else {
            *findRet += tempStats;
        }
    }

    return NETSYS_SUCCESS;
}

int32_t NetsysBpfStats::GetAllStatsInfo(std::vector<OHOS::NetManagerStandard::NetStatsInfo> &stats)
{
    BpfMapper<stats_key, stats_value> uidIfaceStatsMap(APP_UID_IF_STATS_MAP_PATH, BPF_F_RDONLY);
    if (!uidIfaceStatsMap.IsValid()) {
        return STATS_ERR_INVALID_IFACE_NAME_MAP;
    }

    stats.clear();
    char if_name[IFNAME_SIZE] = {0};
    auto keys = uidIfaceStatsMap.GetAllKeys();
    for (const auto &k : keys) {
        stats_value v = {};
        if (uidIfaceStatsMap.Read(k, v) < 0) {
            NETNATIVE_LOGE("Read ifaceStatsMap err");
            return STATS_ERR_READ_BPF_FAIL;
        }

        NetStatsInfo tempStats;
        tempStats.uid_ = k.uId;
        if (memset_s(if_name, sizeof(if_name), 0, sizeof(if_name)) != EOK) {
            return STATS_ERR_READ_BPF_FAIL;
        }

        char *pName = if_indextoname(k.ifIndex, if_name);
        if (pName != nullptr) {
            tempStats.iface_ = pName;
        }
        tempStats.rxBytes_ = v.rxBytes;
        tempStats.txBytes_ = v.txBytes;
        tempStats.rxPackets_ = v.rxPackets;
        tempStats.txPackets_ = v.txPackets;
        stats.emplace_back(tempStats);
    }

    return NETSYS_SUCCESS;
}

int32_t NetsysBpfStats::DeleteStatsInfo(const std::string &path, uint32_t uid)
{
    if (path != APP_UID_IF_STATS_MAP_PATH && path != APP_UID_SIM_STATS_MAP_PATH) {
        NETNATIVE_LOGI("DeleteStatsInfo invalid path");
        return NETSYS_SUCCESS;
    }
    BpfMapper<stats_key, stats_value> uidStatsMap(path, BPF_ANY);
    if (!uidStatsMap.IsValid()) {
        return STATS_ERR_INVALID_IFACE_NAME_MAP;
    }
    auto keys = uidStatsMap.GetAllKeys();
    for (const auto &k : keys) {
        if (k.uId == uid) {
            if (uidStatsMap.Delete(k) < 0) {
                NETNATIVE_LOGE("Delete uidStatsMap err");
                return STATS_ERR_WRITE_BPF_FAIL;
            }
        }
    }
    return NETSYS_SUCCESS;
}

int32_t NetsysBpfStats::GetIfaceStats(uint64_t &stats, const StatsType statsType, const std::string &interfaceName)
{
    stats = 0;
    BpfMapper<iface_stats_key, iface_stats_value> ifaceStatsMap(IFACE_STATS_MAP_PATH, BPF_F_RDONLY);
    if (!ifaceStatsMap.IsValid()) {
        return STATS_ERR_INVALID_IFACE_NAME_MAP;
    }

    auto ifIndex = if_nametoindex(interfaceName.c_str());
    if (ifIndex <= 0) {
        return STATS_ERR_GET_IFACE_NAME_FAILED;
    }

    iface_stats_value ifaceStats = {0};
    if (ifaceStatsMap.Read(ifIndex, ifaceStats) < 0) {
        return STATS_ERR_READ_BPF_FAIL;
    }
    return GetNumberFromStatsValue(stats, statsType, ifaceStats);
}

int32_t NetsysBpfStats::GetCookieStats(uint64_t &stats, StatsType statsType, uint64_t cookie)
{
    NETNATIVE_LOGI("GetCookieStats start");
    stats = 0;
    BpfMapper<socket_cookie_stats_key, app_cookie_stats_value> appUidCookieStatsMap(APP_COOKIE_STATS_MAP_PATH,
                                                                                    BPF_F_RDONLY);
    if (!appUidCookieStatsMap.IsValid()) {
        NETNATIVE_LOGE("GetCookieStats appUidCookieStatsMap is valid");
        return NETMANAGER_ERR_INTERNAL;
    }

    app_cookie_stats_value cookieStats = {0};
    if (appUidCookieStatsMap.Read(cookie, cookieStats) < 0) {
        NETNATIVE_LOGE("GetCookieStats appUidCookieStatsMap read error");
        return NETMANAGER_ERR_INTERNAL;
    }

    int32_t res = GetNumberFromStatsValue(stats, statsType, cookieStats);
    if (res == STATS_ERR_READ_BPF_FAIL) {
        NETNATIVE_LOGE("GetCookieStats GetNumberFromStatsValue error");
        return NETMANAGER_ERR_INTERNAL;
    }
    return NETSYS_SUCCESS;
}
} // namespace OHOS::NetManagerStandard