urm.cpp

// Copyright (c) Qualcomm Technologies, Inc. and/or its subsidiaries.
// SPDX-License-Identifier: BSD-3-Clause-Clear

#include <csignal>
#include <thread>

#include "Utils.h"
#include "Logger.h"
#include "UrmSettings.h"
#include "UrmPlatformAL.h"
#include "ComponentRegistry.h"

static int8_t terminateServer = false;

static void handleSIGINT(int32_t sig) {
    (void)sig;
    terminateServer = true;
}

static void handleSIGTERM(int32_t sig) {
    (void)sig;
    terminateServer = true;
}

static ErrCode initModuleIfPresent(ModuleID moduleId) {
    // Check if the module is plugged in and Initialize it
    ModuleInfo modInfo = ComponentRegistry::getModuleInfo(moduleId);
    if(modInfo.mInit == nullptr) {
        // Module not enabled
        return RC_SUCCESS;
    }

    return modInfo.mInit(nullptr);
}

static ErrCode cleanupModule(ModuleID moduleId) {
    // Check if the module is plugged in and Initialize it
    ModuleInfo modInfo = ComponentRegistry::getModuleInfo(moduleId);
    if(modInfo.mTear == nullptr) {
        // Module not enabled
        return RC_SUCCESS;
    }

    return modInfo.mTear(nullptr);
}

static void serverCleanup() {
    LOGI("RESTUNE_SERVER_INIT", "Server Stopped, Cleanup Initiated");
    UrmSettings::setServerOnlineStatus(false);
}

int32_t main(int32_t argc, char *argv[]) {
    (void)argc;
    (void)argv;
    // Initialize syslog
    openlog(URM_IDENTIFIER, LOG_PID | LOG_CONS | LOG_NDELAY, LOG_DAEMON);
    setlogmask(LOG_UPTO(LOG_DEBUG));

    ErrCode opStatus = RC_SUCCESS;

    std::signal(SIGINT, handleSIGINT);
    std::signal(SIGTERM, handleSIGTERM);

    TYPELOGV(NOTIFY_RESOURCE_TUNER_INIT_START, getpid());

    // Ready for requests
    if(RC_IS_OK(opStatus)) {
        UrmSettings::setServerOnlineStatus(true);
        UrmSettings::targetConfigs.currMode = MODE_RESUME;
    }

    if(RC_IS_OK(opStatus)) {
        opStatus = initModuleIfPresent(ModuleID::MOD_RESTUNE);
        if(RC_IS_NOTOK(opStatus)) {
            TYPELOGV(MODULE_INIT_FAILED, "resource-tuner");
        }
    }

    if(RC_IS_OK(opStatus)) {
        opStatus = initModuleIfPresent(ModuleID::MOD_CLASSIFIER);
        if(RC_IS_NOTOK(opStatus)) {
            TYPELOGV(MODULE_INIT_FAILED, "classifier");
        }
    }

    if(RC_IS_OK(opStatus)) {
        // Listen for Terminal prompts
        while(!terminateServer) {
            std::this_thread::sleep_for(std::chrono::seconds(2));
        }
    }

    // Cleanup the Server, so that it boots up correctly the next time
    // - This includes Closing the Server Communication Endpoints
    // - Resetting all the Sysfs nodes to their original values
    // - Terminating the different listener and Processor threads created to handle Requests.
    serverCleanup();

    cleanupModule(ModuleID::MOD_RESTUNE);
    cleanupModule(ModuleID::MOD_CLASSIFIER);

    closelog();
    return 0;
}