blob: 131408610524c83f676e8f86785d631360f721c9 [file] [log] [blame]
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -07001// Copyright (c) 2010 The Chromium OS Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#include "update_engine/update_attempter.h"
Andrew de los Reyes63b96d72010-05-10 13:08:54 -07006
7// From 'man clock_gettime': feature test macro: _POSIX_C_SOURCE >= 199309L
8#ifndef _POSIX_C_SOURCE
9#define _POSIX_C_SOURCE 199309L
10#endif // _POSIX_C_SOURCE
11#include <time.h>
12
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -070013#include <string>
Darin Petkov9b230572010-10-08 10:20:09 -070014#include <tr1/memory>
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -070015#include <vector>
Darin Petkov9d65b7b2010-07-20 09:13:01 -070016
Andrew de los Reyes45168102010-11-22 11:13:50 -080017#include <base/rand_util.h>
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -070018#include <glib.h>
Darin Petkov1023a602010-08-30 13:47:51 -070019#include <metrics/metrics_library.h>
Darin Petkov9d65b7b2010-07-20 09:13:01 -070020
Andrew de los Reyes63b96d72010-05-10 13:08:54 -070021#include "update_engine/dbus_service.h"
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -070022#include "update_engine/download_action.h"
23#include "update_engine/filesystem_copier_action.h"
24#include "update_engine/libcurl_http_fetcher.h"
Andrew de los Reyes819fef22010-12-17 11:33:58 -080025#include "update_engine/multi_range_http_fetcher.h"
Darin Petkov6a5b3222010-07-13 14:55:28 -070026#include "update_engine/omaha_request_action.h"
Darin Petkova4a8a8c2010-07-15 22:21:12 -070027#include "update_engine/omaha_request_params.h"
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -070028#include "update_engine/omaha_response_handler_action.h"
29#include "update_engine/postinstall_runner_action.h"
Darin Petkov36275772010-10-01 11:40:57 -070030#include "update_engine/prefs_interface.h"
Andrew de los Reyes6dbf30a2011-04-19 10:58:16 -070031#include "update_engine/subprocess.h"
Darin Petkov1023a602010-08-30 13:47:51 -070032#include "update_engine/update_check_scheduler.h"
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -070033
Darin Petkovaf183052010-08-23 12:07:13 -070034using base::TimeDelta;
35using base::TimeTicks;
Andrew de los Reyes21816e12011-04-07 14:18:56 -070036using google::protobuf::NewPermanentCallback;
Darin Petkov9b230572010-10-08 10:20:09 -070037using std::make_pair;
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -070038using std::tr1::shared_ptr;
39using std::string;
40using std::vector;
41
42namespace chromeos_update_engine {
43
Darin Petkov36275772010-10-01 11:40:57 -070044const int UpdateAttempter::kMaxDeltaUpdateFailures = 3;
45
Darin Petkovcd1666f2010-09-23 09:53:44 -070046const char* kUpdateCompletedMarker =
47 "/var/run/update_engine_autoupdate_completed";
Andrew de los Reyes6b78e292010-05-10 15:54:39 -070048
Andrew de los Reyes45168102010-11-22 11:13:50 -080049namespace {
50const int kMaxConsecutiveObeyProxyRequests = 20;
51} // namespace {}
52
Andrew de los Reyes63b96d72010-05-10 13:08:54 -070053const char* UpdateStatusToString(UpdateStatus status) {
54 switch (status) {
55 case UPDATE_STATUS_IDLE:
56 return "UPDATE_STATUS_IDLE";
57 case UPDATE_STATUS_CHECKING_FOR_UPDATE:
58 return "UPDATE_STATUS_CHECKING_FOR_UPDATE";
59 case UPDATE_STATUS_UPDATE_AVAILABLE:
60 return "UPDATE_STATUS_UPDATE_AVAILABLE";
61 case UPDATE_STATUS_DOWNLOADING:
62 return "UPDATE_STATUS_DOWNLOADING";
63 case UPDATE_STATUS_VERIFYING:
64 return "UPDATE_STATUS_VERIFYING";
65 case UPDATE_STATUS_FINALIZING:
66 return "UPDATE_STATUS_FINALIZING";
67 case UPDATE_STATUS_UPDATED_NEED_REBOOT:
68 return "UPDATE_STATUS_UPDATED_NEED_REBOOT";
Darin Petkov09f96c32010-07-20 09:24:57 -070069 case UPDATE_STATUS_REPORTING_ERROR_EVENT:
70 return "UPDATE_STATUS_REPORTING_ERROR_EVENT";
Andrew de los Reyes63b96d72010-05-10 13:08:54 -070071 default:
72 return "unknown status";
73 }
74}
75
Darin Petkov777dbfa2010-07-20 15:03:37 -070076// Turns a generic kActionCodeError to a generic error code specific
77// to |action| (e.g., kActionCodeFilesystemCopierError). If |code| is
78// not kActionCodeError, or the action is not matched, returns |code|
79// unchanged.
80ActionExitCode GetErrorCodeForAction(AbstractAction* action,
81 ActionExitCode code) {
82 if (code != kActionCodeError)
83 return code;
84
85 const string type = action->Type();
86 if (type == OmahaRequestAction::StaticType())
87 return kActionCodeOmahaRequestError;
88 if (type == OmahaResponseHandlerAction::StaticType())
89 return kActionCodeOmahaResponseHandlerError;
90 if (type == FilesystemCopierAction::StaticType())
91 return kActionCodeFilesystemCopierError;
92 if (type == PostinstallRunnerAction::StaticType())
93 return kActionCodePostinstallRunnerError;
Darin Petkov777dbfa2010-07-20 15:03:37 -070094
95 return code;
96}
97
Darin Petkovc6c135c2010-08-11 13:36:18 -070098UpdateAttempter::UpdateAttempter(PrefsInterface* prefs,
Andrew de los Reyes45168102010-11-22 11:13:50 -080099 MetricsLibraryInterface* metrics_lib,
100 DbusGlibInterface* dbus_iface)
Darin Petkovf42cc1c2010-09-01 09:03:02 -0700101 : processor_(new ActionProcessor()),
102 dbus_service_(NULL),
Darin Petkovc6c135c2010-08-11 13:36:18 -0700103 prefs_(prefs),
104 metrics_lib_(metrics_lib),
Darin Petkov1023a602010-08-30 13:47:51 -0700105 update_check_scheduler_(NULL),
Andrew de los Reyesc1d5c932011-04-20 17:15:47 -0700106 fake_update_success_(false),
Darin Petkov1023a602010-08-30 13:47:51 -0700107 http_response_code_(0),
Darin Petkovc6c135c2010-08-11 13:36:18 -0700108 priority_(utils::kProcessPriorityNormal),
109 manage_priority_source_(NULL),
Darin Petkov9d911fa2010-08-19 09:36:08 -0700110 download_active_(false),
Darin Petkovc6c135c2010-08-11 13:36:18 -0700111 status_(UPDATE_STATUS_IDLE),
112 download_progress_(0.0),
113 last_checked_time_(0),
114 new_version_("0.0.0.0"),
Darin Petkov36275772010-10-01 11:40:57 -0700115 new_size_(0),
Andrew de los Reyes45168102010-11-22 11:13:50 -0800116 is_full_update_(false),
117 proxy_manual_checks_(0),
118 obeying_proxies_(true),
Andrew de los Reyes6dbf30a2011-04-19 10:58:16 -0700119 chrome_proxy_resolver_(dbus_iface),
120 updated_boot_flags_(false) {
Darin Petkovc6c135c2010-08-11 13:36:18 -0700121 if (utils::FileExists(kUpdateCompletedMarker))
122 status_ = UPDATE_STATUS_UPDATED_NEED_REBOOT;
123}
124
125UpdateAttempter::~UpdateAttempter() {
126 CleanupPriorityManagement();
127}
128
Darin Petkov5a7f5652010-07-22 21:40:09 -0700129void UpdateAttempter::Update(const std::string& app_version,
Andrew de los Reyes45168102010-11-22 11:13:50 -0800130 const std::string& omaha_url,
131 bool obey_proxies) {
Andrew de los Reyes000d8952011-03-02 15:21:14 -0800132 chrome_proxy_resolver_.Init();
Andrew de los Reyesc1d5c932011-04-20 17:15:47 -0700133 fake_update_success_ = false;
Andrew de los Reyes6dbf30a2011-04-19 10:58:16 -0700134 UpdateBootFlags(); // Just in case we didn't do this yet.
Andrew de los Reyes6b78e292010-05-10 15:54:39 -0700135 if (status_ == UPDATE_STATUS_UPDATED_NEED_REBOOT) {
Thieu Le116fda32011-04-19 11:01:54 -0700136 // Although we have applied an update, we still want to ping Omaha
137 // to ensure the number of active statistics is accurate.
Andrew de los Reyes6b78e292010-05-10 15:54:39 -0700138 LOG(INFO) << "Not updating b/c we already updated and we're waiting for "
Thieu Le116fda32011-04-19 11:01:54 -0700139 << "reboot, we'll ping Omaha instead";
140 PingOmaha();
Andrew de los Reyes6b78e292010-05-10 15:54:39 -0700141 return;
142 }
143 if (status_ != UPDATE_STATUS_IDLE) {
144 // Update in progress. Do nothing
145 return;
146 }
Darin Petkov1023a602010-08-30 13:47:51 -0700147 http_response_code_ = 0;
Darin Petkov5a7f5652010-07-22 21:40:09 -0700148 if (!omaha_request_params_.Init(app_version, omaha_url)) {
Darin Petkova4a8a8c2010-07-15 22:21:12 -0700149 LOG(ERROR) << "Unable to initialize Omaha request device params.";
150 return;
151 }
Darin Petkov3aefa862010-12-07 14:45:00 -0800152
Andrew de los Reyes45168102010-11-22 11:13:50 -0800153 obeying_proxies_ = true;
154 if (obey_proxies || proxy_manual_checks_ == 0) {
155 LOG(INFO) << "forced to obey proxies";
156 // If forced to obey proxies, every 20th request will not use proxies
157 proxy_manual_checks_++;
158 LOG(INFO) << "proxy manual checks: " << proxy_manual_checks_;
159 if (proxy_manual_checks_ >= kMaxConsecutiveObeyProxyRequests) {
160 proxy_manual_checks_ = 0;
161 obeying_proxies_ = false;
162 }
163 } else if (base::RandInt(0, 4) == 0) {
164 obeying_proxies_ = false;
165 }
166 LOG_IF(INFO, !obeying_proxies_) << "To help ensure updates work, this update "
167 "check we are ignoring the proxy settings and using "
168 "direct connections.";
169
Darin Petkov36275772010-10-01 11:40:57 -0700170 DisableDeltaUpdateIfNeeded();
Darin Petkovf42cc1c2010-09-01 09:03:02 -0700171 CHECK(!processor_->IsRunning());
172 processor_->set_delegate(this);
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700173
174 // Actions:
Darin Petkova0929552010-11-29 14:19:06 -0800175 LibcurlHttpFetcher* update_check_fetcher =
176 new LibcurlHttpFetcher(GetProxyResolver());
Andrew de los Reyes5d0783d2010-11-29 18:14:16 -0800177 // Try harder to connect to the network. See comment in
178 // libcurl_http_fetcher.cc.
179 update_check_fetcher->set_no_network_max_retries(3);
Darin Petkov6a5b3222010-07-13 14:55:28 -0700180 shared_ptr<OmahaRequestAction> update_check_action(
Darin Petkov1cbd78f2010-07-29 12:38:34 -0700181 new OmahaRequestAction(prefs_,
182 omaha_request_params_,
Darin Petkova4a8a8c2010-07-15 22:21:12 -0700183 NULL,
Thieu Le116fda32011-04-19 11:01:54 -0700184 update_check_fetcher, // passes ownership
185 false));
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700186 shared_ptr<OmahaResponseHandlerAction> response_handler_action(
Darin Petkov73058b42010-10-06 16:32:19 -0700187 new OmahaResponseHandlerAction(prefs_));
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700188 shared_ptr<FilesystemCopierAction> filesystem_copier_action(
Darin Petkov3aefa862010-12-07 14:45:00 -0800189 new FilesystemCopierAction(false, false));
Andrew de los Reyesf9714432010-05-04 10:21:23 -0700190 shared_ptr<FilesystemCopierAction> kernel_filesystem_copier_action(
Darin Petkov3aefa862010-12-07 14:45:00 -0800191 new FilesystemCopierAction(true, false));
Darin Petkov8c2980e2010-07-16 15:16:49 -0700192 shared_ptr<OmahaRequestAction> download_started_action(
Darin Petkov1cbd78f2010-07-29 12:38:34 -0700193 new OmahaRequestAction(prefs_,
194 omaha_request_params_,
Darin Petkov8c2980e2010-07-16 15:16:49 -0700195 new OmahaEvent(
Darin Petkove17f86b2010-07-20 09:12:01 -0700196 OmahaEvent::kTypeUpdateDownloadStarted),
Thieu Le116fda32011-04-19 11:01:54 -0700197 new LibcurlHttpFetcher(GetProxyResolver()),
198 false));
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700199 shared_ptr<DownloadAction> download_action(
Andrew de los Reyes819fef22010-12-17 11:33:58 -0800200 new DownloadAction(prefs_, new MultiRangeHTTPFetcher(
201 new LibcurlHttpFetcher(GetProxyResolver()))));
Andrew de los Reyes21816e12011-04-07 14:18:56 -0700202 // This action is always initially in place to warn OS vendor of a
203 // signature failure. If it's not needed, it will be told to skip.
204 shared_ptr<OmahaRequestAction> download_signature_warning(
205 new OmahaRequestAction(
206 prefs_,
207 omaha_request_params_,
208 new OmahaEvent(
209 OmahaEvent::kTypeUpdateDownloadFinished,
210 OmahaEvent::kResultError,
211 kActionCodeDownloadPayloadPubKeyVerificationError),
Thieu Le116fda32011-04-19 11:01:54 -0700212 new LibcurlHttpFetcher(GetProxyResolver()),
213 false));
Andrew de los Reyes21816e12011-04-07 14:18:56 -0700214 download_action->set_skip_reporting_signature_fail(
215 NewPermanentCallback(download_signature_warning.get(),
216 &OmahaRequestAction::set_should_skip,
217 true));
Darin Petkov8c2980e2010-07-16 15:16:49 -0700218 shared_ptr<OmahaRequestAction> download_finished_action(
Darin Petkov1cbd78f2010-07-29 12:38:34 -0700219 new OmahaRequestAction(prefs_,
220 omaha_request_params_,
Darin Petkov8c2980e2010-07-16 15:16:49 -0700221 new OmahaEvent(
Darin Petkove17f86b2010-07-20 09:12:01 -0700222 OmahaEvent::kTypeUpdateDownloadFinished),
Thieu Le116fda32011-04-19 11:01:54 -0700223 new LibcurlHttpFetcher(GetProxyResolver()),
224 false));
Darin Petkov3aefa862010-12-07 14:45:00 -0800225 shared_ptr<FilesystemCopierAction> filesystem_verifier_action(
226 new FilesystemCopierAction(false, true));
227 shared_ptr<FilesystemCopierAction> kernel_filesystem_verifier_action(
228 new FilesystemCopierAction(true, true));
Darin Petkov6d5dbf62010-11-08 16:09:55 -0800229 shared_ptr<PostinstallRunnerAction> postinstall_runner_action(
230 new PostinstallRunnerAction);
Darin Petkov8c2980e2010-07-16 15:16:49 -0700231 shared_ptr<OmahaRequestAction> update_complete_action(
Darin Petkov1cbd78f2010-07-29 12:38:34 -0700232 new OmahaRequestAction(prefs_,
233 omaha_request_params_,
Darin Petkove17f86b2010-07-20 09:12:01 -0700234 new OmahaEvent(OmahaEvent::kTypeUpdateComplete),
Thieu Le116fda32011-04-19 11:01:54 -0700235 new LibcurlHttpFetcher(GetProxyResolver()),
236 false));
Darin Petkov6a5b3222010-07-13 14:55:28 -0700237
Andrew de los Reyes63b96d72010-05-10 13:08:54 -0700238 download_action->set_delegate(this);
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700239 response_handler_action_ = response_handler_action;
Darin Petkov9b230572010-10-08 10:20:09 -0700240 download_action_ = download_action;
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700241
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700242 actions_.push_back(shared_ptr<AbstractAction>(update_check_action));
243 actions_.push_back(shared_ptr<AbstractAction>(response_handler_action));
244 actions_.push_back(shared_ptr<AbstractAction>(filesystem_copier_action));
Andrew de los Reyesf9185172010-05-03 11:07:05 -0700245 actions_.push_back(shared_ptr<AbstractAction>(
Andrew de los Reyesf9714432010-05-04 10:21:23 -0700246 kernel_filesystem_copier_action));
Darin Petkov8c2980e2010-07-16 15:16:49 -0700247 actions_.push_back(shared_ptr<AbstractAction>(download_started_action));
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700248 actions_.push_back(shared_ptr<AbstractAction>(download_action));
Andrew de los Reyes21816e12011-04-07 14:18:56 -0700249 actions_.push_back(shared_ptr<AbstractAction>(download_signature_warning));
Darin Petkov8c2980e2010-07-16 15:16:49 -0700250 actions_.push_back(shared_ptr<AbstractAction>(download_finished_action));
Darin Petkov3aefa862010-12-07 14:45:00 -0800251 actions_.push_back(shared_ptr<AbstractAction>(filesystem_verifier_action));
252 actions_.push_back(shared_ptr<AbstractAction>(
253 kernel_filesystem_verifier_action));
Darin Petkov6d5dbf62010-11-08 16:09:55 -0800254 actions_.push_back(shared_ptr<AbstractAction>(postinstall_runner_action));
Darin Petkov8c2980e2010-07-16 15:16:49 -0700255 actions_.push_back(shared_ptr<AbstractAction>(update_complete_action));
Darin Petkov6a5b3222010-07-13 14:55:28 -0700256
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700257 // Enqueue the actions
258 for (vector<shared_ptr<AbstractAction> >::iterator it = actions_.begin();
259 it != actions_.end(); ++it) {
Darin Petkovf42cc1c2010-09-01 09:03:02 -0700260 processor_->EnqueueAction(it->get());
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700261 }
262
263 // Bond them together. We have to use the leaf-types when calling
264 // BondActions().
Andrew de los Reyesf98bff82010-05-06 13:33:25 -0700265 BondActions(update_check_action.get(),
266 response_handler_action.get());
Andrew de los Reyesf9185172010-05-03 11:07:05 -0700267 BondActions(response_handler_action.get(),
Andrew de los Reyesf98bff82010-05-06 13:33:25 -0700268 filesystem_copier_action.get());
269 BondActions(filesystem_copier_action.get(),
Andrew de los Reyesf9714432010-05-04 10:21:23 -0700270 kernel_filesystem_copier_action.get());
271 BondActions(kernel_filesystem_copier_action.get(),
Andrew de los Reyesf9185172010-05-03 11:07:05 -0700272 download_action.get());
Andrew de los Reyesf98bff82010-05-06 13:33:25 -0700273 BondActions(download_action.get(),
Darin Petkov3aefa862010-12-07 14:45:00 -0800274 filesystem_verifier_action.get());
275 BondActions(filesystem_verifier_action.get(),
276 kernel_filesystem_verifier_action.get());
277 BondActions(kernel_filesystem_verifier_action.get(),
Darin Petkov6d5dbf62010-11-08 16:09:55 -0800278 postinstall_runner_action.get());
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700279
Andrew de los Reyes63b96d72010-05-10 13:08:54 -0700280 SetStatusAndNotify(UPDATE_STATUS_CHECKING_FOR_UPDATE);
Darin Petkove6ef2f82011-03-07 17:31:11 -0800281
282 // Start the processing asynchronously to unblock the event loop.
283 g_idle_add(&StaticStartProcessing, this);
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700284}
285
Darin Petkov5a7f5652010-07-22 21:40:09 -0700286void UpdateAttempter::CheckForUpdate(const std::string& app_version,
287 const std::string& omaha_url) {
Andrew de los Reyes63b96d72010-05-10 13:08:54 -0700288 if (status_ != UPDATE_STATUS_IDLE) {
289 LOG(INFO) << "Check for update requested, but status is "
290 << UpdateStatusToString(status_) << ", so not checking.";
291 return;
292 }
Andrew de los Reyes45168102010-11-22 11:13:50 -0800293 Update(app_version, omaha_url, true);
Andrew de los Reyes63b96d72010-05-10 13:08:54 -0700294}
295
Darin Petkov296889c2010-07-23 16:20:54 -0700296bool UpdateAttempter::RebootIfNeeded() {
297 if (status_ != UPDATE_STATUS_UPDATED_NEED_REBOOT) {
298 LOG(INFO) << "Reboot requested, but status is "
299 << UpdateStatusToString(status_) << ", so not rebooting.";
300 return false;
301 }
302 TEST_AND_RETURN_FALSE(utils::Reboot());
303 return true;
304}
305
Andrew de los Reyes63b96d72010-05-10 13:08:54 -0700306// Delegate methods:
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700307void UpdateAttempter::ProcessingDone(const ActionProcessor* processor,
Darin Petkovc1a8b422010-07-19 11:34:49 -0700308 ActionExitCode code) {
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700309 CHECK(response_handler_action_);
Andrew de los Reyes63b96d72010-05-10 13:08:54 -0700310 LOG(INFO) << "Processing Done.";
Andrew de los Reyes6b78e292010-05-10 15:54:39 -0700311 actions_.clear();
Darin Petkov09f96c32010-07-20 09:24:57 -0700312
Darin Petkovc6c135c2010-08-11 13:36:18 -0700313 // Reset process priority back to normal.
314 CleanupPriorityManagement();
315
Darin Petkov09f96c32010-07-20 09:24:57 -0700316 if (status_ == UPDATE_STATUS_REPORTING_ERROR_EVENT) {
317 LOG(INFO) << "Error event sent.";
318 SetStatusAndNotify(UPDATE_STATUS_IDLE);
Andrew de los Reyesc1d5c932011-04-20 17:15:47 -0700319 if (!fake_update_success_) {
320 return;
321 }
322 LOG(INFO) << "Booted from FW B and tried to install new firmware, "
323 "so requesting reboot from user.";
Darin Petkov09f96c32010-07-20 09:24:57 -0700324 }
325
Darin Petkovc1a8b422010-07-19 11:34:49 -0700326 if (code == kActionCodeSuccess) {
Andrew de los Reyes6b78e292010-05-10 15:54:39 -0700327 utils::WriteFile(kUpdateCompletedMarker, "", 0);
Darin Petkov36275772010-10-01 11:40:57 -0700328 prefs_->SetInt64(kPrefsDeltaUpdateFailures, 0);
Darin Petkov95508da2011-01-05 12:42:29 -0800329 prefs_->SetString(kPrefsPreviousVersion, omaha_request_params_.app_version);
Darin Petkov9b230572010-10-08 10:20:09 -0700330 DeltaPerformer::ResetUpdateProgress(prefs_, false);
331 SetStatusAndNotify(UPDATE_STATUS_UPDATED_NEED_REBOOT);
Darin Petkov9d65b7b2010-07-20 09:13:01 -0700332
333 // Report the time it took to update the system.
334 int64_t update_time = time(NULL) - last_checked_time_;
Andrew de los Reyesc1d5c932011-04-20 17:15:47 -0700335 if (!fake_update_success_)
336 metrics_lib_->SendToUMA("Installer.UpdateTime",
337 static_cast<int>(update_time), // sample
338 1, // min = 1 second
339 20 * 60, // max = 20 minutes
340 50); // buckets
Darin Petkov09f96c32010-07-20 09:24:57 -0700341 return;
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700342 }
Darin Petkov09f96c32010-07-20 09:24:57 -0700343
Darin Petkov1023a602010-08-30 13:47:51 -0700344 if (ScheduleErrorEventAction()) {
Darin Petkov09f96c32010-07-20 09:24:57 -0700345 return;
Darin Petkov1023a602010-08-30 13:47:51 -0700346 }
347 LOG(INFO) << "No update.";
Darin Petkov09f96c32010-07-20 09:24:57 -0700348 SetStatusAndNotify(UPDATE_STATUS_IDLE);
Andrew de los Reyes63b96d72010-05-10 13:08:54 -0700349}
350
351void UpdateAttempter::ProcessingStopped(const ActionProcessor* processor) {
Darin Petkovc6c135c2010-08-11 13:36:18 -0700352 // Reset process priority back to normal.
353 CleanupPriorityManagement();
Andrew de los Reyes63b96d72010-05-10 13:08:54 -0700354 download_progress_ = 0.0;
355 SetStatusAndNotify(UPDATE_STATUS_IDLE);
Andrew de los Reyes6b78e292010-05-10 15:54:39 -0700356 actions_.clear();
Darin Petkov09f96c32010-07-20 09:24:57 -0700357 error_event_.reset(NULL);
Andrew de los Reyes63b96d72010-05-10 13:08:54 -0700358}
359
360// Called whenever an action has finished processing, either successfully
361// or otherwise.
362void UpdateAttempter::ActionCompleted(ActionProcessor* processor,
363 AbstractAction* action,
Darin Petkovc1a8b422010-07-19 11:34:49 -0700364 ActionExitCode code) {
Darin Petkov1023a602010-08-30 13:47:51 -0700365 // Reset download progress regardless of whether or not the download
366 // action succeeded. Also, get the response code from HTTP request
367 // actions (update download as well as the initial update check
368 // actions).
Andrew de los Reyes63b96d72010-05-10 13:08:54 -0700369 const string type = action->Type();
Darin Petkov1023a602010-08-30 13:47:51 -0700370 if (type == DownloadAction::StaticType()) {
Andrew de los Reyes63b96d72010-05-10 13:08:54 -0700371 download_progress_ = 0.0;
Darin Petkov1023a602010-08-30 13:47:51 -0700372 DownloadAction* download_action = dynamic_cast<DownloadAction*>(action);
373 http_response_code_ = download_action->GetHTTPResponseCode();
374 } else if (type == OmahaRequestAction::StaticType()) {
375 OmahaRequestAction* omaha_request_action =
376 dynamic_cast<OmahaRequestAction*>(action);
377 // If the request is not an event, then it's the update-check.
378 if (!omaha_request_action->IsEvent()) {
379 http_response_code_ = omaha_request_action->GetHTTPResponseCode();
Darin Petkov85ced132010-09-01 10:20:56 -0700380 // Forward the server-dictated poll interval to the update check
381 // scheduler, if any.
382 if (update_check_scheduler_) {
383 update_check_scheduler_->set_poll_interval(
384 omaha_request_action->GetOutputObject().poll_interval);
385 }
Darin Petkov1023a602010-08-30 13:47:51 -0700386 }
387 }
Darin Petkov09f96c32010-07-20 09:24:57 -0700388 if (code != kActionCodeSuccess) {
Darin Petkov36275772010-10-01 11:40:57 -0700389 // If this was a delta update attempt and the current state is at or past
390 // the download phase, count the failure in case a switch to full update
391 // becomes necessary. Ignore network transfer timeouts and failures.
392 if (status_ >= UPDATE_STATUS_DOWNLOADING &&
393 !is_full_update_ &&
394 code != kActionCodeDownloadTransferError) {
395 MarkDeltaUpdateFailure();
396 }
Darin Petkov777dbfa2010-07-20 15:03:37 -0700397 // On failure, schedule an error event to be sent to Omaha.
398 CreatePendingErrorEvent(action, code);
Andrew de los Reyes63b96d72010-05-10 13:08:54 -0700399 return;
Darin Petkov09f96c32010-07-20 09:24:57 -0700400 }
Andrew de los Reyes63b96d72010-05-10 13:08:54 -0700401 // Find out which action completed.
402 if (type == OmahaResponseHandlerAction::StaticType()) {
Darin Petkov9b230572010-10-08 10:20:09 -0700403 // Note that the status will be updated to DOWNLOADING when some bytes get
404 // actually downloaded from the server and the BytesReceived callback is
405 // invoked. This avoids notifying the user that a download has started in
406 // cases when the server and the client are unable to initiate the download.
407 CHECK(action == response_handler_action_.get());
408 const InstallPlan& plan = response_handler_action_->install_plan();
Andrew de los Reyes63b96d72010-05-10 13:08:54 -0700409 last_checked_time_ = time(NULL);
410 // TODO(adlr): put version in InstallPlan
411 new_version_ = "0.0.0.0";
412 new_size_ = plan.size;
Darin Petkov36275772010-10-01 11:40:57 -0700413 is_full_update_ = plan.is_full_update;
Darin Petkov9b230572010-10-08 10:20:09 -0700414 SetupDownload();
Darin Petkovc6c135c2010-08-11 13:36:18 -0700415 SetupPriorityManagement();
Darin Petkovb00bccc2010-10-26 14:13:08 -0700416 SetStatusAndNotify(UPDATE_STATUS_UPDATE_AVAILABLE);
Andrew de los Reyes63b96d72010-05-10 13:08:54 -0700417 } else if (type == DownloadAction::StaticType()) {
418 SetStatusAndNotify(UPDATE_STATUS_FINALIZING);
419 }
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700420}
421
422// Stop updating. An attempt will be made to record status to the disk
423// so that updates can be resumed later.
424void UpdateAttempter::Terminate() {
425 // TODO(adlr): implement this method.
426 NOTIMPLEMENTED();
427}
428
429// Try to resume from a previously Terminate()d update.
430void UpdateAttempter::ResumeUpdating() {
431 // TODO(adlr): implement this method.
432 NOTIMPLEMENTED();
433}
434
Darin Petkov9d911fa2010-08-19 09:36:08 -0700435void UpdateAttempter::SetDownloadStatus(bool active) {
436 download_active_ = active;
437 LOG(INFO) << "Download status: " << (active ? "active" : "inactive");
438}
439
Andrew de los Reyes63b96d72010-05-10 13:08:54 -0700440void UpdateAttempter::BytesReceived(uint64_t bytes_received, uint64_t total) {
Darin Petkov9d911fa2010-08-19 09:36:08 -0700441 if (!download_active_) {
Andrew de los Reyes63b96d72010-05-10 13:08:54 -0700442 LOG(ERROR) << "BytesReceived called while not downloading.";
443 return;
444 }
Darin Petkovaf183052010-08-23 12:07:13 -0700445 double progress = static_cast<double>(bytes_received) /
Andrew de los Reyes63b96d72010-05-10 13:08:54 -0700446 static_cast<double>(total);
Darin Petkovaf183052010-08-23 12:07:13 -0700447 // Self throttle based on progress. Also send notifications if
448 // progress is too slow.
449 const double kDeltaPercent = 0.01; // 1%
450 if (status_ != UPDATE_STATUS_DOWNLOADING ||
451 bytes_received == total ||
452 progress - download_progress_ >= kDeltaPercent ||
453 TimeTicks::Now() - last_notify_time_ >= TimeDelta::FromSeconds(10)) {
454 download_progress_ = progress;
Andrew de los Reyes63b96d72010-05-10 13:08:54 -0700455 SetStatusAndNotify(UPDATE_STATUS_DOWNLOADING);
456 }
457}
458
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700459bool UpdateAttempter::GetStatus(int64_t* last_checked_time,
460 double* progress,
461 std::string* current_operation,
462 std::string* new_version,
463 int64_t* new_size) {
Andrew de los Reyes63b96d72010-05-10 13:08:54 -0700464 *last_checked_time = last_checked_time_;
465 *progress = download_progress_;
466 *current_operation = UpdateStatusToString(status_);
467 *new_version = new_version_;
468 *new_size = new_size_;
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700469 return true;
470}
471
Andrew de los Reyes6dbf30a2011-04-19 10:58:16 -0700472void UpdateAttempter::UpdateBootFlags() {
473 if (updated_boot_flags_) {
474 LOG(INFO) << "Already updated boot flags. Skipping.";
475 return;
476 }
477 // This is purely best effort. Failures should be logged by Subprocess.
478 int unused = 0;
479 vector<string> cmd(1, "/usr/sbin/chromeos-setgoodkernel");
480 Subprocess::SynchronousExec(cmd, &unused);
481 updated_boot_flags_ = true;
482 return;
483}
484
Andrew de los Reyes63b96d72010-05-10 13:08:54 -0700485void UpdateAttempter::SetStatusAndNotify(UpdateStatus status) {
486 status_ = status;
Darin Petkov1023a602010-08-30 13:47:51 -0700487 if (update_check_scheduler_) {
488 update_check_scheduler_->SetUpdateStatus(status_);
489 }
Andrew de los Reyes63b96d72010-05-10 13:08:54 -0700490 if (!dbus_service_)
491 return;
Darin Petkovaf183052010-08-23 12:07:13 -0700492 last_notify_time_ = TimeTicks::Now();
Andrew de los Reyes63b96d72010-05-10 13:08:54 -0700493 update_engine_service_emit_status_update(
494 dbus_service_,
495 last_checked_time_,
496 download_progress_,
497 UpdateStatusToString(status_),
498 new_version_.c_str(),
499 new_size_);
500}
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700501
Darin Petkov777dbfa2010-07-20 15:03:37 -0700502void UpdateAttempter::CreatePendingErrorEvent(AbstractAction* action,
503 ActionExitCode code) {
Darin Petkov09f96c32010-07-20 09:24:57 -0700504 if (error_event_.get()) {
505 // This shouldn't really happen.
506 LOG(WARNING) << "There's already an existing pending error event.";
507 return;
508 }
Darin Petkov777dbfa2010-07-20 15:03:37 -0700509
Darin Petkovabc7bc02011-02-23 14:39:43 -0800510 // For now assume that a generic Omaha response action failure means that
511 // there's no update so don't send an event. Also, double check that the
512 // failure has not occurred while sending an error event -- in which case
513 // don't schedule another. This shouldn't really happen but just in case...
514 if ((action->Type() == OmahaResponseHandlerAction::StaticType() &&
515 code == kActionCodeError) ||
Darin Petkov777dbfa2010-07-20 15:03:37 -0700516 status_ == UPDATE_STATUS_REPORTING_ERROR_EVENT) {
517 return;
518 }
519
520 code = GetErrorCodeForAction(action, code);
Andrew de los Reyesc1d5c932011-04-20 17:15:47 -0700521 fake_update_success_ = code == kActionCodePostinstallBootedFromFirmwareB;
Darin Petkov09f96c32010-07-20 09:24:57 -0700522 error_event_.reset(new OmahaEvent(OmahaEvent::kTypeUpdateComplete,
523 OmahaEvent::kResultError,
524 code));
525}
526
527bool UpdateAttempter::ScheduleErrorEventAction() {
528 if (error_event_.get() == NULL)
529 return false;
530
Darin Petkov1023a602010-08-30 13:47:51 -0700531 LOG(INFO) << "Update failed -- reporting the error event.";
Darin Petkov09f96c32010-07-20 09:24:57 -0700532 shared_ptr<OmahaRequestAction> error_event_action(
Darin Petkov1cbd78f2010-07-29 12:38:34 -0700533 new OmahaRequestAction(prefs_,
534 omaha_request_params_,
Darin Petkov09f96c32010-07-20 09:24:57 -0700535 error_event_.release(), // Pass ownership.
Thieu Le116fda32011-04-19 11:01:54 -0700536 new LibcurlHttpFetcher(GetProxyResolver()),
537 false));
Darin Petkov09f96c32010-07-20 09:24:57 -0700538 actions_.push_back(shared_ptr<AbstractAction>(error_event_action));
Darin Petkovf42cc1c2010-09-01 09:03:02 -0700539 processor_->EnqueueAction(error_event_action.get());
Darin Petkov09f96c32010-07-20 09:24:57 -0700540 SetStatusAndNotify(UPDATE_STATUS_REPORTING_ERROR_EVENT);
Darin Petkovf42cc1c2010-09-01 09:03:02 -0700541 processor_->StartProcessing();
Darin Petkov09f96c32010-07-20 09:24:57 -0700542 return true;
543}
544
Darin Petkovc6c135c2010-08-11 13:36:18 -0700545void UpdateAttempter::SetPriority(utils::ProcessPriority priority) {
546 if (priority_ == priority) {
547 return;
548 }
549 if (utils::SetProcessPriority(priority)) {
550 priority_ = priority;
551 LOG(INFO) << "Process priority = " << priority_;
552 }
553}
554
555void UpdateAttempter::SetupPriorityManagement() {
556 if (manage_priority_source_) {
557 LOG(ERROR) << "Process priority timeout source hasn't been destroyed.";
558 CleanupPriorityManagement();
559 }
Darin Petkovf622ef72010-10-26 13:49:24 -0700560 const int kPriorityTimeout = 2 * 60 * 60; // 2 hours
Darin Petkovc6c135c2010-08-11 13:36:18 -0700561 manage_priority_source_ = g_timeout_source_new_seconds(kPriorityTimeout);
562 g_source_set_callback(manage_priority_source_,
563 StaticManagePriorityCallback,
564 this,
565 NULL);
566 g_source_attach(manage_priority_source_, NULL);
567 SetPriority(utils::kProcessPriorityLow);
568}
569
570void UpdateAttempter::CleanupPriorityManagement() {
571 if (manage_priority_source_) {
572 g_source_destroy(manage_priority_source_);
573 manage_priority_source_ = NULL;
574 }
575 SetPriority(utils::kProcessPriorityNormal);
576}
577
578gboolean UpdateAttempter::StaticManagePriorityCallback(gpointer data) {
579 return reinterpret_cast<UpdateAttempter*>(data)->ManagePriorityCallback();
580}
581
Darin Petkove6ef2f82011-03-07 17:31:11 -0800582gboolean UpdateAttempter::StaticStartProcessing(gpointer data) {
583 reinterpret_cast<UpdateAttempter*>(data)->processor_->StartProcessing();
584 return FALSE; // Don't call this callback again.
585}
586
Darin Petkovc6c135c2010-08-11 13:36:18 -0700587bool UpdateAttempter::ManagePriorityCallback() {
Darin Petkovf622ef72010-10-26 13:49:24 -0700588 SetPriority(utils::kProcessPriorityNormal);
Darin Petkovc6c135c2010-08-11 13:36:18 -0700589 manage_priority_source_ = NULL;
Darin Petkovf622ef72010-10-26 13:49:24 -0700590 return false; // Destroy the timeout source.
Darin Petkovc6c135c2010-08-11 13:36:18 -0700591}
592
Darin Petkov36275772010-10-01 11:40:57 -0700593void UpdateAttempter::DisableDeltaUpdateIfNeeded() {
594 int64_t delta_failures;
595 if (omaha_request_params_.delta_okay &&
596 prefs_->GetInt64(kPrefsDeltaUpdateFailures, &delta_failures) &&
597 delta_failures >= kMaxDeltaUpdateFailures) {
598 LOG(WARNING) << "Too many delta update failures, forcing full update.";
599 omaha_request_params_.delta_okay = false;
600 }
601}
602
603void UpdateAttempter::MarkDeltaUpdateFailure() {
604 CHECK(!is_full_update_);
Darin Petkov2dd01092010-10-08 15:43:05 -0700605 // Don't try to resume a failed delta update.
606 DeltaPerformer::ResetUpdateProgress(prefs_, false);
Darin Petkov36275772010-10-01 11:40:57 -0700607 int64_t delta_failures;
608 if (!prefs_->GetInt64(kPrefsDeltaUpdateFailures, &delta_failures) ||
609 delta_failures < 0) {
610 delta_failures = 0;
611 }
612 prefs_->SetInt64(kPrefsDeltaUpdateFailures, ++delta_failures);
613}
614
Darin Petkov9b230572010-10-08 10:20:09 -0700615void UpdateAttempter::SetupDownload() {
Andrew de los Reyes819fef22010-12-17 11:33:58 -0800616 MultiRangeHTTPFetcher* fetcher =
617 dynamic_cast<MultiRangeHTTPFetcher*>(download_action_->http_fetcher());
618 fetcher->ClearRanges();
Darin Petkov9b230572010-10-08 10:20:09 -0700619 if (response_handler_action_->install_plan().is_resume) {
Darin Petkovb21ce5d2010-10-21 16:03:05 -0700620 // Resuming an update so fetch the update manifest metadata first.
Darin Petkov9b230572010-10-08 10:20:09 -0700621 int64_t manifest_metadata_size = 0;
622 prefs_->GetInt64(kPrefsManifestMetadataSize, &manifest_metadata_size);
Andrew de los Reyes819fef22010-12-17 11:33:58 -0800623 fetcher->AddRange(0, manifest_metadata_size);
Darin Petkovb21ce5d2010-10-21 16:03:05 -0700624 // If there're remaining unprocessed data blobs, fetch them. Be careful not
625 // to request data beyond the end of the payload to avoid 416 HTTP response
626 // error codes.
Darin Petkov9b230572010-10-08 10:20:09 -0700627 int64_t next_data_offset = 0;
628 prefs_->GetInt64(kPrefsUpdateStateNextDataOffset, &next_data_offset);
Darin Petkovb21ce5d2010-10-21 16:03:05 -0700629 uint64_t resume_offset = manifest_metadata_size + next_data_offset;
630 if (resume_offset < response_handler_action_->install_plan().size) {
Andrew de los Reyes819fef22010-12-17 11:33:58 -0800631 fetcher->AddRange(resume_offset, -1);
Darin Petkovb21ce5d2010-10-21 16:03:05 -0700632 }
Darin Petkov9b230572010-10-08 10:20:09 -0700633 } else {
Andrew de los Reyes819fef22010-12-17 11:33:58 -0800634 fetcher->AddRange(0, -1);
Darin Petkov9b230572010-10-08 10:20:09 -0700635 }
Darin Petkov9b230572010-10-08 10:20:09 -0700636}
637
Thieu Le116fda32011-04-19 11:01:54 -0700638void UpdateAttempter::PingOmaha() {
639 shared_ptr<OmahaRequestAction> ping_action(
640 new OmahaRequestAction(prefs_,
641 omaha_request_params_,
642 NULL,
643 new LibcurlHttpFetcher(GetProxyResolver()),
644 true));
645 actions_.push_back(shared_ptr<OmahaRequestAction>(ping_action));
646 CHECK(!processor_->IsRunning());
647 processor_->set_delegate(NULL);
648 processor_->EnqueueAction(ping_action.get());
649 g_idle_add(&StaticStartProcessing, this);
650 SetStatusAndNotify(UPDATE_STATUS_UPDATED_NEED_REBOOT);
651}
652
Andrew de los Reyes4e9b9f42010-04-26 15:06:43 -0700653} // namespace chromeos_update_engine