/*
 * 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 <cstdlib>
#include <getopt.h>
#include <iosfwd>
#include <iostream>
#include <istream>
#include <memory>
#include <ostream>
#include <string>
#include <unistd.h>

#include "selinux/avc.h"
#include "hdf_service_checker.h"
#include "selinux_error.h"
#include "service_checker.h"

using namespace Selinux;

static std::unique_ptr<ServiceChecker> g_service = nullptr;

struct TestInput {
    char cmd = '\0';
    bool isHdf = false;
    std::string serviceName;
};

static void PrintUsage()
{
    std::cout << "Options:" << std::endl;
    std::cout << " -h (--help)           show the help information.        [eg: service_check -h]" << std::endl;
    std::cout << "***********************optinal*************************************************" << std::endl;
    std::cout << " -d (--isHdf)          service or hdf_service.           [eg: service_check -d]" << std::endl;
    std::cout << " -n (--serviceName)    serviceName.                      [eg: service_check -n service_name]"
              << std::endl;
    std::cout << "***********************requered: 1 in 4****************************************" << std::endl;
    std::cout << " -a (--add)            add service check.                [eg: service_check -a]" << std::endl;
    std::cout << " -g (--get)            get service check.                [eg: service_check -g]" << std::endl;
    std::cout << " -r (--get_remote)     get remote service check.         [eg: service_check -r]" << std::endl;
    std::cout << " -l (--list)           list service check.               [eg: service_check -l]" << 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 << "service_check (-d) -a|-g|-r|-l" << std::endl;
    std::cout << "step 2:" << std::endl;
    std::cout << "input service name and press 'enter' to continue, or ctrl+C to end process" << std::endl;
    std::cout << ">>>>>>> choice 2: single input parameter" << std::endl;
    std::cout << "service_check (-d) -a|-g|-r|-l -n service_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 = "dhlagrn:";
    int para = 0;
    while ((para = getopt_long(argc, argv, optStr, options, &index)) != -1) {
        switch (para) {
            case 'h': {
                PrintUsage();
                exit(0);
            }
            case 'n': {
                input.serviceName = optarg;
                break;
            }
            case 'd': {
                input.isHdf = true;
                break;
            }
            case 'a': {
                input.cmd = 'a';
                break;
            }
            case 'g': {
                input.cmd = 'g';
                break;
            }
            case 'r': {
                input.cmd = 'r';
                break;
            }
            case 'l': {
                input.cmd = 'l';
                break;
            }
            default:
                std::cout << "Try 'service_check -h' for more information." << std::endl;
                exit(-1);
        }
    }
}

static int GetSidForCurrentProcess(std::string &sid)
{
    char *con = nullptr;
    if (getcon(&con) < 0) {
        return -1;
    }
    sid = con;
    freecon(con);
    return 0;
}

static void TestAddService(bool isHdf, const std::string &serviceName)
{
    std::string sid;
    if (GetSidForCurrentProcess(sid) < 0) {
        return;
    }
    if (!serviceName.empty()) {
        std::cout << GetErrStr(isHdf ? HdfAddServiceCheck(sid.c_str(), serviceName.c_str())
                                     : g_service->AddServiceCheck(sid, serviceName))
                  << std::endl;
        exit(0);
    }
    std::string serName;
    while (std::cin >> serName) {
        std::cout << GetErrStr(isHdf ? HdfAddServiceCheck(sid.c_str(), serName.c_str())
                                     : g_service->AddServiceCheck(sid, serName))
                  << std::endl;
    }
}

static void TestGetService(bool isHdf, const std::string &serviceName)
{
    std::string sid;
    if (GetSidForCurrentProcess(sid) < 0) {
        return;
    }
    if (!serviceName.empty()) {
        std::cout << GetErrStr(isHdf ? HdfGetServiceCheck(sid.c_str(), serviceName.c_str())
                                     : g_service->GetServiceCheck(sid, serviceName))
                  << std::endl;
        exit(0);
    }
    std::string serName;
    while (std::cin >> serName) {
        std::cout << GetErrStr(isHdf ? HdfGetServiceCheck(sid.c_str(), serName.c_str())
                                     : g_service->GetServiceCheck(sid, serName))
                  << std::endl;
    }
}

static void TestGetRemoteService(bool isHdf, const std::string &serviceName)
{
    std::string sid;
    if (GetSidForCurrentProcess(sid) < 0) {
        return;
    }
    if (!serviceName.empty()) {
        std::cout << GetErrStr(isHdf ? SELINUX_PERMISSION_DENY
                                     : g_service->GetRemoteServiceCheck(sid, serviceName))
                  << std::endl;
        exit(0);
    }
    std::string serName;
    while (std::cin >> serName) {
        std::cout << GetErrStr(isHdf ? SELINUX_PERMISSION_DENY : g_service->GetRemoteServiceCheck(sid, serName))
                  << std::endl;
    }
}

static void TestListService(bool isHdf)
{
    std::string sid;
    if (GetSidForCurrentProcess(sid) < 0) {
        return;
    }
    std::cout << GetErrStr(isHdf ? HdfListServiceCheck(sid.c_str()) : g_service->ListServiceCheck(sid)) << std::endl;
}

static void Test(const TestInput &testCmd)
{
    switch (testCmd.cmd) {
        case 'a': {
            TestAddService(testCmd.isHdf, testCmd.serviceName);
            exit(0);
        }
        case 'g': {
            TestGetService(testCmd.isHdf, testCmd.serviceName);
            exit(0);
        }
        case 'r': {
            TestGetRemoteService(testCmd.isHdf, testCmd.serviceName);
            exit(0);
        }
        case 'l': {
            TestListService(testCmd.isHdf);
            exit(0);
        }
        default:
            exit(-1);
    }
}

int main(int argc, char *argv[])
{
    struct option options[] = {
        {"help", no_argument, nullptr, 'h'},  {"add", no_argument, nullptr, 'a'},
        {"get", no_argument, nullptr, 'g'},   {"get_remote", no_argument, nullptr, 'r'},
        {"isHdf", no_argument, nullptr, 'd'}, {"list", no_argument, nullptr, 'l'},
        {"serviceName", required_argument, nullptr, 'n'}, {nullptr, no_argument, nullptr, 0},
    };

    if (argc == 1) {
        PrintUsage();
        exit(0);
    }

    TestInput input;
    SetOptions(argc, argv, options, input);
    if (!input.isHdf) {
        g_service = std::make_unique<ServiceChecker>(false);
    }
    Test(input);

    exit(0);
}