xref: /foundation/resourceschedule/work_scheduler/services/zidl/src/work_sched_service_stub.cpp

/*
 * 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 "work_sched_service_stub.h"

#include <message_parcel.h>
#include <string_ex.h>

#include "iwork_sched_service_ipc_interface_code.h"
#include "work_sched_common.h"
#include "work_sched_errors.h"

namespace OHOS {
namespace WorkScheduler {
int32_t WorkSchedServiceStub::OnRemoteRequest(uint32_t code, MessageParcel &data, MessageParcel &reply,
    MessageOption &option)
{
    WS_HILOGD("cmd = %{public}u, flags = %{public}d", code, option.GetFlags());
    std::u16string descriptor = WorkSchedServiceStub::GetDescriptor();
    std::u16string remoteDescriptor = data.ReadInterfaceToken();
    if (descriptor != remoteDescriptor) {
        WS_HILOGE("failed, descriptor is not matched!");
        return E_PARCEL_OPERATION_FAILED;
    }
    switch (code) {
        case static_cast<int32_t>(IWorkSchedServiceInterfaceCode::START_WORK): {
            int32_t ret = StartWorkStub(data);
            reply.WriteInt32(ret);
            return ret;
        }
        case static_cast<int32_t>(IWorkSchedServiceInterfaceCode::STOP_WORK): {
            int32_t ret = StopWorkStub(data);
            reply.WriteInt32(ret);
            return ret;
        }
        case static_cast<int32_t>(IWorkSchedServiceInterfaceCode::STOP_AND_CANCEL_WORK): {
            int32_t ret = StopAndCancelWorkStub(data);
            reply.WriteInt32(ret);
            return ret;
        }
        case static_cast<int32_t>(IWorkSchedServiceInterfaceCode::STOP_AND_CLEAR_WORKS): {
            int32_t ret = StopAndClearWorksStub(data);
            reply.WriteInt32(ret);
            return ret;
        }
        case static_cast<int32_t>(IWorkSchedServiceInterfaceCode::IS_LAST_WORK_TIMEOUT): {
            bool isLastWorkTimeout;
            int32_t ret = IsLastWorkTimeoutStub(data, isLastWorkTimeout);
            reply.WriteInt32(ret);
            reply.WriteBool(isLastWorkTimeout);
            return ret;
        }
        case static_cast<int32_t>(IWorkSchedServiceInterfaceCode::OBTAIN_ALL_WORKS): {
            std::list<std::shared_ptr<WorkInfo>> workInfos;
            int32_t ret = ObtainAllWorksStub(data, workInfos);
            uint32_t worksize = workInfos.size();
            reply.WriteInt32(ret);
            reply.WriteInt32(worksize);
            for (auto workInfo : workInfos) {
                reply.WriteParcelable(&*workInfo);
            }
            return ERR_OK;
        }
        case static_cast<int32_t>(IWorkSchedServiceInterfaceCode::GET_WORK_STATUS): {
            WS_HILOGI("call GetWorkStatus");
            std::shared_ptr<WorkInfo> workInfo;
            int32_t ret = GetWorkStatusStub(data, workInfo);
            reply.WriteInt32(ret);
            reply.WriteParcelable(&*workInfo);
            return ERR_OK;
        }
        case static_cast<int32_t>(IWorkSchedServiceInterfaceCode::GET_ALL_RUNNING_WORKS): {
            std::list<std::shared_ptr<WorkInfo>> workInfos;
            int32_t ret = GetAllRunningWorksStub(workInfos);
            uint32_t worksize = workInfos.size();
            reply.WriteInt32(ret);
            reply.WriteInt32(worksize);
            for (auto workInfo : workInfos) {
                reply.WriteParcelable(&*workInfo);
            }
            return ERR_OK;
        }
        default: {
            WS_HILOGD("OnRemoteRequest switch default, code: %{public}u", code);
            return IPCObjectStub::OnRemoteRequest(code, data, reply, option);
        }
    }
    return ERR_OK;
}

int32_t WorkSchedServiceStub::StartWorkStub(MessageParcel& data)
{
    sptr<WorkInfo> workInfo = data.ReadStrongParcelable<WorkInfo>();
    if (workInfo == nullptr) {
        WS_HILOGD("workInfo is nullptr");
        return E_PARCEL_OPERATION_FAILED;
    }
    return StartWork(*workInfo);
}

int32_t WorkSchedServiceStub::StopWorkStub(MessageParcel& data)
{
    sptr<WorkInfo> workInfo = data.ReadStrongParcelable<WorkInfo>();
    if (workInfo == nullptr) {
        WS_HILOGD("workInfo is nullptr");
        return E_PARCEL_OPERATION_FAILED;
    }
    return StopWork(*workInfo);
}

int32_t WorkSchedServiceStub::StopAndCancelWorkStub(MessageParcel& data)
{
    sptr<WorkInfo> workInfo = data.ReadStrongParcelable<WorkInfo>();
    if (workInfo == nullptr) {
        WS_HILOGD("workInfo is nullptr");
        return E_PARCEL_OPERATION_FAILED;
    }
    return StopAndCancelWork(*workInfo);
}

int32_t WorkSchedServiceStub::StopAndClearWorksStub(MessageParcel& data)
{
    return StopAndClearWorks();
}

int32_t WorkSchedServiceStub::IsLastWorkTimeoutStub(MessageParcel& data, bool &result)
{
    int32_t workId = data.ReadInt32();
    return IsLastWorkTimeout(workId, result);
}

int32_t WorkSchedServiceStub::ObtainAllWorksStub(MessageParcel& data, std::list<std::shared_ptr<WorkInfo>>& workInfos)
{
    int32_t pid = 0;
    int32_t uid = 0;
    READ_PARCEL_WITHOUT_RET(data, Int32, uid);
    READ_PARCEL_WITHOUT_RET(data, Int32, pid);
    return ObtainAllWorks(uid, pid, workInfos);
}

int32_t WorkSchedServiceStub::GetWorkStatusStub(MessageParcel& data, std::shared_ptr<WorkInfo>& workInfo)
{
    int32_t uid;
    int32_t workId;
    READ_PARCEL_WITHOUT_RET(data, Int32, uid);
    READ_PARCEL_WITHOUT_RET(data, Int32, workId);
    return GetWorkStatus(uid, workId, workInfo);
}

int32_t WorkSchedServiceStub::GetAllRunningWorksStub(std::list<std::shared_ptr<WorkInfo>>& workInfos)
{
    return GetAllRunningWorks(workInfos);
}
} // namespace WorkScheduler
} // namespace OHOS