OpenHarmony-v4.0.1-Release/s

/*
 * Copyright (c) 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 <algorithm>

#include "ability_death_recipient.h"
#include "datetime_ex.h"
#include "ipc_skeleton.h"
#include "memory_guard.h"
#ifdef RESSCHED_ENABLE
#include "res_sched_client.h"
#endif
#include "sam_log.h"
#include "string_ex.h"
#include "system_ability_manager.h"
#include "schedule/system_ability_state_scheduler.h"

namespace OHOS {
namespace {
constexpr int64_t RESTART_TIME_INTERVAL_LIMIT = 20 * 1000;
constexpr int32_t RESTART_TIMES_LIMIT = 4;
constexpr int32_t MAX_SUBSCRIBE_COUNT = 256;
constexpr int32_t UNLOAD_TIMEOUT_TIME = 5 * 1000;
const std::string LOCAL_DEVICE = "local";
constexpr int32_t MAX_DELAY_TIME = 5 * 60 * 1000;
constexpr const char* CANCEL_UNLOAD = "cancelUnload";
const std::string KEY_EVENT_ID = "eventId";
const std::string KEY_NAME = "name";
const std::string KEY_VALUE = "value";
const std::string KEY_EXTRA_DATA_ID = "extraDataId";
const std::string KEY_UNLOAD_TIMEOUT = "unloadTimeout";
}
void SystemAbilityStateScheduler::Init(const std::list<SaProfile>& saProfiles)
{
    HILOGI("[SA Scheduler] init start");
    InitStateContext(saProfiles);
    processListenerDeath_ = sptr<IRemoteObject::DeathRecipient>(new SystemProcessListenerDeathRecipient());
    auto unloadRunner = AppExecFwk::EventRunner::Create("UnloadHandler");
    unloadEventHandler_ = std::make_shared<UnloadEventHandler>(unloadRunner, weak_from_this());
    unloadEventHandler_->PostTask([]() { Samgr::MemoryGuard cacheGuard; });

    auto listener =  std::dynamic_pointer_cast<SystemAbilityStateListener>(shared_from_this());
    stateMachine_ = std::make_shared<SystemAbilityStateMachine>(listener);
    stateEventHandler_ = std::make_shared<SystemAbilityEventHandler>(stateMachine_);

    auto processRunner = AppExecFwk::EventRunner::Create("ProcessHandler");
    processHandler_ = std::make_shared<AppExecFwk::EventHandler>(processRunner);
    processHandler_->PostTask([]() { Samgr::MemoryGuard cacheGuard; });
    HILOGI("[SA Scheduler] init end");
}

void SystemAbilityStateScheduler::InitStateContext(const std::list<SaProfile>& saProfiles)
{
    for (auto& saProfile : saProfiles) {
        if (saProfile.process.empty()) {
            continue;
        }
        std::unique_lock<std::shared_mutex> processWriteLock(processMapLock_);
        if (processContextMap_.count(saProfile.process) == 0) {
            auto processContext = std::make_shared<SystemProcessContext>();
            processContext->processName = saProfile.process;
            processContext->abilityStateCountMap[SystemAbilityState::NOT_LOADED] = 0;
            processContext->abilityStateCountMap[SystemAbilityState::LOADING] = 0;
            processContext->abilityStateCountMap[SystemAbilityState::LOADED] = 0;
            processContext->abilityStateCountMap[SystemAbilityState::UNLOADABLE] = 0;
            processContext->abilityStateCountMap[SystemAbilityState::UNLOADING] = 0;
            processContextMap_[saProfile.process] = processContext;
        }
        processContextMap_[saProfile.process]->saList.push_back(saProfile.saId);
        processContextMap_[saProfile.process]->abilityStateCountMap[SystemAbilityState::NOT_LOADED]++;
        auto abilityContext = std::make_shared<SystemAbilityContext>();
        abilityContext->systemAbilityId = saProfile.saId;
        abilityContext->isAutoRestart = saProfile.autoRestart;
        int32_t delayUnloadTime = LimitDelayUnloadTime(saProfile.stopOnDemand.delayTime);
        abilityContext->delayUnloadTime = delayUnloadTime;
        abilityContext->ownProcessContext = processContextMap_[saProfile.process];
        std::unique_lock<std::shared_mutex> abiltyWriteLock(abiltyMapLock_);
        abilityContextMap_[saProfile.saId] = abilityContext;
    }
}

int32_t SystemAbilityStateScheduler::LimitDelayUnloadTime(int32_t delayUnloadTime)
{
    if (delayUnloadTime < 0) {
        return 0;
    }
    if (delayUnloadTime > MAX_DELAY_TIME) {
        return MAX_DELAY_TIME;
    }
    return delayUnloadTime;
}

bool SystemAbilityStateScheduler::GetSystemAbilityContext(int32_t systemAbilityId,
    std::shared_ptr<SystemAbilityContext>& abilityContext)
{
    std::shared_lock<std::shared_mutex> readLock(abiltyMapLock_);
    if (abilityContextMap_.count(systemAbilityId) == 0) {
        HILOGE("[SA Scheduler][SA: %{public}d] not in SA profiles", systemAbilityId);
        return false;
    }
    abilityContext = abilityContextMap_[systemAbilityId];
    if (abilityContext == nullptr) {
        HILOGE("[SA Scheduler][SA: %{public}d] context is nullptr", systemAbilityId);
        return false;
    }
    if (abilityContext->ownProcessContext == nullptr) {
        HILOGE("[SA Scheduler][SA: %{public}d] not in any process", systemAbilityId);
        return false;
    }
    return true;
}

bool SystemAbilityStateScheduler::GetSystemProcessContext(const std::u16string& processName,
    std::shared_ptr<SystemProcessContext>& processContext)
{
    std::shared_lock<std::shared_mutex> readLock(processMapLock_);
    if (processContextMap_.count(processName) == 0) {
        HILOGE("[SA Scheduler][processName: %{public}s] invalid", Str16ToStr8(processName).c_str());
        return false;
    }
    processContext = processContextMap_[processName];
    if (processContext == nullptr) {
        HILOGE("[SA Scheduler][processName: %{public}s] context is nullptr", Str16ToStr8(processName).c_str());
        return false;
    }
    return true;
}

bool SystemAbilityStateScheduler::IsSystemAbilityUnloading(int32_t systemAbilityId)
{
    std::shared_ptr<SystemAbilityContext> abilityContext;
    if (!GetSystemAbilityContext(systemAbilityId, abilityContext)) {
        return false;
    }
    std::lock_guard<std::recursive_mutex> autoLock(abilityContext->ownProcessContext->processLock);
    if (abilityContext->state ==SystemAbilityState::UNLOADING
        || abilityContext->ownProcessContext->state == SystemProcessState::STOPPING) {
        return true;
    }
    return false;
}

int32_t SystemAbilityStateScheduler::HandleLoadAbilityEvent(int32_t systemAbilityId, bool& isExist)
{
    HILOGI("[SA Scheduler][SA: %{public}d] handle load event by check start",
        systemAbilityId);
    std::shared_ptr<SystemAbilityContext> abilityContext;
    if (!GetSystemAbilityContext(systemAbilityId, abilityContext)) {
        isExist = false;
        return ERR_INVALID_VALUE;
    }
    std::lock_guard<std::recursive_mutex> autoLock(abilityContext->ownProcessContext->processLock);
    if (abilityContext->ownProcessContext->state == SystemProcessState::NOT_STARTED) {
        isExist = false;
        return ERR_INVALID_VALUE;
    }
    if (abilityContext->ownProcessContext->state == SystemProcessState::STARTED
        && abilityContext->state ==SystemAbilityState::NOT_LOADED) {
        bool result = SystemAbilityManager::GetInstance()->DoLoadOnDemandAbility(systemAbilityId, isExist);
        if (result) {
            return stateMachine_->AbilityStateTransitionLocked(abilityContext, SystemAbilityState::LOADING);
        }
        return ERR_INVALID_VALUE;
    }
    isExist = true;
    return ERR_OK;
}

int32_t SystemAbilityStateScheduler::HandleLoadAbilityEvent(const LoadRequestInfo& loadRequestInfo)
{
    HILOGI("[SA Scheduler][SA: %{public}d] handle load event start, callingpid: %{public}d",
        loadRequestInfo.systemAbilityId, loadRequestInfo.callingPid);
    std::shared_ptr<SystemAbilityContext> abilityContext;
    if (!GetSystemAbilityContext(loadRequestInfo.systemAbilityId, abilityContext)) {
        return ERR_INVALID_VALUE;
    }
    std::lock_guard<std::recursive_mutex> autoLock(abilityContext->ownProcessContext->processLock);
    return HandleLoadAbilityEventLocked(abilityContext, loadRequestInfo);
}

int32_t SystemAbilityStateScheduler::HandleLoadAbilityEventLocked(
    const std::shared_ptr<SystemAbilityContext>& abilityContext, const LoadRequestInfo& loadRequestInfo)
{
    if (abilityContext->state ==SystemAbilityState::UNLOADING
        || abilityContext->ownProcessContext->state == SystemProcessState::STOPPING) {
        return PendLoadEventLocked(abilityContext, loadRequestInfo);
    }
    nlohmann::json activeReason;
    activeReason[KEY_EVENT_ID] = loadRequestInfo.loadEvent.eventId;
    activeReason[KEY_NAME] = loadRequestInfo.loadEvent.name;
    activeReason[KEY_VALUE] = loadRequestInfo.loadEvent.value;
    activeReason[KEY_EXTRA_DATA_ID] = loadRequestInfo.loadEvent.extraDataId;
    int32_t result = ERR_INVALID_VALUE;
    switch (abilityContext->state) {
        case SystemAbilityState::LOADING:
            result = RemovePendingUnloadEventLocked(abilityContext);
            break;
        case SystemAbilityState::LOADED:
            result = RemoveDelayUnloadEventLocked(abilityContext->systemAbilityId);
            break;
        case SystemAbilityState::UNLOADABLE:
            result = ActiveSystemAbilityLocked(abilityContext, activeReason);
            break;
        case SystemAbilityState::NOT_LOADED:
            result = ERR_OK;
            break;
        default:
            result = ERR_INVALID_VALUE;
            HILOGI("[SA Scheduler][SA: %{public}d] in state %{public}d, cannot load ability",
                loadRequestInfo.systemAbilityId, abilityContext->state);
            break;
    }
    if (result == ERR_OK) {
        return DoLoadSystemAbilityLocked(abilityContext, loadRequestInfo);
    }
    return result;
}

int32_t SystemAbilityStateScheduler::HandleUnloadAbilityEvent(const UnloadRequestInfo& unloadRequestInfo)
{
    HILOGI("[SA Scheduler][SA: %{public}d] handle unload event start", unloadRequestInfo.systemAbilityId);
    std::shared_ptr<SystemAbilityContext> abilityContext;
    if (!GetSystemAbilityContext(unloadRequestInfo.systemAbilityId, abilityContext)) {
        return ERR_INVALID_VALUE;
    }
    std::lock_guard<std::recursive_mutex> autoLock(abilityContext->ownProcessContext->processLock);
    return HandleUnloadAbilityEventLocked(abilityContext, unloadRequestInfo);
}

int32_t SystemAbilityStateScheduler::HandleUnloadAbilityEventLocked(
    const std::shared_ptr<SystemAbilityContext>& abilityContext, const UnloadRequestInfo& unloadRequestInfo)
{
    abilityContext->unloadRequest = unloadRequestInfo;
    int32_t result = ERR_INVALID_VALUE;
    switch (abilityContext->state) {
        case SystemAbilityState::LOADING:
            result = PendUnloadEventLocked(abilityContext, unloadRequestInfo);
            break;
        case SystemAbilityState::LOADED:
            if (unloadRequestInfo.unloadEvent.eventId == INTERFACE_CALL) {
                result = ProcessDelayUnloadEvent(abilityContext->systemAbilityId);
            } else {
                result = SendDelayUnloadEventLocked(abilityContext->systemAbilityId, abilityContext->delayUnloadTime);
            }
            break;
        default:
            result = ERR_OK;
            HILOGI("[SA Scheduler][SA: %{public}d] in state %{public}d, not need unload ability",
                abilityContext->systemAbilityId, abilityContext->state);
            break;
    }
    return result;
}

int32_t SystemAbilityStateScheduler::HandleCancelUnloadAbilityEvent(int32_t systemAbilityId)
{
    HILOGI("[SA Scheduler][SA: %{public}d] cancel unload start", systemAbilityId);
    std::shared_ptr<SystemAbilityContext> abilityContext;
    if (!GetSystemAbilityContext(systemAbilityId, abilityContext)) {
        return ERR_INVALID_VALUE;
    }
    nlohmann::json activeReason;
    activeReason[KEY_EVENT_ID] = INTERFACE_CALL;
    activeReason[KEY_NAME] = CANCEL_UNLOAD;
    activeReason[KEY_VALUE] = "";
    activeReason[KEY_EXTRA_DATA_ID] = -1;
    int32_t result = ERR_INVALID_VALUE;
    std::lock_guard<std::recursive_mutex> autoLock(abilityContext->ownProcessContext->processLock);
    switch (abilityContext->state) {
        case SystemAbilityState::UNLOADABLE:
            result = ActiveSystemAbilityLocked(abilityContext, activeReason);
            break;
        default:
            result = ERR_OK;
            HILOGI("[SA Scheduler][SA: %{public}d] in state %{public}d, not need cancel unload",
                systemAbilityId, abilityContext->state);
            break;
    }
    return result;
}

int32_t SystemAbilityStateScheduler::ActiveSystemAbilityLocked(
    const std::shared_ptr<SystemAbilityContext>& abilityContext,
    const nlohmann::json& activeReason)
{
    bool result = SystemAbilityManager::GetInstance()->ActiveSystemAbility(abilityContext->systemAbilityId,
        abilityContext->ownProcessContext->processName, activeReason);
    if (!result) {
        HILOGE("[SA Scheduler][SA: %{public}d] active ability failed", abilityContext->systemAbilityId);
        return ERR_INVALID_VALUE;
    }
    return stateMachine_->AbilityStateTransitionLocked(abilityContext, SystemAbilityState::LOADED);
}

int32_t SystemAbilityStateScheduler::SendAbilityStateEvent(int32_t systemAbilityId, AbilityStateEvent event)
{
    HILOGD("[SA Scheduler][SA: %{public}d] receive state event", systemAbilityId);
    std::shared_ptr<SystemAbilityContext> abilityContext;
    if (!GetSystemAbilityContext(systemAbilityId, abilityContext)) {
        return ERR_INVALID_VALUE;
    }
    std::lock_guard<std::recursive_mutex> autoLock(abilityContext->ownProcessContext->processLock);
    return stateEventHandler_->HandleAbilityEventLocked(abilityContext, event);
}

int32_t SystemAbilityStateScheduler::SendProcessStateEvent(const ProcessInfo& processInfo, ProcessStateEvent event)
{
    HILOGD("[SA Scheduler][process: %{public}s] receive state event",
        Str16ToStr8(processInfo.processName).c_str());
    std::shared_ptr<SystemProcessContext> processContext;
    if (!GetSystemProcessContext(processInfo.processName, processContext)) {
        return ERR_INVALID_VALUE;
    }
    std::lock_guard<std::recursive_mutex> autoLock(processContext->processLock);
    return stateEventHandler_->HandleProcessEventLocked(processContext, processInfo, event);
}

int32_t SystemAbilityStateScheduler::SendDelayUnloadEventLocked(uint32_t systemAbilityId, int32_t delayTime)
{
    if (unloadEventHandler_ == nullptr) {
        HILOGE("[SA Scheduler] unload handler not initialized!");
        return ERR_INVALID_VALUE;
    }
    if (unloadEventHandler_->HasInnerEvent(systemAbilityId)) {
        return ERR_OK;
    }
    HILOGI("[SA Scheduler][SA: %{public}d] send delay unload event", systemAbilityId);
    bool ret = unloadEventHandler_->SendEvent(systemAbilityId, 0, delayTime);
    if (!ret) {
        HILOGE("[SA Scheduler] send event failed!");
        return ERR_INVALID_VALUE;
    }
    return ERR_OK;
}

int32_t SystemAbilityStateScheduler::RemoveDelayUnloadEventLocked(uint32_t systemAbilityId)
{
    if (unloadEventHandler_ == nullptr) {
        HILOGE("[SA Scheduler] unload handler not initialized!");
        return ERR_INVALID_VALUE;
    }
    if (!unloadEventHandler_->HasInnerEvent(systemAbilityId)) {
        return ERR_OK;
    }
    HILOGI("[SA Scheduler][SA: %{public}d] remove delay unload event", systemAbilityId);
    unloadEventHandler_->RemoveEvent(systemAbilityId);
    return ERR_OK;
}

int32_t SystemAbilityStateScheduler::PendLoadEventLocked(const std::shared_ptr<SystemAbilityContext>& abilityContext,
    const LoadRequestInfo& loadRequestInfo)
{
    HILOGI("[SA Scheduler][SA: %{public}d] save load event", abilityContext->systemAbilityId);
    if (loadRequestInfo.callback == nullptr) {
        HILOGW("[SA Scheduler] callback invalid!");
        return ERR_INVALID_VALUE;
    }
    bool isExist = std::any_of(abilityContext->pendingLoadEventList.begin(),
        abilityContext->pendingLoadEventList.end(), [&loadRequestInfo](const auto& loadEventItem) {
            return loadRequestInfo.callback->AsObject() == loadEventItem.callback->AsObject();
        });
    if (isExist) {
        HILOGI("[SA Scheduler][SA: %{public}d] already existed callback object", abilityContext->systemAbilityId);
        return ERR_OK;
    }
    auto& count = abilityContext->pendingLoadEventCountMap[loadRequestInfo.callingPid];
    if (count >= MAX_SUBSCRIBE_COUNT) {
        HILOGE("[SA Scheduler][SA: %{public}d] pid:%{public}d overflow max callback count!",
            abilityContext->systemAbilityId, loadRequestInfo.callingPid);
        return ERR_PERMISSION_DENIED;
    }
    ++count;
    abilityContext->pendingLoadEventList.emplace_back(loadRequestInfo);
    abilityContext->pendingEvent = PendingEvent::LOAD_ABILITY_EVENT;
    return ERR_OK;
}

int32_t SystemAbilityStateScheduler::PendUnloadEventLocked(
    const std::shared_ptr<SystemAbilityContext>& abilityContext, const UnloadRequestInfo& unloadRequestInfo)
{
    HILOGI("[SA Scheduler][SA: %{public}d] save unload event", abilityContext->systemAbilityId);
    abilityContext->pendingEvent = PendingEvent::UNLOAD_ABILITY_EVENT;
    abilityContext->pendingUnloadEvent = unloadRequestInfo;
    return ERR_OK;
}

int32_t SystemAbilityStateScheduler::RemovePendingUnloadEventLocked(
    const std::shared_ptr<SystemAbilityContext>& abilityContext)
{
    if (abilityContext->pendingEvent == PendingEvent::UNLOAD_ABILITY_EVENT) {
        HILOGI("[SA Scheduler][SA: %{public}d] remove pending unload event", abilityContext->systemAbilityId);
        abilityContext->pendingEvent = PendingEvent::NO_EVENT;
    }
    return ERR_OK;
}

int32_t SystemAbilityStateScheduler::HandlePendingLoadEventLocked(
    const std::shared_ptr<SystemAbilityContext>& abilityContext)
{
    if (abilityContext->pendingEvent != PendingEvent::LOAD_ABILITY_EVENT) {
        HILOGD("[SA Scheduler][SA: %{public}d] no pending load event", abilityContext->systemAbilityId);
        return ERR_OK;
    }
    HILOGI("[SA Scheduler][SA: %{public}d] handle pending load event start", abilityContext->systemAbilityId);
    abilityContext->pendingEvent = PendingEvent::NO_EVENT;
    for (auto& loadRequestInfo : abilityContext->pendingLoadEventList) {
        int32_t result = HandleLoadAbilityEventLocked(abilityContext, loadRequestInfo);
        if (result != ERR_OK) {
            HILOGE("[SA Scheduler][SA: %{public}d] handle pending load event failed, callingPid: %{public}d",
                abilityContext->systemAbilityId, loadRequestInfo.callingPid);
        }
    }
    abilityContext->pendingLoadEventList.clear();
    abilityContext->pendingLoadEventCountMap.clear();
    return ERR_OK;
}

int32_t SystemAbilityStateScheduler::HandlePendingUnloadEventLocked(
    const std::shared_ptr<SystemAbilityContext>& abilityContext)
{
    if (abilityContext->pendingEvent != PendingEvent::UNLOAD_ABILITY_EVENT) {
        HILOGD("[SA Scheduler][SA: %{public}d] no pending unload event", abilityContext->systemAbilityId);
        return ERR_OK;
    }
    HILOGI("[SA Scheduler][SA: %{public}d] handle pending unload event start", abilityContext->systemAbilityId);
    abilityContext->pendingEvent = PendingEvent::NO_EVENT;
    return HandleUnloadAbilityEventLocked(abilityContext, abilityContext->pendingUnloadEvent);
}

int32_t SystemAbilityStateScheduler::DoLoadSystemAbilityLocked(
    const std::shared_ptr<SystemAbilityContext>& abilityContext, const LoadRequestInfo& loadRequestInfo)
{
    int32_t result = ERR_OK;
    if (loadRequestInfo.deviceId == LOCAL_DEVICE) {
        HILOGD("[SA Scheduler][SA: %{public}d] load ability from local start", abilityContext->systemAbilityId);
        result = SystemAbilityManager::GetInstance()->DoLoadSystemAbility(abilityContext->systemAbilityId,
            abilityContext->ownProcessContext->processName, loadRequestInfo.callback, loadRequestInfo.callingPid,
            loadRequestInfo.loadEvent);
    } else {
        HILOGD("[SA Scheduler][SA: %{public}d] load ability from remote start", abilityContext->systemAbilityId);
        result = SystemAbilityManager::GetInstance()->DoLoadSystemAbilityFromRpc(loadRequestInfo.deviceId,
            abilityContext->systemAbilityId, abilityContext->ownProcessContext->processName, loadRequestInfo.callback,
            loadRequestInfo.loadEvent);
    }
    if (result == ERR_OK && abilityContext->state == SystemAbilityState::NOT_LOADED) {
        return stateMachine_->AbilityStateTransitionLocked(abilityContext, SystemAbilityState::LOADING);
    }
    return result;
}

int32_t SystemAbilityStateScheduler::TryUnloadAllSystemAbility(
    const std::shared_ptr<SystemProcessContext>& processContext)
{
    if (processContext == nullptr) {
        HILOGE("[SA Scheduler] process context is nullptr");
        return ERR_INVALID_VALUE;
    }
    std::lock_guard<std::recursive_mutex> autoLock(processContext->processLock);
    if (CanUnloadAllSystemAbility(processContext)) {
        return UnloadAllSystemAbilityLocked(processContext);
    }
    return ERR_OK;
}

bool SystemAbilityStateScheduler::CanUnloadAllSystemAbility(
    const std::shared_ptr<SystemProcessContext>& processContext)
{
    std::shared_lock<std::shared_mutex> sharedLock(processContext->stateCountLock);
    uint32_t notLoadAbilityCount = processContext->abilityStateCountMap[SystemAbilityState::NOT_LOADED];
    uint32_t unloadableAbilityCount = processContext->abilityStateCountMap[SystemAbilityState::UNLOADABLE];
    HILOGI("[SA Scheduler][process: %{public}s] SA num: %{public}zu, notloaded: %{public}d, unloadable: %{public}d",
        Str16ToStr8(processContext->processName).c_str(), processContext->saList.size(), notLoadAbilityCount,
        unloadableAbilityCount);
    if (unloadableAbilityCount == 0) {
        return false;
    }
    if (notLoadAbilityCount + unloadableAbilityCount == processContext->saList.size()) {
        return true;
    }
    return false;
}

int32_t SystemAbilityStateScheduler::PostTryUnloadAllAbilityTask(
    const std::shared_ptr<SystemProcessContext>& processContext)
{
    bool result = processHandler_->PostTask([this, processContext] () {
        int32_t ret = TryUnloadAllSystemAbility(processContext);
        if (ret != ERR_OK) {
            HILOGE("[SA Scheduler][process: %{public}s] unload all SA failed",
                Str16ToStr8(processContext->processName).c_str());
        }
    });
    if (!result) {
        HILOGW("[SA Scheduler][process: %{public}s] post task failed",
            Str16ToStr8(processContext->processName).c_str());
        return ERR_INVALID_VALUE;
    }
    return ERR_OK;
}

int32_t SystemAbilityStateScheduler::PostUnloadTimeoutTask(const std::shared_ptr<SystemProcessContext>& processContext)
{
    auto timeoutTask = [this, processContext] () {
        std::lock_guard<std::recursive_mutex> autoLock(processContext->processLock);
        if (processContext->state == SystemProcessState::STOPPING) {
            HILOGW("[SA Scheduler][process: %{public}s] unload SA timeout",
                Str16ToStr8(processContext->processName).c_str());
            int32_t result = KillSystemProcessLocked(processContext);
            HILOGI("[SA Scheduler][process: %{public}s] kill timeout process ret: %{public}d",
                Str16ToStr8(processContext->processName).c_str(), result);
        }
    };
    bool ret = processHandler_->PostTask(timeoutTask, KEY_UNLOAD_TIMEOUT + Str16ToStr8(processContext->processName),
        UNLOAD_TIMEOUT_TIME);
    if (!ret) {
        HILOGW("[SA Scheduler][process: %{public}s] post timeout task failed",
            Str16ToStr8(processContext->processName).c_str());
        return ERR_INVALID_VALUE;
    }
    return ERR_OK;
}

void SystemAbilityStateScheduler::RemoveUnloadTimeoutTask(const std::shared_ptr<SystemProcessContext>& processContext)
{
    processHandler_->RemoveTask(KEY_UNLOAD_TIMEOUT + Str16ToStr8(processContext->processName));
}

int32_t SystemAbilityStateScheduler::PostTryKillProcessTask(
    const std::shared_ptr<SystemProcessContext>& processContext)
{
    bool result = processHandler_->PostTask([this, processContext] () {
        int32_t ret = TryKillSystemProcess(processContext);
        if (ret != ERR_OK) {
            HILOGE("[SA Scheduler][process: %{public}s] kill process failed",
                Str16ToStr8(processContext->processName).c_str());
        }
    });
    if (!result) {
        HILOGW("[SA Scheduler][process: %{public}s] post task failed",
            Str16ToStr8(processContext->processName).c_str());
        return ERR_INVALID_VALUE;
    }
    return ERR_OK;
}

int32_t SystemAbilityStateScheduler::UnloadAllSystemAbilityLocked(
    const std::shared_ptr<SystemProcessContext>& processContext)
{
    HILOGI("[SA Scheduler][process: %{public}s] unload all SA", Str16ToStr8(processContext->processName).c_str());
    for (auto& saId : processContext->saList) {
        std::shared_ptr<SystemAbilityContext> abilityContext;
        if (!GetSystemAbilityContext(saId, abilityContext)) {
            continue;
        }
        int32_t result = ERR_OK;
        if (abilityContext->state == SystemAbilityState::UNLOADABLE) {
            result = DoUnloadSystemAbilityLocked(abilityContext);
        }
        if (result != ERR_OK) {
            HILOGE("[SA Scheduler][SA: %{public}d] unload failed", saId);
        }
    }
    PostUnloadTimeoutTask(processContext);
    return stateMachine_->ProcessStateTransitionLocked(processContext, SystemProcessState::STOPPING);
}

int32_t SystemAbilityStateScheduler::DoUnloadSystemAbilityLocked(
    const std::shared_ptr<SystemAbilityContext>& abilityContext)
{
    int32_t result = ERR_OK;
    HILOGI("[SA Scheduler][SA: %{public}d] unload ability start", abilityContext->systemAbilityId);
    result = SystemAbilityManager::GetInstance()->DoUnloadSystemAbility(abilityContext->systemAbilityId,
        abilityContext->ownProcessContext->processName, abilityContext->unloadRequest.unloadEvent);
    if (result == ERR_OK) {
        return stateMachine_->AbilityStateTransitionLocked(abilityContext, SystemAbilityState::UNLOADING);
    }
    return result;
}

int32_t SystemAbilityStateScheduler::TryKillSystemProcess(
    const std::shared_ptr<SystemProcessContext>& processContext)
{
    if (processContext == nullptr) {
        HILOGE("[SA Scheduler] process context is nullptr");
        return ERR_INVALID_VALUE;
    }
    std::lock_guard<std::recursive_mutex> autoLock(processContext->processLock);
    if (CanKillSystemProcess(processContext)) {
        return KillSystemProcessLocked(processContext);
    }
    return ERR_OK;
}

bool SystemAbilityStateScheduler::CanKillSystemProcess(
    const std::shared_ptr<SystemProcessContext>& processContext)
{
    std::shared_lock<std::shared_mutex> sharedLock(processContext->stateCountLock);
    uint32_t notLoadAbilityCount = processContext->abilityStateCountMap[SystemAbilityState::NOT_LOADED];
    HILOGI("[SA Scheduler][process: %{public}s] SA num: %{public}zu, not loaded num: %{public}d",
        Str16ToStr8(processContext->processName).c_str(), processContext->saList.size(), notLoadAbilityCount);
    if (notLoadAbilityCount == processContext->saList.size()) {
        return true;
    }
    return false;
}

int32_t SystemAbilityStateScheduler::KillSystemProcessLocked(
    const std::shared_ptr<SystemProcessContext>& processContext)
{
    int32_t result = ERR_OK;
    #ifdef RESSCHED_ENABLE
    std::unordered_map<std::string, std::string> payload;
    payload["pid"] = std::to_string(processContext->pid);
    payload["uid"] = std::to_string(processContext->uid);
    payload["processName"] = Str16ToStr8(processContext->processName);
    result = ResourceSchedule::ResSchedClient::GetInstance().KillProcess(payload);
    HILOGI("[SA Scheduler][process: %{public}s] kill process, pid: %{public}d, uid: %{public}d, result: %{public}d",
        Str16ToStr8(processContext->processName).c_str(), processContext->pid, processContext->uid, result);
    #endif
    return result;
}

bool SystemAbilityStateScheduler::CanRestartProcessLocked(const std::shared_ptr<SystemProcessContext>& processContext)
{
    if (!processContext->enableRestart) {
        return false;
    }
    int64_t curtime = GetTickCount();
    if (processContext->restartCountsCtrl.size() < RESTART_TIMES_LIMIT) {
        processContext->restartCountsCtrl.push_back(curtime);
        return true;
    } else if (processContext->restartCountsCtrl.size() == RESTART_TIMES_LIMIT) {
        if (curtime - processContext->restartCountsCtrl.front() < RESTART_TIME_INTERVAL_LIMIT) {
            processContext->enableRestart = false;
            return false;
        }
        processContext->restartCountsCtrl.push_back(curtime);
        processContext->restartCountsCtrl.pop_front();
        return true;
    } else {
        HILOGE("[SA Scheduler][process: %{public}s] unkown error",
            Str16ToStr8(processContext->processName).c_str());
    }
    return false;
}

int32_t SystemAbilityStateScheduler::GetAbnormallyDiedAbilityLocked(
    std::shared_ptr<SystemProcessContext>& processContext,
    std::list<std::shared_ptr<SystemAbilityContext>>& abnormallyDiedAbilityList)
{
    for (auto& saId : processContext->saList) {
        std::shared_ptr<SystemAbilityContext> abilityContext;
        if (!GetSystemAbilityContext(saId, abilityContext)) {
            continue;
        }
        if (abilityContext->state == SystemAbilityState::LOADED
            || abilityContext->state == SystemAbilityState::LOADING) {
            HILOGI("[SA Scheduler][SA: %{public}d] abnormally died", abilityContext->systemAbilityId);
            if (abilityContext->isAutoRestart) {
                HILOGI("[SA Scheduler][SA: %{public}d] is auto restart", abilityContext->systemAbilityId);
                abnormallyDiedAbilityList.emplace_back(abilityContext);
            }
        }
    }
    return ERR_OK;
}


int32_t SystemAbilityStateScheduler::HandleAbnormallyDiedAbilityLocked(
    std::shared_ptr<SystemProcessContext>& processContext,
    std::list<std::shared_ptr<SystemAbilityContext>>& abnormallyDiedAbilityList)
{
    if (abnormallyDiedAbilityList.empty()) {
        return ERR_OK;
    }
    if (!CanRestartProcessLocked(processContext)) {
        HILOGW("[SA Scheduler][process: %{public}s] can't restart: More than 4 restarts in 20 seconds",
            Str16ToStr8(processContext->processName).c_str());
        return ERR_OK;
    }
    OnDemandEvent onDemandEvent = {INTERFACE_CALL, "restart"};
    sptr<ISystemAbilityLoadCallback> callback(new SystemAbilityLoadCallbackStub());
    for (auto& abilityContext : abnormallyDiedAbilityList) {
        LoadRequestInfo loadRequestInfo = {abilityContext->systemAbilityId,
            LOCAL_DEVICE, callback, -1, onDemandEvent};
        HandleLoadAbilityEventLocked(abilityContext, loadRequestInfo);
    }
    return ERR_OK;
}

void SystemAbilityStateScheduler::NotifyProcessStarted(const std::shared_ptr<SystemProcessContext>& processContext)
{
    std::shared_lock<std::shared_mutex> readLock(listenerSetLock_);
    for (auto& listener : processListeners_) {
        if (listener->AsObject() != nullptr) {
            SystemProcessInfo systemProcessInfo = {Str16ToStr8(processContext->processName), processContext->pid,
                processContext->uid};
            listener->OnSystemProcessStarted(systemProcessInfo);
        }
    }
}

void SystemAbilityStateScheduler::NotifyProcessStopped(const std::shared_ptr<SystemProcessContext>& processContext)
{
    std::shared_lock<std::shared_mutex> readLock(listenerSetLock_);
    for (auto& listener : processListeners_) {
        if (listener->AsObject() != nullptr) {
            SystemProcessInfo systemProcessInfo = {Str16ToStr8(processContext->processName), processContext->pid,
                processContext->uid};
            listener->OnSystemProcessStopped(systemProcessInfo);
        }
    }
}

void SystemAbilityStateScheduler::OnProcessStartedLocked(const std::u16string& processName)
{
    HILOGI("[SA Scheduler][process: %{public}s] started", Str16ToStr8(processName).c_str());
    std::shared_ptr<SystemProcessContext> processContext;
    if (!GetSystemProcessContext(processName, processContext)) {
        return;
    }
    NotifyProcessStarted(processContext);
}

void SystemAbilityStateScheduler::OnProcessNotStartedLocked(const std::u16string& processName)
{
    HILOGI("[SA Scheduler][process: %{public}s] stopped", Str16ToStr8(processName).c_str());
    std::shared_ptr<SystemProcessContext> processContext;
    if (!GetSystemProcessContext(processName, processContext)) {
        return;
    }
    NotifyProcessStopped(processContext);
    RemoveUnloadTimeoutTask(processContext);

    std::list<std::shared_ptr<SystemAbilityContext>> abnormallyDiedAbilityList;
    GetAbnormallyDiedAbilityLocked(processContext, abnormallyDiedAbilityList);
    for (auto& saId : processContext->saList) {
        std::shared_ptr<SystemAbilityContext> abilityContext;
        if (!GetSystemAbilityContext(saId, abilityContext)) {
            continue;
        }
        int32_t result = stateMachine_->AbilityStateTransitionLocked(abilityContext, SystemAbilityState::NOT_LOADED);
        if (result != ERR_OK) {
            continue;
        }
        HandlePendingLoadEventLocked(abilityContext);
    }
    HandleAbnormallyDiedAbilityLocked(processContext, abnormallyDiedAbilityList);
}

int32_t SystemAbilityStateScheduler::HandleAbilityDiedEvent(int32_t systemAbilityId)
{
    HILOGD("[SA Scheduler][SA: %{public}d] handle ability died event", systemAbilityId);
    return ERR_OK;
}

void SystemAbilityStateScheduler::OnAbilityNotLoadedLocked(int32_t systemAbilityId)
{
    HILOGI("[SA Scheduler][SA: %{public}d] not loaded", systemAbilityId);
    std::shared_ptr<SystemAbilityContext> abilityContext;
    if (!GetSystemAbilityContext(systemAbilityId, abilityContext)) {
        return;
    }
    RemoveDelayUnloadEventLocked(abilityContext->systemAbilityId);
    RemovePendingUnloadEventLocked(abilityContext);
    if (abilityContext->ownProcessContext->state == SystemProcessState::STOPPING) {
        PostTryKillProcessTask(abilityContext->ownProcessContext);
    } else if (abilityContext->ownProcessContext->state == SystemProcessState::STARTED) {
        PostTryUnloadAllAbilityTask(abilityContext->ownProcessContext);
    }
}

void SystemAbilityStateScheduler::OnAbilityLoadedLocked(int32_t systemAbilityId)
{
    HILOGI("[SA Scheduler][SA: %{public}d] loaded", systemAbilityId);
    std::shared_ptr<SystemAbilityContext> abilityContext;
    if (!GetSystemAbilityContext(systemAbilityId, abilityContext)) {
        return;
    }
    HandlePendingUnloadEventLocked(abilityContext);
}

void SystemAbilityStateScheduler::OnAbilityUnloadableLocked(int32_t systemAbilityId)
{
    HILOGI("[SA Scheduler][SA: %{public}d] unloadable", systemAbilityId);
    std::shared_ptr<SystemAbilityContext> abilityContext;
    if (!GetSystemAbilityContext(systemAbilityId, abilityContext)) {
        return;
    }
    PostTryUnloadAllAbilityTask(abilityContext->ownProcessContext);
}

int32_t SystemAbilityStateScheduler::GetRunningSystemProcess(std::list<SystemProcessInfo>& systemProcessInfos)
{
    HILOGI("[SA Scheduler] get running process");
    std::shared_lock<std::shared_mutex> readLock(processMapLock_);
    for (auto it : processContextMap_) {
        auto& processContext = it.second;
        if (processContext == nullptr) {
            continue;
        }
        std::lock_guard<std::recursive_mutex> autoLock(processContext->processLock);
        if (processContext->state == SystemProcessState::STARTED) {
            SystemProcessInfo systemProcessInfo = {Str16ToStr8(processContext->processName), processContext->pid,
                processContext->uid};
            systemProcessInfos.emplace_back(systemProcessInfo);
        }
    }
    return ERR_OK;
}

int32_t SystemAbilityStateScheduler::SubscribeSystemProcess(const sptr<ISystemProcessStatusChange>& listener)
{
    std::unique_lock<std::shared_mutex> writeLock(listenerSetLock_);
    auto iter = std::find_if(processListeners_.begin(), processListeners_.end(),
        [listener](sptr<ISystemProcessStatusChange>& item) {
        return item->AsObject() == listener->AsObject();
    });
    if (iter == processListeners_.end()) {
        if (processListenerDeath_ != nullptr) {
            bool ret = listener->AsObject()->AddDeathRecipient(processListenerDeath_);
            HILOGI("SubscribeSystemProcess AddDeathRecipient %{public}s", ret ? "succeed" : "failed");
        }
        processListeners_.emplace_back(listener);
    } else {
        HILOGI("SubscribeSystemProcess listener already exists");
    }
    return ERR_OK;
}

int32_t SystemAbilityStateScheduler::UnSubscribeSystemProcess(const sptr<ISystemProcessStatusChange>& listener)
{
    std::unique_lock<std::shared_mutex> writeLock(listenerSetLock_);
    auto iter = std::find_if(processListeners_.begin(), processListeners_.end(),
        [listener](sptr<ISystemProcessStatusChange>& item) {
        return item->AsObject() == listener->AsObject();
    });
    if (iter != processListeners_.end()) {
        if (processListenerDeath_ != nullptr) {
            listener->AsObject()->RemoveDeathRecipient(processListenerDeath_);
        }
        processListeners_.erase(iter);
        HILOGI("UnSubscribeSystemProcess listener remove success");
    } else {
        HILOGI("UnSubscribeSystemProcess listener not exists");
    }
    return ERR_OK;
}

int32_t SystemAbilityStateScheduler::ProcessDelayUnloadEvent(int32_t systemAbilityId)
{
    std::shared_ptr<SystemAbilityContext> abilityContext;
    if (!GetSystemAbilityContext(systemAbilityId, abilityContext)) {
        return ERR_INVALID_VALUE;
    }
    std::lock_guard<std::recursive_mutex> autoLock(abilityContext->ownProcessContext->processLock);
    if (abilityContext->state != SystemAbilityState::LOADED) {
        HILOGW("[SA Scheduler][SA: %{public}d] cannot process delay unload event", systemAbilityId);
        return ERR_OK;
    }
    HILOGI("[SA Scheduler][SA: %{public}d] process delay unload event", systemAbilityId);
    int32_t delayTime = 0;
    nlohmann::json idleReason;
    idleReason[KEY_EVENT_ID] = abilityContext->unloadRequest.unloadEvent.eventId;
    idleReason[KEY_NAME] = abilityContext->unloadRequest.unloadEvent.name;
    idleReason[KEY_VALUE] = abilityContext->unloadRequest.unloadEvent.value;
    idleReason[KEY_EXTRA_DATA_ID] = abilityContext->unloadRequest.unloadEvent.extraDataId;
    bool result = SystemAbilityManager::GetInstance()->IdleSystemAbility(abilityContext->systemAbilityId,
        abilityContext->ownProcessContext->processName, idleReason, delayTime);
    if (!result) {
        HILOGE("[SA Scheduler][SA: %{public}d] idle system ability failed", systemAbilityId);
        return ERR_INVALID_VALUE;
    }
    if (delayTime < 0) {
        HILOGI("[SA Scheduler][SA: %{public}d] reject unload", systemAbilityId);
        return ERR_OK;
    } else if (delayTime == 0) {
        HILOGI("[SA Scheduler][SA: %{public}d] agree unload", systemAbilityId);
        return stateMachine_->AbilityStateTransitionLocked(abilityContext, SystemAbilityState::UNLOADABLE);
    } else {
        HILOGI("[SA Scheduler][SA: %{public}d] choose delay unload", systemAbilityId);
        return SendDelayUnloadEventLocked(abilityContext->systemAbilityId, fmin(delayTime, MAX_DELAY_TIME));
    }
}

void SystemAbilityStateScheduler::UnloadEventHandler::ProcessEvent(const OHOS::AppExecFwk::InnerEvent::Pointer& event)
{
    if (event == nullptr) {
        HILOGE("[SA Scheduler] ProcessEvent event is nullptr!");
        return;
    }
    auto eventId = event->GetInnerEventId();
    int32_t systemAbilityId = static_cast<int32_t>(eventId);
    auto stateScheduler = stateScheduler_.lock();
    int32_t result = ERR_OK;
    if (stateScheduler != nullptr) {
        result = stateScheduler->ProcessDelayUnloadEvent(systemAbilityId);
    }
    if (result != ERR_OK) {
        HILOGE("[SA Scheduler][SA: %{public}d] process delay unload event failed", systemAbilityId);
    }
}
}  // namespace OHOS