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
16 #include "client_lifecycle.h"
17
18 #include "if_system_ability_manager.h"
19 #include "iservice_registry.h"
20 #include "power_log.h"
21 #include "power_mgr_errors.h"
22 #include "system_ability_definition.h"
23
24 namespace OHOS {
25 namespace PowerMgr {
26 sptr<IPowerMgr> ClientLifeCycle::proxy_;
27 sptr<IRemoteObject::DeathRecipient> ClientLifeCycle::deathRecipient_;
28 std::mutex ClientLifeCycle::mutex_;
29
GetProxy()30 sptr<IPowerMgr> ClientLifeCycle::GetProxy()
31 {
32 std::lock_guard<std::mutex> lock(mutex_);
33 if (proxy_ == nullptr) {
34 auto ret = Connect();
35 if (ret != ERR_OK) {
36 POWER_HILOGE(COMP_FWK, "Failed to connect PowerMgrService");
37 }
38 }
39 return proxy_;
40 }
41
Connect()42 ErrCode ClientLifeCycle::Connect()
43 {
44 if (proxy_ != nullptr) {
45 return ERR_OK;
46 }
47
48 sptr<ISystemAbilityManager> sam = SystemAbilityManagerClient::GetInstance().GetSystemAbilityManager();
49 if (sam == nullptr) {
50 POWER_HILOGE(COMP_FWK, "Failed to obtain SystemAbilityMgr");
51 return E_GET_SYSTEM_ABILITY_MANAGER_FAILED;
52 }
53 sptr<IRemoteObject> remoteObject_ = sam->CheckSystemAbility(POWER_MANAGER_SERVICE_ID);
54 if (remoteObject_ == nullptr) {
55 POWER_HILOGE(COMP_FWK, "Check SystemAbility failed");
56 return E_GET_POWER_SERVICE_FAILED;
57 }
58
59 deathRecipient_ = sptr<IRemoteObject::DeathRecipient>(new LifeCycleRecipient());
60 if (deathRecipient_ == nullptr) {
61 POWER_HILOGE(COMP_FWK, "Failed to create LifeCycleRecipient");
62 return ERR_NO_MEMORY;
63 }
64 if ((remoteObject_->IsProxyObject()) && (!remoteObject_->AddDeathRecipient(deathRecipient_))) {
65 POWER_HILOGE(COMP_FWK, "Add death recipient to PowerMgr service failed");
66 return E_ADD_DEATH_RECIPIENT_FAILED;
67 }
68
69 proxy_ = iface_cast<IPowerMgr>(remoteObject_);
70 POWER_HILOGI(COMP_FWK, "Connecting PowerMgrService success");
71 return ERR_OK;
72 }
73
Reset(const wptr<IRemoteObject> & remote)74 void ClientLifeCycle::Reset(const wptr<IRemoteObject>& remote)
75 {
76 if (proxy_ == nullptr) {
77 return;
78 }
79
80 auto service = proxy_->AsObject();
81 if ((service != nullptr) && (service == remote.promote())) {
82 service->RemoveDeathRecipient(deathRecipient_);
83 proxy_ = nullptr;
84 }
85 }
86
OnRemoteDied(const wptr<IRemoteObject> & remote)87 void ClientLifeCycle::LifeCycleRecipient::OnRemoteDied(const wptr<IRemoteObject>& remote)
88 {
89 if (remote == nullptr) {
90 POWER_HILOGW(COMP_FWK, "OnRemoteDied failed, remote is nullptr");
91 return;
92 }
93 std::lock_guard<std::mutex> lock(mutex_);
94 ClientLifeCycle::Reset(remote);
95 POWER_HILOGW(COMP_FWK, "Recv death notice");
96 }
97 } // namespace PowerMgr
98 } // namespace OHOS
99