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 "dhcp_sa_manager.h"
17 #include "dhcp_errcode.h"
18 #include "iremote_broker.h"
19 #include "iremote_object.h"
20 #include "iservice_registry.h"
21 #include "dhcp_logger.h"
22
23 DEFINE_DHCPLOG_DHCP_LABEL("DhcpSaLoadManager");
24 namespace OHOS {
25 namespace DHCP {
26 static constexpr int32_t DHCP_LOADSA_TIMEOUT_MS = 1000;
GetInstance()27 DhcpSaLoadManager& DhcpSaLoadManager::GetInstance()
28 {
29 static auto instance = new DhcpSaLoadManager();
30 return *instance;
31 }
32
LoadWifiSa(int32_t systemAbilityId)33 ErrCode DhcpSaLoadManager::LoadWifiSa(int32_t systemAbilityId) __attribute__((no_sanitize("cfi")))
34 {
35 DHCP_LOGD("%{public}s enter, systemAbilityId = [%{public}d] loading", __func__, systemAbilityId);
36 sptr<ISystemAbilityManager> samgr =
37 SystemAbilityManagerClient::GetInstance().GetSystemAbilityManager();
38 if (samgr == nullptr) {
39 DHCP_LOGE("%{public}s: get system ability manager failed!", __func__);
40 return DHCP_E_FAILED;
41 }
42 auto object = samgr->CheckSystemAbility(systemAbilityId);
43 if (object != nullptr) {
44 return DHCP_E_SUCCESS;
45 }
46 InitLoadState();
47 sptr<DhcpSaLoadCallback> loadCallback = new (std::nothrow)DhcpSaLoadCallback();
48 if (loadCallback == nullptr) {
49 DHCP_LOGE("%{public}s: dhcp sa load callback failed!", __func__);
50 return DHCP_E_FAILED;
51 }
52 int32_t ret = samgr->LoadSystemAbility(systemAbilityId, loadCallback);
53 if (ret != DHCP_E_SUCCESS) {
54 DHCP_LOGE("%{public}s: Failed to load system ability, SA Id = [%{public}d], ret = [%{public}d].",
55 __func__, systemAbilityId, ret);
56 return DHCP_E_FAILED;
57 }
58 DHCP_LOGI("DhcpSaLoadManager LoadWifiSa");
59 return WaitLoadStateChange(systemAbilityId);
60 }
61
InitLoadState()62 void DhcpSaLoadManager::InitLoadState()
63 {
64 std::unique_lock<std::mutex> lock(locatorMutex_);
65 state_ = false;
66 }
67
WaitLoadStateChange(int32_t systemAbilityId)68 ErrCode DhcpSaLoadManager::WaitLoadStateChange(int32_t systemAbilityId)
69 {
70 std::unique_lock<std::mutex> lock(locatorMutex_);
71 auto wait = locatorCon_.wait_for(lock, std::chrono::milliseconds(DHCP_LOADSA_TIMEOUT_MS), [this] {
72 DHCP_LOGI("DhcpSaLoadManager wait_for");
73 return state_ == true;
74 });
75 if (!wait) {
76 DHCP_LOGE("locator sa [%{public}d] start time out.", systemAbilityId);
77 return DHCP_E_FAILED;
78 }
79 DHCP_LOGI("DhcpSaLoadManager WaitLoadStateChange ok");
80 return DHCP_E_SUCCESS;
81 }
82
UnloadWifiSa(int32_t systemAbilityId)83 ErrCode DhcpSaLoadManager::UnloadWifiSa(int32_t systemAbilityId)
84 {
85 DHCP_LOGI("%{public}s enter, systemAbilityId = [%{public}d] unloading", __func__, systemAbilityId);
86 sptr<ISystemAbilityManager> samgr = SystemAbilityManagerClient::GetInstance().GetSystemAbilityManager();
87 if (samgr == nullptr) {
88 DHCP_LOGE("%{public}s: get system ability manager failed!", __func__);
89 return DHCP_E_FAILED;
90 }
91 int32_t ret = samgr->UnloadSystemAbility(systemAbilityId);
92 if (ret != DHCP_E_SUCCESS) {
93 DHCP_LOGE("%{public}s: Failed to unload system ability, SA Id = [%{public}d], ret = [%{public}d].",
94 __func__, systemAbilityId, ret);
95 return DHCP_E_FAILED;
96 }
97 DHCP_LOGI("DhcpSaLoadManager UnloadWifiSa ok");
98 return DHCP_E_SUCCESS;
99 }
100
LoadSystemAbilitySuccess()101 void DhcpSaLoadManager::LoadSystemAbilitySuccess()
102 {
103 std::unique_lock<std::mutex> lock(locatorMutex_);
104 state_ = true;
105 DHCP_LOGI("DhcpSaLoadManager LoadSystemAbilitySuccess notify_one");
106 locatorCon_.notify_one();
107 }
108
LoadSystemAbilityFail()109 void DhcpSaLoadManager::LoadSystemAbilityFail()
110 {
111 std::unique_lock<std::mutex> lock(locatorMutex_);
112 state_ = false;
113 DHCP_LOGI("DhcpSaLoadManager LoadSystemAbilityFail notify_one");
114 locatorCon_.notify_one();
115 }
116
OnLoadSystemAbilitySuccess(int32_t systemAbilityId,const sptr<IRemoteObject> & remoteObject)117 void DhcpSaLoadCallback::OnLoadSystemAbilitySuccess(int32_t systemAbilityId, const sptr<IRemoteObject> &remoteObject)
118 {
119 DHCP_LOGI("DhcpSaLoadManager Load SA success, systemAbilityId = [%{public}d]", systemAbilityId);
120 DhcpSaLoadManager::GetInstance().LoadSystemAbilitySuccess();
121 }
122
OnLoadSystemAbilityFail(int32_t systemAbilityId)123 void DhcpSaLoadCallback::OnLoadSystemAbilityFail(int32_t systemAbilityId)
124 {
125 DHCP_LOGI("DhcpSaLoadManager Load SA failed, systemAbilityId = [%{public}d]", systemAbilityId);
126 DhcpSaLoadManager::GetInstance().LoadSystemAbilityFail();
127 }
128 } // namespace DHCP
129 } // namespace OHOS
130