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 "startup_manager.h"
17
18 #include "if_system_ability_manager.h"
19 #include "iservice_registry.h"
20 #include "system_ability_definition.h"
21
22 #include "access_factory.h"
23 #include "constant.h"
24 #include "firmware_preferences_utils.h"
25 #include "schedule_config.h"
26 #include "string_utils.h"
27 #include "time_utils.h"
28 #include "update_log.h"
29
30 namespace OHOS {
31 namespace UpdateEngine {
StartupManager()32 StartupManager::StartupManager()
33 {
34 ENGINE_LOGD("StartupManager constructor");
35 scheduleManager_ = std::make_shared<ScheduleManager>(
36 DelayedSingleton<AccessManager>::GetInstance(), DelayedSingleton<StartupSchedule>::GetInstance());
37 lastIdleCheckTime_ = TimeUtils::GetTimestamp();
38 ENGINE_LOGI("init lastIdleCheckTime is %{public}s", TimeUtils::GetPrintTimeStr(lastIdleCheckTime_).c_str());
39 }
40
~StartupManager()41 StartupManager::~StartupManager()
42 {
43 ENGINE_LOGD("StartupManager deConstructor");
44 }
45
Start()46 void StartupManager::Start()
47 {
48 ENGINE_LOGI("Start");
49 Init();
50 IdleLoop();
51 }
52
Init() const53 void StartupManager::Init() const
54 {
55 ENGINE_LOGI("Init");
56 ScheduleConfig::InitConfig();
57 StartupReason startupReason = GetStartupReason();
58 ENGINE_LOGI("startupReason is %{public}d", CAST_INT(startupReason));
59
60 if (startupReason == StartupReason::PROCESS_ENV_RESET) {
61 FileUtils::DestroyBaseDirectory(BASE_DIR_INFOS);
62 }
63 FileUtils::InitAndCreateBaseDirs(BASE_DIR_INFOS);
64
65 for (auto &type : Startup::ACCESS_TYPES) {
66 AccessFactory::GetInstance(type)->Init(startupReason);
67 }
68 }
69
IdleLoop()70 void StartupManager::IdleLoop()
71 {
72 ENGINE_LOGI("IdleLoop");
73 DelayedSingleton<StartupSchedule>::GetInstance()->RegisterLooper([=]() {
74 if (!IdleCheck()) {
75 ENGINE_LOGD("IdleLoop not idle");
76 return;
77 }
78
79 if (!PreExit()) {
80 ENGINE_LOGI("IdleLoop pre exit fail");
81 return;
82 }
83
84 SAExit();
85 });
86 }
87
IdleCheck()88 bool StartupManager::IdleCheck()
89 {
90 if (scheduleManager_ == nullptr) {
91 ENGINE_LOGE("IdleCheck scheduleManager is null, return idle");
92 return true;
93 }
94
95 if (abs(TimeUtils::GetTimestamp() - lastIdleCheckTime_) <
96 static_cast<int64_t>(ScheduleConfig::GetIdleCheckInterval())) {
97 ENGINE_LOGD("IdleCheck check time not arrive: lastIdleCheckTime is %{public}s",
98 TimeUtils::GetPrintTimeStr(lastIdleCheckTime_).c_str());
99 return false;
100 }
101
102 lastIdleCheckTime_ = TimeUtils::GetTimestamp();
103 ENGINE_LOGI(
104 "IdleCheck update lastIdleCheckTime: %{public}s", TimeUtils::GetPrintTimeStr(lastIdleCheckTime_).c_str());
105
106 bool isIdle = scheduleManager_->IdleCheck();
107 ENGINE_LOGI("IdleCheck idleState is %{public}s", StringUtils::GetBoolStr(isIdle).c_str());
108 return isIdle;
109 }
110
PreExit()111 bool StartupManager::PreExit()
112 {
113 if (scheduleManager_ == nullptr) {
114 ENGINE_LOGE("IdleCheck PreExit is null, return true");
115 return true;
116 }
117 return scheduleManager_->Exit();
118 }
119
SAExit() const120 void StartupManager::SAExit() const
121 {
122 ENGINE_LOGI("SAExit");
123 sptr<ISystemAbilityManager> sm = SystemAbilityManagerClient::GetInstance().GetSystemAbilityManager();
124 if (sm == nullptr) {
125 ENGINE_LOGE("GetSystemAbilityManager samgr object null!");
126 return;
127 }
128 int32_t res = sm->UnloadSystemAbility(UPDATE_DISTRIBUTED_SERVICE_ID);
129 ENGINE_LOGI("UnloadSystemAbility res is %{public}d", res);
130 }
131
GetStartupReason() const132 StartupReason StartupManager::GetStartupReason() const
133 {
134 if (TimeUtils::IsInRebootDuration()) {
135 return StartupReason::DEVICE_REBOOT;
136 }
137 // 由于SaMgr暂不支持记录启动原因,因此临时从SP文件中读取启动原因
138 auto preferenceUtil = DelayedSingleton<FirmwarePreferencesUtil>::GetInstance();
139 auto reasonFromSp = static_cast<StartupReason>(
140 preferenceUtil->ObtainInt(Constant::PROCESS_RESTART_REASON, CAST_INT(StartupReason::UNKNOWN)));
141 if (reasonFromSp == StartupReason::UNKNOWN) {
142 return StartupReason::SCHEDULE_TASK;
143 }
144 // 读取后清除SP文件中记录的标记
145 preferenceUtil->Remove(Constant::PROCESS_RESTART_REASON);
146 return reasonFromSp;
147 }
148 } // namespace UpdateEngine
149 } // namespace OHOS