1/*
2 * Copyright (c) 2023 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 *     http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15#include "dump_manager_cpu_client.h"
16#include <iservice_registry.h>
17#include <string_ex.h>
18#include <unistd.h>
19#include "hilog_wrapper.h"
20#include "dump_errors.h"
21#include "inner/dump_service_id.h"
22#include "dump_on_demand_load.h"
23#include "dump_broker_cpu_proxy.h"
24namespace OHOS {
25namespace HiviewDFX {
26DumpManagerCpuClient::DumpManagerCpuClient()
27{
28}
29
30DumpManagerCpuClient::~DumpManagerCpuClient()
31{
32    Reset();
33}
34
35int32_t DumpManagerCpuClient::Request(DumpCpuData &dumpCpuData)
36{
37    if (Connect() != ERR_OK) {
38        DUMPER_HILOGE(MODULE_CPU_CLIENT, "debug|cpu connect error");
39        return DumpStatus::DUMP_FAIL;
40    }
41    int32_t ret = proxy_->Request(dumpCpuData);
42    return ret;
43}
44
45int32_t DumpManagerCpuClient::GetCpuUsageByPid(int32_t pid, double &cpuUsage)
46{
47    if (Connect() != ERR_OK) {
48        DUMPER_HILOGE(MODULE_CPU_CLIENT, "cpu connect error");
49        return DumpStatus::DUMP_FAIL;
50    }
51    int32_t ret = proxy_->GetCpuUsageByPid(pid, cpuUsage);
52    return ret;
53}
54
55ErrCode DumpManagerCpuClient::Connect()
56{
57    std::lock_guard<std::mutex> lock(mutex_);
58    if (proxy_ != nullptr) {
59        return ERR_OK;
60    }
61    sptr<ISystemAbilityManager> sam = SystemAbilityManagerClient::GetInstance().GetSystemAbilityManager();
62    if (sam == nullptr) {
63        DUMPER_HILOGE(MODULE_CPU_CLIENT, "sam is null");
64        return ERROR_GET_SYSTEM_ABILITY_MANAGER;
65    }
66    sptr<IRemoteObject> remoteObject = sam->CheckSystemAbility(DFX_SYS_HIDUMPER_CPU_ABILITY_ID);
67    if (remoteObject == nullptr) {
68        DUMPER_HILOGE(MODULE_CPU_CLIENT, "cpu remoteobject is null");
69        return ERROR_GET_DUMPER_SERVICE;
70    }
71    deathRecipient_ = sptr<IRemoteObject::DeathRecipient>(new DumpManagerCpuDeathRecipient());
72    if (deathRecipient_ == nullptr) {
73        DUMPER_HILOGE(MODULE_CPU_CLIENT, "cpu deathRecipient_ is null");
74        return ERR_NO_MEMORY;
75    }
76    if ((remoteObject->IsProxyObject()) && (!remoteObject->AddDeathRecipient(deathRecipient_))) {
77        DUMPER_HILOGE(MODULE_CPU_CLIENT, "cpu IsProxyObject is null");
78        return ERROR_ADD_DEATH_RECIPIENT;
79    }
80    proxy_ = iface_cast<IDumpCpuBroker>(remoteObject);
81    return ERR_OK;
82}
83
84void DumpManagerCpuClient::Reset()
85{
86    std::lock_guard<std::mutex> lock(mutex_);
87    if (proxy_ == nullptr) {
88        return;
89    }
90    auto serviceRemote = proxy_->AsObject();
91    if (serviceRemote != nullptr) {
92        serviceRemote->RemoveDeathRecipient(deathRecipient_);
93        proxy_ = nullptr;
94        DUMPER_HILOGD(MODULE_CPU_CLIENT, "debug|disconnected");
95    }
96}
97
98void DumpManagerCpuClient::ResetProxy(const wptr<IRemoteObject>& remote)
99{
100    std::lock_guard<std::mutex> lock(mutex_);
101    if (proxy_ == nullptr) {
102        return;
103    }
104    auto serviceRemote = proxy_->AsObject();
105    if ((serviceRemote != nullptr) && (serviceRemote == remote.promote())) {
106        serviceRemote->RemoveDeathRecipient(deathRecipient_);
107        proxy_ = nullptr;
108        DUMPER_HILOGD(MODULE_CPU_CLIENT, "debug|disconnected");
109    }
110}
111
112void DumpManagerCpuClient::DumpManagerCpuDeathRecipient::OnRemoteDied(const wptr<IRemoteObject>& remote)
113{
114    if (remote == nullptr) {
115        return;
116    }
117    DumpManagerCpuClient::GetInstance().ResetProxy(remote);
118}
119} // namespace HiviewDFX
120} // namespace OHOS
121