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 "firmware_check_data_processor.h"
17
18 #include "config_parse.h"
19 #include "constant.h"
20 #include "dupdate_inet_observer.h"
21 #include "dupdate_net_observer.h"
22 #include "firmware_changelog_utils.h"
23 #include "firmware_combine_version_utils.h"
24 #include "firmware_common.h"
25 #include "firmware_component_operator.h"
26 #include "firmware_constant.h"
27 #include "firmware_callback_utils.h"
28 #include "firmware_log.h"
29 #include "firmware_task_operator.h"
30 #include "device_adapter.h"
31 #include "firmware_update_helper.h"
32 #include "string_utils.h"
33 #include "time_utils.h"
34 #include "update_helper.h"
35
36 namespace OHOS {
37 namespace UpdateEngine {
FirmwareCheckDataProcessor()38 FirmwareCheckDataProcessor::FirmwareCheckDataProcessor()
39 {
40 FIRMWARE_LOGD("FirmwareCheckDataProcessor::FirmwareCheckDataProcessor");
41 }
42
~FirmwareCheckDataProcessor()43 FirmwareCheckDataProcessor::~FirmwareCheckDataProcessor()
44 {
45 FIRMWARE_LOGD("FirmwareCheckDataProcessor::~FirmwareCheckDataProcessor");
46 }
47
SetCheckResult(CheckStatus status,const Duration & duration,const std::vector<FirmwareComponent> & componentList)48 void FirmwareCheckDataProcessor::SetCheckResult(
49 CheckStatus status, const Duration &duration, const std::vector<FirmwareComponent> &componentList)
50 {
51 FIRMWARE_LOGI("FirmwareCheckDataProcessor::SetCheckResult status %{public}d duration %{public}d",
52 static_cast<uint32_t>(status), duration.duration);
53 status_ = status;
54 duration_ = duration;
55 componentList_ = componentList;
56 HandleCheckResult();
57 }
58
IsCheckFailed()59 bool FirmwareCheckDataProcessor::IsCheckFailed()
60 {
61 return status_ == CheckStatus::CHECK_FAIL ||
62 status_ == CheckStatus::FILE_LIST_REQUEST_FAIL ||
63 status_ == CheckStatus::CHANGELOG_REQUEST_FAIL;
64 }
65
HasNewVersion()66 bool FirmwareCheckDataProcessor::HasNewVersion()
67 {
68 return !componentList_.empty();
69 }
70
IsSameWithDb()71 bool FirmwareCheckDataProcessor::IsSameWithDb()
72 {
73 return isSameWithDb_;
74 }
75
HandleCheckResult()76 void FirmwareCheckDataProcessor::HandleCheckResult()
77 {
78 FIRMWARE_LOGI("FirmwareCheckDataProcessor HandleCheckResult status %{public}d size: %{public}d",
79 CAST_INT(status_), CAST_INT(componentList_.size()));
80 if (IsCheckFailed()) {
81 // 搜包失败或者获取fileList失败
82 FIRMWARE_LOGI("FirmwareCheckDataProcessor HandleCheckResult fail");
83 return;
84 }
85
86 RefreshPollingCycle();
87 if (HasUndoneTask()) {
88 // 下载中、安装中，不处理
89 FIRMWARE_LOGI("FirmwareCheckDataProcessor HasUndoneTask");
90 HandleUndoneEvent();
91 return;
92 }
93
94 if (HasNewVersion()) {
95 HandleNewVersion();
96 } else {
97 HandleNoNewVersion();
98 }
99 }
100
HasUndoneTask()101 bool FirmwareCheckDataProcessor::HasUndoneTask()
102 {
103 FirmwareTask task;
104 FirmwareTaskOperator().QueryTask(task);
105 if (!task.isExistTask) {
106 FIRMWARE_LOGI("HasUndoneTask no task");
107 return false;
108 }
109
110 FIRMWARE_LOGI("HasUndoneTask status %{public}d", task.status);
111 UpgradeStatus taskStatus = task.status;
112 return taskStatus == UpgradeStatus::DOWNLOADING ||
113 taskStatus == UpgradeStatus::DOWNLOAD_PAUSE ||
114 taskStatus == UpgradeStatus::VERIFYING ||
115 taskStatus == UpgradeStatus::INSTALLING ||
116 taskStatus == UpgradeStatus::UPDATING;
117 }
118
IsUpdateOnStatus()119 bool FirmwareCheckDataProcessor::IsUpdateOnStatus()
120 {
121 FirmwareTask task;
122 FirmwareTaskOperator().QueryTask(task);
123 if (!task.isExistTask) {
124 FIRMWARE_LOGI("IsUpdateOnStatus no task");
125 return false;
126 }
127
128 FIRMWARE_LOGI("IsUpdateOnStatus status %{public}d", task.status);
129 return task.status == UpgradeStatus::UPDATING;
130 }
131
HandleNewVersion()132 void FirmwareCheckDataProcessor::HandleNewVersion()
133 {
134 FIRMWARE_LOGI("FirmwareCheckDataProcessor::HandleNewVersion");
135 isSameWithDb_ = IsVersionSameWithDb();
136 if (isSameWithDb_) {
137 UpdateFirmwareComponent();
138 FIRMWARE_LOGI("FirmwareCheckDataProcessor::HandleNewVersion version is same with db");
139 return;
140 }
141
142 // 删除数据库和本地下载文件
143 FirmwareUpdateHelper::ClearFirmwareInfo();
144
145 BuildComponentSPath();
146 FirmwareComponentOperator().Insert(componentList_);
147 FirmwareTask task;
148 task.taskId = FirmwareUpdateHelper::BuildTaskId(componentList_);
149 task.status = UpgradeStatus::CHECK_VERSION_SUCCESS;
150 task.combinationType = GetCombinationType();
151 FirmwareTaskOperator().Insert(task);
152 }
153
BuildComponentSPath()154 void FirmwareCheckDataProcessor::BuildComponentSPath()
155 {
156 for (auto &component : componentList_) {
157 component.spath = Firmware::PACKAGE_PATH + "/" + component.fileName;
158 }
159 }
160
IsVersionSameWithDb()161 bool FirmwareCheckDataProcessor::IsVersionSameWithDb()
162 {
163 // 查询当前数据库保存数据
164 std::vector<FirmwareComponent> dbComponentList;
165 FirmwareComponentOperator().QueryAll(dbComponentList);
166
167 FIRMWARE_LOGI("FirmwareCheckDataProcessor::IsVersionSameWithDb db size %{public}d server size %{public}d",
168 CAST_INT(dbComponentList.size()), CAST_INT(componentList_.size()));
169 if (dbComponentList.size() != componentList_.size()) {
170 return false;
171 }
172
173 for (const auto &serverResult : componentList_) {
174 std::string versionId = serverResult.versionId;
175 FirmwareComponent dbResult;
176 bool hasResult = FirmwareComponentOperator().QueryByVersionId(versionId, dbResult);
177 if (!hasResult || dbResult.size != serverResult.size || dbResult.verifyInfo != serverResult.verifyInfo) {
178 FIRMWARE_LOGI("FirmwareCheckDataProcessor::IsVersionSameWithDb versionId %{public}s not in db or data "
179 "is updated size %{public}s verifyInfo %{public}s",
180 versionId.c_str(), std::to_string(dbResult.size).c_str(), dbResult.verifyInfo.c_str());
181 return false;
182 }
183 }
184 FIRMWARE_LOGI("FirmwareCheckDataProcessor::IsVersionSameWithDb");
185 return true;
186 }
187
UpdateFirmwareComponent()188 void FirmwareCheckDataProcessor::UpdateFirmwareComponent()
189 {
190 FirmwareTask task;
191 FirmwareTaskOperator().QueryTask(task);
192 if (!task.isExistTask) {
193 FIRMWARE_LOGE("UpdateFirmwareComponent no task");
194 return;
195 }
196 if (task.status != UpgradeStatus::CHECK_VERSION_SUCCESS) {
197 FIRMWARE_LOGI("UpdateFirmwareComponent status not check version success");
198 return;
199 }
200 for (const auto &serverResult : componentList_) {
201 FirmwareComponentOperator().UpdateUrlByVersionId(serverResult.versionId, serverResult.url);
202 }
203 }
204
HandleNoNewVersion()205 void FirmwareCheckDataProcessor::HandleNoNewVersion()
206 {
207 FIRMWARE_LOGI("FirmwareCheckDataProcessor::HandleNoNewVersion");
208 FirmwareTask task;
209 FirmwareTaskOperator().QueryTask(task);
210 if (task.isExistTask) {
211 // 搜包从有到无，推送task_cancel
212 DelayedSingleton<FirmwareCallbackUtils>::GetInstance()->NotifyEvent(
213 task.taskId, EventId::EVENT_TASK_CANCEL, UpgradeStatus::INIT);
214 }
215 FirmwareUpdateHelper::ClearFirmwareInfo();
216 }
217
GetCombinationType()218 CombinationType FirmwareCheckDataProcessor::GetCombinationType()
219 {
220 uint32_t combinationType = CAST_UINT(CombinationType::INVALLID_TYPE);
221 for (const auto &component : componentList_) {
222 if (component.blVersionType == CAST_INT(BlType::HOTA_TYPE)) {
223 combinationType |= CAST_UINT(CombinationType::HOTA);
224 }
225 }
226
227 FIRMWARE_LOGI("CombinationType::%{public}u", combinationType);
228 return static_cast<CombinationType>(combinationType);
229 }
230
RefreshPollingCycle()231 void FirmwareCheckDataProcessor::RefreshPollingCycle()
232 {
233 FIRMWARE_LOGI("FirmwareCheckDataProcessor::RefreshPollingCycle");
234 }
235
BuildCheckResult(CheckResult & checkResult)236 void FirmwareCheckDataProcessor::BuildCheckResult(CheckResult &checkResult)
237 {
238 if (IsCheckFailed()) {
239 FIRMWARE_LOGE("BuildCheckResult check error");
240 return;
241 }
242
243 // 搜包结果返回数据库结果，避免下载中，搜到新包，返回新包结果，此时数据库中保存的是老包
244 std::vector<FirmwareComponent> dbComponentList;
245 FirmwareComponentOperator().QueryAll(dbComponentList);
246 checkResult.isExistNewVersion = (dbComponentList.size() != 0);
247 FIRMWARE_LOGI("BuildCheckResult status %{public}s", StringUtils::GetBoolStr(checkResult.isExistNewVersion).c_str());
248 if (checkResult.isExistNewVersion) {
249 BuildVersionDigest(checkResult.newVersionInfo, dbComponentList);
250 FirmwareUpdateHelper::BuildNewVersionInfo(dbComponentList, checkResult.newVersionInfo.versionComponents);
251 }
252 }
253
BuildVersionDigest(NewVersionInfo & newVersionInfo,const std::vector<FirmwareComponent> & componentList)254 void FirmwareCheckDataProcessor::BuildVersionDigest(
255 NewVersionInfo &newVersionInfo, const std::vector<FirmwareComponent> &componentList)
256 {
257 newVersionInfo.versionDigestInfo.versionDigest = FirmwareUpdateHelper::BuildTaskId(componentList);
258 }
259 } // namespace UpdateEngine
260 } // namespace OHOS