blob: cad78387efb040be409f75e12bef4b455d22c903 [file] [log] [blame]
Alex Deymoaea4c1c2015-08-19 20:24:43 -07001//
2// Copyright (C) 2012 The Android Open Source Project
3//
4// Licensed under the Apache License, Version 2.0 (the "License");
5// you may not use this file except in compliance with the License.
6// You may obtain a copy of the License at
7//
8// http://www.apache.org/licenses/LICENSE-2.0
9//
10// Unless required by applicable law or agreed to in writing, software
11// distributed under the License is distributed on an "AS IS" BASIS,
12// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13// See the License for the specific language governing permissions and
14// limitations under the License.
15//
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -070016
Amin Hassaniec7bc112020-10-29 16:47:58 -070017#include "update_engine/cros/update_attempter.h"
Andrew de los Reyes63b96d72010-05-10 13:08:54 -070018
Ben Chan9abb7632014-08-07 00:10:53 -070019#include <stdint.h>
20
Alex Vakulenko59e253e2014-02-24 10:40:21 -080021#include <algorithm>
Andrewe045aef2020-01-08 16:29:22 -080022#include <map>
Alex Deymobc91a272014-05-20 16:45:33 -070023#include <memory>
24#include <string>
Andrewa8d7df32020-03-15 20:10:01 -070025#include <unordered_set>
Alex Vakulenkod2779df2014-06-16 13:19:00 -070026#include <utility>
Alex Deymo44666f92014-07-22 20:29:24 -070027#include <vector>
Darin Petkov9d65b7b2010-07-20 09:13:01 -070028
Gilad Arnoldec7f9162014-07-15 13:24:46 -070029#include <base/bind.h>
Denis Nikitin8f2fa742019-05-29 10:36:52 -070030#include <base/compiler_specific.h>
Ben Chan06c76a42014-09-05 08:21:06 -070031#include <base/files/file_util.h>
David Zeuthen3c55abd2013-10-14 12:48:03 -070032#include <base/logging.h>
Andrew de los Reyes45168102010-11-22 11:13:50 -080033#include <base/rand_util.h>
Andrewe045aef2020-01-08 16:29:22 -080034#include <base/strings/string_number_conversions.h>
Daniel Erat65f1da02014-06-27 22:05:38 -070035#include <base/strings/string_util.h>
Alex Vakulenko75039d72014-03-25 12:36:28 -070036#include <base/strings/stringprintf.h>
May Lippert60aa3ca2018-08-15 16:55:29 -070037#include <base/time/time.h>
Sen Jiang2703ef42017-03-16 13:36:21 -070038#include <brillo/data_encoding.h>
Sen Jiange67bb5b2016-06-20 15:53:56 -070039#include <brillo/errors/error_codes.h>
Alex Vakulenko3f39d5c2015-10-13 09:27:13 -070040#include <brillo/message_loops/message_loop.h>
Patrick Dubroy7fbbe8a2011-08-01 17:28:22 +020041#include <policy/device_policy.h>
Alex Deymobc91a272014-05-20 16:45:33 -070042#include <policy/libpolicy.h>
Alex Deymod6deb1d2015-08-28 15:54:37 -070043#include <update_engine/dbus-constants.h>
Darin Petkov9d65b7b2010-07-20 09:13:01 -070044
Alex Deymo14c0da82016-07-20 16:45:45 -070045#include "update_engine/certificate_checker.h"
Alex Deymo39910dc2015-11-09 17:04:30 -080046#include "update_engine/common/boot_control_interface.h"
Alex Deymo39910dc2015-11-09 17:04:30 -080047#include "update_engine/common/constants.h"
Xiaochu Liu8ba486f2018-11-06 11:14:10 -080048#include "update_engine/common/dlcservice_interface.h"
Amin Hassaniec7bc112020-10-29 16:47:58 -070049#include "update_engine/common/download_action.h"
Jae Hoon Kim5e8e30b2020-05-06 14:59:06 -070050#include "update_engine/common/excluder_interface.h"
Alex Deymo39910dc2015-11-09 17:04:30 -080051#include "update_engine/common/hardware_interface.h"
Amin Hassaniec7bc112020-10-29 16:47:58 -070052#include "update_engine/common/metrics_reporter_interface.h"
Alex Deymo39910dc2015-11-09 17:04:30 -080053#include "update_engine/common/platform_constants.h"
Andrewe045aef2020-01-08 16:29:22 -080054#include "update_engine/common/prefs.h"
Alex Deymo39910dc2015-11-09 17:04:30 -080055#include "update_engine/common/prefs_interface.h"
56#include "update_engine/common/subprocess.h"
Amin Hassaniec7bc112020-10-29 16:47:58 -070057#include "update_engine/common/system_state.h"
Alex Deymo39910dc2015-11-09 17:04:30 -080058#include "update_engine/common/utils.h"
Amin Hassaniec7bc112020-10-29 16:47:58 -070059#include "update_engine/cros/omaha_request_action.h"
60#include "update_engine/cros/omaha_request_params.h"
61#include "update_engine/cros/omaha_response_handler_action.h"
62#include "update_engine/cros/omaha_utils.h"
63#include "update_engine/cros/p2p_manager.h"
64#include "update_engine/cros/payload_state_interface.h"
65#include "update_engine/cros/power_manager_interface.h"
Alex Deymo14c0da82016-07-20 16:45:45 -070066#include "update_engine/libcurl_http_fetcher.h"
Alex Deymo39910dc2015-11-09 17:04:30 -080067#include "update_engine/payload_consumer/filesystem_verifier_action.h"
68#include "update_engine/payload_consumer/postinstall_runner_action.h"
Amin Hassani0882a512018-04-05 16:25:44 -070069#include "update_engine/update_boot_flags_action.h"
Gilad Arnoldec7f9162014-07-15 13:24:46 -070070#include "update_engine/update_manager/policy.h"
Marton Hunyadye58bddb2018-04-10 20:27:26 +020071#include "update_engine/update_manager/policy_utils.h"
Gilad Arnoldec7f9162014-07-15 13:24:46 -070072#include "update_engine/update_manager/update_manager.h"
Christopher Wileycc8ce0e2015-10-01 16:48:47 -070073#include "update_engine/update_status_utils.h"
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -070074
Gilad Arnoldec7f9162014-07-15 13:24:46 -070075using base::Bind;
76using base::Callback;
Andrewe045aef2020-01-08 16:29:22 -080077using base::FilePath;
David Zeuthen3c55abd2013-10-14 12:48:03 -070078using base::Time;
Darin Petkovaf183052010-08-23 12:07:13 -070079using base::TimeDelta;
80using base::TimeTicks;
Alex Vakulenko3f39d5c2015-10-13 09:27:13 -070081using brillo::MessageLoop;
Amin Hassani7cc8bb02019-01-14 16:29:47 -080082using chromeos_update_manager::CalculateStagingCase;
Gilad Arnoldec7f9162014-07-15 13:24:46 -070083using chromeos_update_manager::EvalStatus;
84using chromeos_update_manager::Policy;
Adolfo Victoria497044c2018-07-18 07:51:42 -070085using chromeos_update_manager::StagingCase;
Amin Hassani7cc8bb02019-01-14 16:29:47 -080086using chromeos_update_manager::UpdateCheckParams;
Andrew065d78d2020-04-07 15:43:07 -070087using std::map;
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -070088using std::string;
89using std::vector;
Aaron Woodbf5a2522017-10-04 10:58:36 -070090using update_engine::UpdateAttemptFlags;
Aaron Wood7f92e2b2017-08-28 14:51:21 -070091using update_engine::UpdateEngineStatus;
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -070092
93namespace chromeos_update_engine {
94
Darin Petkov36275772010-10-01 11:40:57 -070095const int UpdateAttempter::kMaxDeltaUpdateFailures = 3;
96
Andrew de los Reyes45168102010-11-22 11:13:50 -080097namespace {
98const int kMaxConsecutiveObeyProxyRequests = 20;
Gilad Arnold70e476e2013-07-30 16:01:13 -070099
Alex Deymo0d298542016-03-30 18:31:49 -0700100// Minimum threshold to broadcast an status update in progress and time.
101const double kBroadcastThresholdProgress = 0.01; // 1%
102const int kBroadcastThresholdSeconds = 10;
103
David Pursell02c18642014-11-06 11:26:11 -0800104// By default autest bypasses scattering. If we want to test scattering,
105// use kScheduledAUTestURLRequest. The URL used is same in both cases, but
106// different params are passed to CheckForUpdate().
107const char kAUTestURLRequest[] = "autest";
108const char kScheduledAUTestURLRequest[] = "autest-scheduled";
Alex Vakulenkod2779df2014-06-16 13:19:00 -0700109} // namespace
Andrew de los Reyes45168102010-11-22 11:13:50 -0800110
Sen Jiangdaeaa432018-10-09 18:18:45 -0700111ErrorCode GetErrorCodeForAction(AbstractAction* action, ErrorCode code) {
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -0700112 if (code != ErrorCode::kError)
Darin Petkov777dbfa2010-07-20 15:03:37 -0700113 return code;
114
115 const string type = action->Type();
116 if (type == OmahaRequestAction::StaticType())
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -0700117 return ErrorCode::kOmahaRequestError;
Darin Petkov777dbfa2010-07-20 15:03:37 -0700118 if (type == OmahaResponseHandlerAction::StaticType())
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -0700119 return ErrorCode::kOmahaResponseHandlerError;
Allie Woodeb9e6d82015-04-17 13:55:30 -0700120 if (type == FilesystemVerifierAction::StaticType())
121 return ErrorCode::kFilesystemVerifierError;
Darin Petkov777dbfa2010-07-20 15:03:37 -0700122 if (type == PostinstallRunnerAction::StaticType())
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -0700123 return ErrorCode::kPostinstallRunnerError;
Darin Petkov777dbfa2010-07-20 15:03:37 -0700124
125 return code;
126}
127
Amin Hassani538bd592020-11-04 20:46:08 -0800128UpdateAttempter::UpdateAttempter(CertificateChecker* cert_checker)
Gilad Arnold1f847232014-04-07 12:07:49 -0700129 : processor_(new ActionProcessor()),
Colin Howes978c1082018-12-03 11:46:12 -0800130 cert_checker_(cert_checker),
131 is_install_(false) {}
Darin Petkovc6c135c2010-08-11 13:36:18 -0700132
Gilad Arnoldec7f9162014-07-15 13:24:46 -0700133UpdateAttempter::~UpdateAttempter() {
Amin Hassanic288bf62020-11-04 19:33:43 -0800134 // Prevent any DBus communication from UpdateAttempter when shutting down the
135 // daemon.
136 ClearObservers();
137
Alex Deymo33e91e72015-12-01 18:26:08 -0300138 // CertificateChecker might not be initialized in unittests.
139 if (cert_checker_)
140 cert_checker_->SetObserver(nullptr);
Alex Deymo2b4268c2015-12-04 13:56:25 -0800141 // Release ourselves as the ActionProcessor's delegate to prevent
142 // re-scheduling the updates due to the processing stopped.
143 processor_->set_delegate(nullptr);
Gilad Arnoldec7f9162014-07-15 13:24:46 -0700144}
145
Gilad Arnold1f847232014-04-07 12:07:49 -0700146void UpdateAttempter::Init() {
147 // Pulling from the SystemState can only be done after construction, since
148 // this is an aggregate of various objects (such as the UpdateAttempter),
149 // which requires them all to be constructed prior to it being used.
Amin Hassani538bd592020-11-04 20:46:08 -0800150 prefs_ = SystemState::Get()->prefs();
151 omaha_request_params_ = SystemState::Get()->request_params();
Alex Deymo906191f2015-10-12 12:22:44 -0700152
Alex Deymo33e91e72015-12-01 18:26:08 -0300153 if (cert_checker_)
154 cert_checker_->SetObserver(this);
155
Alex Deymo906191f2015-10-12 12:22:44 -0700156 // In case of update_engine restart without a reboot we need to restore the
157 // reboot needed state.
Vyshu Khota4c5413d2020-11-04 16:17:25 -0800158 if (GetBootTimeAtUpdate(nullptr)) {
Alex Deymo906191f2015-10-12 12:22:44 -0700159 status_ = UpdateStatus::UPDATED_NEED_REBOOT;
Vyshu Khota4c5413d2020-11-04 16:17:25 -0800160 } else {
Alex Deymo906191f2015-10-12 12:22:44 -0700161 status_ = UpdateStatus::IDLE;
Vyshu Khota4c5413d2020-11-04 16:17:25 -0800162 prefs_->Delete(kPrefsLastFp, {kDlcPrefsSubDir});
163 }
Gilad Arnold1f847232014-04-07 12:07:49 -0700164}
165
Xiaochu Liu88d90382018-08-29 16:09:11 -0700166bool UpdateAttempter::ScheduleUpdates() {
Jae Hoon Kimba2fdce2019-07-11 13:18:58 -0700167 if (IsBusyOrUpdateScheduled())
Xiaochu Liu88d90382018-08-29 16:09:11 -0700168 return false;
Gilad Arnold54fa66d2014-09-29 13:14:29 -0700169
Gilad Arnoldec7f9162014-07-15 13:24:46 -0700170 chromeos_update_manager::UpdateManager* const update_manager =
Amin Hassani538bd592020-11-04 20:46:08 -0800171 SystemState::Get()->update_manager();
Gilad Arnoldec7f9162014-07-15 13:24:46 -0700172 CHECK(update_manager);
Amin Hassani7cc8bb02019-01-14 16:29:47 -0800173 Callback<void(EvalStatus, const UpdateCheckParams&)> callback =
174 Bind(&UpdateAttempter::OnUpdateScheduled, base::Unretained(this));
Gilad Arnoldec7f9162014-07-15 13:24:46 -0700175 // We limit the async policy request to a reasonably short time, to avoid a
176 // starvation due to a transient bug.
Jae Hoon Kim504c3cb2019-07-02 11:17:24 -0700177 update_manager->AsyncPolicyRequestUpdateCheckAllowed(
178 callback, &Policy::UpdateCheckAllowed);
Gilad Arnoldec7f9162014-07-15 13:24:46 -0700179 waiting_for_scheduled_check_ = true;
Xiaochu Liu88d90382018-08-29 16:09:11 -0700180 return true;
Darin Petkovc6c135c2010-08-11 13:36:18 -0700181}
182
Amin Hassanic288bf62020-11-04 19:33:43 -0800183bool UpdateAttempter::StartUpdater() {
184 // Initiate update checks.
185 ScheduleUpdates();
186
Amin Hassani538bd592020-11-04 20:46:08 -0800187 auto update_boot_flags_action = std::make_unique<UpdateBootFlagsAction>(
188 SystemState::Get()->boot_control());
Amin Hassani1e19be32020-11-16 14:12:33 -0800189 aux_processor_.EnqueueAction(std::move(update_boot_flags_action));
Amin Hassanic288bf62020-11-04 19:33:43 -0800190 // Update boot flags after 45 seconds.
191 MessageLoop::current()->PostDelayedTask(
192 FROM_HERE,
193 base::Bind(&ActionProcessor::StartProcessing,
Amin Hassani1e19be32020-11-16 14:12:33 -0800194 base::Unretained(&aux_processor_)),
Amin Hassanic288bf62020-11-04 19:33:43 -0800195 base::TimeDelta::FromSeconds(45));
196
197 // Broadcast the update engine status on startup to ensure consistent system
198 // state on crashes.
199 MessageLoop::current()->PostTask(
200 FROM_HERE,
201 base::Bind(&UpdateAttempter::BroadcastStatus, base::Unretained(this)));
202
203 MessageLoop::current()->PostTask(
204 FROM_HERE,
205 base::Bind(&UpdateAttempter::UpdateEngineStarted,
206 base::Unretained(this)));
207 return true;
208}
209
Alex Deymoc1c17b42015-11-23 03:53:15 -0300210void UpdateAttempter::CertificateChecked(ServerToCheck server_to_check,
211 CertificateCheckResult result) {
Amin Hassani538bd592020-11-04 20:46:08 -0800212 SystemState::Get()->metrics_reporter()->ReportCertificateCheckMetrics(
Tianjie Xu282aa1f2017-09-05 13:42:45 -0700213 server_to_check, result);
Alex Deymoc1c17b42015-11-23 03:53:15 -0300214}
215
David Zeuthen985b1122013-10-09 12:13:15 -0700216bool UpdateAttempter::CheckAndReportDailyMetrics() {
217 int64_t stored_value;
Amin Hassani538bd592020-11-04 20:46:08 -0800218 Time now = SystemState::Get()->clock()->GetWallclockTime();
219 if (SystemState::Get()->prefs()->Exists(kPrefsDailyMetricsLastReportedAt) &&
220 SystemState::Get()->prefs()->GetInt64(kPrefsDailyMetricsLastReportedAt,
221 &stored_value)) {
Alex Deymof329b932014-10-30 01:37:48 -0700222 Time last_reported_at = Time::FromInternalValue(stored_value);
223 TimeDelta time_reported_since = now - last_reported_at;
David Zeuthen985b1122013-10-09 12:13:15 -0700224 if (time_reported_since.InSeconds() < 0) {
225 LOG(WARNING) << "Last reported daily metrics "
226 << utils::FormatTimeDelta(time_reported_since) << " ago "
227 << "which is negative. Either the system clock is wrong or "
228 << "the kPrefsDailyMetricsLastReportedAt state variable "
229 << "is wrong.";
230 // In this case, report daily metrics to reset.
231 } else {
Amin Hassani7cc8bb02019-01-14 16:29:47 -0800232 if (time_reported_since.InSeconds() < 24 * 60 * 60) {
David Zeuthen985b1122013-10-09 12:13:15 -0700233 LOG(INFO) << "Last reported daily metrics "
234 << utils::FormatTimeDelta(time_reported_since) << " ago.";
235 return false;
236 }
237 LOG(INFO) << "Last reported daily metrics "
238 << utils::FormatTimeDelta(time_reported_since) << " ago, "
239 << "which is more than 24 hours ago.";
240 }
241 }
242
243 LOG(INFO) << "Reporting daily metrics.";
Amin Hassani538bd592020-11-04 20:46:08 -0800244 SystemState::Get()->prefs()->SetInt64(kPrefsDailyMetricsLastReportedAt,
245 now.ToInternalValue());
David Zeuthen985b1122013-10-09 12:13:15 -0700246
247 ReportOSAge();
248
249 return true;
250}
251
252void UpdateAttempter::ReportOSAge() {
253 struct stat sb;
David Zeuthen985b1122013-10-09 12:13:15 -0700254 if (stat("/etc/lsb-release", &sb) != 0) {
David Zeuthend9aca3b2014-08-21 13:11:21 -0400255 PLOG(ERROR) << "Error getting file status for /etc/lsb-release "
256 << "(Note: this may happen in some unit tests)";
David Zeuthen985b1122013-10-09 12:13:15 -0700257 return;
258 }
259
Sen Jiang1c3db7d2019-01-10 15:48:55 -0800260 Time lsb_release_timestamp = Time::FromTimeSpec(sb.st_ctim);
Amin Hassani538bd592020-11-04 20:46:08 -0800261 Time now = SystemState::Get()->clock()->GetWallclockTime();
Alex Deymof329b932014-10-30 01:37:48 -0700262 TimeDelta age = now - lsb_release_timestamp;
David Zeuthen985b1122013-10-09 12:13:15 -0700263 if (age.InSeconds() < 0) {
264 LOG(ERROR) << "The OS age (" << utils::FormatTimeDelta(age)
David Zeuthend9aca3b2014-08-21 13:11:21 -0400265 << ") is negative. Maybe the clock is wrong? "
266 << "(Note: this may happen in some unit tests.)";
David Zeuthen985b1122013-10-09 12:13:15 -0700267 return;
268 }
269
Amin Hassani538bd592020-11-04 20:46:08 -0800270 SystemState::Get()->metrics_reporter()->ReportDailyMetrics(age);
David Zeuthen985b1122013-10-09 12:13:15 -0700271}
272
Amin Hassanie8153632020-10-27 15:11:28 -0700273void UpdateAttempter::Update(const UpdateCheckParams& params) {
Gilad Arnoldec7f9162014-07-15 13:24:46 -0700274 // This is normally called frequently enough so it's appropriate to use as a
275 // hook for reporting daily metrics.
276 // TODO(garnold) This should be hooked to a separate (reliable and consistent)
277 // timeout event.
David Zeuthen985b1122013-10-09 12:13:15 -0700278 CheckAndReportDailyMetrics();
279
Andrew de los Reyesc1d5c932011-04-20 17:15:47 -0700280 fake_update_success_ = false;
Christopher Wileycc8ce0e2015-10-01 16:48:47 -0700281 if (status_ == UpdateStatus::UPDATED_NEED_REBOOT) {
Thieu Le116fda32011-04-19 11:01:54 -0700282 // Although we have applied an update, we still want to ping Omaha
283 // to ensure the number of active statistics is accurate.
David Zeuthen33bae492014-02-25 16:16:18 -0800284 //
285 // Also convey to the UpdateEngine.Check.Result metric that we're
286 // not performing an update check because of this.
Andrew de los Reyes6b78e292010-05-10 15:54:39 -0700287 LOG(INFO) << "Not updating b/c we already updated and we're waiting for "
Thieu Le116fda32011-04-19 11:01:54 -0700288 << "reboot, we'll ping Omaha instead";
Amin Hassani538bd592020-11-04 20:46:08 -0800289 SystemState::Get()->metrics_reporter()->ReportUpdateCheckMetrics(
Tianjie Xu282aa1f2017-09-05 13:42:45 -0700290 metrics::CheckResult::kRebootPending,
291 metrics::CheckReaction::kUnset,
292 metrics::DownloadErrorCode::kUnset);
Thieu Le116fda32011-04-19 11:01:54 -0700293 PingOmaha();
Andrew de los Reyes6b78e292010-05-10 15:54:39 -0700294 return;
295 }
Christopher Wileycc8ce0e2015-10-01 16:48:47 -0700296 if (status_ != UpdateStatus::IDLE) {
Andrew de los Reyes6b78e292010-05-10 15:54:39 -0700297 // Update in progress. Do nothing
298 return;
299 }
Jay Srinivasan480ddfa2012-06-01 19:15:26 -0700300
Amin Hassanie8153632020-10-27 15:11:28 -0700301 if (!CalculateUpdateParams(params)) {
Jay Srinivasan480ddfa2012-06-01 19:15:26 -0700302 return;
303 }
304
Amin Hassanie8153632020-10-27 15:11:28 -0700305 BuildUpdateActions(params.interactive);
Jay Srinivasan480ddfa2012-06-01 19:15:26 -0700306
Christopher Wileycc8ce0e2015-10-01 16:48:47 -0700307 SetStatusAndNotify(UpdateStatus::CHECKING_FOR_UPDATE);
Jay Srinivasan480ddfa2012-06-01 19:15:26 -0700308
Gilad Arnoldec7f9162014-07-15 13:24:46 -0700309 // Update the last check time here; it may be re-updated when an Omaha
310 // response is received, but this will prevent us from repeatedly scheduling
311 // checks in the case where a response is not received.
312 UpdateLastCheckedTime();
313
Amin Hassani0882a512018-04-05 16:25:44 -0700314 ScheduleProcessingStart();
Jay Srinivasan480ddfa2012-06-01 19:15:26 -0700315}
316
Jay Srinivasan1c0fe792013-03-28 16:45:25 -0700317void UpdateAttempter::RefreshDevicePolicy() {
318 // Lazy initialize the policy provider, or reload the latest policy data.
319 if (!policy_provider_.get())
320 policy_provider_.reset(new policy::PolicyProvider());
321 policy_provider_->Reload();
322
Alex Vakulenko88b591f2014-08-28 16:48:57 -0700323 const policy::DevicePolicy* device_policy = nullptr;
Jay Srinivasan1c0fe792013-03-28 16:45:25 -0700324 if (policy_provider_->device_policy_is_loaded())
325 device_policy = &policy_provider_->GetDevicePolicy();
326
327 if (device_policy)
328 LOG(INFO) << "Device policies/settings present";
329 else
330 LOG(INFO) << "No device policies/settings present.";
331
Amin Hassani538bd592020-11-04 20:46:08 -0800332 SystemState::Get()->set_device_policy(device_policy);
333 SystemState::Get()->p2p_manager()->SetDevicePolicy(device_policy);
Jay Srinivasan1c0fe792013-03-28 16:45:25 -0700334}
335
David Zeuthen8f191b22013-08-06 12:27:50 -0700336void UpdateAttempter::CalculateP2PParams(bool interactive) {
337 bool use_p2p_for_downloading = false;
338 bool use_p2p_for_sharing = false;
339
Thiemo Nagel27d9d532017-05-26 16:35:20 +0200340 // Never use p2p for downloading in interactive checks unless the developer
341 // has opted in for it via a marker file.
David Zeuthen8f191b22013-08-06 12:27:50 -0700342 //
Thiemo Nagel27d9d532017-05-26 16:35:20 +0200343 // (Why would a developer want to opt in? If they are working on the
344 // update_engine or p2p codebases so they can actually test their code.)
David Zeuthen8f191b22013-08-06 12:27:50 -0700345
Amin Hassani538bd592020-11-04 20:46:08 -0800346 if (!SystemState::Get()->p2p_manager()->IsP2PEnabled()) {
347 LOG(INFO) << "p2p is not enabled - disallowing p2p for both"
348 << " downloading and sharing.";
349 } else {
350 // Allow p2p for sharing, even in interactive checks.
351 use_p2p_for_sharing = true;
352 if (!interactive) {
353 LOG(INFO) << "Non-interactive check - allowing p2p for downloading";
354 use_p2p_for_downloading = true;
David Zeuthen8f191b22013-08-06 12:27:50 -0700355 } else {
Amin Hassani538bd592020-11-04 20:46:08 -0800356 LOG(INFO) << "Forcibly disabling use of p2p for downloading "
357 << "since this update attempt is interactive.";
David Zeuthen8f191b22013-08-06 12:27:50 -0700358 }
359 }
360
Amin Hassani538bd592020-11-04 20:46:08 -0800361 PayloadStateInterface* const payload_state =
362 SystemState::Get()->payload_state();
Gilad Arnold74b5f552014-10-07 08:17:16 -0700363 payload_state->SetUsingP2PForDownloading(use_p2p_for_downloading);
364 payload_state->SetUsingP2PForSharing(use_p2p_for_sharing);
David Zeuthen8f191b22013-08-06 12:27:50 -0700365}
366
Amin Hassanie8153632020-10-27 15:11:28 -0700367bool UpdateAttempter::CalculateUpdateParams(const UpdateCheckParams& params) {
Darin Petkov1023a602010-08-30 13:47:51 -0700368 http_response_code_ = 0;
Amin Hassani538bd592020-11-04 20:46:08 -0800369 PayloadStateInterface* const payload_state =
370 SystemState::Get()->payload_state();
Patrick Dubroy7fbbe8a2011-08-01 17:28:22 +0200371
Alex Deymo749ecf12014-10-21 20:06:57 -0700372 // Refresh the policy before computing all the update parameters.
373 RefreshDevicePolicy();
374
Marton Hunyadye58bddb2018-04-10 20:27:26 +0200375 // Check whether we need to clear the rollback-happened preference after
376 // policy is available again.
377 UpdateRollbackHappened();
378
Amin Hassanie8153632020-10-27 15:11:28 -0700379 CalculateStagingParams(params.interactive);
Adolfo Victoria497044c2018-07-18 07:51:42 -0700380 // If staging_wait_time_ wasn't set, staging is off, use scattering instead.
381 if (staging_wait_time_.InSeconds() == 0) {
Amin Hassanie8153632020-10-27 15:11:28 -0700382 CalculateScatteringParams(params.interactive);
Adolfo Victoria497044c2018-07-18 07:51:42 -0700383 }
Patrick Dubroy7fbbe8a2011-08-01 17:28:22 +0200384
Amin Hassanie8153632020-10-27 15:11:28 -0700385 CalculateP2PParams(params.interactive);
Gilad Arnold74b5f552014-10-07 08:17:16 -0700386 if (payload_state->GetUsingP2PForDownloading() ||
387 payload_state->GetUsingP2PForSharing()) {
David Zeuthen8f191b22013-08-06 12:27:50 -0700388 // OK, p2p is to be used - start it and perform housekeeping.
389 if (!StartP2PAndPerformHousekeeping()) {
390 // If this fails, disable p2p for this attempt
391 LOG(INFO) << "Forcibly disabling use of p2p since starting p2p or "
392 << "performing housekeeping failed.";
Gilad Arnold74b5f552014-10-07 08:17:16 -0700393 payload_state->SetUsingP2PForDownloading(false);
394 payload_state->SetUsingP2PForSharing(false);
David Zeuthen8f191b22013-08-06 12:27:50 -0700395 }
396 }
397
Amin Hassanie8153632020-10-27 15:11:28 -0700398 if (!omaha_request_params_->Init(
399 forced_app_version_, forced_omaha_url_, params)) {
Jay Srinivasan1c0fe792013-03-28 16:45:25 -0700400 LOG(ERROR) << "Unable to initialize Omaha request params.";
Jay Srinivasan480ddfa2012-06-01 19:15:26 -0700401 return false;
Darin Petkova4a8a8c2010-07-15 22:21:12 -0700402 }
Darin Petkov3aefa862010-12-07 14:45:00 -0800403
Andrewe045aef2020-01-08 16:29:22 -0800404 // The function |CalculateDlcParams| makes use of the function |GetAppId| from
405 // |OmahaRequestParams|, so to ensure that the return from |GetAppId|
406 // doesn't change, no changes to the values |download_channel_|,
407 // |image_props_.product_id| and |image_props_.canary_product_id| from
408 // |omaha_request_params_| shall be made below this line.
409 CalculateDlcParams();
410
Gilad Arnoldafd70ed2014-09-30 16:27:06 -0700411 LOG(INFO) << "target_version_prefix = "
Jay Srinivasanae4697c2013-03-18 17:08:08 -0700412 << omaha_request_params_->target_version_prefix()
Marton Hunyadyba51c3f2018-04-25 15:18:10 +0200413 << ", rollback_allowed = "
414 << omaha_request_params_->rollback_allowed()
Jay Srinivasan480ddfa2012-06-01 19:15:26 -0700415 << ", scatter_factor_in_seconds = "
416 << utils::FormatSecs(scatter_factor_.InSeconds());
417
418 LOG(INFO) << "Wall Clock Based Wait Enabled = "
Jay Srinivasanae4697c2013-03-18 17:08:08 -0700419 << omaha_request_params_->wall_clock_based_wait_enabled()
Jay Srinivasan480ddfa2012-06-01 19:15:26 -0700420 << ", Update Check Count Wait Enabled = "
Jay Srinivasanae4697c2013-03-18 17:08:08 -0700421 << omaha_request_params_->update_check_count_wait_enabled()
Amin Hassani7cc8bb02019-01-14 16:29:47 -0800422 << ", Waiting Period = "
423 << utils::FormatSecs(
424 omaha_request_params_->waiting_period().InSeconds());
Jay Srinivasan0a708742012-03-20 11:26:12 -0700425
David Zeuthen8f191b22013-08-06 12:27:50 -0700426 LOG(INFO) << "Use p2p For Downloading = "
Gilad Arnold74b5f552014-10-07 08:17:16 -0700427 << payload_state->GetUsingP2PForDownloading()
David Zeuthen8f191b22013-08-06 12:27:50 -0700428 << ", Use p2p For Sharing = "
Gilad Arnold74b5f552014-10-07 08:17:16 -0700429 << payload_state->GetUsingP2PForSharing();
David Zeuthen8f191b22013-08-06 12:27:50 -0700430
Andrew de los Reyes45168102010-11-22 11:13:50 -0800431 obeying_proxies_ = true;
Amin Hassanie8153632020-10-27 15:11:28 -0700432 if (proxy_manual_checks_ == 0) {
Andrew de los Reyes45168102010-11-22 11:13:50 -0800433 LOG(INFO) << "forced to obey proxies";
434 // If forced to obey proxies, every 20th request will not use proxies
435 proxy_manual_checks_++;
436 LOG(INFO) << "proxy manual checks: " << proxy_manual_checks_;
437 if (proxy_manual_checks_ >= kMaxConsecutiveObeyProxyRequests) {
438 proxy_manual_checks_ = 0;
439 obeying_proxies_ = false;
440 }
441 } else if (base::RandInt(0, 4) == 0) {
442 obeying_proxies_ = false;
443 }
Amin Hassani7cc8bb02019-01-14 16:29:47 -0800444 LOG_IF(INFO, !obeying_proxies_)
445 << "To help ensure updates work, this update check we are ignoring the "
446 << "proxy settings and using direct connections.";
Andrew de los Reyes45168102010-11-22 11:13:50 -0800447
Darin Petkov36275772010-10-01 11:40:57 -0700448 DisableDeltaUpdateIfNeeded();
Jay Srinivasan480ddfa2012-06-01 19:15:26 -0700449 return true;
450}
451
Gilad Arnoldb92f0df2013-01-10 16:32:45 -0800452void UpdateAttempter::CalculateScatteringParams(bool interactive) {
Jay Srinivasan21be0752012-07-25 15:44:56 -0700453 // Take a copy of the old scatter value before we update it, as
454 // we need to update the waiting period if this value changes.
455 TimeDelta old_scatter_factor = scatter_factor_;
Amin Hassani538bd592020-11-04 20:46:08 -0800456 const policy::DevicePolicy* device_policy =
457 SystemState::Get()->device_policy();
Jay Srinivasan21be0752012-07-25 15:44:56 -0700458 if (device_policy) {
Ben Chan9abb7632014-08-07 00:10:53 -0700459 int64_t new_scatter_factor_in_secs = 0;
Jay Srinivasan21be0752012-07-25 15:44:56 -0700460 device_policy->GetScatterFactorInSeconds(&new_scatter_factor_in_secs);
Alex Vakulenkod2779df2014-06-16 13:19:00 -0700461 if (new_scatter_factor_in_secs < 0) // sanitize input, just in case.
Amin Hassani7cc8bb02019-01-14 16:29:47 -0800462 new_scatter_factor_in_secs = 0;
Jay Srinivasan21be0752012-07-25 15:44:56 -0700463 scatter_factor_ = TimeDelta::FromSeconds(new_scatter_factor_in_secs);
464 }
465
466 bool is_scatter_enabled = false;
467 if (scatter_factor_.InSeconds() == 0) {
468 LOG(INFO) << "Scattering disabled since scatter factor is set to 0";
Gilad Arnoldb92f0df2013-01-10 16:32:45 -0800469 } else if (interactive) {
470 LOG(INFO) << "Scattering disabled as this is an interactive update check";
Amin Hassani538bd592020-11-04 20:46:08 -0800471 } else if (SystemState::Get()->hardware()->IsOOBEEnabled() &&
472 !SystemState::Get()->hardware()->IsOOBEComplete(nullptr)) {
Alex Deymo46a9aae2016-05-04 20:20:11 -0700473 LOG(INFO) << "Scattering disabled since OOBE is enabled but not complete "
474 "yet";
Jay Srinivasan21be0752012-07-25 15:44:56 -0700475 } else {
476 is_scatter_enabled = true;
477 LOG(INFO) << "Scattering is enabled";
478 }
479
480 if (is_scatter_enabled) {
481 // This means the scattering policy is turned on.
482 // Now check if we need to update the waiting period. The two cases
483 // in which we'd need to update the waiting period are:
484 // 1. First time in process or a scheduled check after a user-initiated one.
Jay Srinivasanae4697c2013-03-18 17:08:08 -0700485 // (omaha_request_params_->waiting_period will be zero in this case).
Jay Srinivasan21be0752012-07-25 15:44:56 -0700486 // 2. Admin has changed the scattering policy value.
487 // (new scattering value will be different from old one in this case).
Ben Chan9abb7632014-08-07 00:10:53 -0700488 int64_t wait_period_in_secs = 0;
Jay Srinivasanae4697c2013-03-18 17:08:08 -0700489 if (omaha_request_params_->waiting_period().InSeconds() == 0) {
Jay Srinivasan21be0752012-07-25 15:44:56 -0700490 // First case. Check if we have a suitable value to set for
491 // the waiting period.
Adolfo Victoriad3a1e352018-07-16 11:40:47 -0700492 if (prefs_->GetInt64(kPrefsWallClockScatteringWaitPeriod,
493 &wait_period_in_secs) &&
Jay Srinivasan21be0752012-07-25 15:44:56 -0700494 wait_period_in_secs > 0 &&
495 wait_period_in_secs <= scatter_factor_.InSeconds()) {
496 // This means:
497 // 1. There's a persisted value for the waiting period available.
498 // 2. And that persisted value is still valid.
499 // So, in this case, we should reuse the persisted value instead of
500 // generating a new random value to improve the chances of a good
501 // distribution for scattering.
Jay Srinivasanae4697c2013-03-18 17:08:08 -0700502 omaha_request_params_->set_waiting_period(
Amin Hassani7cc8bb02019-01-14 16:29:47 -0800503 TimeDelta::FromSeconds(wait_period_in_secs));
504 LOG(INFO) << "Using persisted wall-clock waiting period: "
505 << utils::FormatSecs(
506 omaha_request_params_->waiting_period().InSeconds());
Alex Vakulenkod2779df2014-06-16 13:19:00 -0700507 } else {
Jay Srinivasan21be0752012-07-25 15:44:56 -0700508 // This means there's no persisted value for the waiting period
509 // available or its value is invalid given the new scatter_factor value.
510 // So, we should go ahead and regenerate a new value for the
511 // waiting period.
512 LOG(INFO) << "Persisted value not present or not valid ("
513 << utils::FormatSecs(wait_period_in_secs)
514 << ") for wall-clock waiting period.";
515 GenerateNewWaitingPeriod();
516 }
517 } else if (scatter_factor_ != old_scatter_factor) {
518 // This means there's already a waiting period value, but we detected
519 // a change in the scattering policy value. So, we should regenerate the
520 // waiting period to make sure it's within the bounds of the new scatter
521 // factor value.
522 GenerateNewWaitingPeriod();
523 } else {
524 // Neither the first time scattering is enabled nor the scattering value
525 // changed. Nothing to do.
Amin Hassani7cc8bb02019-01-14 16:29:47 -0800526 LOG(INFO) << "Keeping current wall-clock waiting period: "
527 << utils::FormatSecs(
528 omaha_request_params_->waiting_period().InSeconds());
Jay Srinivasan21be0752012-07-25 15:44:56 -0700529 }
530
Jay Srinivasanae4697c2013-03-18 17:08:08 -0700531 // The invariant at this point is that omaha_request_params_->waiting_period
Jay Srinivasan21be0752012-07-25 15:44:56 -0700532 // is non-zero no matter which path we took above.
Jay Srinivasanae4697c2013-03-18 17:08:08 -0700533 LOG_IF(ERROR, omaha_request_params_->waiting_period().InSeconds() == 0)
Jay Srinivasan21be0752012-07-25 15:44:56 -0700534 << "Waiting Period should NOT be zero at this point!!!";
535
536 // Since scattering is enabled, wall clock based wait will always be
537 // enabled.
Jay Srinivasanae4697c2013-03-18 17:08:08 -0700538 omaha_request_params_->set_wall_clock_based_wait_enabled(true);
Jay Srinivasan21be0752012-07-25 15:44:56 -0700539
540 // If we don't have any issues in accessing the file system to update
541 // the update check count value, we'll turn that on as well.
542 bool decrement_succeeded = DecrementUpdateCheckCount();
Jay Srinivasanae4697c2013-03-18 17:08:08 -0700543 omaha_request_params_->set_update_check_count_wait_enabled(
Amin Hassani7cc8bb02019-01-14 16:29:47 -0800544 decrement_succeeded);
Jay Srinivasan21be0752012-07-25 15:44:56 -0700545 } else {
546 // This means the scattering feature is turned off or disabled for
547 // this particular update check. Make sure to disable
548 // all the knobs and artifacts so that we don't invoke any scattering
549 // related code.
Jay Srinivasanae4697c2013-03-18 17:08:08 -0700550 omaha_request_params_->set_wall_clock_based_wait_enabled(false);
551 omaha_request_params_->set_update_check_count_wait_enabled(false);
552 omaha_request_params_->set_waiting_period(TimeDelta::FromSeconds(0));
Adolfo Victoriad3a1e352018-07-16 11:40:47 -0700553 prefs_->Delete(kPrefsWallClockScatteringWaitPeriod);
Jay Srinivasan21be0752012-07-25 15:44:56 -0700554 prefs_->Delete(kPrefsUpdateCheckCount);
555 // Don't delete the UpdateFirstSeenAt file as we don't want manual checks
556 // that result in no-updates (e.g. due to server side throttling) to
557 // cause update starvation by having the client generate a new
558 // UpdateFirstSeenAt for each scheduled check that follows a manual check.
559 }
560}
561
562void UpdateAttempter::GenerateNewWaitingPeriod() {
Amin Hassani7cc8bb02019-01-14 16:29:47 -0800563 omaha_request_params_->set_waiting_period(
564 TimeDelta::FromSeconds(base::RandInt(1, scatter_factor_.InSeconds())));
Jay Srinivasan21be0752012-07-25 15:44:56 -0700565
Amin Hassani7cc8bb02019-01-14 16:29:47 -0800566 LOG(INFO) << "Generated new wall-clock waiting period: "
567 << utils::FormatSecs(
568 omaha_request_params_->waiting_period().InSeconds());
Jay Srinivasan21be0752012-07-25 15:44:56 -0700569
570 // Do a best-effort to persist this in all cases. Even if the persistence
571 // fails, we'll still be able to scatter based on our in-memory value.
572 // The persistence only helps in ensuring a good overall distribution
573 // across multiple devices if they tend to reboot too often.
Amin Hassani538bd592020-11-04 20:46:08 -0800574 SystemState::Get()->payload_state()->SetScatteringWaitPeriod(
Gilad Arnold519cfc72014-10-02 10:34:54 -0700575 omaha_request_params_->waiting_period());
Jay Srinivasan21be0752012-07-25 15:44:56 -0700576}
577
Adolfo Victoria497044c2018-07-18 07:51:42 -0700578void UpdateAttempter::CalculateStagingParams(bool interactive) {
Amin Hassani538bd592020-11-04 20:46:08 -0800579 bool oobe_complete = SystemState::Get()->hardware()->IsOOBEEnabled() &&
580 SystemState::Get()->hardware()->IsOOBEComplete(nullptr);
581 auto device_policy = SystemState::Get()->device_policy();
Adolfo Victoria497044c2018-07-18 07:51:42 -0700582 StagingCase staging_case = StagingCase::kOff;
583 if (device_policy && !interactive && oobe_complete) {
584 staging_wait_time_ = omaha_request_params_->waiting_period();
585 staging_case = CalculateStagingCase(
Amin Hassani90e9f192020-11-18 14:20:56 -0800586 device_policy, &staging_wait_time_, &staging_schedule_);
Adolfo Victoria497044c2018-07-18 07:51:42 -0700587 }
588 switch (staging_case) {
589 case StagingCase::kOff:
590 // Staging is off, get rid of persisted value.
591 prefs_->Delete(kPrefsWallClockStagingWaitPeriod);
592 // Set |staging_wait_time_| to its default value so scattering can still
593 // be turned on
594 staging_wait_time_ = TimeDelta();
595 break;
596 // Let the cases fall through since they just add, and never remove, steps
597 // to turning staging on.
598 case StagingCase::kNoSavedValue:
599 prefs_->SetInt64(kPrefsWallClockStagingWaitPeriod,
600 staging_wait_time_.InDays());
Denis Nikitin8f2fa742019-05-29 10:36:52 -0700601 FALLTHROUGH;
Adolfo Victoria497044c2018-07-18 07:51:42 -0700602 case StagingCase::kSetStagingFromPref:
603 omaha_request_params_->set_waiting_period(staging_wait_time_);
Denis Nikitin8f2fa742019-05-29 10:36:52 -0700604 FALLTHROUGH;
Adolfo Victoria497044c2018-07-18 07:51:42 -0700605 case StagingCase::kNoAction:
606 // Staging is on, enable wallclock based wait so that its values get used.
607 omaha_request_params_->set_wall_clock_based_wait_enabled(true);
608 // Use UpdateCheckCount if possible to prevent devices updating all at
609 // once.
610 omaha_request_params_->set_update_check_count_wait_enabled(
611 DecrementUpdateCheckCount());
612 // Scattering should not be turned on if staging is on, delete the
613 // existing scattering configuration.
614 prefs_->Delete(kPrefsWallClockScatteringWaitPeriod);
615 scatter_factor_ = TimeDelta();
616 }
Chris Sosad317e402013-06-12 13:47:09 -0700617}
618
Andrew065d78d2020-04-07 15:43:07 -0700619bool UpdateAttempter::ResetDlcPrefs(const string& dlc_id) {
620 vector<string> failures;
Andrew065d78d2020-04-07 15:43:07 -0700621 for (auto& sub_key :
622 {kPrefsPingActive, kPrefsPingLastActive, kPrefsPingLastRollcall}) {
Amin Hassani90e9f192020-11-18 14:20:56 -0800623 auto key = prefs_->CreateSubKey({kDlcPrefsSubDir, dlc_id, sub_key});
624 if (!prefs_->Delete(key))
Andrew065d78d2020-04-07 15:43:07 -0700625 failures.emplace_back(sub_key);
626 }
627 if (failures.size() != 0)
628 PLOG(ERROR) << "Failed to delete prefs (" << base::JoinString(failures, ",")
629 << " for DLC (" << dlc_id << ").";
630
631 return failures.size() == 0;
632}
633
Vyshu Khota4c5413d2020-11-04 16:17:25 -0800634void UpdateAttempter::SetPref(const string& pref_key,
635 const string& pref_value,
636 const string& payload_id) {
637 string dlc_id;
638 if (!omaha_request_params_->GetDlcId(payload_id, &dlc_id)) {
639 // Not a DLC ID, set fingerprint in perf for platform ID.
640 prefs_->SetString(pref_key, pref_value);
641 } else {
642 // Set fingerprint in pref for DLC ID.
643 auto key = prefs_->CreateSubKey({kDlcPrefsSubDir, dlc_id, pref_key});
644 prefs_->SetString(key, pref_value);
645 }
646}
647
Andrewa8d7df32020-03-15 20:10:01 -0700648bool UpdateAttempter::SetDlcActiveValue(bool is_active, const string& dlc_id) {
649 if (dlc_id.empty()) {
650 LOG(ERROR) << "Empty DLC ID passed.";
651 return false;
652 }
653 LOG(INFO) << "Set DLC (" << dlc_id << ") to "
654 << (is_active ? "Active" : "Inactive");
Andrewa8d7df32020-03-15 20:10:01 -0700655 if (is_active) {
Andrew065d78d2020-04-07 15:43:07 -0700656 auto ping_active_key =
Amin Hassani90e9f192020-11-18 14:20:56 -0800657 prefs_->CreateSubKey({kDlcPrefsSubDir, dlc_id, kPrefsPingActive});
658 if (!prefs_->SetInt64(ping_active_key, kPingActiveValue)) {
Andrewa8d7df32020-03-15 20:10:01 -0700659 LOG(ERROR) << "Failed to set the value of ping metadata '"
660 << kPrefsPingActive << "'.";
661 return false;
662 }
663 } else {
Andrew065d78d2020-04-07 15:43:07 -0700664 return ResetDlcPrefs(dlc_id);
Andrewa8d7df32020-03-15 20:10:01 -0700665 }
666 return true;
667}
668
Andrew065d78d2020-04-07 15:43:07 -0700669int64_t UpdateAttempter::GetPingMetadata(const string& metadata_key) const {
Andrewe045aef2020-01-08 16:29:22 -0800670 // The first time a ping is sent, the metadata files containing the values
671 // sent back by the server still don't exist. A value of -1 is used to
672 // indicate this.
Amin Hassani538bd592020-11-04 20:46:08 -0800673 if (!SystemState::Get()->prefs()->Exists(metadata_key))
Andrewe045aef2020-01-08 16:29:22 -0800674 return kPingNeverPinged;
675
676 int64_t value;
Amin Hassani538bd592020-11-04 20:46:08 -0800677 if (SystemState::Get()->prefs()->GetInt64(metadata_key, &value))
Andrewe045aef2020-01-08 16:29:22 -0800678 return value;
679
680 // Return -2 when the file exists and there is a problem reading from it, or
681 // the value cannot be converted to an integer.
682 return kPingUnknownValue;
683}
684
685void UpdateAttempter::CalculateDlcParams() {
Amin Hassani2b68e6b2020-04-17 10:49:12 -0700686 // Set the |dlc_ids_| only for an update. This is required to get the
Andrewe045aef2020-01-08 16:29:22 -0800687 // currently installed DLC(s).
688 if (!is_install_ &&
Amin Hassani538bd592020-11-04 20:46:08 -0800689 !SystemState::Get()->dlcservice()->GetDlcsToUpdate(&dlc_ids_)) {
Andrewe045aef2020-01-08 16:29:22 -0800690 LOG(INFO) << "Failed to retrieve DLC module IDs from dlcservice. Check the "
691 "state of dlcservice, will not update DLC modules.";
692 }
Andrew065d78d2020-04-07 15:43:07 -0700693 map<string, OmahaRequestParams::AppParams> dlc_apps_params;
Amin Hassani2b68e6b2020-04-17 10:49:12 -0700694 for (const auto& dlc_id : dlc_ids_) {
Andrewe045aef2020-01-08 16:29:22 -0800695 OmahaRequestParams::AppParams dlc_params{
696 .active_counting_type = OmahaRequestParams::kDateBased,
697 .name = dlc_id,
698 .send_ping = false};
Andrew065d78d2020-04-07 15:43:07 -0700699 if (is_install_) {
700 // In some cases, |SetDlcActiveValue| might fail to reset the DLC prefs
701 // when a DLC is uninstalled. To avoid having stale values from that
702 // scenario, we reset the metadata values on a new install request.
703 // Ignore failure to delete stale prefs.
704 ResetDlcPrefs(dlc_id);
705 SetDlcActiveValue(true, dlc_id);
706 } else {
707 // Only send the ping when the request is to update DLCs. When installing
708 // DLCs, we don't want to send the ping yet, since the DLCs might fail to
709 // install or might not really be active yet.
710 dlc_params.ping_active = kPingActiveValue;
711 auto ping_active_key =
Amin Hassani90e9f192020-11-18 14:20:56 -0800712 prefs_->CreateSubKey({kDlcPrefsSubDir, dlc_id, kPrefsPingActive});
713 if (!prefs_->GetInt64(ping_active_key, &dlc_params.ping_active) ||
Andrew065d78d2020-04-07 15:43:07 -0700714 dlc_params.ping_active != kPingActiveValue) {
715 dlc_params.ping_active = kPingInactiveValue;
Andrewe045aef2020-01-08 16:29:22 -0800716 }
Andrew065d78d2020-04-07 15:43:07 -0700717 auto ping_last_active_key =
Amin Hassani90e9f192020-11-18 14:20:56 -0800718 prefs_->CreateSubKey({kDlcPrefsSubDir, dlc_id, kPrefsPingLastActive});
Andrew065d78d2020-04-07 15:43:07 -0700719 dlc_params.ping_date_last_active = GetPingMetadata(ping_last_active_key);
720
Amin Hassani90e9f192020-11-18 14:20:56 -0800721 auto ping_last_rollcall_key = prefs_->CreateSubKey(
Jae Hoon Kimc1f36922020-05-11 18:20:18 -0700722 {kDlcPrefsSubDir, dlc_id, kPrefsPingLastRollcall});
Andrew065d78d2020-04-07 15:43:07 -0700723 dlc_params.ping_date_last_rollcall =
724 GetPingMetadata(ping_last_rollcall_key);
725
726 dlc_params.send_ping = true;
Andrewe045aef2020-01-08 16:29:22 -0800727 }
728 dlc_apps_params[omaha_request_params_->GetDlcAppId(dlc_id)] = dlc_params;
729 }
730 omaha_request_params_->set_dlc_apps_params(dlc_apps_params);
Amin Hassanie8153632020-10-27 15:11:28 -0700731 omaha_request_params_->set_is_install(is_install_);
Andrewe045aef2020-01-08 16:29:22 -0800732}
733
Jay Srinivasan480ddfa2012-06-01 19:15:26 -0700734void UpdateAttempter::BuildUpdateActions(bool interactive) {
Darin Petkovf42cc1c2010-09-01 09:03:02 -0700735 CHECK(!processor_->IsRunning());
736 processor_->set_delegate(this);
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700737
Jae Hoon Kimedb65502019-06-14 11:52:17 -0700738 // The session ID needs to be kept throughout the update flow. The value
739 // of the session ID will reset/update only when it is a new update flow.
740 session_id_ = base::GenerateGUID();
741
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700742 // Actions:
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700743 auto update_check_fetcher = std::make_unique<LibcurlHttpFetcher>(
Amin Hassani538bd592020-11-04 20:46:08 -0800744 GetProxyResolver(), SystemState::Get()->hardware());
Alex Deymo33e91e72015-12-01 18:26:08 -0300745 update_check_fetcher->set_server_to_check(ServerToCheck::kUpdate);
Andrew de los Reyesfb2f4612011-06-09 18:21:49 -0700746 // Try harder to connect to the network, esp when not interactive.
747 // See comment in libcurl_http_fetcher.cc.
748 update_check_fetcher->set_no_network_max_retries(interactive ? 1 : 3);
Xiaochu Liu4a1173a2019-04-10 10:49:08 -0700749 update_check_fetcher->set_is_update_check(true);
Amin Hassani538bd592020-11-04 20:46:08 -0800750 auto update_check_action = std::make_unique<OmahaRequestAction>(
751 nullptr, std::move(update_check_fetcher), false, session_id_);
752 auto response_handler_action = std::make_unique<OmahaResponseHandlerAction>();
753 auto update_boot_flags_action = std::make_unique<UpdateBootFlagsAction>(
754 SystemState::Get()->boot_control());
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700755 auto download_started_action = std::make_unique<OmahaRequestAction>(
Sen Jiang18414082018-01-11 14:50:36 -0800756 new OmahaEvent(OmahaEvent::kTypeUpdateDownloadStarted),
757 std::make_unique<LibcurlHttpFetcher>(GetProxyResolver(),
Amin Hassani538bd592020-11-04 20:46:08 -0800758 SystemState::Get()->hardware()),
Jae Hoon Kimedb65502019-06-14 11:52:17 -0700759 false,
760 session_id_);
Alex Deymo33e91e72015-12-01 18:26:08 -0300761
Amin Hassani538bd592020-11-04 20:46:08 -0800762 LibcurlHttpFetcher* download_fetcher = new LibcurlHttpFetcher(
763 GetProxyResolver(), SystemState::Get()->hardware());
Alex Deymo33e91e72015-12-01 18:26:08 -0300764 download_fetcher->set_server_to_check(ServerToCheck::kDownload);
Sen Jiangee174a12017-12-21 17:38:08 -0800765 if (interactive)
766 download_fetcher->set_max_retry_count(kDownloadMaxRetryCountInteractive);
Jae Hoon Kim0ae8fe12019-06-26 14:32:50 -0700767 download_fetcher->SetHeader(kXGoogleUpdateSessionId, session_id_);
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700768 auto download_action =
769 std::make_unique<DownloadAction>(prefs_,
Amin Hassani538bd592020-11-04 20:46:08 -0800770 SystemState::Get()->boot_control(),
771 SystemState::Get()->hardware(),
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700772 download_fetcher, // passes ownership
773 interactive);
Andrew de los Reyes63b96d72010-05-10 13:08:54 -0700774 download_action->set_delegate(this);
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700775
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700776 auto download_finished_action = std::make_unique<OmahaRequestAction>(
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700777 new OmahaEvent(OmahaEvent::kTypeUpdateDownloadFinished),
778 std::make_unique<LibcurlHttpFetcher>(GetProxyResolver(),
Amin Hassani538bd592020-11-04 20:46:08 -0800779 SystemState::Get()->hardware()),
Jae Hoon Kimedb65502019-06-14 11:52:17 -0700780 false,
781 session_id_);
Tianjie24f96092020-06-30 12:26:25 -0700782 auto filesystem_verifier_action = std::make_unique<FilesystemVerifierAction>(
Amin Hassani538bd592020-11-04 20:46:08 -0800783 SystemState::Get()->boot_control()->GetDynamicPartitionControl());
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700784 auto update_complete_action = std::make_unique<OmahaRequestAction>(
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700785 new OmahaEvent(OmahaEvent::kTypeUpdateComplete),
786 std::make_unique<LibcurlHttpFetcher>(GetProxyResolver(),
Amin Hassani538bd592020-11-04 20:46:08 -0800787 SystemState::Get()->hardware()),
Jae Hoon Kimedb65502019-06-14 11:52:17 -0700788 false,
789 session_id_);
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700790
791 auto postinstall_runner_action = std::make_unique<PostinstallRunnerAction>(
Amin Hassani538bd592020-11-04 20:46:08 -0800792 SystemState::Get()->boot_control(), SystemState::Get()->hardware());
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700793 postinstall_runner_action->set_delegate(this);
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700794
795 // Bond them together. We have to use the leaf-types when calling
796 // BondActions().
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700797 BondActions(update_check_action.get(), response_handler_action.get());
798 BondActions(response_handler_action.get(), download_action.get());
799 BondActions(download_action.get(), filesystem_verifier_action.get());
800 BondActions(filesystem_verifier_action.get(),
801 postinstall_runner_action.get());
Chris Sosad317e402013-06-12 13:47:09 -0700802
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700803 processor_->EnqueueAction(std::move(update_check_action));
804 processor_->EnqueueAction(std::move(response_handler_action));
Amin Hassani0882a512018-04-05 16:25:44 -0700805 processor_->EnqueueAction(std::move(update_boot_flags_action));
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700806 processor_->EnqueueAction(std::move(download_started_action));
807 processor_->EnqueueAction(std::move(download_action));
808 processor_->EnqueueAction(std::move(download_finished_action));
809 processor_->EnqueueAction(std::move(filesystem_verifier_action));
810 processor_->EnqueueAction(std::move(postinstall_runner_action));
811 processor_->EnqueueAction(std::move(update_complete_action));
Chris Sosad317e402013-06-12 13:47:09 -0700812}
813
Chris Sosa44b9b7e2014-04-02 13:53:46 -0700814bool UpdateAttempter::Rollback(bool powerwash) {
Colin Howes978c1082018-12-03 11:46:12 -0800815 is_install_ = false;
Chris Sosa44b9b7e2014-04-02 13:53:46 -0700816 if (!CanRollback()) {
817 return false;
818 }
Chris Sosad317e402013-06-12 13:47:09 -0700819
Chris Sosa44b9b7e2014-04-02 13:53:46 -0700820 // Extra check for enterprise-enrolled devices since they don't support
821 // powerwash.
822 if (powerwash) {
823 // Enterprise-enrolled devices have an empty owner in their device policy.
824 string owner;
825 RefreshDevicePolicy();
Amin Hassani538bd592020-11-04 20:46:08 -0800826 const policy::DevicePolicy* device_policy =
827 SystemState::Get()->device_policy();
Chris Sosa44b9b7e2014-04-02 13:53:46 -0700828 if (device_policy && (!device_policy->GetOwner(&owner) || owner.empty())) {
829 LOG(ERROR) << "Enterprise device detected. "
830 << "Cannot perform a powerwash for enterprise devices.";
831 return false;
832 }
833 }
834
835 processor_->set_delegate(this);
Chris Sosaaa18e162013-06-20 13:20:30 -0700836
Chris Sosa28e479c2013-07-12 11:39:53 -0700837 // Initialize the default request params.
Amin Hassanie8153632020-10-27 15:11:28 -0700838 if (!omaha_request_params_->Init("", "", {.interactive = true})) {
Chris Sosa28e479c2013-07-12 11:39:53 -0700839 LOG(ERROR) << "Unable to initialize Omaha request params.";
840 return false;
841 }
842
Chris Sosad317e402013-06-12 13:47:09 -0700843 LOG(INFO) << "Setting rollback options.";
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700844 install_plan_.reset(new InstallPlan());
845 install_plan_->target_slot = GetRollbackSlot();
Amin Hassani538bd592020-11-04 20:46:08 -0800846 install_plan_->source_slot =
847 SystemState::Get()->boot_control()->GetCurrentSlot();
Chris Sosa76a29ae2013-07-11 17:59:24 -0700848
Amin Hassani538bd592020-11-04 20:46:08 -0800849 TEST_AND_RETURN_FALSE(install_plan_->LoadPartitionsFromSlots(
850 SystemState::Get()->boot_control()));
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700851 install_plan_->powerwash_required = powerwash;
Chris Sosad317e402013-06-12 13:47:09 -0700852
853 LOG(INFO) << "Using this install plan:";
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700854 install_plan_->Dump();
Chris Sosad317e402013-06-12 13:47:09 -0700855
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700856 auto install_plan_action =
857 std::make_unique<InstallPlanAction>(*install_plan_);
858 auto postinstall_runner_action = std::make_unique<PostinstallRunnerAction>(
Amin Hassani538bd592020-11-04 20:46:08 -0800859 SystemState::Get()->boot_control(), SystemState::Get()->hardware());
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700860 postinstall_runner_action->set_delegate(this);
861 BondActions(install_plan_action.get(), postinstall_runner_action.get());
862 processor_->EnqueueAction(std::move(install_plan_action));
863 processor_->EnqueueAction(std::move(postinstall_runner_action));
Chris Sosaaa18e162013-06-20 13:20:30 -0700864
865 // Update the payload state for Rollback.
Amin Hassani538bd592020-11-04 20:46:08 -0800866 SystemState::Get()->payload_state()->Rollback();
Chris Sosaaa18e162013-06-20 13:20:30 -0700867
Christopher Wileycc8ce0e2015-10-01 16:48:47 -0700868 SetStatusAndNotify(UpdateStatus::ATTEMPTING_ROLLBACK);
Chris Sosad317e402013-06-12 13:47:09 -0700869
Amin Hassani0882a512018-04-05 16:25:44 -0700870 ScheduleProcessingStart();
Chris Sosaaa18e162013-06-20 13:20:30 -0700871 return true;
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700872}
873
Alex Vakulenko59e253e2014-02-24 10:40:21 -0800874bool UpdateAttempter::CanRollback() const {
Chris Sosa44b9b7e2014-04-02 13:53:46 -0700875 // We can only rollback if the update_engine isn't busy and we have a valid
876 // rollback partition.
Christopher Wileycc8ce0e2015-10-01 16:48:47 -0700877 return (status_ == UpdateStatus::IDLE &&
Alex Deymo763e7db2015-08-27 21:08:08 -0700878 GetRollbackSlot() != BootControlInterface::kInvalidSlot);
Alex Vakulenko2bddadd2014-03-27 13:23:46 -0700879}
880
Alex Deymo763e7db2015-08-27 21:08:08 -0700881BootControlInterface::Slot UpdateAttempter::GetRollbackSlot() const {
882 LOG(INFO) << "UpdateAttempter::GetRollbackSlot";
Amin Hassani538bd592020-11-04 20:46:08 -0800883 const unsigned int num_slots =
884 SystemState::Get()->boot_control()->GetNumSlots();
Alex Deymo763e7db2015-08-27 21:08:08 -0700885 const BootControlInterface::Slot current_slot =
Amin Hassani538bd592020-11-04 20:46:08 -0800886 SystemState::Get()->boot_control()->GetCurrentSlot();
Alex Vakulenko59e253e2014-02-24 10:40:21 -0800887
Alex Deymo763e7db2015-08-27 21:08:08 -0700888 LOG(INFO) << " Installed slots: " << num_slots;
889 LOG(INFO) << " Booted from slot: "
890 << BootControlInterface::SlotName(current_slot);
Alex Vakulenko59e253e2014-02-24 10:40:21 -0800891
Alex Deymo763e7db2015-08-27 21:08:08 -0700892 if (current_slot == BootControlInterface::kInvalidSlot || num_slots < 2) {
893 LOG(INFO) << "Device is not updateable.";
894 return BootControlInterface::kInvalidSlot;
Alex Vakulenko59e253e2014-02-24 10:40:21 -0800895 }
896
Alex Deymo763e7db2015-08-27 21:08:08 -0700897 vector<BootControlInterface::Slot> bootable_slots;
Alex Deymof7ead812015-10-23 17:37:27 -0700898 for (BootControlInterface::Slot slot = 0; slot < num_slots; slot++) {
Alex Deymo763e7db2015-08-27 21:08:08 -0700899 if (slot != current_slot &&
Amin Hassani538bd592020-11-04 20:46:08 -0800900 SystemState::Get()->boot_control()->IsSlotBootable(slot)) {
Alex Deymo763e7db2015-08-27 21:08:08 -0700901 LOG(INFO) << "Found bootable slot "
902 << BootControlInterface::SlotName(slot);
903 return slot;
Alex Vakulenko59e253e2014-02-24 10:40:21 -0800904 }
905 }
Alex Deymo763e7db2015-08-27 21:08:08 -0700906 LOG(INFO) << "No other bootable slot found.";
907 return BootControlInterface::kInvalidSlot;
Alex Vakulenko2bddadd2014-03-27 13:23:46 -0700908}
909
Aaron Wood081c0232017-10-19 17:14:58 -0700910bool UpdateAttempter::CheckForUpdate(const string& app_version,
Jay Srinivasane73acab2012-07-10 14:34:03 -0700911 const string& omaha_url,
Aaron Wood081c0232017-10-19 17:14:58 -0700912 UpdateAttemptFlags flags) {
Jae Hoon Kimc437ea52019-07-11 11:20:38 -0700913 if (status_ != UpdateStatus::IDLE) {
914 LOG(INFO) << "Refusing to do an update as there is an "
915 << (is_install_ ? "install" : "update")
916 << " already in progress.";
Aaron Wood081c0232017-10-19 17:14:58 -0700917 return false;
918 }
919
Jae Hoon Kimc437ea52019-07-11 11:20:38 -0700920 bool interactive = !(flags & UpdateAttemptFlags::kFlagNonInteractive);
Jae Hoon Kimc437ea52019-07-11 11:20:38 -0700921 is_install_ = false;
922
Gilad Arnoldec7f9162014-07-15 13:24:46 -0700923 LOG(INFO) << "Forced update check requested.";
David Pursell02c18642014-11-06 11:26:11 -0800924 forced_app_version_.clear();
925 forced_omaha_url_.clear();
926
927 // Certain conditions must be met to allow setting custom version and update
928 // server URLs. However, kScheduledAUTestURLRequest and kAUTestURLRequest are
929 // always allowed regardless of device state.
930 if (IsAnyUpdateSourceAllowed()) {
931 forced_app_version_ = app_version;
932 forced_omaha_url_ = omaha_url;
933 }
934 if (omaha_url == kScheduledAUTestURLRequest) {
Alex Deymoac41a822015-09-15 20:52:53 -0700935 forced_omaha_url_ = constants::kOmahaDefaultAUTestURL;
David Pursell02c18642014-11-06 11:26:11 -0800936 // Pretend that it's not user-initiated even though it is,
937 // so as to test scattering logic, etc. which get kicked off
938 // only in scheduled update checks.
939 interactive = false;
940 } else if (omaha_url == kAUTestURLRequest) {
Alex Deymoac41a822015-09-15 20:52:53 -0700941 forced_omaha_url_ = constants::kOmahaDefaultAUTestURL;
David Pursell02c18642014-11-06 11:26:11 -0800942 }
943
Aaron Wood081c0232017-10-19 17:14:58 -0700944 if (interactive) {
945 // Use the passed-in update attempt flags for this update attempt instead
946 // of the previously set ones.
947 current_update_attempt_flags_ = flags;
948 // Note: The caching for non-interactive update checks happens in
Jae Hoon Kim75daa382019-07-02 11:17:24 -0700949 // |OnUpdateScheduled()|.
Aaron Wood081c0232017-10-19 17:14:58 -0700950 }
951
Jae Hoon Kim2b73ac22019-07-02 11:17:24 -0700952 // |forced_update_pending_callback_| should always be set, but even in the
953 // case that it is not, we still return true indicating success because the
954 // scheduled periodic check will pick up these changes.
Gilad Arnold54fa66d2014-09-29 13:14:29 -0700955 if (forced_update_pending_callback_.get()) {
Jae Hoon Kim2b73ac22019-07-02 11:17:24 -0700956 // Always call |ScheduleUpdates()| before forcing an update. This is because
957 // we need an update to be scheduled for the
958 // |forced_update_pending_callback_| to have an effect. Here we don't need
959 // to care about the return value from |ScheduleUpdate()|.
Gilad Arnold54fa66d2014-09-29 13:14:29 -0700960 ScheduleUpdates();
Gilad Arnoldec7f9162014-07-15 13:24:46 -0700961 forced_update_pending_callback_->Run(true, interactive);
Gilad Arnold54fa66d2014-09-29 13:14:29 -0700962 }
Aaron Wood081c0232017-10-19 17:14:58 -0700963 return true;
Andrew de los Reyes63b96d72010-05-10 13:08:54 -0700964}
965
Amin Hassani2b68e6b2020-04-17 10:49:12 -0700966bool UpdateAttempter::CheckForInstall(const vector<string>& dlc_ids,
Xiaochu Liu88d90382018-08-29 16:09:11 -0700967 const string& omaha_url) {
Jae Hoon Kimc437ea52019-07-11 11:20:38 -0700968 if (status_ != UpdateStatus::IDLE) {
969 LOG(INFO) << "Refusing to do an install as there is an "
970 << (is_install_ ? "install" : "update")
971 << " already in progress.";
972 return false;
973 }
974
Amin Hassani2b68e6b2020-04-17 10:49:12 -0700975 dlc_ids_ = dlc_ids;
Xiaochu Liu88d90382018-08-29 16:09:11 -0700976 is_install_ = true;
977 forced_omaha_url_.clear();
978
979 // Certain conditions must be met to allow setting custom version and update
980 // server URLs. However, kScheduledAUTestURLRequest and kAUTestURLRequest are
981 // always allowed regardless of device state.
982 if (IsAnyUpdateSourceAllowed()) {
983 forced_omaha_url_ = omaha_url;
984 }
Jae Hoon Kimc43f6bb2019-07-03 12:56:52 -0700985
986 if (omaha_url == kScheduledAUTestURLRequest ||
987 omaha_url == kAUTestURLRequest) {
Xiaochu Liu88d90382018-08-29 16:09:11 -0700988 forced_omaha_url_ = constants::kOmahaDefaultAUTestURL;
989 }
990
Jae Hoon Kim2b73ac22019-07-02 11:17:24 -0700991 // |forced_update_pending_callback_| should always be set, but even in the
992 // case that it is not, we still return true indicating success because the
993 // scheduled periodic check will pick up these changes.
994 if (forced_update_pending_callback_.get()) {
995 // Always call |ScheduleUpdates()| before forcing an update. This is because
996 // we need an update to be scheduled for the
997 // |forced_update_pending_callback_| to have an effect. Here we don't need
998 // to care about the return value from |ScheduleUpdate()|.
999 ScheduleUpdates();
1000 forced_update_pending_callback_->Run(true, true);
Xiaochu Liu88d90382018-08-29 16:09:11 -07001001 }
1002 return true;
1003}
1004
Darin Petkov296889c2010-07-23 16:20:54 -07001005bool UpdateAttempter::RebootIfNeeded() {
Amin Hassani538bd592020-11-04 20:46:08 -08001006 if (SystemState::Get()->power_manager()->RequestReboot())
Daniel Erat65f1da02014-06-27 22:05:38 -07001007 return true;
1008
1009 return RebootDirectly();
Darin Petkov296889c2010-07-23 16:20:54 -07001010}
1011
David Zeuthen3c55abd2013-10-14 12:48:03 -07001012void UpdateAttempter::WriteUpdateCompletedMarker() {
Alex Deymo906191f2015-10-12 12:22:44 -07001013 string boot_id;
1014 if (!utils::GetBootId(&boot_id))
David Zeuthen3c55abd2013-10-14 12:48:03 -07001015 return;
Alex Deymo906191f2015-10-12 12:22:44 -07001016 prefs_->SetString(kPrefsUpdateCompletedOnBootId, boot_id);
David Zeuthen3c55abd2013-10-14 12:48:03 -07001017
Amin Hassani538bd592020-11-04 20:46:08 -08001018 int64_t value = SystemState::Get()->clock()->GetBootTime().ToInternalValue();
Alex Deymo906191f2015-10-12 12:22:44 -07001019 prefs_->SetInt64(kPrefsUpdateCompletedBootTime, value);
David Zeuthen3c55abd2013-10-14 12:48:03 -07001020}
1021
Daniel Erat65f1da02014-06-27 22:05:38 -07001022bool UpdateAttempter::RebootDirectly() {
Amin Hassani3a4caa12019-11-06 11:12:28 -08001023 vector<string> command = {"/sbin/shutdown", "-r", "now"};
Daniel Erat65f1da02014-06-27 22:05:38 -07001024 int rc = 0;
Amin Hassani3a4caa12019-11-06 11:12:28 -08001025 Subprocess::SynchronousExec(command, &rc, nullptr, nullptr);
Daniel Erat65f1da02014-06-27 22:05:38 -07001026 return rc == 0;
1027}
1028
Gilad Arnoldec7f9162014-07-15 13:24:46 -07001029void UpdateAttempter::OnUpdateScheduled(EvalStatus status,
1030 const UpdateCheckParams& params) {
1031 waiting_for_scheduled_check_ = false;
1032
1033 if (status == EvalStatus::kSucceeded) {
1034 if (!params.updates_enabled) {
1035 LOG(WARNING) << "Updates permanently disabled.";
Gilad Arnold54fa66d2014-09-29 13:14:29 -07001036 // Signal disabled status, then switch right back to idle. This is
1037 // necessary for ensuring that observers waiting for a signal change will
1038 // actually notice one on subsequent calls. Note that we don't need to
1039 // re-schedule a check in this case as updates are permanently disabled;
1040 // further (forced) checks may still initiate a scheduling call.
Christopher Wileycc8ce0e2015-10-01 16:48:47 -07001041 SetStatusAndNotify(UpdateStatus::DISABLED);
1042 SetStatusAndNotify(UpdateStatus::IDLE);
Gilad Arnoldec7f9162014-07-15 13:24:46 -07001043 return;
1044 }
1045
Amin Hassanied37d682018-04-06 13:22:00 -07001046 LOG(INFO) << "Running " << (params.interactive ? "interactive" : "periodic")
Gilad Arnoldec7f9162014-07-15 13:24:46 -07001047 << " update.";
1048
Amin Hassanied37d682018-04-06 13:22:00 -07001049 if (!params.interactive) {
Aaron Wood081c0232017-10-19 17:14:58 -07001050 // Cache the update attempt flags that will be used by this update attempt
1051 // so that they can't be changed mid-way through.
1052 current_update_attempt_flags_ = update_attempt_flags_;
1053 }
1054
Aaron Woodbf5a2522017-10-04 10:58:36 -07001055 LOG(INFO) << "Update attempt flags in use = 0x" << std::hex
1056 << current_update_attempt_flags_;
1057
Amin Hassanie8153632020-10-27 15:11:28 -07001058 Update(params);
Alex Deymo71479082016-03-25 17:54:28 -07001059 // Always clear the forced app_version and omaha_url after an update attempt
1060 // so the next update uses the defaults.
1061 forced_app_version_.clear();
1062 forced_omaha_url_.clear();
Gilad Arnoldec7f9162014-07-15 13:24:46 -07001063 } else {
1064 LOG(WARNING)
1065 << "Update check scheduling failed (possibly timed out); retrying.";
1066 ScheduleUpdates();
1067 }
1068
1069 // This check ensures that future update checks will be or are already
1070 // scheduled. The check should never fail. A check failure means that there's
1071 // a bug that will most likely prevent further automatic update checks. It
1072 // seems better to crash in such cases and restart the update_engine daemon
1073 // into, hopefully, a known good state.
Jae Hoon Kimba2fdce2019-07-11 13:18:58 -07001074 CHECK(IsBusyOrUpdateScheduled());
Gilad Arnoldec7f9162014-07-15 13:24:46 -07001075}
1076
1077void UpdateAttempter::UpdateLastCheckedTime() {
Amin Hassani538bd592020-11-04 20:46:08 -08001078 last_checked_time_ =
1079 SystemState::Get()->clock()->GetWallclockTime().ToTimeT();
Gilad Arnoldec7f9162014-07-15 13:24:46 -07001080}
1081
Marton Hunyadye58bddb2018-04-10 20:27:26 +02001082void UpdateAttempter::UpdateRollbackHappened() {
Amin Hassani538bd592020-11-04 20:46:08 -08001083 DCHECK(SystemState::Get()->payload_state());
Marton Hunyadye58bddb2018-04-10 20:27:26 +02001084 DCHECK(policy_provider_);
Amin Hassani538bd592020-11-04 20:46:08 -08001085 if (SystemState::Get()->payload_state()->GetRollbackHappened() &&
Marton Hunyadye58bddb2018-04-10 20:27:26 +02001086 (policy_provider_->device_policy_is_loaded() ||
1087 policy_provider_->IsConsumerDevice())) {
1088 // Rollback happened, but we already went through OOBE and policy is
1089 // present or it's a consumer device.
Amin Hassani538bd592020-11-04 20:46:08 -08001090 SystemState::Get()->payload_state()->SetRollbackHappened(false);
Marton Hunyadye58bddb2018-04-10 20:27:26 +02001091 }
1092}
1093
Jae Hoon Kimed3fcc02019-07-11 14:35:38 -07001094void UpdateAttempter::ProcessingDoneInternal(const ActionProcessor* processor,
1095 ErrorCode code) {
Chris Sosa4f8ee272012-11-30 13:01:54 -08001096 // Reset cpu shares back to normal.
Alex Deymoab0d9762016-02-02 10:52:56 -08001097 cpu_limiter_.StopLimiter();
Darin Petkovc6c135c2010-08-11 13:36:18 -07001098
Amin Hassaniecb60d32019-06-17 18:09:10 -07001099 ResetInteractivityFlags();
Adolfo Victoria20262ad2018-08-06 10:40:11 -07001100
Christopher Wileycc8ce0e2015-10-01 16:48:47 -07001101 if (status_ == UpdateStatus::REPORTING_ERROR_EVENT) {
Darin Petkov09f96c32010-07-20 09:24:57 -07001102 LOG(INFO) << "Error event sent.";
Gilad Arnold1ebd8132012-03-05 10:19:29 -08001103
Amin Hassaniecb60d32019-06-17 18:09:10 -07001104 // Inform scheduler of new status.
Christopher Wileycc8ce0e2015-10-01 16:48:47 -07001105 SetStatusAndNotify(UpdateStatus::IDLE);
Gilad Arnold54fa66d2014-09-29 13:14:29 -07001106 ScheduleUpdates();
Gilad Arnold1ebd8132012-03-05 10:19:29 -08001107
Andrew de los Reyesc1d5c932011-04-20 17:15:47 -07001108 if (!fake_update_success_) {
1109 return;
1110 }
1111 LOG(INFO) << "Booted from FW B and tried to install new firmware, "
Amin Hassani7cc8bb02019-01-14 16:29:47 -08001112 "so requesting reboot from user.";
Darin Petkov09f96c32010-07-20 09:24:57 -07001113 }
1114
Sen Jiang89e24c12018-03-22 18:05:44 -07001115 attempt_error_code_ = utils::GetBaseErrorCode(code);
1116
Jae Hoon Kimed3fcc02019-07-11 14:35:38 -07001117 if (code != ErrorCode::kSuccess) {
1118 if (ScheduleErrorEventAction()) {
Colin Howesac170d92018-11-20 16:29:28 -08001119 return;
1120 }
Jae Hoon Kimed3fcc02019-07-11 14:35:38 -07001121 LOG(INFO) << "No update.";
1122 SetStatusAndNotify(UpdateStatus::IDLE);
Gilad Arnold54fa66d2014-09-29 13:14:29 -07001123 ScheduleUpdates();
Darin Petkov09f96c32010-07-20 09:24:57 -07001124 return;
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -07001125 }
Darin Petkov09f96c32010-07-20 09:24:57 -07001126
Jae Hoon Kimed3fcc02019-07-11 14:35:38 -07001127 ReportTimeToUpdateAppliedMetric();
1128 prefs_->SetInt64(kPrefsDeltaUpdateFailures, 0);
1129 prefs_->SetString(kPrefsPreviousVersion,
1130 omaha_request_params_->app_version());
1131 DeltaPerformer::ResetUpdateProgress(prefs_, false);
1132
Amin Hassani538bd592020-11-04 20:46:08 -08001133 SystemState::Get()->payload_state()->UpdateSucceeded();
Jae Hoon Kimed3fcc02019-07-11 14:35:38 -07001134
1135 // Since we're done with scattering fully at this point, this is the
1136 // safest point delete the state files, as we're sure that the status is
1137 // set to reboot (which means no more updates will be applied until reboot)
1138 // This deletion is required for correctness as we want the next update
1139 // check to re-create a new random number for the update check count.
1140 // Similarly, we also delete the wall-clock-wait period that was persisted
1141 // so that we start with a new random value for the next update check
1142 // after reboot so that the same device is not favored or punished in any
1143 // way.
1144 prefs_->Delete(kPrefsUpdateCheckCount);
Amin Hassani538bd592020-11-04 20:46:08 -08001145 SystemState::Get()->payload_state()->SetScatteringWaitPeriod(TimeDelta());
1146 SystemState::Get()->payload_state()->SetStagingWaitPeriod(TimeDelta());
Jae Hoon Kimed3fcc02019-07-11 14:35:38 -07001147 prefs_->Delete(kPrefsUpdateFirstSeenAt);
1148
1149 // Note: below this comment should only be on |ErrorCode::kSuccess|.
1150 if (is_install_) {
1151 ProcessingDoneInstall(processor, code);
1152 } else {
1153 ProcessingDoneUpdate(processor, code);
Darin Petkov1023a602010-08-30 13:47:51 -07001154 }
Jae Hoon Kimed3fcc02019-07-11 14:35:38 -07001155}
1156
Jae Hoon Kim7fdfbf12020-04-10 18:15:50 -07001157vector<string> UpdateAttempter::GetSuccessfulDlcIds() {
1158 vector<string> dlc_ids;
1159 for (const auto& pr : omaha_request_params_->dlc_apps_params())
1160 if (pr.second.updated)
1161 dlc_ids.push_back(pr.second.name);
1162 return dlc_ids;
1163}
1164
Jae Hoon Kimed3fcc02019-07-11 14:35:38 -07001165void UpdateAttempter::ProcessingDoneInstall(const ActionProcessor* processor,
1166 ErrorCode code) {
Amin Hassani538bd592020-11-04 20:46:08 -08001167 if (!SystemState::Get()->dlcservice()->InstallCompleted(
1168 GetSuccessfulDlcIds()))
Jae Hoon Kim7fdfbf12020-04-10 18:15:50 -07001169 LOG(WARNING) << "dlcservice didn't successfully handle install completion.";
Christopher Wileycc8ce0e2015-10-01 16:48:47 -07001170 SetStatusAndNotify(UpdateStatus::IDLE);
Gilad Arnold54fa66d2014-09-29 13:14:29 -07001171 ScheduleUpdates();
Jae Hoon Kimed3fcc02019-07-11 14:35:38 -07001172 LOG(INFO) << "DLC successfully installed, no reboot needed.";
1173}
1174
1175void UpdateAttempter::ProcessingDoneUpdate(const ActionProcessor* processor,
1176 ErrorCode code) {
1177 WriteUpdateCompletedMarker();
1178
Amin Hassani538bd592020-11-04 20:46:08 -08001179 if (!SystemState::Get()->dlcservice()->UpdateCompleted(GetSuccessfulDlcIds()))
Jae Hoon Kim7fdfbf12020-04-10 18:15:50 -07001180 LOG(WARNING) << "dlcservice didn't successfully handle update completion.";
Jae Hoon Kimed3fcc02019-07-11 14:35:38 -07001181 SetStatusAndNotify(UpdateStatus::UPDATED_NEED_REBOOT);
1182 ScheduleUpdates();
1183 LOG(INFO) << "Update successfully applied, waiting to reboot.";
1184
1185 // |install_plan_| is null during rollback operations, and the stats don't
1186 // make much sense then anyway.
1187 if (install_plan_) {
1188 // Generate an unique payload identifier.
1189 string target_version_uid;
1190 for (const auto& payload : install_plan_->payloads) {
1191 target_version_uid += brillo::data_encoding::Base64Encode(payload.hash) +
1192 ":" + payload.metadata_signature + ":";
Vyshu Khota4c5413d2020-11-04 16:17:25 -08001193 // Set fingerprint value for updates only.
1194 if (!is_install_)
1195 SetPref(kPrefsLastFp, payload.fp, payload.app_id);
Jae Hoon Kimed3fcc02019-07-11 14:35:38 -07001196 }
1197
1198 // If we just downloaded a rollback image, we should preserve this fact
1199 // over the following powerwash.
1200 if (install_plan_->is_rollback) {
Amin Hassani538bd592020-11-04 20:46:08 -08001201 SystemState::Get()->payload_state()->SetRollbackHappened(true);
1202 SystemState::Get()->metrics_reporter()->ReportEnterpriseRollbackMetrics(
Jae Hoon Kimed3fcc02019-07-11 14:35:38 -07001203 /*success=*/true, install_plan_->version);
1204 }
1205
1206 // Expect to reboot into the new version to send the proper metric during
1207 // next boot.
Amin Hassani538bd592020-11-04 20:46:08 -08001208 SystemState::Get()->payload_state()->ExpectRebootInNewVersion(
Jae Hoon Kimed3fcc02019-07-11 14:35:38 -07001209 target_version_uid);
1210 } else {
1211 // If we just finished a rollback, then we expect to have no Omaha
1212 // response. Otherwise, it's an error.
Amin Hassani538bd592020-11-04 20:46:08 -08001213 if (SystemState::Get()->payload_state()->GetRollbackVersion().empty()) {
Jae Hoon Kimed3fcc02019-07-11 14:35:38 -07001214 LOG(ERROR) << "Can't send metrics because there was no Omaha response";
1215 }
1216 }
1217}
1218
1219// Delegate methods:
1220void UpdateAttempter::ProcessingDone(const ActionProcessor* processor,
1221 ErrorCode code) {
1222 LOG(INFO) << "Processing Done.";
1223 ProcessingDoneInternal(processor, code);
1224
1225 // Note: do cleanups here for any variables that need to be reset after a
1226 // failure, error, update, or install.
1227 is_install_ = false;
Andrew de los Reyes63b96d72010-05-10 13:08:54 -07001228}
1229
1230void UpdateAttempter::ProcessingStopped(const ActionProcessor* processor) {
Chris Sosa4f8ee272012-11-30 13:01:54 -08001231 // Reset cpu shares back to normal.
Alex Deymoab0d9762016-02-02 10:52:56 -08001232 cpu_limiter_.StopLimiter();
Andrew de los Reyes63b96d72010-05-10 13:08:54 -07001233 download_progress_ = 0.0;
Amin Hassaniecb60d32019-06-17 18:09:10 -07001234
1235 ResetInteractivityFlags();
1236
Christopher Wileycc8ce0e2015-10-01 16:48:47 -07001237 SetStatusAndNotify(UpdateStatus::IDLE);
Gilad Arnold54fa66d2014-09-29 13:14:29 -07001238 ScheduleUpdates();
Alex Vakulenko88b591f2014-08-28 16:48:57 -07001239 error_event_.reset(nullptr);
Andrew de los Reyes63b96d72010-05-10 13:08:54 -07001240}
1241
1242// Called whenever an action has finished processing, either successfully
1243// or otherwise.
1244void UpdateAttempter::ActionCompleted(ActionProcessor* processor,
1245 AbstractAction* action,
David Zeuthena99981f2013-04-29 13:42:47 -07001246 ErrorCode code) {
Darin Petkov1023a602010-08-30 13:47:51 -07001247 // Reset download progress regardless of whether or not the download
1248 // action succeeded. Also, get the response code from HTTP request
1249 // actions (update download as well as the initial update check
1250 // actions).
Andrew de los Reyes63b96d72010-05-10 13:08:54 -07001251 const string type = action->Type();
Darin Petkov1023a602010-08-30 13:47:51 -07001252 if (type == DownloadAction::StaticType()) {
Andrew de los Reyes63b96d72010-05-10 13:08:54 -07001253 download_progress_ = 0.0;
Gilad Arnoldcf175a02014-07-10 16:48:47 -07001254 DownloadAction* download_action = static_cast<DownloadAction*>(action);
Darin Petkov1023a602010-08-30 13:47:51 -07001255 http_response_code_ = download_action->GetHTTPResponseCode();
1256 } else if (type == OmahaRequestAction::StaticType()) {
1257 OmahaRequestAction* omaha_request_action =
Gilad Arnoldcf175a02014-07-10 16:48:47 -07001258 static_cast<OmahaRequestAction*>(action);
Darin Petkov1023a602010-08-30 13:47:51 -07001259 // If the request is not an event, then it's the update-check.
1260 if (!omaha_request_action->IsEvent()) {
1261 http_response_code_ = omaha_request_action->GetHTTPResponseCode();
Gilad Arnolda6dab942014-04-25 11:46:03 -07001262
1263 // Record the number of consecutive failed update checks.
1264 if (http_response_code_ == kHttpResponseInternalServerError ||
1265 http_response_code_ == kHttpResponseServiceUnavailable) {
1266 consecutive_failed_update_checks_++;
1267 } else {
1268 consecutive_failed_update_checks_ = 0;
1269 }
1270
Weidong Guo421ff332017-04-17 10:08:38 -07001271 const OmahaResponse& omaha_response =
1272 omaha_request_action->GetOutputObject();
Gilad Arnolda0258a52014-07-10 16:21:19 -07001273 // Store the server-dictated poll interval, if any.
1274 server_dictated_poll_interval_ =
Weidong Guo421ff332017-04-17 10:08:38 -07001275 std::max(0, omaha_response.poll_interval);
1276
1277 // This update is ignored by omaha request action because update over
1278 // cellular connection is not allowed. Needs to ask for user's permissions
1279 // to update.
1280 if (code == ErrorCode::kOmahaUpdateIgnoredOverCellular) {
1281 new_version_ = omaha_response.version;
1282 new_payload_size_ = 0;
1283 for (const auto& package : omaha_response.packages) {
1284 new_payload_size_ += package.size;
1285 }
1286 SetStatusAndNotify(UpdateStatus::NEED_PERMISSION_TO_UPDATE);
1287 }
Darin Petkov1023a602010-08-30 13:47:51 -07001288 }
Aaron Wood23bd3392017-10-06 14:48:25 -07001289 } else if (type == OmahaResponseHandlerAction::StaticType()) {
1290 // Depending on the returned error code, note that an update is available.
1291 if (code == ErrorCode::kOmahaUpdateDeferredPerPolicy ||
1292 code == ErrorCode::kSuccess) {
1293 // Note that the status will be updated to DOWNLOADING when some bytes
1294 // get actually downloaded from the server and the BytesReceived
1295 // callback is invoked. This avoids notifying the user that a download
1296 // has started in cases when the server and the client are unable to
1297 // initiate the download.
Amin Hassanid3f4bea2018-04-30 14:52:40 -07001298 auto omaha_response_handler_action =
1299 static_cast<OmahaResponseHandlerAction*>(action);
1300 install_plan_.reset(
1301 new InstallPlan(omaha_response_handler_action->install_plan()));
Aaron Wood23bd3392017-10-06 14:48:25 -07001302 UpdateLastCheckedTime();
Amin Hassanid3f4bea2018-04-30 14:52:40 -07001303 new_version_ = install_plan_->version;
Aaron Wood23bd3392017-10-06 14:48:25 -07001304 new_payload_size_ = 0;
Amin Hassanid3f4bea2018-04-30 14:52:40 -07001305 for (const auto& payload : install_plan_->payloads)
Aaron Wood23bd3392017-10-06 14:48:25 -07001306 new_payload_size_ += payload.size;
1307 cpu_limiter_.StartLimiter();
1308 SetStatusAndNotify(UpdateStatus::UPDATE_AVAILABLE);
1309 }
Darin Petkov1023a602010-08-30 13:47:51 -07001310 }
Aaron Wood23bd3392017-10-06 14:48:25 -07001311 // General failure cases.
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001312 if (code != ErrorCode::kSuccess) {
Darin Petkov7ed561b2011-10-04 02:59:03 -07001313 // If the current state is at or past the download phase, count the failure
1314 // in case a switch to full update becomes necessary. Ignore network
1315 // transfer timeouts and failures.
Amin Hassani89a37122018-03-23 12:59:24 -07001316 if (code != ErrorCode::kDownloadTransferError) {
1317 switch (status_) {
1318 case UpdateStatus::IDLE:
1319 case UpdateStatus::CHECKING_FOR_UPDATE:
1320 case UpdateStatus::UPDATE_AVAILABLE:
1321 case UpdateStatus::NEED_PERMISSION_TO_UPDATE:
1322 break;
1323 case UpdateStatus::DOWNLOADING:
1324 case UpdateStatus::VERIFYING:
1325 case UpdateStatus::FINALIZING:
1326 case UpdateStatus::UPDATED_NEED_REBOOT:
1327 case UpdateStatus::REPORTING_ERROR_EVENT:
1328 case UpdateStatus::ATTEMPTING_ROLLBACK:
1329 case UpdateStatus::DISABLED:
Amin Hassani70a90f52020-09-15 15:30:09 -07001330 case UpdateStatus::CLEANUP_PREVIOUS_UPDATE:
Amin Hassani89a37122018-03-23 12:59:24 -07001331 MarkDeltaUpdateFailure();
1332 break;
1333 }
Darin Petkov36275772010-10-01 11:40:57 -07001334 }
Sen Jiang89e24c12018-03-22 18:05:44 -07001335 if (code != ErrorCode::kNoUpdate) {
1336 // On failure, schedule an error event to be sent to Omaha.
1337 CreatePendingErrorEvent(action, code);
1338 }
Andrew de los Reyes63b96d72010-05-10 13:08:54 -07001339 return;
Darin Petkov09f96c32010-07-20 09:24:57 -07001340 }
Aaron Wood23bd3392017-10-06 14:48:25 -07001341 // Find out which action completed (successfully).
1342 if (type == DownloadAction::StaticType()) {
Christopher Wileycc8ce0e2015-10-01 16:48:47 -07001343 SetStatusAndNotify(UpdateStatus::FINALIZING);
Amin Hassaniafd8cea2017-12-04 14:20:00 -08001344 } else if (type == FilesystemVerifierAction::StaticType()) {
1345 // Log the system properties before the postinst and after the file system
1346 // is verified. It used to be done in the postinst itself. But postinst
1347 // cannot do this anymore. On the other hand, these logs are frequently
1348 // looked at and it is preferable not to scatter them in random location in
1349 // the log and rather log it right before the postinst. The reason not do
1350 // this in the |PostinstallRunnerAction| is to prevent dependency from
1351 // libpayload_consumer to libupdate_engine.
1352 LogImageProperties();
Andrew de los Reyes63b96d72010-05-10 13:08:54 -07001353 }
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -07001354}
1355
Alex Deymo542c19b2015-12-03 07:43:31 -03001356void UpdateAttempter::BytesReceived(uint64_t bytes_progressed,
1357 uint64_t bytes_received,
1358 uint64_t total) {
1359 // The PayloadState keeps track of how many bytes were actually downloaded
1360 // from a given URL for the URL skipping logic.
Amin Hassani538bd592020-11-04 20:46:08 -08001361 SystemState::Get()->payload_state()->DownloadProgress(bytes_progressed);
Alex Deymo542c19b2015-12-03 07:43:31 -03001362
Alex Deymo0d298542016-03-30 18:31:49 -07001363 double progress = 0;
1364 if (total)
1365 progress = static_cast<double>(bytes_received) / static_cast<double>(total);
1366 if (status_ != UpdateStatus::DOWNLOADING || bytes_received == total) {
Darin Petkovaf183052010-08-23 12:07:13 -07001367 download_progress_ = progress;
Christopher Wileycc8ce0e2015-10-01 16:48:47 -07001368 SetStatusAndNotify(UpdateStatus::DOWNLOADING);
Alex Deymo0d298542016-03-30 18:31:49 -07001369 } else {
1370 ProgressUpdate(progress);
Andrew de los Reyes63b96d72010-05-10 13:08:54 -07001371 }
1372}
1373
Alex Deymo542c19b2015-12-03 07:43:31 -03001374void UpdateAttempter::DownloadComplete() {
Amin Hassani538bd592020-11-04 20:46:08 -08001375 SystemState::Get()->payload_state()->DownloadComplete();
Alex Deymo542c19b2015-12-03 07:43:31 -03001376}
1377
Alex Deymo0d298542016-03-30 18:31:49 -07001378void UpdateAttempter::ProgressUpdate(double progress) {
1379 // Self throttle based on progress. Also send notifications if progress is
1380 // too slow.
1381 if (progress == 1.0 ||
1382 progress - download_progress_ >= kBroadcastThresholdProgress ||
1383 TimeTicks::Now() - last_notify_time_ >=
1384 TimeDelta::FromSeconds(kBroadcastThresholdSeconds)) {
1385 download_progress_ = progress;
1386 BroadcastStatus();
1387 }
1388}
1389
Amin Hassaniecb60d32019-06-17 18:09:10 -07001390void UpdateAttempter::ResetInteractivityFlags() {
1391 // Reset the state that's only valid for a single update pass.
1392 current_update_attempt_flags_ = UpdateAttemptFlags::kNone;
1393
1394 if (forced_update_pending_callback_.get())
1395 // Clear prior interactive requests once the processor is done.
1396 forced_update_pending_callback_->Run(false, false);
1397}
1398
Jay Srinivasanc1ba09a2012-08-14 14:15:57 -07001399bool UpdateAttempter::ResetStatus() {
1400 LOG(INFO) << "Attempting to reset state from "
Christopher Wileycc8ce0e2015-10-01 16:48:47 -07001401 << UpdateStatusToString(status_) << " to UpdateStatus::IDLE";
Jay Srinivasanc1ba09a2012-08-14 14:15:57 -07001402
1403 switch (status_) {
Christopher Wileycc8ce0e2015-10-01 16:48:47 -07001404 case UpdateStatus::IDLE:
Jay Srinivasanc1ba09a2012-08-14 14:15:57 -07001405 // no-op.
1406 return true;
1407
Amin Hassani7cc8bb02019-01-14 16:29:47 -08001408 case UpdateStatus::UPDATED_NEED_REBOOT: {
Jay Srinivasan1c0fe792013-03-28 16:45:25 -07001409 bool ret_value = true;
Christopher Wileycc8ce0e2015-10-01 16:48:47 -07001410 status_ = UpdateStatus::IDLE;
Jay Srinivasanc1ba09a2012-08-14 14:15:57 -07001411
Jay Srinivasan1c0fe792013-03-28 16:45:25 -07001412 // Remove the reboot marker so that if the machine is rebooted
Jay Srinivasanc1ba09a2012-08-14 14:15:57 -07001413 // after resetting to idle state, it doesn't go back to
Christopher Wileycc8ce0e2015-10-01 16:48:47 -07001414 // UpdateStatus::UPDATED_NEED_REBOOT state.
Alex Deymo906191f2015-10-12 12:22:44 -07001415 ret_value = prefs_->Delete(kPrefsUpdateCompletedOnBootId) && ret_value;
1416 ret_value = prefs_->Delete(kPrefsUpdateCompletedBootTime) && ret_value;
Vyshu Khota4c5413d2020-11-04 16:17:25 -08001417 ret_value = prefs_->Delete(kPrefsLastFp, {kDlcPrefsSubDir}) && ret_value;
Jay Srinivasan1c0fe792013-03-28 16:45:25 -07001418
Alex Deymo9870c0e2015-09-23 13:58:31 -07001419 // Update the boot flags so the current slot has higher priority.
Amin Hassani538bd592020-11-04 20:46:08 -08001420 BootControlInterface* boot_control = SystemState::Get()->boot_control();
Alex Deymo9870c0e2015-09-23 13:58:31 -07001421 if (!boot_control->SetActiveBootSlot(boot_control->GetCurrentSlot()))
1422 ret_value = false;
1423
Alex Deymo52590332016-11-29 18:29:13 -08001424 // Mark the current slot as successful again, since marking it as active
1425 // may reset the successful bit. We ignore the result of whether marking
1426 // the current slot as successful worked.
Amin Hassani7cc8bb02019-01-14 16:29:47 -08001427 if (!boot_control->MarkBootSuccessfulAsync(Bind([](bool successful) {})))
Alex Deymo52590332016-11-29 18:29:13 -08001428 ret_value = false;
1429
Alex Deymo42432912013-07-12 20:21:15 -07001430 // Notify the PayloadState that the successful payload was canceled.
Amin Hassani538bd592020-11-04 20:46:08 -08001431 SystemState::Get()->payload_state()->ResetUpdateStatus();
Alex Deymo42432912013-07-12 20:21:15 -07001432
Alex Deymo87c08862015-10-30 21:56:55 -07001433 // The previous version is used to report back to omaha after reboot that
1434 // we actually rebooted into the new version from this "prev-version". We
1435 // need to clear out this value now to prevent it being sent on the next
1436 // updatecheck request.
1437 ret_value = prefs_->SetString(kPrefsPreviousVersion, "") && ret_value;
1438
Alex Deymo906191f2015-10-12 12:22:44 -07001439 LOG(INFO) << "Reset status " << (ret_value ? "successful" : "failed");
Jay Srinivasan1c0fe792013-03-28 16:45:25 -07001440 return ret_value;
Jay Srinivasanc1ba09a2012-08-14 14:15:57 -07001441 }
1442
1443 default:
1444 LOG(ERROR) << "Reset not allowed in this state.";
1445 return false;
1446 }
1447}
1448
Aaron Wood7f92e2b2017-08-28 14:51:21 -07001449bool UpdateAttempter::GetStatus(UpdateEngineStatus* out_status) {
Aaron Wood795c5b42017-12-05 16:06:13 -08001450 out_status->last_checked_time = last_checked_time_;
Aaron Wood7f92e2b2017-08-28 14:51:21 -07001451 out_status->status = status_;
1452 out_status->current_version = omaha_request_params_->app_version();
Aaron Wood7f92e2b2017-08-28 14:51:21 -07001453 out_status->progress = download_progress_;
1454 out_status->new_size_bytes = new_payload_size_;
1455 out_status->new_version = new_version_;
Amin Hassani9be122e2019-08-29 09:20:12 -07001456 out_status->is_enterprise_rollback =
1457 install_plan_ && install_plan_->is_rollback;
Jae Hoon Kim2f78c1c2019-07-25 13:20:43 -07001458 out_status->is_install = is_install_;
Jae Hoon Kim051627a2019-09-03 12:56:32 -07001459
1460 string str_eol_date;
Amin Hassani538bd592020-11-04 20:46:08 -08001461 if (SystemState::Get()->prefs()->Exists(kPrefsOmahaEolDate) &&
1462 !SystemState::Get()->prefs()->GetString(kPrefsOmahaEolDate,
1463 &str_eol_date))
Jae Hoon Kim2dfd35d2020-06-02 10:53:13 -07001464 LOG(ERROR) << "Failed to retrieve kPrefsOmahaEolDate pref.";
Jae Hoon Kim051627a2019-09-03 12:56:32 -07001465 out_status->eol_date = StringToEolDate(str_eol_date);
1466
Miriam Polzer0cf1acb2020-04-29 17:39:51 +02001467 // A powerwash will take place either if the install plan says it is required
1468 // or if an enterprise rollback is happening.
1469 out_status->will_powerwash_after_reboot =
1470 install_plan_ &&
1471 (install_plan_->powerwash_required || install_plan_->is_rollback);
1472
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -07001473 return true;
1474}
1475
Darin Petkov61635a92011-05-18 16:20:36 -07001476void UpdateAttempter::BroadcastStatus() {
Aaron Wood7f92e2b2017-08-28 14:51:21 -07001477 UpdateEngineStatus broadcast_status;
1478 // Use common method for generating the current status.
1479 GetStatus(&broadcast_status);
1480
Alex Deymofa78f142016-01-26 21:36:16 -08001481 for (const auto& observer : service_observers_) {
Aaron Wood7f92e2b2017-08-28 14:51:21 -07001482 observer->SendStatusUpdate(broadcast_status);
Alex Deymofa78f142016-01-26 21:36:16 -08001483 }
Darin Petkovaf183052010-08-23 12:07:13 -07001484 last_notify_time_ = TimeTicks::Now();
Alex Deymofa78f142016-01-26 21:36:16 -08001485}
1486
Amin Hassani7cc8bb02019-01-14 16:29:47 -08001487uint32_t UpdateAttempter::GetErrorCodeFlags() {
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001488 uint32_t flags = 0;
1489
Amin Hassani538bd592020-11-04 20:46:08 -08001490 if (!SystemState::Get()->hardware()->IsNormalBootMode())
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001491 flags |= static_cast<uint32_t>(ErrorCode::kDevModeFlag);
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001492
Amin Hassanid3f4bea2018-04-30 14:52:40 -07001493 if (install_plan_ && install_plan_->is_resume)
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001494 flags |= static_cast<uint32_t>(ErrorCode::kResumedFlag);
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001495
Amin Hassani538bd592020-11-04 20:46:08 -08001496 if (!SystemState::Get()->hardware()->IsOfficialBuild())
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001497 flags |= static_cast<uint32_t>(ErrorCode::kTestImageFlag);
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001498
Sen Jiangdaeaa432018-10-09 18:18:45 -07001499 if (!omaha_request_params_->IsUpdateUrlOfficial()) {
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001500 flags |= static_cast<uint32_t>(ErrorCode::kTestOmahaUrlFlag);
Alex Deymoac41a822015-09-15 20:52:53 -07001501 }
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001502
1503 return flags;
1504}
1505
David Zeuthena99981f2013-04-29 13:42:47 -07001506bool UpdateAttempter::ShouldCancel(ErrorCode* cancel_reason) {
Jay Srinivasan1c0fe792013-03-28 16:45:25 -07001507 // Check if the channel we're attempting to update to is the same as the
1508 // target channel currently chosen by the user.
Amin Hassani538bd592020-11-04 20:46:08 -08001509 OmahaRequestParams* params = SystemState::Get()->request_params();
Jay Srinivasan1c0fe792013-03-28 16:45:25 -07001510 if (params->download_channel() != params->target_channel()) {
1511 LOG(ERROR) << "Aborting download as target channel: "
1512 << params->target_channel()
1513 << " is different from the download channel: "
1514 << params->download_channel();
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001515 *cancel_reason = ErrorCode::kUpdateCanceledByChannelChange;
Jay Srinivasan1c0fe792013-03-28 16:45:25 -07001516 return true;
1517 }
1518
1519 return false;
1520}
1521
Nam T. Nguyen7d623eb2014-05-13 16:06:28 -07001522void UpdateAttempter::SetStatusAndNotify(UpdateStatus status) {
Darin Petkov61635a92011-05-18 16:20:36 -07001523 status_ = status;
Darin Petkov61635a92011-05-18 16:20:36 -07001524 BroadcastStatus();
1525}
1526
Darin Petkov777dbfa2010-07-20 15:03:37 -07001527void UpdateAttempter::CreatePendingErrorEvent(AbstractAction* action,
David Zeuthena99981f2013-04-29 13:42:47 -07001528 ErrorCode code) {
Sen Jiang89e24c12018-03-22 18:05:44 -07001529 if (error_event_.get() || status_ == UpdateStatus::REPORTING_ERROR_EVENT) {
Darin Petkov09f96c32010-07-20 09:24:57 -07001530 // This shouldn't really happen.
1531 LOG(WARNING) << "There's already an existing pending error event.";
1532 return;
1533 }
Darin Petkov777dbfa2010-07-20 15:03:37 -07001534
Jay Srinivasan56d5aa42012-03-26 14:27:59 -07001535 // Classify the code to generate the appropriate result so that
1536 // the Borgmon charts show up the results correctly.
1537 // Do this before calling GetErrorCodeForAction which could potentially
1538 // augment the bit representation of code and thus cause no matches for
1539 // the switch cases below.
1540 OmahaEvent::Result event_result;
1541 switch (code) {
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001542 case ErrorCode::kOmahaUpdateIgnoredPerPolicy:
1543 case ErrorCode::kOmahaUpdateDeferredPerPolicy:
1544 case ErrorCode::kOmahaUpdateDeferredForBackoff:
Jay Srinivasan56d5aa42012-03-26 14:27:59 -07001545 event_result = OmahaEvent::kResultUpdateDeferred;
1546 break;
1547 default:
1548 event_result = OmahaEvent::kResultError;
1549 break;
1550 }
1551
Darin Petkov777dbfa2010-07-20 15:03:37 -07001552 code = GetErrorCodeForAction(action, code);
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001553 fake_update_success_ = code == ErrorCode::kPostinstallBootedFromFirmwareB;
Darin Petkov18c7bce2011-06-16 14:07:00 -07001554
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001555 // Compute the final error code with all the bit flags to be sent to Omaha.
Amin Hassani7cc8bb02019-01-14 16:29:47 -08001556 code =
1557 static_cast<ErrorCode>(static_cast<uint32_t>(code) | GetErrorCodeFlags());
1558 error_event_.reset(
1559 new OmahaEvent(OmahaEvent::kTypeUpdateComplete, event_result, code));
Darin Petkov09f96c32010-07-20 09:24:57 -07001560}
1561
1562bool UpdateAttempter::ScheduleErrorEventAction() {
Alex Vakulenko88b591f2014-08-28 16:48:57 -07001563 if (error_event_.get() == nullptr)
Darin Petkov09f96c32010-07-20 09:24:57 -07001564 return false;
1565
Jay Srinivasan6f6ea002012-12-14 11:26:28 -08001566 LOG(ERROR) << "Update failed.";
Amin Hassani538bd592020-11-04 20:46:08 -08001567 SystemState::Get()->payload_state()->UpdateFailed(error_event_->error_code);
Jay Srinivasan6f6ea002012-12-14 11:26:28 -08001568
Marton Hunyadya0302682018-05-16 18:52:13 +02001569 // Send metrics if it was a rollback.
Amin Hassanid3f4bea2018-04-30 14:52:40 -07001570 if (install_plan_ && install_plan_->is_rollback) {
Amin Hassani538bd592020-11-04 20:46:08 -08001571 SystemState::Get()->metrics_reporter()->ReportEnterpriseRollbackMetrics(
Amin Hassanid3f4bea2018-04-30 14:52:40 -07001572 /*success=*/false, install_plan_->version);
Marton Hunyadya0302682018-05-16 18:52:13 +02001573 }
1574
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001575 // Send it to Omaha.
Alex Deymoaf9a8632015-09-23 18:51:48 -07001576 LOG(INFO) << "Reporting the error event";
Amin Hassanid3f4bea2018-04-30 14:52:40 -07001577 auto error_event_action = std::make_unique<OmahaRequestAction>(
Amin Hassanid3f4bea2018-04-30 14:52:40 -07001578 error_event_.release(), // Pass ownership.
1579 std::make_unique<LibcurlHttpFetcher>(GetProxyResolver(),
Amin Hassani538bd592020-11-04 20:46:08 -08001580 SystemState::Get()->hardware()),
Jae Hoon Kimedb65502019-06-14 11:52:17 -07001581 false,
1582 session_id_);
Amin Hassanid3f4bea2018-04-30 14:52:40 -07001583 processor_->EnqueueAction(std::move(error_event_action));
Christopher Wileycc8ce0e2015-10-01 16:48:47 -07001584 SetStatusAndNotify(UpdateStatus::REPORTING_ERROR_EVENT);
Darin Petkovf42cc1c2010-09-01 09:03:02 -07001585 processor_->StartProcessing();
Darin Petkov09f96c32010-07-20 09:24:57 -07001586 return true;
1587}
1588
Darin Petkov58dd1342011-05-06 12:05:13 -07001589void UpdateAttempter::ScheduleProcessingStart() {
1590 LOG(INFO) << "Scheduling an action processor start.";
Alex Deymo60ca1a72015-06-18 18:19:15 -07001591 MessageLoop::current()->PostTask(
1592 FROM_HERE,
Luis Hector Chavezf1cf3482016-07-19 14:29:19 -07001593 Bind([](ActionProcessor* processor) { processor->StartProcessing(); },
1594 base::Unretained(processor_.get())));
Darin Petkov58dd1342011-05-06 12:05:13 -07001595}
1596
Darin Petkov36275772010-10-01 11:40:57 -07001597void UpdateAttempter::DisableDeltaUpdateIfNeeded() {
1598 int64_t delta_failures;
Jay Srinivasanae4697c2013-03-18 17:08:08 -07001599 if (omaha_request_params_->delta_okay() &&
Darin Petkov36275772010-10-01 11:40:57 -07001600 prefs_->GetInt64(kPrefsDeltaUpdateFailures, &delta_failures) &&
1601 delta_failures >= kMaxDeltaUpdateFailures) {
1602 LOG(WARNING) << "Too many delta update failures, forcing full update.";
Jay Srinivasanae4697c2013-03-18 17:08:08 -07001603 omaha_request_params_->set_delta_okay(false);
Darin Petkov36275772010-10-01 11:40:57 -07001604 }
1605}
1606
1607void UpdateAttempter::MarkDeltaUpdateFailure() {
Darin Petkov2dd01092010-10-08 15:43:05 -07001608 // Don't try to resume a failed delta update.
1609 DeltaPerformer::ResetUpdateProgress(prefs_, false);
Darin Petkov36275772010-10-01 11:40:57 -07001610 int64_t delta_failures;
1611 if (!prefs_->GetInt64(kPrefsDeltaUpdateFailures, &delta_failures) ||
1612 delta_failures < 0) {
1613 delta_failures = 0;
1614 }
1615 prefs_->SetInt64(kPrefsDeltaUpdateFailures, ++delta_failures);
1616}
1617
Thieu Le116fda32011-04-19 11:01:54 -07001618void UpdateAttempter::PingOmaha() {
Thieu Led88a8572011-05-26 09:09:19 -07001619 if (!processor_->IsRunning()) {
Amin Hassaniecb60d32019-06-17 18:09:10 -07001620 ResetInteractivityFlags();
1621
Amin Hassanid3f4bea2018-04-30 14:52:40 -07001622 auto ping_action = std::make_unique<OmahaRequestAction>(
Alex Deymoc1c17b42015-11-23 03:53:15 -03001623 nullptr,
Ben Chanab5a0af2017-10-12 14:57:50 -07001624 std::make_unique<LibcurlHttpFetcher>(GetProxyResolver(),
Amin Hassani538bd592020-11-04 20:46:08 -08001625 SystemState::Get()->hardware()),
Jae Hoon Kimedb65502019-06-14 11:52:17 -07001626 true,
1627 "" /* session_id */);
Alex Vakulenko88b591f2014-08-28 16:48:57 -07001628 processor_->set_delegate(nullptr);
Amin Hassanid3f4bea2018-04-30 14:52:40 -07001629 processor_->EnqueueAction(std::move(ping_action));
Thieu Led88a8572011-05-26 09:09:19 -07001630 // Call StartProcessing() synchronously here to avoid any race conditions
1631 // caused by multiple outstanding ping Omaha requests. If we call
1632 // StartProcessing() asynchronously, the device can be suspended before we
1633 // get a chance to callback to StartProcessing(). When the device resumes
1634 // (assuming the device sleeps longer than the next update check period),
1635 // StartProcessing() is called back and at the same time, the next update
1636 // check is fired which eventually invokes StartProcessing(). A crash
1637 // can occur because StartProcessing() checks to make sure that the
1638 // processor is idle which it isn't due to the two concurrent ping Omaha
1639 // requests.
1640 processor_->StartProcessing();
1641 } else {
Darin Petkov58dd1342011-05-06 12:05:13 -07001642 LOG(WARNING) << "Action processor running, Omaha ping suppressed.";
Darin Petkov58dd1342011-05-06 12:05:13 -07001643 }
Thieu Led88a8572011-05-26 09:09:19 -07001644
Gilad Arnoldec7f9162014-07-15 13:24:46 -07001645 // Update the last check time here; it may be re-updated when an Omaha
1646 // response is received, but this will prevent us from repeatedly scheduling
1647 // checks in the case where a response is not received.
1648 UpdateLastCheckedTime();
1649
Thieu Led88a8572011-05-26 09:09:19 -07001650 // Update the status which will schedule the next update check
Christopher Wileycc8ce0e2015-10-01 16:48:47 -07001651 SetStatusAndNotify(UpdateStatus::UPDATED_NEED_REBOOT);
Gilad Arnold54fa66d2014-09-29 13:14:29 -07001652 ScheduleUpdates();
Thieu Le116fda32011-04-19 11:01:54 -07001653}
1654
Jay Srinivasan480ddfa2012-06-01 19:15:26 -07001655bool UpdateAttempter::DecrementUpdateCheckCount() {
Ben Chan9abb7632014-08-07 00:10:53 -07001656 int64_t update_check_count_value;
Jay Srinivasan480ddfa2012-06-01 19:15:26 -07001657
1658 if (!prefs_->Exists(kPrefsUpdateCheckCount)) {
1659 // This file does not exist. This means we haven't started our update
1660 // check count down yet, so nothing more to do. This file will be created
1661 // later when we first satisfy the wall-clock-based-wait period.
1662 LOG(INFO) << "No existing update check count. That's normal.";
1663 return true;
1664 }
1665
1666 if (prefs_->GetInt64(kPrefsUpdateCheckCount, &update_check_count_value)) {
1667 // Only if we're able to read a proper integer value, then go ahead
1668 // and decrement and write back the result in the same file, if needed.
1669 LOG(INFO) << "Update check count = " << update_check_count_value;
1670
1671 if (update_check_count_value == 0) {
1672 // It could be 0, if, for some reason, the file didn't get deleted
1673 // when we set our status to waiting for reboot. so we just leave it
1674 // as is so that we can prevent another update_check wait for this client.
1675 LOG(INFO) << "Not decrementing update check count as it's already 0.";
1676 return true;
1677 }
1678
1679 if (update_check_count_value > 0)
1680 update_check_count_value--;
1681 else
1682 update_check_count_value = 0;
1683
1684 // Write out the new value of update_check_count_value.
1685 if (prefs_->SetInt64(kPrefsUpdateCheckCount, update_check_count_value)) {
Sen Jiang771f6482018-04-04 17:59:10 -07001686 // We successfully wrote out the new value, so enable the
Jay Srinivasan480ddfa2012-06-01 19:15:26 -07001687 // update check based wait.
1688 LOG(INFO) << "New update check count = " << update_check_count_value;
1689 return true;
1690 }
1691 }
1692
1693 LOG(INFO) << "Deleting update check count state due to read/write errors.";
1694
1695 // We cannot read/write to the file, so disable the update check based wait
1696 // so that we don't get stuck in this OS version by any chance (which could
1697 // happen if there's some bug that causes to read/write incorrectly).
1698 // Also attempt to delete the file to do our best effort to cleanup.
1699 prefs_->Delete(kPrefsUpdateCheckCount);
1700 return false;
1701}
Chris Sosad317e402013-06-12 13:47:09 -07001702
David Zeuthene4c58bf2013-06-18 17:26:50 -07001703void UpdateAttempter::UpdateEngineStarted() {
Alex Vakulenkodea2eac2014-03-14 15:56:59 -07001704 // If we just booted into a new update, keep the previous OS version
1705 // in case we rebooted because of a crash of the old version, so we
Alex Vakulenko88b591f2014-08-28 16:48:57 -07001706 // can do a proper crash report with correct information.
Alex Vakulenkodea2eac2014-03-14 15:56:59 -07001707 // This must be done before calling
Amin Hassani538bd592020-11-04 20:46:08 -08001708 // SystemState::Get()->payload_state()->UpdateEngineStarted() since it will
Alex Vakulenkodea2eac2014-03-14 15:56:59 -07001709 // delete SystemUpdated marker file.
Amin Hassani538bd592020-11-04 20:46:08 -08001710 if (SystemState::Get()->system_rebooted() &&
Alex Vakulenkodea2eac2014-03-14 15:56:59 -07001711 prefs_->Exists(kPrefsSystemUpdatedMarker)) {
1712 if (!prefs_->GetString(kPrefsPreviousVersion, &prev_version_)) {
1713 // If we fail to get the version string, make sure it stays empty.
1714 prev_version_.clear();
1715 }
1716 }
1717
Amin Hassani538bd592020-11-04 20:46:08 -08001718 SystemState::Get()->payload_state()->UpdateEngineStarted();
David Zeuthen8f191b22013-08-06 12:27:50 -07001719 StartP2PAtStartup();
Jae Hoon Kim5e8e30b2020-05-06 14:59:06 -07001720
Amin Hassani90e9f192020-11-18 14:20:56 -08001721 excluder_ = CreateExcluder();
David Zeuthen8f191b22013-08-06 12:27:50 -07001722}
1723
1724bool UpdateAttempter::StartP2PAtStartup() {
Amin Hassani538bd592020-11-04 20:46:08 -08001725 if (!SystemState::Get()->p2p_manager()->IsP2PEnabled()) {
David Zeuthen8f191b22013-08-06 12:27:50 -07001726 LOG(INFO) << "Not starting p2p at startup since it's not enabled.";
1727 return false;
1728 }
1729
Amin Hassani538bd592020-11-04 20:46:08 -08001730 if (SystemState::Get()->p2p_manager()->CountSharedFiles() < 1) {
David Zeuthen8f191b22013-08-06 12:27:50 -07001731 LOG(INFO) << "Not starting p2p at startup since our application "
1732 << "is not sharing any files.";
1733 return false;
1734 }
1735
1736 return StartP2PAndPerformHousekeeping();
1737}
1738
1739bool UpdateAttempter::StartP2PAndPerformHousekeeping() {
Amin Hassani538bd592020-11-04 20:46:08 -08001740 if (!SystemState::Get()->p2p_manager()->IsP2PEnabled()) {
David Zeuthen8f191b22013-08-06 12:27:50 -07001741 LOG(INFO) << "Not starting p2p since it's not enabled.";
1742 return false;
1743 }
1744
1745 LOG(INFO) << "Ensuring that p2p is running.";
Amin Hassani538bd592020-11-04 20:46:08 -08001746 if (!SystemState::Get()->p2p_manager()->EnsureP2PRunning()) {
David Zeuthen8f191b22013-08-06 12:27:50 -07001747 LOG(ERROR) << "Error starting p2p.";
1748 return false;
1749 }
1750
1751 LOG(INFO) << "Performing p2p housekeeping.";
Amin Hassani538bd592020-11-04 20:46:08 -08001752 if (!SystemState::Get()->p2p_manager()->PerformHousekeeping()) {
David Zeuthen8f191b22013-08-06 12:27:50 -07001753 LOG(ERROR) << "Error performing housekeeping for p2p.";
1754 return false;
1755 }
1756
1757 LOG(INFO) << "Done performing p2p housekeeping.";
1758 return true;
David Zeuthene4c58bf2013-06-18 17:26:50 -07001759}
1760
Amin Hassani7cc8bb02019-01-14 16:29:47 -08001761bool UpdateAttempter::GetBootTimeAtUpdate(Time* out_boot_time) {
Alex Deymo906191f2015-10-12 12:22:44 -07001762 // In case of an update_engine restart without a reboot, we stored the boot_id
1763 // when the update was completed by setting a pref, so we can check whether
1764 // the last update was on this boot or a previous one.
1765 string boot_id;
1766 TEST_AND_RETURN_FALSE(utils::GetBootId(&boot_id));
1767
1768 string update_completed_on_boot_id;
1769 if (!prefs_->Exists(kPrefsUpdateCompletedOnBootId) ||
1770 !prefs_->GetString(kPrefsUpdateCompletedOnBootId,
1771 &update_completed_on_boot_id) ||
1772 update_completed_on_boot_id != boot_id)
David Zeuthen3c55abd2013-10-14 12:48:03 -07001773 return false;
1774
Alex Deymo906191f2015-10-12 12:22:44 -07001775 // Short-circuit avoiding the read in case out_boot_time is nullptr.
1776 if (out_boot_time) {
1777 int64_t boot_time = 0;
1778 // Since the kPrefsUpdateCompletedOnBootId was correctly set, this pref
1779 // should not fail.
1780 TEST_AND_RETURN_FALSE(
1781 prefs_->GetInt64(kPrefsUpdateCompletedBootTime, &boot_time));
1782 *out_boot_time = Time::FromInternalValue(boot_time);
David Zeuthen3c55abd2013-10-14 12:48:03 -07001783 }
David Zeuthen3c55abd2013-10-14 12:48:03 -07001784 return true;
1785}
1786
Jae Hoon Kimba2fdce2019-07-11 13:18:58 -07001787bool UpdateAttempter::IsBusyOrUpdateScheduled() {
Christopher Wileycc8ce0e2015-10-01 16:48:47 -07001788 return ((status_ != UpdateStatus::IDLE &&
1789 status_ != UpdateStatus::UPDATED_NEED_REBOOT) ||
Gilad Arnold54fa66d2014-09-29 13:14:29 -07001790 waiting_for_scheduled_check_);
1791}
1792
Sen Jiangdaeaa432018-10-09 18:18:45 -07001793bool UpdateAttempter::IsAnyUpdateSourceAllowed() const {
David Pursell907b4fa2015-01-27 10:27:38 -08001794 // We allow updates from any source if either of these are true:
1795 // * The device is running an unofficial (dev/test) image.
1796 // * The debugd dev features are accessible (i.e. in devmode with no owner).
1797 // This protects users running a base image, while still allowing a specific
1798 // window (gated by the debug dev features) where `cros flash` is usable.
Amin Hassani538bd592020-11-04 20:46:08 -08001799 if (!SystemState::Get()->hardware()->IsOfficialBuild()) {
David Pursell02c18642014-11-06 11:26:11 -08001800 LOG(INFO) << "Non-official build; allowing any update source.";
1801 return true;
1802 }
1803
Amin Hassani538bd592020-11-04 20:46:08 -08001804 if (SystemState::Get()->hardware()->AreDevFeaturesEnabled()) {
Sen Jiange67bb5b2016-06-20 15:53:56 -07001805 LOG(INFO) << "Developer features enabled; allowing custom update sources.";
David Pursell02c18642014-11-06 11:26:11 -08001806 return true;
1807 }
Sen Jiange67bb5b2016-06-20 15:53:56 -07001808
1809 LOG(INFO)
1810 << "Developer features disabled; disallowing custom update sources.";
David Pursell02c18642014-11-06 11:26:11 -08001811 return false;
1812}
1813
May Lippert60aa3ca2018-08-15 16:55:29 -07001814void UpdateAttempter::ReportTimeToUpdateAppliedMetric() {
Amin Hassani538bd592020-11-04 20:46:08 -08001815 const policy::DevicePolicy* device_policy =
1816 SystemState::Get()->device_policy();
May Lippert60aa3ca2018-08-15 16:55:29 -07001817 if (device_policy && device_policy->IsEnterpriseEnrolled()) {
1818 vector<policy::DevicePolicy::WeeklyTimeInterval> parsed_intervals;
1819 bool has_time_restrictions =
1820 device_policy->GetDisallowedTimeIntervals(&parsed_intervals);
1821
1822 int64_t update_first_seen_at_int;
Amin Hassani538bd592020-11-04 20:46:08 -08001823 if (SystemState::Get()->prefs()->Exists(kPrefsUpdateFirstSeenAt)) {
1824 if (SystemState::Get()->prefs()->GetInt64(kPrefsUpdateFirstSeenAt,
1825 &update_first_seen_at_int)) {
May Lippert60aa3ca2018-08-15 16:55:29 -07001826 TimeDelta update_delay =
Amin Hassani538bd592020-11-04 20:46:08 -08001827 SystemState::Get()->clock()->GetWallclockTime() -
May Lippert60aa3ca2018-08-15 16:55:29 -07001828 Time::FromInternalValue(update_first_seen_at_int);
Amin Hassani538bd592020-11-04 20:46:08 -08001829 SystemState::Get()
1830 ->metrics_reporter()
May Lippert60aa3ca2018-08-15 16:55:29 -07001831 ->ReportEnterpriseUpdateSeenToDownloadDays(has_time_restrictions,
1832 update_delay.InDays());
1833 }
1834 }
1835 }
1836}
1837
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -07001838} // namespace chromeos_update_engine