/*
 * Copyright (c) 2022 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 <getopt.h>
#include <iostream>
#include <selinux/selinux.h>
#include <unistd.h>
#ifdef TIME_DISPLAY
#include <sys/time.h>
#endif
#include "selinux_error.h"
#include "selinux_parameter.h"

using namespace Selinux;

#ifdef TIME_DISPLAY
const static long USEC_PER_SEC = 1000000L;
#endif

struct testInput {
    std::string paraName;
    char cmd = '\0';
};

static void TestLoadList()
{
    std::string path = "/dev/__parameters__/";
    ParamContextsList *buff = nullptr;
#ifdef TIME_DISPLAY
    struct timeval start, end, diff;
    gettimeofday(&start, nullptr);
#endif
    buff = GetParamList();
    if (buff == nullptr) {
        std::cout << "buff empty" << std::endl;
        return;
    }
#ifdef TIME_DISPLAY
    gettimeofday(&end, nullptr);
    timersub(&end, &start, &diff);
    int runtime_us = diff.tv_sec * USEC_PER_SEC + diff.tv_usec;
    std::cout << "GetParamList time use: " << runtime_us << std::endl;
#endif

    ParamContextsList *head = buff;
    while (buff != nullptr) {
        if (security_check_context(buff->info.paraContext) < 0) {
            std::cout << "failed check context: " << buff->info.paraContext << std::endl;
            buff = buff->next;
            continue;
        }
        std::string name = path + std::string(buff->info.paraContext);
        FILE *fp = fopen(name.c_str(), "w");
        if (fp == nullptr) {
            std::cout << "failed: " << name << std::endl;
            buff = buff->next;
            continue;
        }
        (void)fclose(fp);
        if (setfilecon(name.c_str(), buff->info.paraContext) < 0) {
            std::cout << "setcon failed: " << name << std::endl;
        }
        buff = buff->next;
    }
#ifdef TIME_DISPLAY
    gettimeofday(&start, nullptr);
#endif
    DestroyParamList(&head);
#ifdef TIME_DISPLAY
    gettimeofday(&end, nullptr);
    timersub(&end, &start, &diff);
    runtime_us = diff.tv_sec * USEC_PER_SEC + diff.tv_usec;
    std::cout << "DestroyParamList time use: " << runtime_us << std::endl;
#endif
}

static void TestGetContext(std::string &paraName)
{
    char *context = nullptr;
#ifdef TIME_DISPLAY
    struct timeval start, end, diff;
    gettimeofday(&start, nullptr);
#endif
    int res = GetParamLabel(paraName.c_str(), &context);
#ifdef TIME_DISPLAY
    gettimeofday(&end, nullptr);
    timersub(&end, &start, &diff);
    int runtime_us = diff.tv_sec * USEC_PER_SEC + diff.tv_usec;
    std::cout << "time use: " << runtime_us << std::endl;
#endif
    if (res == 0) {
        std::cout << "para " << paraName.c_str() << "'s context is " << context << std::endl;
    } else {
        std::cout << "para " << paraName.c_str() << "'s context get fail, " << GetErrStr(res) << std::endl;
    }
    if (!context) {
        free(context);
    }
}

static void TestReadPara(std::string &paraName)
{
#ifdef TIME_DISPLAY
    struct timeval start, end, diff;
    gettimeofday(&start, nullptr);
#endif
    std::cout << GetErrStr(ReadParamCheck(paraName.c_str())) << std::endl;
#ifdef TIME_DISPLAY
    gettimeofday(&end, nullptr);
    timersub(&end, &start, &diff);
    int runtime_us = diff.tv_sec * USEC_PER_SEC + diff.tv_usec;
    std::cout << "time use: " << runtime_us << std::endl;
#endif
}

static void TestSetPara(std::string &paraName, struct ucred *uc)
{
#ifdef TIME_DISPLAY
    struct timeval start, end, diff;
    gettimeofday(&start, nullptr);
#endif
    std::cout << GetErrStr(SetParamCheck(paraName.c_str(), uc)) << std::endl;
#ifdef TIME_DISPLAY
    gettimeofday(&end, nullptr);
    timersub(&end, &start, &diff);
    int runtime_us = diff.tv_sec * USEC_PER_SEC + diff.tv_usec;
    std::cout << "time use: " << runtime_us << std::endl;
#endif
}

static void PrintUsage()
{
    std::cout << "Options:" << std::endl;
    std::cout << " -h (--help)           show the help information.      [eg: param_check -h]" << std::endl;
    std::cout << " -g (--getContext)     get context for paraName.       [eg: param_check -g]" << std::endl;
    std::cout << " -r (--read)           read para perm.                 [eg: param_check -r]" << std::endl;
    std::cout << " -w (--write)          write para perm.                [eg: param_check -w]" << std::endl;
    std::cout << " -l (--list)           load para list.                 [eg: param_check -l]" << std::endl;
    std::cout << " -n (--paraName)       paraName.                       [eg: param_check -r|-w|-g -n para_name]"
              << std::endl;
    std::cout << "" << std::endl;
    std::cout << "Usage:" << std::endl;
    std::cout << ">>>>>>> choice 1: continuous input parameters" << std::endl;
    std::cout << "step 1:" << std::endl;
    std::cout << "param_check -r|-w|-g|-l" << std::endl;
    std::cout << "step 2:" << std::endl;
    std::cout << "input param name and press 'enter' to continue, or ctrl+C to end process" << std::endl;
    std::cout << "" << std::endl;
    std::cout << ">>>>>>> choice 2: single input parameter" << std::endl;
    std::cout << "param_check -r|-w|-g -n para_name" << std::endl;
    std::cout << "" << std::endl;
}

static void SetOptions(int argc, char *argv[], const option *options, testInput &input)
{
    int index = 0;
    const char *optStr = "hgrwln:";
    int para = 0;
    while ((para = getopt_long(argc, argv, optStr, options, &index)) != -1) {
        switch (para) {
            case 'h': {
                PrintUsage();
                exit(0);
            }
            case 'n': {
                input.paraName = optarg;
                break;
            }
            case 'g': {
                input.cmd = 'g';
                break;
            }
            case 'r': {
                input.cmd = 'r';
                break;
            }
            case 'w': {
                input.cmd = 'w';
                break;
            }
            case 'l': {
                TestLoadList();
                exit(0);
            }
            default:
                std::cout << "Try 'param_check -h' for more information." << std::endl;
                exit(-1);
        }
    }
}

static void Test(testInput &testCmd)
{
    std::string paraName;
    switch (testCmd.cmd) {
        case 'g': {
            if (!testCmd.paraName.empty()) {
                TestGetContext(testCmd.paraName);
                exit(0);
            }
            while (std::cin >> paraName) {
                TestGetContext(paraName);
            }
            exit(0);
        }
        case 'r': {
            if (!testCmd.paraName.empty()) {
                TestReadPara(testCmd.paraName);
                exit(0);
            }
            while (std::cin >> paraName) {
                TestReadPara(paraName);
            }
            exit(0);
        }
        case 'w': {
            struct ucred uc;
            uc.pid = getpid();
            uc.uid = getuid();
            uc.gid = getgid();
            if (!testCmd.paraName.empty()) {
                TestSetPara(testCmd.paraName, &uc);
                exit(0);
            }
            while (std::cin >> paraName) {
                TestSetPara(paraName, &uc);
            }
            exit(0);
        }
        default:
            PrintUsage();
            exit(-1);
    }
}

int main(int argc, char *argv[])
{
    struct option options[] = {
        {"help", no_argument, nullptr, 'h'}, {"paraName", required_argument, nullptr, 'n'},
        {"read", no_argument, nullptr, 'r'}, {"write", no_argument, nullptr, 'w'},
        {"list", no_argument, nullptr, 'l'}, {"getContext", no_argument, nullptr, 'g'},
        {nullptr, no_argument, nullptr, 0},
    };

    if (argc == 1) {
        PrintUsage();
        exit(0);
    }
    SetSelinuxLogCallback();
    testInput testCmd;
    SetOptions(argc, argv, options, testCmd);
    Test(testCmd);
    exit(0);
}