OpenHarmony-v3.2-Release/s

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

#include <chrono>
#include <new>

#include "errors.h"
#include "hitrace_meter.h"
#include "iremote_broker.h"
#include "if_system_ability_manager.h"
#include "iservice_registry.h"
#include "isystem_ability_load_callback.h"

#include "dscreen_constants.h"
#include "dscreen_errcode.h"
#include "dscreen_hitrace.h"
#include "dscreen_hisysevent.h"
#include "dscreen_log.h"
#include "dscreen_sink_load_callback.h"
#include "dscreen_util.h"

namespace OHOS {
namespace DistributedHardware {
IMPLEMENT_SINGLE_INSTANCE(DScreenSinkHandler);

DScreenSinkHandler::DScreenSinkHandler()
{
    DHLOGI("DScreenSinkHandler construct.");
    std::lock_guard<std::mutex> lock(proxyMutex_);
    if (sinkSvrRecipient_ == nullptr) {
        sinkSvrRecipient_ = new (std::nothrow) DScreenSinkSvrRecipient();
    }
}

DScreenSinkHandler::~DScreenSinkHandler()
{
    DHLOGI("~DScreenSinkHandler.");
}

int32_t DScreenSinkHandler::InitSink(const std::string &params)
{
    if (params.empty()) {
        DHLOGE("InitSink params is invalid.");
        return ERR_DH_SCREEN_INPUT_PARAM_INVALID;
    }
    DHLOGI("DScreenSinkHandler InitSink");
    std::unique_lock<std::mutex> lock(proxyMutex_);
    if (dScreenSinkProxy_ == nullptr) {
        sptr<ISystemAbilityManager> samgr = SystemAbilityManagerClient::GetInstance().GetSystemAbilityManager();
        if (samgr == nullptr) {
            DHLOGE("Failed to get system ability mgr.");
            ReportSaFail(DSCREEN_INIT_FAIL, ERR_DH_SCREEN_SA_GET_SAMGR_FAIL, DISTRIBUTED_HARDWARE_SCREEN_SINK_SA_ID,
                "dscreen sink get samgr failed.");
            return ERR_DH_SCREEN_SA_GET_SAMGR_FAIL;
        }
        sptr<DScreenSinkLoadCallback> loadCallback = new DScreenSinkLoadCallback(params);
        if (loadCallback == nullptr) {
            DHLOGE("loadCallback is nullptr.");
            ReportSaFail(DSCREEN_INIT_FAIL, ERR_DH_SCREEN_SA_GET_SAMGR_FAIL, DISTRIBUTED_HARDWARE_SCREEN_SINK_SA_ID,
                "loadCallback is nullptr.");
            return ERR_DH_SCREEN_SA_GET_SAMGR_FAIL;
        }
        StartTrace(DSCREEN_HITRACE_LABEL, DSCREEN_SINK_LOAD_SYSTEM_ABILITY_START);
        int32_t ret = samgr->LoadSystemAbility(DISTRIBUTED_HARDWARE_SCREEN_SINK_SA_ID, loadCallback);
        if (ret != ERR_OK) {
            DHLOGE("Failed to Load systemAbility, systemAbilityId:%d, ret code:%d",
                DISTRIBUTED_HARDWARE_SCREEN_SINK_SA_ID, ret);
            ReportSaFail(DSCREEN_INIT_FAIL, ret, DISTRIBUTED_HARDWARE_SCREEN_SINK_SA_ID,
                "dscreen sink LoadSystemAbility call failed.");
            return ERR_DH_SCREEN_SA_GET_SINKPROXY_FAIL;
        }
    }

    auto waitStatus = proxyConVar_.wait_for(lock, std::chrono::milliseconds(SCREEN_LOADSA_TIMEOUT_MS),
        [this]() { return dScreenSinkProxy_ != nullptr; });
    if (!waitStatus) {
        DHLOGE("screen load sa timeout");
        ReportSaFail(DSCREEN_INIT_FAIL, ERR_DH_SCREEN_SA_LOAD_TIMEOUT, DISTRIBUTED_HARDWARE_SCREEN_SINK_SA_ID,
            "dscreen sink sa load timeout.");
        return ERR_DH_SCREEN_SA_LOAD_TIMEOUT;
    }

    FinishTrace(DSCREEN_HITRACE_LABEL);
    return DH_SUCCESS;
}

void DScreenSinkHandler::FinishStartSA(const std::string &params, const sptr<IRemoteObject> &remoteObject)
{
    DHLOGI("DScreenSinkHandler FinishStartSA");
    std::lock_guard<std::mutex> lock(proxyMutex_);
    if (sinkSvrRecipient_ == nullptr) {
        DHLOGE("sinkSvrRecipient is nullptr.");
        ReportSaFail(DSCREEN_INIT_FAIL, ERR_DH_SCREEN_SA_SINKPROXY_NOT_INIT, DISTRIBUTED_HARDWARE_SCREEN_SINK_SA_ID,
            "sinkSvrRecipient is nullptr.");
        return;
    }
    remoteObject->AddDeathRecipient(sinkSvrRecipient_);
    dScreenSinkProxy_ = iface_cast<IDScreenSink>(remoteObject);
    if ((dScreenSinkProxy_== nullptr) || (dScreenSinkProxy_->AsObject() == nullptr)) {
        DHLOGE("Failed to get dscreen sink proxy.");
        ReportSaFail(DSCREEN_INIT_FAIL, ERR_DH_SCREEN_SA_SINKPROXY_NOT_INIT, DISTRIBUTED_HARDWARE_SCREEN_SINK_SA_ID,
            "dscreen sink get proxy failed.");
        return;
    }
    dScreenSinkProxy_->InitSink(params);
    proxyConVar_.notify_one();
    ReportSaEvent(DSCREEN_INIT, DISTRIBUTED_HARDWARE_SCREEN_SINK_SA_ID, "dscreen sink sa load success.");
}

int32_t DScreenSinkHandler::ReleaseSink()
{
    DHLOGI("DScreenSinkHandler ReleaseSink");
    std::lock_guard<std::mutex> lock(proxyMutex_);
    if (dScreenSinkProxy_ == nullptr) {
        DHLOGE("screen sink proxy not init.");
        ReportSaFail(DSCREEN_INIT_FAIL, ERR_DH_SCREEN_SA_SINKPROXY_NOT_INIT, DISTRIBUTED_HARDWARE_SCREEN_SINK_SA_ID,
            "dscreen sink proxy not init.");
        return ERR_DH_SCREEN_SA_SINKPROXY_NOT_INIT;
    }

    int32_t ret = dScreenSinkProxy_->ReleaseSink();
    dScreenSinkProxy_ = nullptr;
    return ret;
}

int32_t DScreenSinkHandler::SubscribeLocalHardware(const std::string &dhId, const std::string &param)
{
    DHLOGI("DScreenSinkHandler SubscribeLocalHardware dhId: %s", GetAnonyString(dhId).c_str());
    std::lock_guard<std::mutex> lock(proxyMutex_);
    if (dScreenSinkProxy_ == nullptr) {
        DHLOGE("screen sink proxy not init.");
        return ERR_DH_SCREEN_SA_SINKPROXY_NOT_INIT;
    }

    return dScreenSinkProxy_->SubscribeLocalHardware(dhId, param);
}

int32_t DScreenSinkHandler::UnsubscribeLocalHardware(const std::string &dhId)
{
    DHLOGI("DScreenSinkHandler UnsubscribeLocalHardware dhId: %s", GetAnonyString(dhId).c_str());
    std::lock_guard<std::mutex> lock(proxyMutex_);
    if (dScreenSinkProxy_ == nullptr) {
        DHLOGE("screen sink proxy not init.");
        return ERR_DH_SCREEN_SA_SINKPROXY_NOT_INIT;
    }

    return dScreenSinkProxy_->UnsubscribeLocalHardware(dhId);
}

void DScreenSinkHandler::DScreenSinkSvrRecipient::OnRemoteDied(const wptr<IRemoteObject> &remote)
{
    if (remote == nullptr) {
        DHLOGI("OnRemoteDied remote is nullptr.");
        return;
    }
    DHLOGI("DScreenSinkSvrRecipient OnRemoteDied.");
    DScreenSinkHandler::GetInstance().OnRemoteSinkSvrDied(remote);
}

void DScreenSinkHandler::OnRemoteSinkSvrDied(const wptr<IRemoteObject> &remote)
{
    DHLOGI("DScreenSinkHandler OnRemoteSinkSvrDied");
    std::lock_guard<std::mutex> lock(proxyMutex_);
    if (dScreenSinkProxy_ == nullptr) {
        DHLOGE("dScreenSinkProxy is nullptr.");
        return;
    }
    sptr<IRemoteObject> remoteObject = remote.promote();
    if (remoteObject == nullptr) {
        DHLOGE("OnRemoteDied remote promoted failed");
        return;
    }

    if (dScreenSinkProxy_->AsObject() != remoteObject) {
        DHLOGE("OnRemoteSinkSvrDied not found remote object.");
        return;
    }

    dScreenSinkProxy_->AsObject()->RemoveDeathRecipient(sinkSvrRecipient_);
    dScreenSinkProxy_ = nullptr;
}

IDistributedHardwareSink *GetSinkHardwareHandler()
{
    DHLOGI("GetSinkHardwareHandler");
    return &DScreenSinkHandler::GetInstance();
}
} // namespace DistributedHardware
} // namespace OHOS