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