blob: e039480779370de86707938890ea72fa95abeabd [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"
Kelvin Zhang97cb0582020-12-02 16:42:15 -050059#include "update_engine/cros/download_action_chromeos.h"
Amin Hassaniec7bc112020-10-29 16:47:58 -070060#include "update_engine/cros/omaha_request_action.h"
61#include "update_engine/cros/omaha_request_params.h"
62#include "update_engine/cros/omaha_response_handler_action.h"
63#include "update_engine/cros/omaha_utils.h"
64#include "update_engine/cros/p2p_manager.h"
65#include "update_engine/cros/payload_state_interface.h"
66#include "update_engine/cros/power_manager_interface.h"
Alex Deymo14c0da82016-07-20 16:45:45 -070067#include "update_engine/libcurl_http_fetcher.h"
Alex Deymo39910dc2015-11-09 17:04:30 -080068#include "update_engine/payload_consumer/filesystem_verifier_action.h"
69#include "update_engine/payload_consumer/postinstall_runner_action.h"
Amin Hassani0882a512018-04-05 16:25:44 -070070#include "update_engine/update_boot_flags_action.h"
Gilad Arnoldec7f9162014-07-15 13:24:46 -070071#include "update_engine/update_manager/policy.h"
Marton Hunyadye58bddb2018-04-10 20:27:26 +020072#include "update_engine/update_manager/policy_utils.h"
Gilad Arnoldec7f9162014-07-15 13:24:46 -070073#include "update_engine/update_manager/update_manager.h"
Christopher Wileycc8ce0e2015-10-01 16:48:47 -070074#include "update_engine/update_status_utils.h"
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -070075
Gilad Arnoldec7f9162014-07-15 13:24:46 -070076using base::Bind;
77using base::Callback;
Andrewe045aef2020-01-08 16:29:22 -080078using base::FilePath;
David Zeuthen3c55abd2013-10-14 12:48:03 -070079using base::Time;
Darin Petkovaf183052010-08-23 12:07:13 -070080using base::TimeDelta;
81using base::TimeTicks;
Alex Vakulenko3f39d5c2015-10-13 09:27:13 -070082using brillo::MessageLoop;
Amin Hassani7cc8bb02019-01-14 16:29:47 -080083using chromeos_update_manager::CalculateStagingCase;
Gilad Arnoldec7f9162014-07-15 13:24:46 -070084using chromeos_update_manager::EvalStatus;
85using chromeos_update_manager::Policy;
Adolfo Victoria497044c2018-07-18 07:51:42 -070086using chromeos_update_manager::StagingCase;
Amin Hassani7cc8bb02019-01-14 16:29:47 -080087using chromeos_update_manager::UpdateCheckParams;
Andrew065d78d2020-04-07 15:43:07 -070088using std::map;
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -070089using std::string;
90using std::vector;
Aaron Woodbf5a2522017-10-04 10:58:36 -070091using update_engine::UpdateAttemptFlags;
Aaron Wood7f92e2b2017-08-28 14:51:21 -070092using update_engine::UpdateEngineStatus;
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -070093
94namespace chromeos_update_engine {
95
Darin Petkov36275772010-10-01 11:40:57 -070096const int UpdateAttempter::kMaxDeltaUpdateFailures = 3;
97
Andrew de los Reyes45168102010-11-22 11:13:50 -080098namespace {
99const int kMaxConsecutiveObeyProxyRequests = 20;
Gilad Arnold70e476e2013-07-30 16:01:13 -0700100
Alex Deymo0d298542016-03-30 18:31:49 -0700101// Minimum threshold to broadcast an status update in progress and time.
102const double kBroadcastThresholdProgress = 0.01; // 1%
103const int kBroadcastThresholdSeconds = 10;
104
David Pursell02c18642014-11-06 11:26:11 -0800105// By default autest bypasses scattering. If we want to test scattering,
106// use kScheduledAUTestURLRequest. The URL used is same in both cases, but
107// different params are passed to CheckForUpdate().
108const char kAUTestURLRequest[] = "autest";
109const char kScheduledAUTestURLRequest[] = "autest-scheduled";
Alex Vakulenkod2779df2014-06-16 13:19:00 -0700110} // namespace
Andrew de los Reyes45168102010-11-22 11:13:50 -0800111
Sen Jiangdaeaa432018-10-09 18:18:45 -0700112ErrorCode GetErrorCodeForAction(AbstractAction* action, ErrorCode code) {
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -0700113 if (code != ErrorCode::kError)
Darin Petkov777dbfa2010-07-20 15:03:37 -0700114 return code;
115
116 const string type = action->Type();
117 if (type == OmahaRequestAction::StaticType())
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -0700118 return ErrorCode::kOmahaRequestError;
Darin Petkov777dbfa2010-07-20 15:03:37 -0700119 if (type == OmahaResponseHandlerAction::StaticType())
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -0700120 return ErrorCode::kOmahaResponseHandlerError;
Allie Woodeb9e6d82015-04-17 13:55:30 -0700121 if (type == FilesystemVerifierAction::StaticType())
122 return ErrorCode::kFilesystemVerifierError;
Darin Petkov777dbfa2010-07-20 15:03:37 -0700123 if (type == PostinstallRunnerAction::StaticType())
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -0700124 return ErrorCode::kPostinstallRunnerError;
Darin Petkov777dbfa2010-07-20 15:03:37 -0700125
126 return code;
127}
128
Amin Hassani538bd592020-11-04 20:46:08 -0800129UpdateAttempter::UpdateAttempter(CertificateChecker* cert_checker)
Gilad Arnold1f847232014-04-07 12:07:49 -0700130 : processor_(new ActionProcessor()),
Colin Howes978c1082018-12-03 11:46:12 -0800131 cert_checker_(cert_checker),
132 is_install_(false) {}
Darin Petkovc6c135c2010-08-11 13:36:18 -0700133
Gilad Arnoldec7f9162014-07-15 13:24:46 -0700134UpdateAttempter::~UpdateAttempter() {
Amin Hassanic288bf62020-11-04 19:33:43 -0800135 // Prevent any DBus communication from UpdateAttempter when shutting down the
136 // daemon.
137 ClearObservers();
138
Alex Deymo33e91e72015-12-01 18:26:08 -0300139 // CertificateChecker might not be initialized in unittests.
140 if (cert_checker_)
141 cert_checker_->SetObserver(nullptr);
Alex Deymo2b4268c2015-12-04 13:56:25 -0800142 // Release ourselves as the ActionProcessor's delegate to prevent
143 // re-scheduling the updates due to the processing stopped.
144 processor_->set_delegate(nullptr);
Gilad Arnoldec7f9162014-07-15 13:24:46 -0700145}
146
Gilad Arnold1f847232014-04-07 12:07:49 -0700147void UpdateAttempter::Init() {
148 // Pulling from the SystemState can only be done after construction, since
149 // this is an aggregate of various objects (such as the UpdateAttempter),
150 // which requires them all to be constructed prior to it being used.
Amin Hassani538bd592020-11-04 20:46:08 -0800151 prefs_ = SystemState::Get()->prefs();
152 omaha_request_params_ = SystemState::Get()->request_params();
Alex Deymo906191f2015-10-12 12:22:44 -0700153
Alex Deymo33e91e72015-12-01 18:26:08 -0300154 if (cert_checker_)
155 cert_checker_->SetObserver(this);
156
Alex Deymo906191f2015-10-12 12:22:44 -0700157 // In case of update_engine restart without a reboot we need to restore the
158 // reboot needed state.
Vyshu Khota4c5413d2020-11-04 16:17:25 -0800159 if (GetBootTimeAtUpdate(nullptr)) {
Alex Deymo906191f2015-10-12 12:22:44 -0700160 status_ = UpdateStatus::UPDATED_NEED_REBOOT;
Vyshu Khota4c5413d2020-11-04 16:17:25 -0800161 } else {
Alex Deymo906191f2015-10-12 12:22:44 -0700162 status_ = UpdateStatus::IDLE;
Vyshu Khota4c5413d2020-11-04 16:17:25 -0800163 prefs_->Delete(kPrefsLastFp, {kDlcPrefsSubDir});
164 }
Gilad Arnold1f847232014-04-07 12:07:49 -0700165}
166
Xiaochu Liu88d90382018-08-29 16:09:11 -0700167bool UpdateAttempter::ScheduleUpdates() {
Jae Hoon Kimba2fdce2019-07-11 13:18:58 -0700168 if (IsBusyOrUpdateScheduled())
Xiaochu Liu88d90382018-08-29 16:09:11 -0700169 return false;
Gilad Arnold54fa66d2014-09-29 13:14:29 -0700170
Gilad Arnoldec7f9162014-07-15 13:24:46 -0700171 chromeos_update_manager::UpdateManager* const update_manager =
Amin Hassani538bd592020-11-04 20:46:08 -0800172 SystemState::Get()->update_manager();
Gilad Arnoldec7f9162014-07-15 13:24:46 -0700173 CHECK(update_manager);
Amin Hassani7cc8bb02019-01-14 16:29:47 -0800174 Callback<void(EvalStatus, const UpdateCheckParams&)> callback =
175 Bind(&UpdateAttempter::OnUpdateScheduled, base::Unretained(this));
Gilad Arnoldec7f9162014-07-15 13:24:46 -0700176 // We limit the async policy request to a reasonably short time, to avoid a
177 // starvation due to a transient bug.
Jae Hoon Kim504c3cb2019-07-02 11:17:24 -0700178 update_manager->AsyncPolicyRequestUpdateCheckAllowed(
179 callback, &Policy::UpdateCheckAllowed);
Gilad Arnoldec7f9162014-07-15 13:24:46 -0700180 waiting_for_scheduled_check_ = true;
Xiaochu Liu88d90382018-08-29 16:09:11 -0700181 return true;
Darin Petkovc6c135c2010-08-11 13:36:18 -0700182}
183
Amin Hassanic288bf62020-11-04 19:33:43 -0800184bool UpdateAttempter::StartUpdater() {
185 // Initiate update checks.
186 ScheduleUpdates();
187
Amin Hassani538bd592020-11-04 20:46:08 -0800188 auto update_boot_flags_action = std::make_unique<UpdateBootFlagsAction>(
189 SystemState::Get()->boot_control());
Amin Hassani1e19be32020-11-16 14:12:33 -0800190 aux_processor_.EnqueueAction(std::move(update_boot_flags_action));
Amin Hassanic288bf62020-11-04 19:33:43 -0800191 // Update boot flags after 45 seconds.
192 MessageLoop::current()->PostDelayedTask(
193 FROM_HERE,
194 base::Bind(&ActionProcessor::StartProcessing,
Amin Hassani1e19be32020-11-16 14:12:33 -0800195 base::Unretained(&aux_processor_)),
Amin Hassanic288bf62020-11-04 19:33:43 -0800196 base::TimeDelta::FromSeconds(45));
197
198 // Broadcast the update engine status on startup to ensure consistent system
199 // state on crashes.
200 MessageLoop::current()->PostTask(
201 FROM_HERE,
202 base::Bind(&UpdateAttempter::BroadcastStatus, base::Unretained(this)));
203
204 MessageLoop::current()->PostTask(
205 FROM_HERE,
206 base::Bind(&UpdateAttempter::UpdateEngineStarted,
207 base::Unretained(this)));
208 return true;
209}
210
Alex Deymoc1c17b42015-11-23 03:53:15 -0300211void UpdateAttempter::CertificateChecked(ServerToCheck server_to_check,
212 CertificateCheckResult result) {
Amin Hassani538bd592020-11-04 20:46:08 -0800213 SystemState::Get()->metrics_reporter()->ReportCertificateCheckMetrics(
Tianjie Xu282aa1f2017-09-05 13:42:45 -0700214 server_to_check, result);
Alex Deymoc1c17b42015-11-23 03:53:15 -0300215}
216
David Zeuthen985b1122013-10-09 12:13:15 -0700217bool UpdateAttempter::CheckAndReportDailyMetrics() {
218 int64_t stored_value;
Amin Hassani538bd592020-11-04 20:46:08 -0800219 Time now = SystemState::Get()->clock()->GetWallclockTime();
220 if (SystemState::Get()->prefs()->Exists(kPrefsDailyMetricsLastReportedAt) &&
221 SystemState::Get()->prefs()->GetInt64(kPrefsDailyMetricsLastReportedAt,
222 &stored_value)) {
Alex Deymof329b932014-10-30 01:37:48 -0700223 Time last_reported_at = Time::FromInternalValue(stored_value);
224 TimeDelta time_reported_since = now - last_reported_at;
David Zeuthen985b1122013-10-09 12:13:15 -0700225 if (time_reported_since.InSeconds() < 0) {
226 LOG(WARNING) << "Last reported daily metrics "
227 << utils::FormatTimeDelta(time_reported_since) << " ago "
228 << "which is negative. Either the system clock is wrong or "
229 << "the kPrefsDailyMetricsLastReportedAt state variable "
230 << "is wrong.";
231 // In this case, report daily metrics to reset.
232 } else {
Amin Hassani7cc8bb02019-01-14 16:29:47 -0800233 if (time_reported_since.InSeconds() < 24 * 60 * 60) {
David Zeuthen985b1122013-10-09 12:13:15 -0700234 LOG(INFO) << "Last reported daily metrics "
235 << utils::FormatTimeDelta(time_reported_since) << " ago.";
236 return false;
237 }
238 LOG(INFO) << "Last reported daily metrics "
239 << utils::FormatTimeDelta(time_reported_since) << " ago, "
240 << "which is more than 24 hours ago.";
241 }
242 }
243
244 LOG(INFO) << "Reporting daily metrics.";
Amin Hassani538bd592020-11-04 20:46:08 -0800245 SystemState::Get()->prefs()->SetInt64(kPrefsDailyMetricsLastReportedAt,
246 now.ToInternalValue());
David Zeuthen985b1122013-10-09 12:13:15 -0700247
248 ReportOSAge();
249
250 return true;
251}
252
253void UpdateAttempter::ReportOSAge() {
254 struct stat sb;
David Zeuthen985b1122013-10-09 12:13:15 -0700255 if (stat("/etc/lsb-release", &sb) != 0) {
David Zeuthend9aca3b2014-08-21 13:11:21 -0400256 PLOG(ERROR) << "Error getting file status for /etc/lsb-release "
257 << "(Note: this may happen in some unit tests)";
David Zeuthen985b1122013-10-09 12:13:15 -0700258 return;
259 }
260
Sen Jiang1c3db7d2019-01-10 15:48:55 -0800261 Time lsb_release_timestamp = Time::FromTimeSpec(sb.st_ctim);
Amin Hassani538bd592020-11-04 20:46:08 -0800262 Time now = SystemState::Get()->clock()->GetWallclockTime();
Alex Deymof329b932014-10-30 01:37:48 -0700263 TimeDelta age = now - lsb_release_timestamp;
David Zeuthen985b1122013-10-09 12:13:15 -0700264 if (age.InSeconds() < 0) {
265 LOG(ERROR) << "The OS age (" << utils::FormatTimeDelta(age)
David Zeuthend9aca3b2014-08-21 13:11:21 -0400266 << ") is negative. Maybe the clock is wrong? "
267 << "(Note: this may happen in some unit tests.)";
David Zeuthen985b1122013-10-09 12:13:15 -0700268 return;
269 }
270
Amin Hassani538bd592020-11-04 20:46:08 -0800271 SystemState::Get()->metrics_reporter()->ReportDailyMetrics(age);
David Zeuthen985b1122013-10-09 12:13:15 -0700272}
273
Amin Hassanie8153632020-10-27 15:11:28 -0700274void UpdateAttempter::Update(const UpdateCheckParams& params) {
Gilad Arnoldec7f9162014-07-15 13:24:46 -0700275 // This is normally called frequently enough so it's appropriate to use as a
276 // hook for reporting daily metrics.
277 // TODO(garnold) This should be hooked to a separate (reliable and consistent)
278 // timeout event.
David Zeuthen985b1122013-10-09 12:13:15 -0700279 CheckAndReportDailyMetrics();
280
Andrew de los Reyesc1d5c932011-04-20 17:15:47 -0700281 fake_update_success_ = false;
Christopher Wileycc8ce0e2015-10-01 16:48:47 -0700282 if (status_ == UpdateStatus::UPDATED_NEED_REBOOT) {
Thieu Le116fda32011-04-19 11:01:54 -0700283 // Although we have applied an update, we still want to ping Omaha
284 // to ensure the number of active statistics is accurate.
David Zeuthen33bae492014-02-25 16:16:18 -0800285 //
286 // Also convey to the UpdateEngine.Check.Result metric that we're
287 // not performing an update check because of this.
Andrew de los Reyes6b78e292010-05-10 15:54:39 -0700288 LOG(INFO) << "Not updating b/c we already updated and we're waiting for "
Thieu Le116fda32011-04-19 11:01:54 -0700289 << "reboot, we'll ping Omaha instead";
Amin Hassani538bd592020-11-04 20:46:08 -0800290 SystemState::Get()->metrics_reporter()->ReportUpdateCheckMetrics(
Tianjie Xu282aa1f2017-09-05 13:42:45 -0700291 metrics::CheckResult::kRebootPending,
292 metrics::CheckReaction::kUnset,
293 metrics::DownloadErrorCode::kUnset);
Thieu Le116fda32011-04-19 11:01:54 -0700294 PingOmaha();
Andrew de los Reyes6b78e292010-05-10 15:54:39 -0700295 return;
296 }
Christopher Wileycc8ce0e2015-10-01 16:48:47 -0700297 if (status_ != UpdateStatus::IDLE) {
Andrew de los Reyes6b78e292010-05-10 15:54:39 -0700298 // Update in progress. Do nothing
299 return;
300 }
Jay Srinivasan480ddfa2012-06-01 19:15:26 -0700301
Amin Hassanie8153632020-10-27 15:11:28 -0700302 if (!CalculateUpdateParams(params)) {
Jay Srinivasan480ddfa2012-06-01 19:15:26 -0700303 return;
304 }
305
Amin Hassanie8153632020-10-27 15:11:28 -0700306 BuildUpdateActions(params.interactive);
Jay Srinivasan480ddfa2012-06-01 19:15:26 -0700307
Christopher Wileycc8ce0e2015-10-01 16:48:47 -0700308 SetStatusAndNotify(UpdateStatus::CHECKING_FOR_UPDATE);
Jay Srinivasan480ddfa2012-06-01 19:15:26 -0700309
Gilad Arnoldec7f9162014-07-15 13:24:46 -0700310 // Update the last check time here; it may be re-updated when an Omaha
311 // response is received, but this will prevent us from repeatedly scheduling
312 // checks in the case where a response is not received.
313 UpdateLastCheckedTime();
314
Amin Hassani0882a512018-04-05 16:25:44 -0700315 ScheduleProcessingStart();
Jay Srinivasan480ddfa2012-06-01 19:15:26 -0700316}
317
Jay Srinivasan1c0fe792013-03-28 16:45:25 -0700318void UpdateAttempter::RefreshDevicePolicy() {
319 // Lazy initialize the policy provider, or reload the latest policy data.
320 if (!policy_provider_.get())
321 policy_provider_.reset(new policy::PolicyProvider());
322 policy_provider_->Reload();
323
Alex Vakulenko88b591f2014-08-28 16:48:57 -0700324 const policy::DevicePolicy* device_policy = nullptr;
Jay Srinivasan1c0fe792013-03-28 16:45:25 -0700325 if (policy_provider_->device_policy_is_loaded())
326 device_policy = &policy_provider_->GetDevicePolicy();
327
328 if (device_policy)
329 LOG(INFO) << "Device policies/settings present";
330 else
331 LOG(INFO) << "No device policies/settings present.";
332
Amin Hassani538bd592020-11-04 20:46:08 -0800333 SystemState::Get()->set_device_policy(device_policy);
334 SystemState::Get()->p2p_manager()->SetDevicePolicy(device_policy);
Jay Srinivasan1c0fe792013-03-28 16:45:25 -0700335}
336
David Zeuthen8f191b22013-08-06 12:27:50 -0700337void UpdateAttempter::CalculateP2PParams(bool interactive) {
338 bool use_p2p_for_downloading = false;
339 bool use_p2p_for_sharing = false;
340
Thiemo Nagel27d9d532017-05-26 16:35:20 +0200341 // Never use p2p for downloading in interactive checks unless the developer
342 // has opted in for it via a marker file.
David Zeuthen8f191b22013-08-06 12:27:50 -0700343 //
Thiemo Nagel27d9d532017-05-26 16:35:20 +0200344 // (Why would a developer want to opt in? If they are working on the
345 // update_engine or p2p codebases so they can actually test their code.)
David Zeuthen8f191b22013-08-06 12:27:50 -0700346
Amin Hassani538bd592020-11-04 20:46:08 -0800347 if (!SystemState::Get()->p2p_manager()->IsP2PEnabled()) {
348 LOG(INFO) << "p2p is not enabled - disallowing p2p for both"
349 << " downloading and sharing.";
350 } else {
351 // Allow p2p for sharing, even in interactive checks.
352 use_p2p_for_sharing = true;
353 if (!interactive) {
354 LOG(INFO) << "Non-interactive check - allowing p2p for downloading";
355 use_p2p_for_downloading = true;
David Zeuthen8f191b22013-08-06 12:27:50 -0700356 } else {
Amin Hassani538bd592020-11-04 20:46:08 -0800357 LOG(INFO) << "Forcibly disabling use of p2p for downloading "
358 << "since this update attempt is interactive.";
David Zeuthen8f191b22013-08-06 12:27:50 -0700359 }
360 }
361
Amin Hassani538bd592020-11-04 20:46:08 -0800362 PayloadStateInterface* const payload_state =
363 SystemState::Get()->payload_state();
Gilad Arnold74b5f552014-10-07 08:17:16 -0700364 payload_state->SetUsingP2PForDownloading(use_p2p_for_downloading);
365 payload_state->SetUsingP2PForSharing(use_p2p_for_sharing);
David Zeuthen8f191b22013-08-06 12:27:50 -0700366}
367
Amin Hassanie8153632020-10-27 15:11:28 -0700368bool UpdateAttempter::CalculateUpdateParams(const UpdateCheckParams& params) {
Darin Petkov1023a602010-08-30 13:47:51 -0700369 http_response_code_ = 0;
Amin Hassani538bd592020-11-04 20:46:08 -0800370 PayloadStateInterface* const payload_state =
371 SystemState::Get()->payload_state();
Patrick Dubroy7fbbe8a2011-08-01 17:28:22 +0200372
Alex Deymo749ecf12014-10-21 20:06:57 -0700373 // Refresh the policy before computing all the update parameters.
374 RefreshDevicePolicy();
375
Marton Hunyadye58bddb2018-04-10 20:27:26 +0200376 // Check whether we need to clear the rollback-happened preference after
377 // policy is available again.
378 UpdateRollbackHappened();
379
Amin Hassanie8153632020-10-27 15:11:28 -0700380 CalculateStagingParams(params.interactive);
Adolfo Victoria497044c2018-07-18 07:51:42 -0700381 // If staging_wait_time_ wasn't set, staging is off, use scattering instead.
382 if (staging_wait_time_.InSeconds() == 0) {
Amin Hassanie8153632020-10-27 15:11:28 -0700383 CalculateScatteringParams(params.interactive);
Adolfo Victoria497044c2018-07-18 07:51:42 -0700384 }
Patrick Dubroy7fbbe8a2011-08-01 17:28:22 +0200385
Amin Hassanie8153632020-10-27 15:11:28 -0700386 CalculateP2PParams(params.interactive);
Gilad Arnold74b5f552014-10-07 08:17:16 -0700387 if (payload_state->GetUsingP2PForDownloading() ||
388 payload_state->GetUsingP2PForSharing()) {
David Zeuthen8f191b22013-08-06 12:27:50 -0700389 // OK, p2p is to be used - start it and perform housekeeping.
390 if (!StartP2PAndPerformHousekeeping()) {
391 // If this fails, disable p2p for this attempt
392 LOG(INFO) << "Forcibly disabling use of p2p since starting p2p or "
393 << "performing housekeeping failed.";
Gilad Arnold74b5f552014-10-07 08:17:16 -0700394 payload_state->SetUsingP2PForDownloading(false);
395 payload_state->SetUsingP2PForSharing(false);
David Zeuthen8f191b22013-08-06 12:27:50 -0700396 }
397 }
398
Amin Hassanie8153632020-10-27 15:11:28 -0700399 if (!omaha_request_params_->Init(
400 forced_app_version_, forced_omaha_url_, params)) {
Jay Srinivasan1c0fe792013-03-28 16:45:25 -0700401 LOG(ERROR) << "Unable to initialize Omaha request params.";
Jay Srinivasan480ddfa2012-06-01 19:15:26 -0700402 return false;
Darin Petkova4a8a8c2010-07-15 22:21:12 -0700403 }
Darin Petkov3aefa862010-12-07 14:45:00 -0800404
Andrewe045aef2020-01-08 16:29:22 -0800405 // The function |CalculateDlcParams| makes use of the function |GetAppId| from
406 // |OmahaRequestParams|, so to ensure that the return from |GetAppId|
407 // doesn't change, no changes to the values |download_channel_|,
408 // |image_props_.product_id| and |image_props_.canary_product_id| from
409 // |omaha_request_params_| shall be made below this line.
410 CalculateDlcParams();
411
Gilad Arnoldafd70ed2014-09-30 16:27:06 -0700412 LOG(INFO) << "target_version_prefix = "
Jay Srinivasanae4697c2013-03-18 17:08:08 -0700413 << omaha_request_params_->target_version_prefix()
Marton Hunyadyba51c3f2018-04-25 15:18:10 +0200414 << ", rollback_allowed = "
415 << omaha_request_params_->rollback_allowed()
Jay Srinivasan480ddfa2012-06-01 19:15:26 -0700416 << ", scatter_factor_in_seconds = "
417 << utils::FormatSecs(scatter_factor_.InSeconds());
418
419 LOG(INFO) << "Wall Clock Based Wait Enabled = "
Jay Srinivasanae4697c2013-03-18 17:08:08 -0700420 << omaha_request_params_->wall_clock_based_wait_enabled()
Jay Srinivasan480ddfa2012-06-01 19:15:26 -0700421 << ", Update Check Count Wait Enabled = "
Jay Srinivasanae4697c2013-03-18 17:08:08 -0700422 << omaha_request_params_->update_check_count_wait_enabled()
Amin Hassani7cc8bb02019-01-14 16:29:47 -0800423 << ", Waiting Period = "
424 << utils::FormatSecs(
425 omaha_request_params_->waiting_period().InSeconds());
Jay Srinivasan0a708742012-03-20 11:26:12 -0700426
David Zeuthen8f191b22013-08-06 12:27:50 -0700427 LOG(INFO) << "Use p2p For Downloading = "
Gilad Arnold74b5f552014-10-07 08:17:16 -0700428 << payload_state->GetUsingP2PForDownloading()
David Zeuthen8f191b22013-08-06 12:27:50 -0700429 << ", Use p2p For Sharing = "
Gilad Arnold74b5f552014-10-07 08:17:16 -0700430 << payload_state->GetUsingP2PForSharing();
David Zeuthen8f191b22013-08-06 12:27:50 -0700431
Andrew de los Reyes45168102010-11-22 11:13:50 -0800432 obeying_proxies_ = true;
Amin Hassanie8153632020-10-27 15:11:28 -0700433 if (proxy_manual_checks_ == 0) {
Andrew de los Reyes45168102010-11-22 11:13:50 -0800434 LOG(INFO) << "forced to obey proxies";
435 // If forced to obey proxies, every 20th request will not use proxies
436 proxy_manual_checks_++;
437 LOG(INFO) << "proxy manual checks: " << proxy_manual_checks_;
438 if (proxy_manual_checks_ >= kMaxConsecutiveObeyProxyRequests) {
439 proxy_manual_checks_ = 0;
440 obeying_proxies_ = false;
441 }
442 } else if (base::RandInt(0, 4) == 0) {
443 obeying_proxies_ = false;
444 }
Amin Hassani7cc8bb02019-01-14 16:29:47 -0800445 LOG_IF(INFO, !obeying_proxies_)
446 << "To help ensure updates work, this update check we are ignoring the "
447 << "proxy settings and using direct connections.";
Andrew de los Reyes45168102010-11-22 11:13:50 -0800448
Darin Petkov36275772010-10-01 11:40:57 -0700449 DisableDeltaUpdateIfNeeded();
Jay Srinivasan480ddfa2012-06-01 19:15:26 -0700450 return true;
451}
452
Gilad Arnoldb92f0df2013-01-10 16:32:45 -0800453void UpdateAttempter::CalculateScatteringParams(bool interactive) {
Jay Srinivasan21be0752012-07-25 15:44:56 -0700454 // Take a copy of the old scatter value before we update it, as
455 // we need to update the waiting period if this value changes.
456 TimeDelta old_scatter_factor = scatter_factor_;
Amin Hassani538bd592020-11-04 20:46:08 -0800457 const policy::DevicePolicy* device_policy =
458 SystemState::Get()->device_policy();
Jay Srinivasan21be0752012-07-25 15:44:56 -0700459 if (device_policy) {
Ben Chan9abb7632014-08-07 00:10:53 -0700460 int64_t new_scatter_factor_in_secs = 0;
Jay Srinivasan21be0752012-07-25 15:44:56 -0700461 device_policy->GetScatterFactorInSeconds(&new_scatter_factor_in_secs);
Alex Vakulenkod2779df2014-06-16 13:19:00 -0700462 if (new_scatter_factor_in_secs < 0) // sanitize input, just in case.
Amin Hassani7cc8bb02019-01-14 16:29:47 -0800463 new_scatter_factor_in_secs = 0;
Jay Srinivasan21be0752012-07-25 15:44:56 -0700464 scatter_factor_ = TimeDelta::FromSeconds(new_scatter_factor_in_secs);
465 }
466
467 bool is_scatter_enabled = false;
468 if (scatter_factor_.InSeconds() == 0) {
469 LOG(INFO) << "Scattering disabled since scatter factor is set to 0";
Gilad Arnoldb92f0df2013-01-10 16:32:45 -0800470 } else if (interactive) {
471 LOG(INFO) << "Scattering disabled as this is an interactive update check";
Amin Hassani538bd592020-11-04 20:46:08 -0800472 } else if (SystemState::Get()->hardware()->IsOOBEEnabled() &&
473 !SystemState::Get()->hardware()->IsOOBEComplete(nullptr)) {
Alex Deymo46a9aae2016-05-04 20:20:11 -0700474 LOG(INFO) << "Scattering disabled since OOBE is enabled but not complete "
475 "yet";
Jay Srinivasan21be0752012-07-25 15:44:56 -0700476 } else {
477 is_scatter_enabled = true;
478 LOG(INFO) << "Scattering is enabled";
479 }
480
481 if (is_scatter_enabled) {
482 // This means the scattering policy is turned on.
483 // Now check if we need to update the waiting period. The two cases
484 // in which we'd need to update the waiting period are:
485 // 1. First time in process or a scheduled check after a user-initiated one.
Jay Srinivasanae4697c2013-03-18 17:08:08 -0700486 // (omaha_request_params_->waiting_period will be zero in this case).
Jay Srinivasan21be0752012-07-25 15:44:56 -0700487 // 2. Admin has changed the scattering policy value.
488 // (new scattering value will be different from old one in this case).
Ben Chan9abb7632014-08-07 00:10:53 -0700489 int64_t wait_period_in_secs = 0;
Jay Srinivasanae4697c2013-03-18 17:08:08 -0700490 if (omaha_request_params_->waiting_period().InSeconds() == 0) {
Jay Srinivasan21be0752012-07-25 15:44:56 -0700491 // First case. Check if we have a suitable value to set for
492 // the waiting period.
Adolfo Victoriad3a1e352018-07-16 11:40:47 -0700493 if (prefs_->GetInt64(kPrefsWallClockScatteringWaitPeriod,
494 &wait_period_in_secs) &&
Jay Srinivasan21be0752012-07-25 15:44:56 -0700495 wait_period_in_secs > 0 &&
496 wait_period_in_secs <= scatter_factor_.InSeconds()) {
497 // This means:
498 // 1. There's a persisted value for the waiting period available.
499 // 2. And that persisted value is still valid.
500 // So, in this case, we should reuse the persisted value instead of
501 // generating a new random value to improve the chances of a good
502 // distribution for scattering.
Jay Srinivasanae4697c2013-03-18 17:08:08 -0700503 omaha_request_params_->set_waiting_period(
Amin Hassani7cc8bb02019-01-14 16:29:47 -0800504 TimeDelta::FromSeconds(wait_period_in_secs));
505 LOG(INFO) << "Using persisted wall-clock waiting period: "
506 << utils::FormatSecs(
507 omaha_request_params_->waiting_period().InSeconds());
Alex Vakulenkod2779df2014-06-16 13:19:00 -0700508 } else {
Jay Srinivasan21be0752012-07-25 15:44:56 -0700509 // This means there's no persisted value for the waiting period
510 // available or its value is invalid given the new scatter_factor value.
511 // So, we should go ahead and regenerate a new value for the
512 // waiting period.
513 LOG(INFO) << "Persisted value not present or not valid ("
514 << utils::FormatSecs(wait_period_in_secs)
515 << ") for wall-clock waiting period.";
516 GenerateNewWaitingPeriod();
517 }
518 } else if (scatter_factor_ != old_scatter_factor) {
519 // This means there's already a waiting period value, but we detected
520 // a change in the scattering policy value. So, we should regenerate the
521 // waiting period to make sure it's within the bounds of the new scatter
522 // factor value.
523 GenerateNewWaitingPeriod();
524 } else {
525 // Neither the first time scattering is enabled nor the scattering value
526 // changed. Nothing to do.
Amin Hassani7cc8bb02019-01-14 16:29:47 -0800527 LOG(INFO) << "Keeping current wall-clock waiting period: "
528 << utils::FormatSecs(
529 omaha_request_params_->waiting_period().InSeconds());
Jay Srinivasan21be0752012-07-25 15:44:56 -0700530 }
531
Jay Srinivasanae4697c2013-03-18 17:08:08 -0700532 // The invariant at this point is that omaha_request_params_->waiting_period
Jay Srinivasan21be0752012-07-25 15:44:56 -0700533 // is non-zero no matter which path we took above.
Jay Srinivasanae4697c2013-03-18 17:08:08 -0700534 LOG_IF(ERROR, omaha_request_params_->waiting_period().InSeconds() == 0)
Jay Srinivasan21be0752012-07-25 15:44:56 -0700535 << "Waiting Period should NOT be zero at this point!!!";
536
537 // Since scattering is enabled, wall clock based wait will always be
538 // enabled.
Jay Srinivasanae4697c2013-03-18 17:08:08 -0700539 omaha_request_params_->set_wall_clock_based_wait_enabled(true);
Jay Srinivasan21be0752012-07-25 15:44:56 -0700540
541 // If we don't have any issues in accessing the file system to update
542 // the update check count value, we'll turn that on as well.
543 bool decrement_succeeded = DecrementUpdateCheckCount();
Jay Srinivasanae4697c2013-03-18 17:08:08 -0700544 omaha_request_params_->set_update_check_count_wait_enabled(
Amin Hassani7cc8bb02019-01-14 16:29:47 -0800545 decrement_succeeded);
Jay Srinivasan21be0752012-07-25 15:44:56 -0700546 } else {
547 // This means the scattering feature is turned off or disabled for
548 // this particular update check. Make sure to disable
549 // all the knobs and artifacts so that we don't invoke any scattering
550 // related code.
Jay Srinivasanae4697c2013-03-18 17:08:08 -0700551 omaha_request_params_->set_wall_clock_based_wait_enabled(false);
552 omaha_request_params_->set_update_check_count_wait_enabled(false);
553 omaha_request_params_->set_waiting_period(TimeDelta::FromSeconds(0));
Adolfo Victoriad3a1e352018-07-16 11:40:47 -0700554 prefs_->Delete(kPrefsWallClockScatteringWaitPeriod);
Jay Srinivasan21be0752012-07-25 15:44:56 -0700555 prefs_->Delete(kPrefsUpdateCheckCount);
556 // Don't delete the UpdateFirstSeenAt file as we don't want manual checks
557 // that result in no-updates (e.g. due to server side throttling) to
558 // cause update starvation by having the client generate a new
559 // UpdateFirstSeenAt for each scheduled check that follows a manual check.
560 }
561}
562
563void UpdateAttempter::GenerateNewWaitingPeriod() {
Amin Hassani7cc8bb02019-01-14 16:29:47 -0800564 omaha_request_params_->set_waiting_period(
565 TimeDelta::FromSeconds(base::RandInt(1, scatter_factor_.InSeconds())));
Jay Srinivasan21be0752012-07-25 15:44:56 -0700566
Amin Hassani7cc8bb02019-01-14 16:29:47 -0800567 LOG(INFO) << "Generated new wall-clock waiting period: "
568 << utils::FormatSecs(
569 omaha_request_params_->waiting_period().InSeconds());
Jay Srinivasan21be0752012-07-25 15:44:56 -0700570
571 // Do a best-effort to persist this in all cases. Even if the persistence
572 // fails, we'll still be able to scatter based on our in-memory value.
573 // The persistence only helps in ensuring a good overall distribution
574 // across multiple devices if they tend to reboot too often.
Amin Hassani538bd592020-11-04 20:46:08 -0800575 SystemState::Get()->payload_state()->SetScatteringWaitPeriod(
Gilad Arnold519cfc72014-10-02 10:34:54 -0700576 omaha_request_params_->waiting_period());
Jay Srinivasan21be0752012-07-25 15:44:56 -0700577}
578
Adolfo Victoria497044c2018-07-18 07:51:42 -0700579void UpdateAttempter::CalculateStagingParams(bool interactive) {
Amin Hassani538bd592020-11-04 20:46:08 -0800580 bool oobe_complete = SystemState::Get()->hardware()->IsOOBEEnabled() &&
581 SystemState::Get()->hardware()->IsOOBEComplete(nullptr);
582 auto device_policy = SystemState::Get()->device_policy();
Adolfo Victoria497044c2018-07-18 07:51:42 -0700583 StagingCase staging_case = StagingCase::kOff;
584 if (device_policy && !interactive && oobe_complete) {
585 staging_wait_time_ = omaha_request_params_->waiting_period();
586 staging_case = CalculateStagingCase(
Amin Hassani90e9f192020-11-18 14:20:56 -0800587 device_policy, &staging_wait_time_, &staging_schedule_);
Adolfo Victoria497044c2018-07-18 07:51:42 -0700588 }
589 switch (staging_case) {
590 case StagingCase::kOff:
591 // Staging is off, get rid of persisted value.
592 prefs_->Delete(kPrefsWallClockStagingWaitPeriod);
593 // Set |staging_wait_time_| to its default value so scattering can still
594 // be turned on
595 staging_wait_time_ = TimeDelta();
596 break;
597 // Let the cases fall through since they just add, and never remove, steps
598 // to turning staging on.
599 case StagingCase::kNoSavedValue:
600 prefs_->SetInt64(kPrefsWallClockStagingWaitPeriod,
601 staging_wait_time_.InDays());
Denis Nikitin8f2fa742019-05-29 10:36:52 -0700602 FALLTHROUGH;
Adolfo Victoria497044c2018-07-18 07:51:42 -0700603 case StagingCase::kSetStagingFromPref:
604 omaha_request_params_->set_waiting_period(staging_wait_time_);
Denis Nikitin8f2fa742019-05-29 10:36:52 -0700605 FALLTHROUGH;
Adolfo Victoria497044c2018-07-18 07:51:42 -0700606 case StagingCase::kNoAction:
607 // Staging is on, enable wallclock based wait so that its values get used.
608 omaha_request_params_->set_wall_clock_based_wait_enabled(true);
609 // Use UpdateCheckCount if possible to prevent devices updating all at
610 // once.
611 omaha_request_params_->set_update_check_count_wait_enabled(
612 DecrementUpdateCheckCount());
613 // Scattering should not be turned on if staging is on, delete the
614 // existing scattering configuration.
615 prefs_->Delete(kPrefsWallClockScatteringWaitPeriod);
616 scatter_factor_ = TimeDelta();
617 }
Chris Sosad317e402013-06-12 13:47:09 -0700618}
619
Andrew065d78d2020-04-07 15:43:07 -0700620bool UpdateAttempter::ResetDlcPrefs(const string& dlc_id) {
621 vector<string> failures;
Andrew065d78d2020-04-07 15:43:07 -0700622 for (auto& sub_key :
623 {kPrefsPingActive, kPrefsPingLastActive, kPrefsPingLastRollcall}) {
Amin Hassani90e9f192020-11-18 14:20:56 -0800624 auto key = prefs_->CreateSubKey({kDlcPrefsSubDir, dlc_id, sub_key});
625 if (!prefs_->Delete(key))
Andrew065d78d2020-04-07 15:43:07 -0700626 failures.emplace_back(sub_key);
627 }
628 if (failures.size() != 0)
629 PLOG(ERROR) << "Failed to delete prefs (" << base::JoinString(failures, ",")
630 << " for DLC (" << dlc_id << ").";
631
632 return failures.size() == 0;
633}
634
Vyshu Khota4c5413d2020-11-04 16:17:25 -0800635void UpdateAttempter::SetPref(const string& pref_key,
636 const string& pref_value,
637 const string& payload_id) {
638 string dlc_id;
639 if (!omaha_request_params_->GetDlcId(payload_id, &dlc_id)) {
640 // Not a DLC ID, set fingerprint in perf for platform ID.
641 prefs_->SetString(pref_key, pref_value);
642 } else {
643 // Set fingerprint in pref for DLC ID.
644 auto key = prefs_->CreateSubKey({kDlcPrefsSubDir, dlc_id, pref_key});
645 prefs_->SetString(key, pref_value);
646 }
647}
648
Andrewa8d7df32020-03-15 20:10:01 -0700649bool UpdateAttempter::SetDlcActiveValue(bool is_active, const string& dlc_id) {
650 if (dlc_id.empty()) {
651 LOG(ERROR) << "Empty DLC ID passed.";
652 return false;
653 }
654 LOG(INFO) << "Set DLC (" << dlc_id << ") to "
655 << (is_active ? "Active" : "Inactive");
Andrewa8d7df32020-03-15 20:10:01 -0700656 if (is_active) {
Andrew065d78d2020-04-07 15:43:07 -0700657 auto ping_active_key =
Amin Hassani90e9f192020-11-18 14:20:56 -0800658 prefs_->CreateSubKey({kDlcPrefsSubDir, dlc_id, kPrefsPingActive});
659 if (!prefs_->SetInt64(ping_active_key, kPingActiveValue)) {
Andrewa8d7df32020-03-15 20:10:01 -0700660 LOG(ERROR) << "Failed to set the value of ping metadata '"
661 << kPrefsPingActive << "'.";
662 return false;
663 }
664 } else {
Andrew065d78d2020-04-07 15:43:07 -0700665 return ResetDlcPrefs(dlc_id);
Andrewa8d7df32020-03-15 20:10:01 -0700666 }
667 return true;
668}
669
Andrew065d78d2020-04-07 15:43:07 -0700670int64_t UpdateAttempter::GetPingMetadata(const string& metadata_key) const {
Andrewe045aef2020-01-08 16:29:22 -0800671 // The first time a ping is sent, the metadata files containing the values
672 // sent back by the server still don't exist. A value of -1 is used to
673 // indicate this.
Amin Hassani538bd592020-11-04 20:46:08 -0800674 if (!SystemState::Get()->prefs()->Exists(metadata_key))
Andrewe045aef2020-01-08 16:29:22 -0800675 return kPingNeverPinged;
676
677 int64_t value;
Amin Hassani538bd592020-11-04 20:46:08 -0800678 if (SystemState::Get()->prefs()->GetInt64(metadata_key, &value))
Andrewe045aef2020-01-08 16:29:22 -0800679 return value;
680
681 // Return -2 when the file exists and there is a problem reading from it, or
682 // the value cannot be converted to an integer.
683 return kPingUnknownValue;
684}
685
686void UpdateAttempter::CalculateDlcParams() {
Amin Hassani2b68e6b2020-04-17 10:49:12 -0700687 // Set the |dlc_ids_| only for an update. This is required to get the
Andrewe045aef2020-01-08 16:29:22 -0800688 // currently installed DLC(s).
689 if (!is_install_ &&
Amin Hassani538bd592020-11-04 20:46:08 -0800690 !SystemState::Get()->dlcservice()->GetDlcsToUpdate(&dlc_ids_)) {
Andrewe045aef2020-01-08 16:29:22 -0800691 LOG(INFO) << "Failed to retrieve DLC module IDs from dlcservice. Check the "
692 "state of dlcservice, will not update DLC modules.";
693 }
Andrew065d78d2020-04-07 15:43:07 -0700694 map<string, OmahaRequestParams::AppParams> dlc_apps_params;
Amin Hassani2b68e6b2020-04-17 10:49:12 -0700695 for (const auto& dlc_id : dlc_ids_) {
Andrewe045aef2020-01-08 16:29:22 -0800696 OmahaRequestParams::AppParams dlc_params{
697 .active_counting_type = OmahaRequestParams::kDateBased,
698 .name = dlc_id,
699 .send_ping = false};
Andrew065d78d2020-04-07 15:43:07 -0700700 if (is_install_) {
701 // In some cases, |SetDlcActiveValue| might fail to reset the DLC prefs
702 // when a DLC is uninstalled. To avoid having stale values from that
703 // scenario, we reset the metadata values on a new install request.
704 // Ignore failure to delete stale prefs.
705 ResetDlcPrefs(dlc_id);
706 SetDlcActiveValue(true, dlc_id);
707 } else {
708 // Only send the ping when the request is to update DLCs. When installing
709 // DLCs, we don't want to send the ping yet, since the DLCs might fail to
710 // install or might not really be active yet.
711 dlc_params.ping_active = kPingActiveValue;
712 auto ping_active_key =
Amin Hassani90e9f192020-11-18 14:20:56 -0800713 prefs_->CreateSubKey({kDlcPrefsSubDir, dlc_id, kPrefsPingActive});
714 if (!prefs_->GetInt64(ping_active_key, &dlc_params.ping_active) ||
Andrew065d78d2020-04-07 15:43:07 -0700715 dlc_params.ping_active != kPingActiveValue) {
716 dlc_params.ping_active = kPingInactiveValue;
Andrewe045aef2020-01-08 16:29:22 -0800717 }
Andrew065d78d2020-04-07 15:43:07 -0700718 auto ping_last_active_key =
Amin Hassani90e9f192020-11-18 14:20:56 -0800719 prefs_->CreateSubKey({kDlcPrefsSubDir, dlc_id, kPrefsPingLastActive});
Andrew065d78d2020-04-07 15:43:07 -0700720 dlc_params.ping_date_last_active = GetPingMetadata(ping_last_active_key);
721
Amin Hassani90e9f192020-11-18 14:20:56 -0800722 auto ping_last_rollcall_key = prefs_->CreateSubKey(
Jae Hoon Kimc1f36922020-05-11 18:20:18 -0700723 {kDlcPrefsSubDir, dlc_id, kPrefsPingLastRollcall});
Andrew065d78d2020-04-07 15:43:07 -0700724 dlc_params.ping_date_last_rollcall =
725 GetPingMetadata(ping_last_rollcall_key);
726
727 dlc_params.send_ping = true;
Andrewe045aef2020-01-08 16:29:22 -0800728 }
729 dlc_apps_params[omaha_request_params_->GetDlcAppId(dlc_id)] = dlc_params;
730 }
731 omaha_request_params_->set_dlc_apps_params(dlc_apps_params);
Amin Hassanie8153632020-10-27 15:11:28 -0700732 omaha_request_params_->set_is_install(is_install_);
Andrewe045aef2020-01-08 16:29:22 -0800733}
734
Jay Srinivasan480ddfa2012-06-01 19:15:26 -0700735void UpdateAttempter::BuildUpdateActions(bool interactive) {
Darin Petkovf42cc1c2010-09-01 09:03:02 -0700736 CHECK(!processor_->IsRunning());
737 processor_->set_delegate(this);
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700738
Jae Hoon Kimedb65502019-06-14 11:52:17 -0700739 // The session ID needs to be kept throughout the update flow. The value
740 // of the session ID will reset/update only when it is a new update flow.
741 session_id_ = base::GenerateGUID();
742
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700743 // Actions:
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700744 auto update_check_fetcher = std::make_unique<LibcurlHttpFetcher>(
Amin Hassani538bd592020-11-04 20:46:08 -0800745 GetProxyResolver(), SystemState::Get()->hardware());
Alex Deymo33e91e72015-12-01 18:26:08 -0300746 update_check_fetcher->set_server_to_check(ServerToCheck::kUpdate);
Andrew de los Reyesfb2f4612011-06-09 18:21:49 -0700747 // Try harder to connect to the network, esp when not interactive.
748 // See comment in libcurl_http_fetcher.cc.
749 update_check_fetcher->set_no_network_max_retries(interactive ? 1 : 3);
Xiaochu Liu4a1173a2019-04-10 10:49:08 -0700750 update_check_fetcher->set_is_update_check(true);
Amin Hassani538bd592020-11-04 20:46:08 -0800751 auto update_check_action = std::make_unique<OmahaRequestAction>(
752 nullptr, std::move(update_check_fetcher), false, session_id_);
753 auto response_handler_action = std::make_unique<OmahaResponseHandlerAction>();
754 auto update_boot_flags_action = std::make_unique<UpdateBootFlagsAction>(
755 SystemState::Get()->boot_control());
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700756 auto download_started_action = std::make_unique<OmahaRequestAction>(
Sen Jiang18414082018-01-11 14:50:36 -0800757 new OmahaEvent(OmahaEvent::kTypeUpdateDownloadStarted),
758 std::make_unique<LibcurlHttpFetcher>(GetProxyResolver(),
Amin Hassani538bd592020-11-04 20:46:08 -0800759 SystemState::Get()->hardware()),
Jae Hoon Kimedb65502019-06-14 11:52:17 -0700760 false,
761 session_id_);
Alex Deymo33e91e72015-12-01 18:26:08 -0300762
Amin Hassani538bd592020-11-04 20:46:08 -0800763 LibcurlHttpFetcher* download_fetcher = new LibcurlHttpFetcher(
764 GetProxyResolver(), SystemState::Get()->hardware());
Alex Deymo33e91e72015-12-01 18:26:08 -0300765 download_fetcher->set_server_to_check(ServerToCheck::kDownload);
Sen Jiangee174a12017-12-21 17:38:08 -0800766 if (interactive)
767 download_fetcher->set_max_retry_count(kDownloadMaxRetryCountInteractive);
Jae Hoon Kim0ae8fe12019-06-26 14:32:50 -0700768 download_fetcher->SetHeader(kXGoogleUpdateSessionId, session_id_);
Kelvin Zhang97cb0582020-12-02 16:42:15 -0500769 auto download_action = std::make_unique<DownloadActionChromeos>(
770 prefs_,
771 SystemState::Get()->boot_control(),
772 SystemState::Get()->hardware(),
773 download_fetcher, // passes ownership
774 interactive);
Andrew de los Reyes63b96d72010-05-10 13:08:54 -0700775 download_action->set_delegate(this);
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700776
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700777 auto download_finished_action = std::make_unique<OmahaRequestAction>(
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700778 new OmahaEvent(OmahaEvent::kTypeUpdateDownloadFinished),
779 std::make_unique<LibcurlHttpFetcher>(GetProxyResolver(),
Amin Hassani538bd592020-11-04 20:46:08 -0800780 SystemState::Get()->hardware()),
Jae Hoon Kimedb65502019-06-14 11:52:17 -0700781 false,
782 session_id_);
Tianjie24f96092020-06-30 12:26:25 -0700783 auto filesystem_verifier_action = std::make_unique<FilesystemVerifierAction>(
Amin Hassani538bd592020-11-04 20:46:08 -0800784 SystemState::Get()->boot_control()->GetDynamicPartitionControl());
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700785 auto update_complete_action = std::make_unique<OmahaRequestAction>(
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700786 new OmahaEvent(OmahaEvent::kTypeUpdateComplete),
787 std::make_unique<LibcurlHttpFetcher>(GetProxyResolver(),
Amin Hassani538bd592020-11-04 20:46:08 -0800788 SystemState::Get()->hardware()),
Jae Hoon Kimedb65502019-06-14 11:52:17 -0700789 false,
790 session_id_);
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700791
792 auto postinstall_runner_action = std::make_unique<PostinstallRunnerAction>(
Amin Hassani538bd592020-11-04 20:46:08 -0800793 SystemState::Get()->boot_control(), SystemState::Get()->hardware());
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700794 postinstall_runner_action->set_delegate(this);
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700795
796 // Bond them together. We have to use the leaf-types when calling
797 // BondActions().
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700798 BondActions(update_check_action.get(), response_handler_action.get());
799 BondActions(response_handler_action.get(), download_action.get());
800 BondActions(download_action.get(), filesystem_verifier_action.get());
801 BondActions(filesystem_verifier_action.get(),
802 postinstall_runner_action.get());
Chris Sosad317e402013-06-12 13:47:09 -0700803
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700804 processor_->EnqueueAction(std::move(update_check_action));
805 processor_->EnqueueAction(std::move(response_handler_action));
Amin Hassani0882a512018-04-05 16:25:44 -0700806 processor_->EnqueueAction(std::move(update_boot_flags_action));
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700807 processor_->EnqueueAction(std::move(download_started_action));
808 processor_->EnqueueAction(std::move(download_action));
809 processor_->EnqueueAction(std::move(download_finished_action));
810 processor_->EnqueueAction(std::move(filesystem_verifier_action));
811 processor_->EnqueueAction(std::move(postinstall_runner_action));
812 processor_->EnqueueAction(std::move(update_complete_action));
Chris Sosad317e402013-06-12 13:47:09 -0700813}
814
Chris Sosa44b9b7e2014-04-02 13:53:46 -0700815bool UpdateAttempter::Rollback(bool powerwash) {
Colin Howes978c1082018-12-03 11:46:12 -0800816 is_install_ = false;
Chris Sosa44b9b7e2014-04-02 13:53:46 -0700817 if (!CanRollback()) {
818 return false;
819 }
Chris Sosad317e402013-06-12 13:47:09 -0700820
Chris Sosa44b9b7e2014-04-02 13:53:46 -0700821 // Extra check for enterprise-enrolled devices since they don't support
822 // powerwash.
823 if (powerwash) {
824 // Enterprise-enrolled devices have an empty owner in their device policy.
825 string owner;
826 RefreshDevicePolicy();
Amin Hassani538bd592020-11-04 20:46:08 -0800827 const policy::DevicePolicy* device_policy =
828 SystemState::Get()->device_policy();
Chris Sosa44b9b7e2014-04-02 13:53:46 -0700829 if (device_policy && (!device_policy->GetOwner(&owner) || owner.empty())) {
830 LOG(ERROR) << "Enterprise device detected. "
831 << "Cannot perform a powerwash for enterprise devices.";
832 return false;
833 }
834 }
835
836 processor_->set_delegate(this);
Chris Sosaaa18e162013-06-20 13:20:30 -0700837
Chris Sosa28e479c2013-07-12 11:39:53 -0700838 // Initialize the default request params.
Amin Hassanie8153632020-10-27 15:11:28 -0700839 if (!omaha_request_params_->Init("", "", {.interactive = true})) {
Chris Sosa28e479c2013-07-12 11:39:53 -0700840 LOG(ERROR) << "Unable to initialize Omaha request params.";
841 return false;
842 }
843
Chris Sosad317e402013-06-12 13:47:09 -0700844 LOG(INFO) << "Setting rollback options.";
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700845 install_plan_.reset(new InstallPlan());
846 install_plan_->target_slot = GetRollbackSlot();
Amin Hassani538bd592020-11-04 20:46:08 -0800847 install_plan_->source_slot =
848 SystemState::Get()->boot_control()->GetCurrentSlot();
Chris Sosa76a29ae2013-07-11 17:59:24 -0700849
Amin Hassani538bd592020-11-04 20:46:08 -0800850 TEST_AND_RETURN_FALSE(install_plan_->LoadPartitionsFromSlots(
851 SystemState::Get()->boot_control()));
Amin Hassanid3f4bea2018-04-30 14:52:40 -0700852 install_plan_->powerwash_required = powerwash;
Chris Sosad317e402013-06-12 13:47:09 -0700853
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();
Kelvin Zhang97cb0582020-12-02 16:42:15 -05001252 if (type == DownloadActionChromeos::StaticType()) {
Andrew de los Reyes63b96d72010-05-10 13:08:54 -07001253 download_progress_ = 0.0;
Kelvin Zhang97cb0582020-12-02 16:42:15 -05001254 DownloadActionChromeos* download_action =
1255 static_cast<DownloadActionChromeos*>(action);
Darin Petkov1023a602010-08-30 13:47:51 -07001256 http_response_code_ = download_action->GetHTTPResponseCode();
1257 } else if (type == OmahaRequestAction::StaticType()) {
1258 OmahaRequestAction* omaha_request_action =
Gilad Arnoldcf175a02014-07-10 16:48:47 -07001259 static_cast<OmahaRequestAction*>(action);
Darin Petkov1023a602010-08-30 13:47:51 -07001260 // If the request is not an event, then it's the update-check.
1261 if (!omaha_request_action->IsEvent()) {
1262 http_response_code_ = omaha_request_action->GetHTTPResponseCode();
Gilad Arnolda6dab942014-04-25 11:46:03 -07001263
1264 // Record the number of consecutive failed update checks.
1265 if (http_response_code_ == kHttpResponseInternalServerError ||
1266 http_response_code_ == kHttpResponseServiceUnavailable) {
1267 consecutive_failed_update_checks_++;
1268 } else {
1269 consecutive_failed_update_checks_ = 0;
1270 }
1271
Weidong Guo421ff332017-04-17 10:08:38 -07001272 const OmahaResponse& omaha_response =
1273 omaha_request_action->GetOutputObject();
Gilad Arnolda0258a52014-07-10 16:21:19 -07001274 // Store the server-dictated poll interval, if any.
1275 server_dictated_poll_interval_ =
Weidong Guo421ff332017-04-17 10:08:38 -07001276 std::max(0, omaha_response.poll_interval);
1277
1278 // This update is ignored by omaha request action because update over
1279 // cellular connection is not allowed. Needs to ask for user's permissions
1280 // to update.
1281 if (code == ErrorCode::kOmahaUpdateIgnoredOverCellular) {
1282 new_version_ = omaha_response.version;
1283 new_payload_size_ = 0;
1284 for (const auto& package : omaha_response.packages) {
1285 new_payload_size_ += package.size;
1286 }
1287 SetStatusAndNotify(UpdateStatus::NEED_PERMISSION_TO_UPDATE);
1288 }
Darin Petkov1023a602010-08-30 13:47:51 -07001289 }
Aaron Wood23bd3392017-10-06 14:48:25 -07001290 } else if (type == OmahaResponseHandlerAction::StaticType()) {
1291 // Depending on the returned error code, note that an update is available.
1292 if (code == ErrorCode::kOmahaUpdateDeferredPerPolicy ||
1293 code == ErrorCode::kSuccess) {
1294 // Note that the status will be updated to DOWNLOADING when some bytes
1295 // get actually downloaded from the server and the BytesReceived
1296 // callback is invoked. This avoids notifying the user that a download
1297 // has started in cases when the server and the client are unable to
1298 // initiate the download.
Amin Hassanid3f4bea2018-04-30 14:52:40 -07001299 auto omaha_response_handler_action =
1300 static_cast<OmahaResponseHandlerAction*>(action);
1301 install_plan_.reset(
1302 new InstallPlan(omaha_response_handler_action->install_plan()));
Aaron Wood23bd3392017-10-06 14:48:25 -07001303 UpdateLastCheckedTime();
Amin Hassanid3f4bea2018-04-30 14:52:40 -07001304 new_version_ = install_plan_->version;
Aaron Wood23bd3392017-10-06 14:48:25 -07001305 new_payload_size_ = 0;
Amin Hassanid3f4bea2018-04-30 14:52:40 -07001306 for (const auto& payload : install_plan_->payloads)
Aaron Wood23bd3392017-10-06 14:48:25 -07001307 new_payload_size_ += payload.size;
1308 cpu_limiter_.StartLimiter();
1309 SetStatusAndNotify(UpdateStatus::UPDATE_AVAILABLE);
1310 }
Darin Petkov1023a602010-08-30 13:47:51 -07001311 }
Aaron Wood23bd3392017-10-06 14:48:25 -07001312 // General failure cases.
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001313 if (code != ErrorCode::kSuccess) {
Darin Petkov7ed561b2011-10-04 02:59:03 -07001314 // If the current state is at or past the download phase, count the failure
1315 // in case a switch to full update becomes necessary. Ignore network
1316 // transfer timeouts and failures.
Amin Hassani89a37122018-03-23 12:59:24 -07001317 if (code != ErrorCode::kDownloadTransferError) {
1318 switch (status_) {
1319 case UpdateStatus::IDLE:
1320 case UpdateStatus::CHECKING_FOR_UPDATE:
1321 case UpdateStatus::UPDATE_AVAILABLE:
1322 case UpdateStatus::NEED_PERMISSION_TO_UPDATE:
1323 break;
1324 case UpdateStatus::DOWNLOADING:
1325 case UpdateStatus::VERIFYING:
1326 case UpdateStatus::FINALIZING:
1327 case UpdateStatus::UPDATED_NEED_REBOOT:
1328 case UpdateStatus::REPORTING_ERROR_EVENT:
1329 case UpdateStatus::ATTEMPTING_ROLLBACK:
1330 case UpdateStatus::DISABLED:
Amin Hassani70a90f52020-09-15 15:30:09 -07001331 case UpdateStatus::CLEANUP_PREVIOUS_UPDATE:
Amin Hassani89a37122018-03-23 12:59:24 -07001332 MarkDeltaUpdateFailure();
1333 break;
1334 }
Darin Petkov36275772010-10-01 11:40:57 -07001335 }
Sen Jiang89e24c12018-03-22 18:05:44 -07001336 if (code != ErrorCode::kNoUpdate) {
1337 // On failure, schedule an error event to be sent to Omaha.
1338 CreatePendingErrorEvent(action, code);
1339 }
Andrew de los Reyes63b96d72010-05-10 13:08:54 -07001340 return;
Darin Petkov09f96c32010-07-20 09:24:57 -07001341 }
Aaron Wood23bd3392017-10-06 14:48:25 -07001342 // Find out which action completed (successfully).
1343 if (type == DownloadAction::StaticType()) {
Christopher Wileycc8ce0e2015-10-01 16:48:47 -07001344 SetStatusAndNotify(UpdateStatus::FINALIZING);
Amin Hassaniafd8cea2017-12-04 14:20:00 -08001345 } else if (type == FilesystemVerifierAction::StaticType()) {
1346 // Log the system properties before the postinst and after the file system
1347 // is verified. It used to be done in the postinst itself. But postinst
1348 // cannot do this anymore. On the other hand, these logs are frequently
1349 // looked at and it is preferable not to scatter them in random location in
1350 // the log and rather log it right before the postinst. The reason not do
1351 // this in the |PostinstallRunnerAction| is to prevent dependency from
1352 // libpayload_consumer to libupdate_engine.
1353 LogImageProperties();
Andrew de los Reyes63b96d72010-05-10 13:08:54 -07001354 }
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -07001355}
1356
Alex Deymo542c19b2015-12-03 07:43:31 -03001357void UpdateAttempter::BytesReceived(uint64_t bytes_progressed,
1358 uint64_t bytes_received,
1359 uint64_t total) {
1360 // The PayloadState keeps track of how many bytes were actually downloaded
1361 // from a given URL for the URL skipping logic.
Amin Hassani538bd592020-11-04 20:46:08 -08001362 SystemState::Get()->payload_state()->DownloadProgress(bytes_progressed);
Alex Deymo542c19b2015-12-03 07:43:31 -03001363
Alex Deymo0d298542016-03-30 18:31:49 -07001364 double progress = 0;
1365 if (total)
1366 progress = static_cast<double>(bytes_received) / static_cast<double>(total);
1367 if (status_ != UpdateStatus::DOWNLOADING || bytes_received == total) {
Darin Petkovaf183052010-08-23 12:07:13 -07001368 download_progress_ = progress;
Christopher Wileycc8ce0e2015-10-01 16:48:47 -07001369 SetStatusAndNotify(UpdateStatus::DOWNLOADING);
Alex Deymo0d298542016-03-30 18:31:49 -07001370 } else {
1371 ProgressUpdate(progress);
Andrew de los Reyes63b96d72010-05-10 13:08:54 -07001372 }
1373}
1374
Alex Deymo542c19b2015-12-03 07:43:31 -03001375void UpdateAttempter::DownloadComplete() {
Amin Hassani538bd592020-11-04 20:46:08 -08001376 SystemState::Get()->payload_state()->DownloadComplete();
Alex Deymo542c19b2015-12-03 07:43:31 -03001377}
1378
Alex Deymo0d298542016-03-30 18:31:49 -07001379void UpdateAttempter::ProgressUpdate(double progress) {
1380 // Self throttle based on progress. Also send notifications if progress is
1381 // too slow.
1382 if (progress == 1.0 ||
1383 progress - download_progress_ >= kBroadcastThresholdProgress ||
1384 TimeTicks::Now() - last_notify_time_ >=
1385 TimeDelta::FromSeconds(kBroadcastThresholdSeconds)) {
1386 download_progress_ = progress;
1387 BroadcastStatus();
1388 }
1389}
1390
Amin Hassaniecb60d32019-06-17 18:09:10 -07001391void UpdateAttempter::ResetInteractivityFlags() {
1392 // Reset the state that's only valid for a single update pass.
1393 current_update_attempt_flags_ = UpdateAttemptFlags::kNone;
1394
1395 if (forced_update_pending_callback_.get())
1396 // Clear prior interactive requests once the processor is done.
1397 forced_update_pending_callback_->Run(false, false);
1398}
1399
Jay Srinivasanc1ba09a2012-08-14 14:15:57 -07001400bool UpdateAttempter::ResetStatus() {
1401 LOG(INFO) << "Attempting to reset state from "
Christopher Wileycc8ce0e2015-10-01 16:48:47 -07001402 << UpdateStatusToString(status_) << " to UpdateStatus::IDLE";
Jay Srinivasanc1ba09a2012-08-14 14:15:57 -07001403
1404 switch (status_) {
Christopher Wileycc8ce0e2015-10-01 16:48:47 -07001405 case UpdateStatus::IDLE:
Jay Srinivasanc1ba09a2012-08-14 14:15:57 -07001406 // no-op.
1407 return true;
1408
Amin Hassani7cc8bb02019-01-14 16:29:47 -08001409 case UpdateStatus::UPDATED_NEED_REBOOT: {
Jay Srinivasan1c0fe792013-03-28 16:45:25 -07001410 bool ret_value = true;
Christopher Wileycc8ce0e2015-10-01 16:48:47 -07001411 status_ = UpdateStatus::IDLE;
Jay Srinivasanc1ba09a2012-08-14 14:15:57 -07001412
Jay Srinivasan1c0fe792013-03-28 16:45:25 -07001413 // Remove the reboot marker so that if the machine is rebooted
Jay Srinivasanc1ba09a2012-08-14 14:15:57 -07001414 // after resetting to idle state, it doesn't go back to
Christopher Wileycc8ce0e2015-10-01 16:48:47 -07001415 // UpdateStatus::UPDATED_NEED_REBOOT state.
Alex Deymo906191f2015-10-12 12:22:44 -07001416 ret_value = prefs_->Delete(kPrefsUpdateCompletedOnBootId) && ret_value;
1417 ret_value = prefs_->Delete(kPrefsUpdateCompletedBootTime) && ret_value;
Vyshu Khota4c5413d2020-11-04 16:17:25 -08001418 ret_value = prefs_->Delete(kPrefsLastFp, {kDlcPrefsSubDir}) && ret_value;
Jay Srinivasan1c0fe792013-03-28 16:45:25 -07001419
Alex Deymo9870c0e2015-09-23 13:58:31 -07001420 // Update the boot flags so the current slot has higher priority.
Amin Hassani538bd592020-11-04 20:46:08 -08001421 BootControlInterface* boot_control = SystemState::Get()->boot_control();
Alex Deymo9870c0e2015-09-23 13:58:31 -07001422 if (!boot_control->SetActiveBootSlot(boot_control->GetCurrentSlot()))
1423 ret_value = false;
1424
Alex Deymo52590332016-11-29 18:29:13 -08001425 // Mark the current slot as successful again, since marking it as active
1426 // may reset the successful bit. We ignore the result of whether marking
1427 // the current slot as successful worked.
Amin Hassani7cc8bb02019-01-14 16:29:47 -08001428 if (!boot_control->MarkBootSuccessfulAsync(Bind([](bool successful) {})))
Alex Deymo52590332016-11-29 18:29:13 -08001429 ret_value = false;
1430
Alex Deymo42432912013-07-12 20:21:15 -07001431 // Notify the PayloadState that the successful payload was canceled.
Amin Hassani538bd592020-11-04 20:46:08 -08001432 SystemState::Get()->payload_state()->ResetUpdateStatus();
Alex Deymo42432912013-07-12 20:21:15 -07001433
Alex Deymo87c08862015-10-30 21:56:55 -07001434 // The previous version is used to report back to omaha after reboot that
1435 // we actually rebooted into the new version from this "prev-version". We
1436 // need to clear out this value now to prevent it being sent on the next
1437 // updatecheck request.
1438 ret_value = prefs_->SetString(kPrefsPreviousVersion, "") && ret_value;
1439
Alex Deymo906191f2015-10-12 12:22:44 -07001440 LOG(INFO) << "Reset status " << (ret_value ? "successful" : "failed");
Jay Srinivasan1c0fe792013-03-28 16:45:25 -07001441 return ret_value;
Jay Srinivasanc1ba09a2012-08-14 14:15:57 -07001442 }
1443
1444 default:
1445 LOG(ERROR) << "Reset not allowed in this state.";
1446 return false;
1447 }
1448}
1449
Aaron Wood7f92e2b2017-08-28 14:51:21 -07001450bool UpdateAttempter::GetStatus(UpdateEngineStatus* out_status) {
Aaron Wood795c5b42017-12-05 16:06:13 -08001451 out_status->last_checked_time = last_checked_time_;
Aaron Wood7f92e2b2017-08-28 14:51:21 -07001452 out_status->status = status_;
1453 out_status->current_version = omaha_request_params_->app_version();
Aaron Wood7f92e2b2017-08-28 14:51:21 -07001454 out_status->progress = download_progress_;
1455 out_status->new_size_bytes = new_payload_size_;
1456 out_status->new_version = new_version_;
Amin Hassani9be122e2019-08-29 09:20:12 -07001457 out_status->is_enterprise_rollback =
1458 install_plan_ && install_plan_->is_rollback;
Jae Hoon Kim2f78c1c2019-07-25 13:20:43 -07001459 out_status->is_install = is_install_;
Jae Hoon Kim051627a2019-09-03 12:56:32 -07001460
1461 string str_eol_date;
Amin Hassani538bd592020-11-04 20:46:08 -08001462 if (SystemState::Get()->prefs()->Exists(kPrefsOmahaEolDate) &&
1463 !SystemState::Get()->prefs()->GetString(kPrefsOmahaEolDate,
1464 &str_eol_date))
Jae Hoon Kim2dfd35d2020-06-02 10:53:13 -07001465 LOG(ERROR) << "Failed to retrieve kPrefsOmahaEolDate pref.";
Jae Hoon Kim051627a2019-09-03 12:56:32 -07001466 out_status->eol_date = StringToEolDate(str_eol_date);
1467
Miriam Polzer0cf1acb2020-04-29 17:39:51 +02001468 // A powerwash will take place either if the install plan says it is required
1469 // or if an enterprise rollback is happening.
1470 out_status->will_powerwash_after_reboot =
1471 install_plan_ &&
1472 (install_plan_->powerwash_required || install_plan_->is_rollback);
1473
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -07001474 return true;
1475}
1476
Darin Petkov61635a92011-05-18 16:20:36 -07001477void UpdateAttempter::BroadcastStatus() {
Aaron Wood7f92e2b2017-08-28 14:51:21 -07001478 UpdateEngineStatus broadcast_status;
1479 // Use common method for generating the current status.
1480 GetStatus(&broadcast_status);
1481
Alex Deymofa78f142016-01-26 21:36:16 -08001482 for (const auto& observer : service_observers_) {
Aaron Wood7f92e2b2017-08-28 14:51:21 -07001483 observer->SendStatusUpdate(broadcast_status);
Alex Deymofa78f142016-01-26 21:36:16 -08001484 }
Darin Petkovaf183052010-08-23 12:07:13 -07001485 last_notify_time_ = TimeTicks::Now();
Alex Deymofa78f142016-01-26 21:36:16 -08001486}
1487
Amin Hassani7cc8bb02019-01-14 16:29:47 -08001488uint32_t UpdateAttempter::GetErrorCodeFlags() {
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001489 uint32_t flags = 0;
1490
Amin Hassani538bd592020-11-04 20:46:08 -08001491 if (!SystemState::Get()->hardware()->IsNormalBootMode())
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001492 flags |= static_cast<uint32_t>(ErrorCode::kDevModeFlag);
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001493
Amin Hassanid3f4bea2018-04-30 14:52:40 -07001494 if (install_plan_ && install_plan_->is_resume)
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001495 flags |= static_cast<uint32_t>(ErrorCode::kResumedFlag);
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001496
Amin Hassani538bd592020-11-04 20:46:08 -08001497 if (!SystemState::Get()->hardware()->IsOfficialBuild())
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001498 flags |= static_cast<uint32_t>(ErrorCode::kTestImageFlag);
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001499
Sen Jiangdaeaa432018-10-09 18:18:45 -07001500 if (!omaha_request_params_->IsUpdateUrlOfficial()) {
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001501 flags |= static_cast<uint32_t>(ErrorCode::kTestOmahaUrlFlag);
Alex Deymoac41a822015-09-15 20:52:53 -07001502 }
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001503
1504 return flags;
1505}
1506
David Zeuthena99981f2013-04-29 13:42:47 -07001507bool UpdateAttempter::ShouldCancel(ErrorCode* cancel_reason) {
Jay Srinivasan1c0fe792013-03-28 16:45:25 -07001508 // Check if the channel we're attempting to update to is the same as the
1509 // target channel currently chosen by the user.
Amin Hassani538bd592020-11-04 20:46:08 -08001510 OmahaRequestParams* params = SystemState::Get()->request_params();
Jay Srinivasan1c0fe792013-03-28 16:45:25 -07001511 if (params->download_channel() != params->target_channel()) {
1512 LOG(ERROR) << "Aborting download as target channel: "
1513 << params->target_channel()
1514 << " is different from the download channel: "
1515 << params->download_channel();
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001516 *cancel_reason = ErrorCode::kUpdateCanceledByChannelChange;
Jay Srinivasan1c0fe792013-03-28 16:45:25 -07001517 return true;
1518 }
1519
1520 return false;
1521}
1522
Nam T. Nguyen7d623eb2014-05-13 16:06:28 -07001523void UpdateAttempter::SetStatusAndNotify(UpdateStatus status) {
Darin Petkov61635a92011-05-18 16:20:36 -07001524 status_ = status;
Darin Petkov61635a92011-05-18 16:20:36 -07001525 BroadcastStatus();
1526}
1527
Darin Petkov777dbfa2010-07-20 15:03:37 -07001528void UpdateAttempter::CreatePendingErrorEvent(AbstractAction* action,
David Zeuthena99981f2013-04-29 13:42:47 -07001529 ErrorCode code) {
Sen Jiang89e24c12018-03-22 18:05:44 -07001530 if (error_event_.get() || status_ == UpdateStatus::REPORTING_ERROR_EVENT) {
Darin Petkov09f96c32010-07-20 09:24:57 -07001531 // This shouldn't really happen.
1532 LOG(WARNING) << "There's already an existing pending error event.";
1533 return;
1534 }
Darin Petkov777dbfa2010-07-20 15:03:37 -07001535
Jay Srinivasan56d5aa42012-03-26 14:27:59 -07001536 // Classify the code to generate the appropriate result so that
1537 // the Borgmon charts show up the results correctly.
1538 // Do this before calling GetErrorCodeForAction which could potentially
1539 // augment the bit representation of code and thus cause no matches for
1540 // the switch cases below.
1541 OmahaEvent::Result event_result;
1542 switch (code) {
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001543 case ErrorCode::kOmahaUpdateIgnoredPerPolicy:
1544 case ErrorCode::kOmahaUpdateDeferredPerPolicy:
1545 case ErrorCode::kOmahaUpdateDeferredForBackoff:
Jay Srinivasan56d5aa42012-03-26 14:27:59 -07001546 event_result = OmahaEvent::kResultUpdateDeferred;
1547 break;
1548 default:
1549 event_result = OmahaEvent::kResultError;
1550 break;
1551 }
1552
Darin Petkov777dbfa2010-07-20 15:03:37 -07001553 code = GetErrorCodeForAction(action, code);
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -07001554 fake_update_success_ = code == ErrorCode::kPostinstallBootedFromFirmwareB;
Darin Petkov18c7bce2011-06-16 14:07:00 -07001555
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001556 // Compute the final error code with all the bit flags to be sent to Omaha.
Amin Hassani7cc8bb02019-01-14 16:29:47 -08001557 code =
1558 static_cast<ErrorCode>(static_cast<uint32_t>(code) | GetErrorCodeFlags());
1559 error_event_.reset(
1560 new OmahaEvent(OmahaEvent::kTypeUpdateComplete, event_result, code));
Darin Petkov09f96c32010-07-20 09:24:57 -07001561}
1562
1563bool UpdateAttempter::ScheduleErrorEventAction() {
Alex Vakulenko88b591f2014-08-28 16:48:57 -07001564 if (error_event_.get() == nullptr)
Darin Petkov09f96c32010-07-20 09:24:57 -07001565 return false;
1566
Jay Srinivasan6f6ea002012-12-14 11:26:28 -08001567 LOG(ERROR) << "Update failed.";
Amin Hassani538bd592020-11-04 20:46:08 -08001568 SystemState::Get()->payload_state()->UpdateFailed(error_event_->error_code);
Jay Srinivasan6f6ea002012-12-14 11:26:28 -08001569
Marton Hunyadya0302682018-05-16 18:52:13 +02001570 // Send metrics if it was a rollback.
Amin Hassanid3f4bea2018-04-30 14:52:40 -07001571 if (install_plan_ && install_plan_->is_rollback) {
Amin Hassani538bd592020-11-04 20:46:08 -08001572 SystemState::Get()->metrics_reporter()->ReportEnterpriseRollbackMetrics(
Amin Hassanid3f4bea2018-04-30 14:52:40 -07001573 /*success=*/false, install_plan_->version);
Marton Hunyadya0302682018-05-16 18:52:13 +02001574 }
1575
Jay Srinivasan55f50c22013-01-10 19:24:35 -08001576 // Send it to Omaha.
Alex Deymoaf9a8632015-09-23 18:51:48 -07001577 LOG(INFO) << "Reporting the error event";
Amin Hassanid3f4bea2018-04-30 14:52:40 -07001578 auto error_event_action = std::make_unique<OmahaRequestAction>(
Amin Hassanid3f4bea2018-04-30 14:52:40 -07001579 error_event_.release(), // Pass ownership.
1580 std::make_unique<LibcurlHttpFetcher>(GetProxyResolver(),
Amin Hassani538bd592020-11-04 20:46:08 -08001581 SystemState::Get()->hardware()),
Jae Hoon Kimedb65502019-06-14 11:52:17 -07001582 false,
1583 session_id_);
Amin Hassanid3f4bea2018-04-30 14:52:40 -07001584 processor_->EnqueueAction(std::move(error_event_action));
Christopher Wileycc8ce0e2015-10-01 16:48:47 -07001585 SetStatusAndNotify(UpdateStatus::REPORTING_ERROR_EVENT);
Darin Petkovf42cc1c2010-09-01 09:03:02 -07001586 processor_->StartProcessing();
Darin Petkov09f96c32010-07-20 09:24:57 -07001587 return true;
1588}
1589
Darin Petkov58dd1342011-05-06 12:05:13 -07001590void UpdateAttempter::ScheduleProcessingStart() {
1591 LOG(INFO) << "Scheduling an action processor start.";
Alex Deymo60ca1a72015-06-18 18:19:15 -07001592 MessageLoop::current()->PostTask(
1593 FROM_HERE,
Luis Hector Chavezf1cf3482016-07-19 14:29:19 -07001594 Bind([](ActionProcessor* processor) { processor->StartProcessing(); },
1595 base::Unretained(processor_.get())));
Darin Petkov58dd1342011-05-06 12:05:13 -07001596}
1597
Darin Petkov36275772010-10-01 11:40:57 -07001598void UpdateAttempter::DisableDeltaUpdateIfNeeded() {
1599 int64_t delta_failures;
Jay Srinivasanae4697c2013-03-18 17:08:08 -07001600 if (omaha_request_params_->delta_okay() &&
Darin Petkov36275772010-10-01 11:40:57 -07001601 prefs_->GetInt64(kPrefsDeltaUpdateFailures, &delta_failures) &&
1602 delta_failures >= kMaxDeltaUpdateFailures) {
1603 LOG(WARNING) << "Too many delta update failures, forcing full update.";
Jay Srinivasanae4697c2013-03-18 17:08:08 -07001604 omaha_request_params_->set_delta_okay(false);
Darin Petkov36275772010-10-01 11:40:57 -07001605 }
1606}
1607
1608void UpdateAttempter::MarkDeltaUpdateFailure() {
Darin Petkov2dd01092010-10-08 15:43:05 -07001609 // Don't try to resume a failed delta update.
1610 DeltaPerformer::ResetUpdateProgress(prefs_, false);
Darin Petkov36275772010-10-01 11:40:57 -07001611 int64_t delta_failures;
1612 if (!prefs_->GetInt64(kPrefsDeltaUpdateFailures, &delta_failures) ||
1613 delta_failures < 0) {
1614 delta_failures = 0;
1615 }
1616 prefs_->SetInt64(kPrefsDeltaUpdateFailures, ++delta_failures);
1617}
1618
Thieu Le116fda32011-04-19 11:01:54 -07001619void UpdateAttempter::PingOmaha() {
Thieu Led88a8572011-05-26 09:09:19 -07001620 if (!processor_->IsRunning()) {
Amin Hassaniecb60d32019-06-17 18:09:10 -07001621 ResetInteractivityFlags();
1622
Amin Hassanid3f4bea2018-04-30 14:52:40 -07001623 auto ping_action = std::make_unique<OmahaRequestAction>(
Alex Deymoc1c17b42015-11-23 03:53:15 -03001624 nullptr,
Ben Chanab5a0af2017-10-12 14:57:50 -07001625 std::make_unique<LibcurlHttpFetcher>(GetProxyResolver(),
Amin Hassani538bd592020-11-04 20:46:08 -08001626 SystemState::Get()->hardware()),
Jae Hoon Kimedb65502019-06-14 11:52:17 -07001627 true,
1628 "" /* session_id */);
Alex Vakulenko88b591f2014-08-28 16:48:57 -07001629 processor_->set_delegate(nullptr);
Amin Hassanid3f4bea2018-04-30 14:52:40 -07001630 processor_->EnqueueAction(std::move(ping_action));
Thieu Led88a8572011-05-26 09:09:19 -07001631 // Call StartProcessing() synchronously here to avoid any race conditions
1632 // caused by multiple outstanding ping Omaha requests. If we call
1633 // StartProcessing() asynchronously, the device can be suspended before we
1634 // get a chance to callback to StartProcessing(). When the device resumes
1635 // (assuming the device sleeps longer than the next update check period),
1636 // StartProcessing() is called back and at the same time, the next update
1637 // check is fired which eventually invokes StartProcessing(). A crash
1638 // can occur because StartProcessing() checks to make sure that the
1639 // processor is idle which it isn't due to the two concurrent ping Omaha
1640 // requests.
1641 processor_->StartProcessing();
1642 } else {
Darin Petkov58dd1342011-05-06 12:05:13 -07001643 LOG(WARNING) << "Action processor running, Omaha ping suppressed.";
Darin Petkov58dd1342011-05-06 12:05:13 -07001644 }
Thieu Led88a8572011-05-26 09:09:19 -07001645
Gilad Arnoldec7f9162014-07-15 13:24:46 -07001646 // Update the last check time here; it may be re-updated when an Omaha
1647 // response is received, but this will prevent us from repeatedly scheduling
1648 // checks in the case where a response is not received.
1649 UpdateLastCheckedTime();
1650
Thieu Led88a8572011-05-26 09:09:19 -07001651 // Update the status which will schedule the next update check
Christopher Wileycc8ce0e2015-10-01 16:48:47 -07001652 SetStatusAndNotify(UpdateStatus::UPDATED_NEED_REBOOT);
Gilad Arnold54fa66d2014-09-29 13:14:29 -07001653 ScheduleUpdates();
Thieu Le116fda32011-04-19 11:01:54 -07001654}
1655
Jay Srinivasan480ddfa2012-06-01 19:15:26 -07001656bool UpdateAttempter::DecrementUpdateCheckCount() {
Ben Chan9abb7632014-08-07 00:10:53 -07001657 int64_t update_check_count_value;
Jay Srinivasan480ddfa2012-06-01 19:15:26 -07001658
1659 if (!prefs_->Exists(kPrefsUpdateCheckCount)) {
1660 // This file does not exist. This means we haven't started our update
1661 // check count down yet, so nothing more to do. This file will be created
1662 // later when we first satisfy the wall-clock-based-wait period.
1663 LOG(INFO) << "No existing update check count. That's normal.";
1664 return true;
1665 }
1666
1667 if (prefs_->GetInt64(kPrefsUpdateCheckCount, &update_check_count_value)) {
1668 // Only if we're able to read a proper integer value, then go ahead
1669 // and decrement and write back the result in the same file, if needed.
1670 LOG(INFO) << "Update check count = " << update_check_count_value;
1671
1672 if (update_check_count_value == 0) {
1673 // It could be 0, if, for some reason, the file didn't get deleted
1674 // when we set our status to waiting for reboot. so we just leave it
1675 // as is so that we can prevent another update_check wait for this client.
1676 LOG(INFO) << "Not decrementing update check count as it's already 0.";
1677 return true;
1678 }
1679
1680 if (update_check_count_value > 0)
1681 update_check_count_value--;
1682 else
1683 update_check_count_value = 0;
1684
1685 // Write out the new value of update_check_count_value.
1686 if (prefs_->SetInt64(kPrefsUpdateCheckCount, update_check_count_value)) {
Sen Jiang771f6482018-04-04 17:59:10 -07001687 // We successfully wrote out the new value, so enable the
Jay Srinivasan480ddfa2012-06-01 19:15:26 -07001688 // update check based wait.
1689 LOG(INFO) << "New update check count = " << update_check_count_value;
1690 return true;
1691 }
1692 }
1693
1694 LOG(INFO) << "Deleting update check count state due to read/write errors.";
1695
1696 // We cannot read/write to the file, so disable the update check based wait
1697 // so that we don't get stuck in this OS version by any chance (which could
1698 // happen if there's some bug that causes to read/write incorrectly).
1699 // Also attempt to delete the file to do our best effort to cleanup.
1700 prefs_->Delete(kPrefsUpdateCheckCount);
1701 return false;
1702}
Chris Sosad317e402013-06-12 13:47:09 -07001703
David Zeuthene4c58bf2013-06-18 17:26:50 -07001704void UpdateAttempter::UpdateEngineStarted() {
Alex Vakulenkodea2eac2014-03-14 15:56:59 -07001705 // If we just booted into a new update, keep the previous OS version
1706 // in case we rebooted because of a crash of the old version, so we
Alex Vakulenko88b591f2014-08-28 16:48:57 -07001707 // can do a proper crash report with correct information.
Alex Vakulenkodea2eac2014-03-14 15:56:59 -07001708 // This must be done before calling
Amin Hassani538bd592020-11-04 20:46:08 -08001709 // SystemState::Get()->payload_state()->UpdateEngineStarted() since it will
Alex Vakulenkodea2eac2014-03-14 15:56:59 -07001710 // delete SystemUpdated marker file.
Amin Hassani538bd592020-11-04 20:46:08 -08001711 if (SystemState::Get()->system_rebooted() &&
Alex Vakulenkodea2eac2014-03-14 15:56:59 -07001712 prefs_->Exists(kPrefsSystemUpdatedMarker)) {
1713 if (!prefs_->GetString(kPrefsPreviousVersion, &prev_version_)) {
1714 // If we fail to get the version string, make sure it stays empty.
1715 prev_version_.clear();
1716 }
1717 }
1718
Vyshu6e8a37b2020-11-25 10:31:00 -08001719 MoveToPrefs({kPrefsLastRollCallPingDay, kPrefsLastActivePingDay});
1720
Amin Hassani538bd592020-11-04 20:46:08 -08001721 SystemState::Get()->payload_state()->UpdateEngineStarted();
David Zeuthen8f191b22013-08-06 12:27:50 -07001722 StartP2PAtStartup();
Jae Hoon Kim5e8e30b2020-05-06 14:59:06 -07001723
Amin Hassani90e9f192020-11-18 14:20:56 -08001724 excluder_ = CreateExcluder();
David Zeuthen8f191b22013-08-06 12:27:50 -07001725}
1726
Vyshu6e8a37b2020-11-25 10:31:00 -08001727void UpdateAttempter::MoveToPrefs(const vector<string>& keys) {
1728 auto* powerwash_safe_prefs = SystemState::Get()->powerwash_safe_prefs();
1729 for (const auto& key : keys) {
1730 // Do not overwrite existing pref key with powerwash prefs.
1731 if (!prefs_->Exists(key) && powerwash_safe_prefs->Exists(key)) {
1732 string value;
1733 if (!powerwash_safe_prefs->GetString(key, &value) ||
1734 !prefs_->SetString(key, value)) {
1735 PLOG(ERROR) << "Unable to add powerwash safe key " << key
1736 << " to prefs. Powerwash safe key will be deleted.";
1737 }
1738 }
1739 // Delete keys regardless of operation success to preserve privacy.
1740 powerwash_safe_prefs->Delete(key);
1741 }
1742}
1743
David Zeuthen8f191b22013-08-06 12:27:50 -07001744bool UpdateAttempter::StartP2PAtStartup() {
Amin Hassani538bd592020-11-04 20:46:08 -08001745 if (!SystemState::Get()->p2p_manager()->IsP2PEnabled()) {
David Zeuthen8f191b22013-08-06 12:27:50 -07001746 LOG(INFO) << "Not starting p2p at startup since it's not enabled.";
1747 return false;
1748 }
1749
Amin Hassani538bd592020-11-04 20:46:08 -08001750 if (SystemState::Get()->p2p_manager()->CountSharedFiles() < 1) {
David Zeuthen8f191b22013-08-06 12:27:50 -07001751 LOG(INFO) << "Not starting p2p at startup since our application "
1752 << "is not sharing any files.";
1753 return false;
1754 }
1755
1756 return StartP2PAndPerformHousekeeping();
1757}
1758
1759bool UpdateAttempter::StartP2PAndPerformHousekeeping() {
Amin Hassani538bd592020-11-04 20:46:08 -08001760 if (!SystemState::Get()->p2p_manager()->IsP2PEnabled()) {
David Zeuthen8f191b22013-08-06 12:27:50 -07001761 LOG(INFO) << "Not starting p2p since it's not enabled.";
1762 return false;
1763 }
1764
1765 LOG(INFO) << "Ensuring that p2p is running.";
Amin Hassani538bd592020-11-04 20:46:08 -08001766 if (!SystemState::Get()->p2p_manager()->EnsureP2PRunning()) {
David Zeuthen8f191b22013-08-06 12:27:50 -07001767 LOG(ERROR) << "Error starting p2p.";
1768 return false;
1769 }
1770
1771 LOG(INFO) << "Performing p2p housekeeping.";
Amin Hassani538bd592020-11-04 20:46:08 -08001772 if (!SystemState::Get()->p2p_manager()->PerformHousekeeping()) {
David Zeuthen8f191b22013-08-06 12:27:50 -07001773 LOG(ERROR) << "Error performing housekeeping for p2p.";
1774 return false;
1775 }
1776
1777 LOG(INFO) << "Done performing p2p housekeeping.";
1778 return true;
David Zeuthene4c58bf2013-06-18 17:26:50 -07001779}
1780
Amin Hassani7cc8bb02019-01-14 16:29:47 -08001781bool UpdateAttempter::GetBootTimeAtUpdate(Time* out_boot_time) {
Alex Deymo906191f2015-10-12 12:22:44 -07001782 // In case of an update_engine restart without a reboot, we stored the boot_id
1783 // when the update was completed by setting a pref, so we can check whether
1784 // the last update was on this boot or a previous one.
1785 string boot_id;
1786 TEST_AND_RETURN_FALSE(utils::GetBootId(&boot_id));
1787
1788 string update_completed_on_boot_id;
1789 if (!prefs_->Exists(kPrefsUpdateCompletedOnBootId) ||
1790 !prefs_->GetString(kPrefsUpdateCompletedOnBootId,
1791 &update_completed_on_boot_id) ||
1792 update_completed_on_boot_id != boot_id)
David Zeuthen3c55abd2013-10-14 12:48:03 -07001793 return false;
1794
Alex Deymo906191f2015-10-12 12:22:44 -07001795 // Short-circuit avoiding the read in case out_boot_time is nullptr.
1796 if (out_boot_time) {
1797 int64_t boot_time = 0;
1798 // Since the kPrefsUpdateCompletedOnBootId was correctly set, this pref
1799 // should not fail.
1800 TEST_AND_RETURN_FALSE(
1801 prefs_->GetInt64(kPrefsUpdateCompletedBootTime, &boot_time));
1802 *out_boot_time = Time::FromInternalValue(boot_time);
David Zeuthen3c55abd2013-10-14 12:48:03 -07001803 }
David Zeuthen3c55abd2013-10-14 12:48:03 -07001804 return true;
1805}
1806
Jae Hoon Kimba2fdce2019-07-11 13:18:58 -07001807bool UpdateAttempter::IsBusyOrUpdateScheduled() {
Christopher Wileycc8ce0e2015-10-01 16:48:47 -07001808 return ((status_ != UpdateStatus::IDLE &&
1809 status_ != UpdateStatus::UPDATED_NEED_REBOOT) ||
Gilad Arnold54fa66d2014-09-29 13:14:29 -07001810 waiting_for_scheduled_check_);
1811}
1812
Sen Jiangdaeaa432018-10-09 18:18:45 -07001813bool UpdateAttempter::IsAnyUpdateSourceAllowed() const {
David Pursell907b4fa2015-01-27 10:27:38 -08001814 // We allow updates from any source if either of these are true:
1815 // * The device is running an unofficial (dev/test) image.
1816 // * The debugd dev features are accessible (i.e. in devmode with no owner).
1817 // This protects users running a base image, while still allowing a specific
1818 // window (gated by the debug dev features) where `cros flash` is usable.
Amin Hassani538bd592020-11-04 20:46:08 -08001819 if (!SystemState::Get()->hardware()->IsOfficialBuild()) {
David Pursell02c18642014-11-06 11:26:11 -08001820 LOG(INFO) << "Non-official build; allowing any update source.";
1821 return true;
1822 }
1823
Amin Hassani538bd592020-11-04 20:46:08 -08001824 if (SystemState::Get()->hardware()->AreDevFeaturesEnabled()) {
Sen Jiange67bb5b2016-06-20 15:53:56 -07001825 LOG(INFO) << "Developer features enabled; allowing custom update sources.";
David Pursell02c18642014-11-06 11:26:11 -08001826 return true;
1827 }
Sen Jiange67bb5b2016-06-20 15:53:56 -07001828
1829 LOG(INFO)
1830 << "Developer features disabled; disallowing custom update sources.";
David Pursell02c18642014-11-06 11:26:11 -08001831 return false;
1832}
1833
May Lippert60aa3ca2018-08-15 16:55:29 -07001834void UpdateAttempter::ReportTimeToUpdateAppliedMetric() {
Amin Hassani538bd592020-11-04 20:46:08 -08001835 const policy::DevicePolicy* device_policy =
1836 SystemState::Get()->device_policy();
May Lippert60aa3ca2018-08-15 16:55:29 -07001837 if (device_policy && device_policy->IsEnterpriseEnrolled()) {
1838 vector<policy::DevicePolicy::WeeklyTimeInterval> parsed_intervals;
1839 bool has_time_restrictions =
1840 device_policy->GetDisallowedTimeIntervals(&parsed_intervals);
1841
1842 int64_t update_first_seen_at_int;
Amin Hassani538bd592020-11-04 20:46:08 -08001843 if (SystemState::Get()->prefs()->Exists(kPrefsUpdateFirstSeenAt)) {
1844 if (SystemState::Get()->prefs()->GetInt64(kPrefsUpdateFirstSeenAt,
1845 &update_first_seen_at_int)) {
May Lippert60aa3ca2018-08-15 16:55:29 -07001846 TimeDelta update_delay =
Amin Hassani538bd592020-11-04 20:46:08 -08001847 SystemState::Get()->clock()->GetWallclockTime() -
May Lippert60aa3ca2018-08-15 16:55:29 -07001848 Time::FromInternalValue(update_first_seen_at_int);
Amin Hassani538bd592020-11-04 20:46:08 -08001849 SystemState::Get()
1850 ->metrics_reporter()
May Lippert60aa3ca2018-08-15 16:55:29 -07001851 ->ReportEnterpriseUpdateSeenToDownloadDays(has_time_restrictions,
1852 update_delay.InDays());
1853 }
1854 }
1855 }
1856}
1857
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -07001858} // namespace chromeos_update_engine