/*
    This file is part of driver-loader
    Copyright (C) 2017 @maldevel

    driver-loader - Load a Windows Kernel Driver.
    https://github.com/maldevel/driver-loader

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
    For more see the file 'LICENSE' for copying permission.
*/

#include "dlservices.h"

static SC_HANDLE scManager;

bool Services::init(void) {
    scManager = OpenSCManager(NULL, SERVICES_ACTIVE_DATABASE, SC_MANAGER_ALL_ACCESS);
    if (scManager == NULL) {
        return false;
    }

    return true;
}

SC_HANDLE Services::Open(QString service) {
    if (service == NULL || scManager == NULL || service.trimmed().isEmpty() ||
            service.trimmed().length() > 256) return NULL;

    return OpenServiceA(scManager, service.toStdString().c_str(), SERVICE_ALL_ACCESS);
}

void Services::uninit(void) {
    if (scManager == NULL) return;

    CloseServiceHandle(scManager);
}

unsigned long Services::Register(QString driver, QString serviceName, QString displayName,
                                 QString startTypeStr, QString error) {
    if (driver == NULL || serviceName == NULL || scManager == NULL ||
            displayName == NULL || startTypeStr == NULL || error == NULL ||
            driver.trimmed().isEmpty() || serviceName.trimmed().isEmpty() ||
            displayName.trimmed().isEmpty() || startTypeStr.trimmed().isEmpty() ||
            error.trimmed().isEmpty() || serviceName.trimmed().length() > 256 ||
            displayName.trimmed().length() > 256) return 1;

    SC_HANDLE scService;
    unsigned long startType = SERVICE_DEMAND_START;
    unsigned long errorControl = SERVICE_ERROR_NORMAL;

    //"Automatic" "Boot" "Demand" "Disabled" "System"
    if (startTypeStr.trimmed().compare("Automatic", Qt::CaseSensitive) == 0) {
        startType = SERVICE_AUTO_START;
    } else if (startTypeStr.trimmed().compare("Boot", Qt::CaseSensitive) == 0) {
        startType = SERVICE_BOOT_START;
    } else if (startTypeStr.trimmed().compare("Demand", Qt::CaseSensitive) == 0) {
        startType = SERVICE_DEMAND_START;
    } else if (startTypeStr.trimmed().compare("Disabled", Qt::CaseSensitive) == 0) {
        startType = SERVICE_DISABLED;
    } else if (startTypeStr.trimmed().compare("System", Qt::CaseSensitive) == 0) {
        startType = SERVICE_SYSTEM_START;
    }

    //"Critical" "Ignore" "Normal" "Severe"
    if (error.trimmed().compare("Critical", Qt::CaseSensitive) == 0) {
        errorControl = SERVICE_ERROR_CRITICAL;
    } else if (error.trimmed().compare("Ignore", Qt::CaseSensitive) == 0) {
        errorControl = SERVICE_ERROR_IGNORE;
    } else if (error.trimmed().compare("Normal", Qt::CaseSensitive) == 0) {
        errorControl = SERVICE_ERROR_NORMAL;
    } else if (error.trimmed().compare("Severe", Qt::CaseSensitive) == 0) {
        errorControl = SERVICE_ERROR_SEVERE;
    }

    if ((scService = CreateServiceA(scManager, serviceName.trimmed().toStdString().c_str(),
                                    displayName.trimmed().toStdString().c_str(),
                                    SERVICE_ALL_ACCESS, SERVICE_KERNEL_DRIVER,
                                    startType, errorControl,
                                    driver.trimmed().toStdString().c_str(),
                                    NULL, NULL, NULL, NULL, NULL)) == NULL) {
        return GetLastError();
    }

    CloseServiceHandle(scService);

    return 0;
}

unsigned long Services::Unregister(QString service) {
    if (service == NULL || scManager == NULL || service.trimmed().isEmpty() ||
            service.trimmed().length() > 256) return 1;

    SC_HANDLE srvHandle;
    unsigned long error = 0;

    if ((srvHandle = Open(service)) == NULL) {
        return GetLastError();
    }

    if ((error = Stop(srvHandle)) != 0) {
        if (error != ERROR_SERVICE_NOT_ACTIVE) {
            CloseServiceHandle(srvHandle);
            return error;
        }
    }

    if (DeleteService(srvHandle) == 0) {
        CloseServiceHandle(srvHandle);
        return GetLastError();
    }

    CloseServiceHandle(srvHandle);

    return 0;
}

unsigned long Services::Start(SC_HANDLE service) {
    if (service == NULL || scManager == NULL) return 1;

    if (StartService(service, 0, NULL) == 0) {
        return GetLastError();
    }

    return 0;
}

unsigned long Services::Start(QString service) {
    if (service == NULL || scManager == NULL || service.trimmed().isEmpty() ||
            service.trimmed().length() > 256) return 1;

    unsigned long error = 0;

    SC_HANDLE srvHandle;

    if ((srvHandle = Open(service)) == NULL) {
        return GetLastError();
    }

    if ((error = Start(srvHandle)) != 0) {
        CloseServiceHandle(srvHandle);
        return error;
    }

    CloseServiceHandle(srvHandle);

    return 0;
}

unsigned long Services::Stop(SC_HANDLE service) {
    if (service == NULL || scManager == NULL) return 1;

    SERVICE_STATUS serviceStatus;

    if (ControlService(service, SERVICE_CONTROL_STOP, &serviceStatus) == 0) {
        return GetLastError();
    }

    return 0;
}

unsigned long Services::Stop(QString service) {
    if (service == NULL || scManager == NULL || service.trimmed().isEmpty() ||
            service.trimmed().length() > 256) return 1;

    SC_HANDLE srvHandle;
    unsigned long error = 0;

    if ((srvHandle = Open(service)) == NULL) {
        return GetLastError();
    }

    if ((error = Stop(srvHandle)) != 0) {
        CloseServiceHandle(srvHandle);
        return error;
    }

    CloseServiceHandle(srvHandle);
    return 0;
}