Jay Srinivasan | 480ddfa | 2012-06-01 19:15:26 -0700 | [diff] [blame] | 1 | // Copyright (c) 2012 The Chromium OS Authors. All rights reserved. |
Andrew de los Reyes | 4e9b9f4 | 2010-04-26 15:06:43 -0700 | [diff] [blame] | 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
| 4 | |
| 5 | #ifndef CHROMEOS_PLATFORM_UPDATE_ENGINE_UPDATE_ATTEMPTER_H__ |
| 6 | #define CHROMEOS_PLATFORM_UPDATE_ENGINE_UPDATE_ATTEMPTER_H__ |
| 7 | |
Andrew de los Reyes | 63b96d7 | 2010-05-10 13:08:54 -0700 | [diff] [blame] | 8 | #include <time.h> |
Darin Petkov | c6c135c | 2010-08-11 13:36:18 -0700 | [diff] [blame] | 9 | |
Andrew de los Reyes | 4e9b9f4 | 2010-04-26 15:06:43 -0700 | [diff] [blame] | 10 | #include <tr1/memory> |
| 11 | #include <string> |
| 12 | #include <vector> |
Darin Petkov | c6c135c | 2010-08-11 13:36:18 -0700 | [diff] [blame] | 13 | |
Darin Petkov | 85ced13 | 2010-09-01 10:20:56 -0700 | [diff] [blame] | 14 | #include <base/time.h> |
Andrew de los Reyes | 4e9b9f4 | 2010-04-26 15:06:43 -0700 | [diff] [blame] | 15 | #include <glib.h> |
Darin Petkov | f42cc1c | 2010-09-01 09:03:02 -0700 | [diff] [blame] | 16 | #include <gtest/gtest_prod.h> // for FRIEND_TEST |
Darin Petkov | c6c135c | 2010-08-11 13:36:18 -0700 | [diff] [blame] | 17 | |
Andrew de los Reyes | 4e9b9f4 | 2010-04-26 15:06:43 -0700 | [diff] [blame] | 18 | #include "update_engine/action_processor.h" |
Andrew de los Reyes | 000d895 | 2011-03-02 15:21:14 -0800 | [diff] [blame] | 19 | #include "update_engine/chrome_browser_proxy_resolver.h" |
Andrew de los Reyes | 63b96d7 | 2010-05-10 13:08:54 -0700 | [diff] [blame] | 20 | #include "update_engine/download_action.h" |
Darin Petkov | a4a8a8c | 2010-07-15 22:21:12 -0700 | [diff] [blame] | 21 | #include "update_engine/omaha_request_params.h" |
Andrew de los Reyes | 4e9b9f4 | 2010-04-26 15:06:43 -0700 | [diff] [blame] | 22 | #include "update_engine/omaha_response_handler_action.h" |
Andrew de los Reyes | 4516810 | 2010-11-22 11:13:50 -0800 | [diff] [blame] | 23 | #include "update_engine/proxy_resolver.h" |
Jay Srinivasan | 4348879 | 2012-06-19 00:25:31 -0700 | [diff] [blame] | 24 | #include "update_engine/system_state.h" |
Andrew de los Reyes | 4e9b9f4 | 2010-04-26 15:06:43 -0700 | [diff] [blame] | 25 | |
Darin Petkov | 9d65b7b | 2010-07-20 09:13:01 -0700 | [diff] [blame] | 26 | class MetricsLibraryInterface; |
Andrew de los Reyes | 63b96d7 | 2010-05-10 13:08:54 -0700 | [diff] [blame] | 27 | struct UpdateEngineService; |
| 28 | |
Patrick Dubroy | 7fbbe8a | 2011-08-01 17:28:22 +0200 | [diff] [blame] | 29 | namespace policy { |
| 30 | class PolicyProvider; |
| 31 | } |
| 32 | |
Andrew de los Reyes | 4e9b9f4 | 2010-04-26 15:06:43 -0700 | [diff] [blame] | 33 | namespace chromeos_update_engine { |
| 34 | |
Darin Petkov | 1023a60 | 2010-08-30 13:47:51 -0700 | [diff] [blame] | 35 | class UpdateCheckScheduler; |
Darin Petkov | c6c135c | 2010-08-11 13:36:18 -0700 | [diff] [blame] | 36 | |
Andrew de los Reyes | 6b78e29 | 2010-05-10 15:54:39 -0700 | [diff] [blame] | 37 | extern const char* kUpdateCompletedMarker; |
| 38 | |
Andrew de los Reyes | 63b96d7 | 2010-05-10 13:08:54 -0700 | [diff] [blame] | 39 | enum UpdateStatus { |
| 40 | UPDATE_STATUS_IDLE = 0, |
| 41 | UPDATE_STATUS_CHECKING_FOR_UPDATE, |
| 42 | UPDATE_STATUS_UPDATE_AVAILABLE, |
| 43 | UPDATE_STATUS_DOWNLOADING, |
| 44 | UPDATE_STATUS_VERIFYING, |
| 45 | UPDATE_STATUS_FINALIZING, |
Darin Petkov | 09f96c3 | 2010-07-20 09:24:57 -0700 | [diff] [blame] | 46 | UPDATE_STATUS_UPDATED_NEED_REBOOT, |
| 47 | UPDATE_STATUS_REPORTING_ERROR_EVENT, |
Andrew de los Reyes | 63b96d7 | 2010-05-10 13:08:54 -0700 | [diff] [blame] | 48 | }; |
| 49 | |
Gilad Arnold | 1ebd813 | 2012-03-05 10:19:29 -0800 | [diff] [blame] | 50 | enum UpdateNotice { |
| 51 | kUpdateNoticeUnspecified = 0, |
| 52 | kUpdateNoticeTestAddrFailed, |
| 53 | }; |
| 54 | |
Andrew de los Reyes | 63b96d7 | 2010-05-10 13:08:54 -0700 | [diff] [blame] | 55 | const char* UpdateStatusToString(UpdateStatus status); |
| 56 | |
| 57 | class UpdateAttempter : public ActionProcessorDelegate, |
| 58 | public DownloadActionDelegate { |
Andrew de los Reyes | 4e9b9f4 | 2010-04-26 15:06:43 -0700 | [diff] [blame] | 59 | public: |
Darin Petkov | 3627577 | 2010-10-01 11:40:57 -0700 | [diff] [blame] | 60 | static const int kMaxDeltaUpdateFailures; |
| 61 | |
Jay Srinivasan | 6f6ea00 | 2012-12-14 11:26:28 -0800 | [diff] [blame] | 62 | UpdateAttempter(SystemState* system_state, |
Gilad Arnold | bf7919b | 2013-01-08 13:07:37 -0800 | [diff] [blame] | 63 | DbusGlibInterface* dbus_iface); |
Darin Petkov | 1023a60 | 2010-08-30 13:47:51 -0700 | [diff] [blame] | 64 | virtual ~UpdateAttempter(); |
Darin Petkov | c6c135c | 2010-08-11 13:36:18 -0700 | [diff] [blame] | 65 | |
Gilad Arnold | 1ebd813 | 2012-03-05 10:19:29 -0800 | [diff] [blame] | 66 | // Checks for update and, if a newer version is available, attempts to update |
| 67 | // the system. Non-empty |in_app_version| or |in_update_url| prevents |
| 68 | // automatic detection of the parameter. If |obey_proxies| is true, the |
| 69 | // update will likely respect Chrome's proxy setting. For security reasons, we |
| 70 | // may still not honor them. Interactive should be true if this was called |
| 71 | // from the user (ie dbus). |is_test| will lead to using an alternative test |
Jay Srinivasan | 08fce04 | 2012-06-07 16:31:01 -0700 | [diff] [blame] | 72 | // server URL, if |omaha_url| is empty. |is_user_initiated| will be true |
| 73 | // only if the update is being kicked off through dbus and will be false for |
| 74 | // other types of kick off such as scheduled updates. |
Darin Petkov | 1023a60 | 2010-08-30 13:47:51 -0700 | [diff] [blame] | 75 | virtual void Update(const std::string& app_version, |
Andrew de los Reyes | 4516810 | 2010-11-22 11:13:50 -0800 | [diff] [blame] | 76 | const std::string& omaha_url, |
Andrew de los Reyes | fb2f461 | 2011-06-09 18:21:49 -0700 | [diff] [blame] | 77 | bool obey_proxies, |
Gilad Arnold | 1ebd813 | 2012-03-05 10:19:29 -0800 | [diff] [blame] | 78 | bool interactive, |
Gilad Arnold | b92f0df | 2013-01-10 16:32:45 -0800 | [diff] [blame] | 79 | bool is_test_mode); |
Darin Petkov | a4a8a8c | 2010-07-15 22:21:12 -0700 | [diff] [blame] | 80 | |
Andrew de los Reyes | 63b96d7 | 2010-05-10 13:08:54 -0700 | [diff] [blame] | 81 | // ActionProcessorDelegate methods: |
Darin Petkov | c1a8b42 | 2010-07-19 11:34:49 -0700 | [diff] [blame] | 82 | void ProcessingDone(const ActionProcessor* processor, ActionExitCode code); |
Andrew de los Reyes | 63b96d7 | 2010-05-10 13:08:54 -0700 | [diff] [blame] | 83 | void ProcessingStopped(const ActionProcessor* processor); |
| 84 | void ActionCompleted(ActionProcessor* processor, |
| 85 | AbstractAction* action, |
Darin Petkov | c1a8b42 | 2010-07-19 11:34:49 -0700 | [diff] [blame] | 86 | ActionExitCode code); |
Darin Petkov | a4a8a8c | 2010-07-15 22:21:12 -0700 | [diff] [blame] | 87 | |
Andrew de los Reyes | 4e9b9f4 | 2010-04-26 15:06:43 -0700 | [diff] [blame] | 88 | // Stop updating. An attempt will be made to record status to the disk |
| 89 | // so that updates can be resumed later. |
| 90 | void Terminate(); |
Darin Petkov | a4a8a8c | 2010-07-15 22:21:12 -0700 | [diff] [blame] | 91 | |
Andrew de los Reyes | 4e9b9f4 | 2010-04-26 15:06:43 -0700 | [diff] [blame] | 92 | // Try to resume from a previously Terminate()d update. |
| 93 | void ResumeUpdating(); |
Darin Petkov | a4a8a8c | 2010-07-15 22:21:12 -0700 | [diff] [blame] | 94 | |
Jay Srinivasan | c1ba09a | 2012-08-14 14:15:57 -0700 | [diff] [blame] | 95 | // Resets the current state to UPDATE_STATUS_IDLE. |
| 96 | // Used by update_engine_client for restarting a new update without |
| 97 | // having to reboot once the previous update has reached |
| 98 | // UPDATE_STATUS_UPDATED_NEED_REBOOT state. This is used only |
| 99 | // for testing purposes. |
| 100 | bool ResetStatus(); |
| 101 | |
Andrew de los Reyes | 4e9b9f4 | 2010-04-26 15:06:43 -0700 | [diff] [blame] | 102 | // Returns the current status in the out params. Returns true on success. |
| 103 | bool GetStatus(int64_t* last_checked_time, |
| 104 | double* progress, |
| 105 | std::string* current_operation, |
| 106 | std::string* new_version, |
| 107 | int64_t* new_size); |
| 108 | |
Darin Petkov | 58dd134 | 2011-05-06 12:05:13 -0700 | [diff] [blame] | 109 | // Runs chromeos-setgoodkernel, whose responsibility it is to mark the |
| 110 | // currently booted partition has high priority/permanent/etc. The execution |
| 111 | // is asynchronous. On completion, the action processor may be started |
| 112 | // depending on the |start_action_processor_| field. Note that every update |
| 113 | // attempt goes through this method. |
Andrew de los Reyes | 6dbf30a | 2011-04-19 10:58:16 -0700 | [diff] [blame] | 114 | void UpdateBootFlags(); |
| 115 | |
Darin Petkov | 58dd134 | 2011-05-06 12:05:13 -0700 | [diff] [blame] | 116 | // Subprocess::Exec callback. |
| 117 | void CompleteUpdateBootFlags(int return_code); |
| 118 | static void StaticCompleteUpdateBootFlags(int return_code, |
| 119 | const std::string& output, |
| 120 | void* p); |
| 121 | |
Darin Petkov | 1023a60 | 2010-08-30 13:47:51 -0700 | [diff] [blame] | 122 | UpdateStatus status() const { return status_; } |
| 123 | |
| 124 | int http_response_code() const { return http_response_code_; } |
| 125 | void set_http_response_code(int code) { http_response_code_ = code; } |
| 126 | |
Andrew de los Reyes | 63b96d7 | 2010-05-10 13:08:54 -0700 | [diff] [blame] | 127 | void set_dbus_service(struct UpdateEngineService* dbus_service) { |
| 128 | dbus_service_ = dbus_service; |
| 129 | } |
| 130 | |
Darin Petkov | 1023a60 | 2010-08-30 13:47:51 -0700 | [diff] [blame] | 131 | UpdateCheckScheduler* update_check_scheduler() const { |
| 132 | return update_check_scheduler_; |
| 133 | } |
| 134 | void set_update_check_scheduler(UpdateCheckScheduler* scheduler) { |
| 135 | update_check_scheduler_ = scheduler; |
| 136 | } |
| 137 | |
Jay Srinivasan | e73acab | 2012-07-10 14:34:03 -0700 | [diff] [blame] | 138 | // This is the internal entry point for going through an |
Darin Petkov | 5a7f565 | 2010-07-22 21:40:09 -0700 | [diff] [blame] | 139 | // update. If the current status is idle invokes Update. |
Jay Srinivasan | e73acab | 2012-07-10 14:34:03 -0700 | [diff] [blame] | 140 | // This is called by the DBus implementation. |
Darin Petkov | 5a7f565 | 2010-07-22 21:40:09 -0700 | [diff] [blame] | 141 | void CheckForUpdate(const std::string& app_version, |
Jay Srinivasan | e73acab | 2012-07-10 14:34:03 -0700 | [diff] [blame] | 142 | const std::string& omaha_url, |
Gilad Arnold | b92f0df | 2013-01-10 16:32:45 -0800 | [diff] [blame] | 143 | bool is_interactive); |
Andrew de los Reyes | 63b96d7 | 2010-05-10 13:08:54 -0700 | [diff] [blame] | 144 | |
Darin Petkov | 296889c | 2010-07-23 16:20:54 -0700 | [diff] [blame] | 145 | // Initiates a reboot if the current state is |
| 146 | // UPDATED_NEED_REBOOT. Returns true on sucess, false otherwise. |
| 147 | bool RebootIfNeeded(); |
| 148 | |
Darin Petkov | 9d911fa | 2010-08-19 09:36:08 -0700 | [diff] [blame] | 149 | // DownloadActionDelegate methods |
| 150 | void SetDownloadStatus(bool active); |
Andrew de los Reyes | 63b96d7 | 2010-05-10 13:08:54 -0700 | [diff] [blame] | 151 | void BytesReceived(uint64_t bytes_received, uint64_t total); |
| 152 | |
Darin Petkov | 61635a9 | 2011-05-18 16:20:36 -0700 | [diff] [blame] | 153 | // Broadcasts the current status over D-Bus. |
| 154 | void BroadcastStatus(); |
| 155 | |
Jay Srinivasan | 55f50c2 | 2013-01-10 19:24:35 -0800 | [diff] [blame] | 156 | // Returns the special flags to be added to ActionExitCode values based on the |
| 157 | // parameters used in the current update attempt. |
| 158 | uint32_t GetErrorCodeFlags(); |
| 159 | |
Andrew de los Reyes | 4e9b9f4 | 2010-04-26 15:06:43 -0700 | [diff] [blame] | 160 | private: |
Gilad Arnold | 28e2f39 | 2012-02-09 14:36:46 -0800 | [diff] [blame] | 161 | // Update server URL for automated lab test. |
| 162 | static const char* const kTestUpdateUrl; |
| 163 | |
Darin Petkov | f42cc1c | 2010-09-01 09:03:02 -0700 | [diff] [blame] | 164 | friend class UpdateAttempterTest; |
Darin Petkov | 1b00310 | 2010-11-30 10:18:36 -0800 | [diff] [blame] | 165 | FRIEND_TEST(UpdateAttempterTest, ActionCompletedDownloadTest); |
| 166 | FRIEND_TEST(UpdateAttempterTest, ActionCompletedErrorTest); |
| 167 | FRIEND_TEST(UpdateAttempterTest, ActionCompletedOmahaRequestTest); |
Darin Petkov | 18c7bce | 2011-06-16 14:07:00 -0700 | [diff] [blame] | 168 | FRIEND_TEST(UpdateAttempterTest, CreatePendingErrorEventTest); |
| 169 | FRIEND_TEST(UpdateAttempterTest, CreatePendingErrorEventResumedTest); |
Darin Petkov | 3627577 | 2010-10-01 11:40:57 -0700 | [diff] [blame] | 170 | FRIEND_TEST(UpdateAttempterTest, DisableDeltaUpdateIfNeededTest); |
| 171 | FRIEND_TEST(UpdateAttempterTest, MarkDeltaUpdateFailureTest); |
Patrick Dubroy | 7fbbe8a | 2011-08-01 17:28:22 +0200 | [diff] [blame] | 172 | FRIEND_TEST(UpdateAttempterTest, ReadTrackFromPolicy); |
Darin Petkov | 18c7bce | 2011-06-16 14:07:00 -0700 | [diff] [blame] | 173 | FRIEND_TEST(UpdateAttempterTest, PingOmahaTest); |
Darin Petkov | 1b00310 | 2010-11-30 10:18:36 -0800 | [diff] [blame] | 174 | FRIEND_TEST(UpdateAttempterTest, ScheduleErrorEventActionNoEventTest); |
| 175 | FRIEND_TEST(UpdateAttempterTest, ScheduleErrorEventActionTest); |
Darin Petkov | f42cc1c | 2010-09-01 09:03:02 -0700 | [diff] [blame] | 176 | FRIEND_TEST(UpdateAttempterTest, UpdateTest); |
| 177 | |
Gilad Arnold | 1ebd813 | 2012-03-05 10:19:29 -0800 | [diff] [blame] | 178 | // Sets the status to the given status and notifies a status update over dbus. |
| 179 | // Also accepts a supplement notice, which is delegated to the scheduler and |
| 180 | // used for making better informed scheduling decisions (e.g. retry timeout). |
| 181 | void SetStatusAndNotify(UpdateStatus status, UpdateNotice notice); |
Darin Petkov | a4a8a8c | 2010-07-15 22:21:12 -0700 | [diff] [blame] | 182 | |
Darin Petkov | 9b23057 | 2010-10-08 10:20:09 -0700 | [diff] [blame] | 183 | // Sets up the download parameters after receiving the update check response. |
| 184 | void SetupDownload(); |
| 185 | |
| 186 | // Creates an error event object in |error_event_| to be included in an |
| 187 | // OmahaRequestAction once the current action processor is done. |
Darin Petkov | 777dbfa | 2010-07-20 15:03:37 -0700 | [diff] [blame] | 188 | void CreatePendingErrorEvent(AbstractAction* action, ActionExitCode code); |
Darin Petkov | 09f96c3 | 2010-07-20 09:24:57 -0700 | [diff] [blame] | 189 | |
Darin Petkov | 9b23057 | 2010-10-08 10:20:09 -0700 | [diff] [blame] | 190 | // If there's a pending error event allocated in |error_event_|, schedules an |
| 191 | // OmahaRequestAction with that event in the current processor, clears the |
| 192 | // pending event, updates the status and returns true. Returns false |
| 193 | // otherwise. |
Darin Petkov | 09f96c3 | 2010-07-20 09:24:57 -0700 | [diff] [blame] | 194 | bool ScheduleErrorEventAction(); |
| 195 | |
Chris Sosa | 4f8ee27 | 2012-11-30 13:01:54 -0800 | [diff] [blame] | 196 | // Sets the cpu shares to |shares| and updates |shares_| if the new |
| 197 | // |shares| is different than the current |shares_|, otherwise simply |
Darin Petkov | 9b23057 | 2010-10-08 10:20:09 -0700 | [diff] [blame] | 198 | // returns. |
Chris Sosa | 4f8ee27 | 2012-11-30 13:01:54 -0800 | [diff] [blame] | 199 | void SetCpuShares(utils::CpuShares shares); |
Darin Petkov | c6c135c | 2010-08-11 13:36:18 -0700 | [diff] [blame] | 200 | |
Chris Sosa | 4f8ee27 | 2012-11-30 13:01:54 -0800 | [diff] [blame] | 201 | // Sets the cpu shares to low and sets up timeout events to increase it. |
| 202 | void SetupCpuSharesManagement(); |
Darin Petkov | c6c135c | 2010-08-11 13:36:18 -0700 | [diff] [blame] | 203 | |
Chris Sosa | 4f8ee27 | 2012-11-30 13:01:54 -0800 | [diff] [blame] | 204 | // Resets the cpu shares to normal and destroys any scheduled timeout |
Darin Petkov | 9b23057 | 2010-10-08 10:20:09 -0700 | [diff] [blame] | 205 | // sources. |
Chris Sosa | 4f8ee27 | 2012-11-30 13:01:54 -0800 | [diff] [blame] | 206 | void CleanupCpuSharesManagement(); |
Darin Petkov | c6c135c | 2010-08-11 13:36:18 -0700 | [diff] [blame] | 207 | |
Chris Sosa | 4f8ee27 | 2012-11-30 13:01:54 -0800 | [diff] [blame] | 208 | // The cpu shares timeout source callback sets the current cpu shares to |
Darin Petkov | f622ef7 | 2010-10-26 13:49:24 -0700 | [diff] [blame] | 209 | // normal. Returns false so that GLib destroys the timeout source. |
Chris Sosa | 4f8ee27 | 2012-11-30 13:01:54 -0800 | [diff] [blame] | 210 | static gboolean StaticManageCpuSharesCallback(gpointer data); |
| 211 | bool ManageCpuSharesCallback(); |
Darin Petkov | c6c135c | 2010-08-11 13:36:18 -0700 | [diff] [blame] | 212 | |
Darin Petkov | e6ef2f8 | 2011-03-07 17:31:11 -0800 | [diff] [blame] | 213 | // Callback to start the action processor. |
| 214 | static gboolean StaticStartProcessing(gpointer data); |
| 215 | |
Darin Petkov | 58dd134 | 2011-05-06 12:05:13 -0700 | [diff] [blame] | 216 | // Schedules an event loop callback to start the action processor. This is |
| 217 | // scheduled asynchronously to unblock the event loop. |
| 218 | void ScheduleProcessingStart(); |
| 219 | |
Darin Petkov | 3627577 | 2010-10-01 11:40:57 -0700 | [diff] [blame] | 220 | // Checks if a full update is needed and forces it by updating the Omaha |
| 221 | // request params. |
| 222 | void DisableDeltaUpdateIfNeeded(); |
| 223 | |
| 224 | // If this was a delta update attempt that failed, count it so that a full |
| 225 | // update can be tried when needed. |
| 226 | void MarkDeltaUpdateFailure(); |
Darin Petkov | 1b00310 | 2010-11-30 10:18:36 -0800 | [diff] [blame] | 227 | |
Andrew de los Reyes | 4516810 | 2010-11-22 11:13:50 -0800 | [diff] [blame] | 228 | ProxyResolver* GetProxyResolver() { |
| 229 | return obeying_proxies_ ? |
| 230 | reinterpret_cast<ProxyResolver*>(&chrome_proxy_resolver_) : |
| 231 | reinterpret_cast<ProxyResolver*>(&direct_proxy_resolver_); |
| 232 | } |
Darin Petkov | 3627577 | 2010-10-01 11:40:57 -0700 | [diff] [blame] | 233 | |
Thieu Le | 116fda3 | 2011-04-19 11:01:54 -0700 | [diff] [blame] | 234 | // Sends a ping to Omaha. |
| 235 | // This is used after an update has been applied and we're waiting for the |
| 236 | // user to reboot. This ping helps keep the number of actives count |
| 237 | // accurate in case a user takes a long time to reboot the device after an |
| 238 | // update has been applied. |
| 239 | void PingOmaha(); |
| 240 | |
Jay Srinivasan | 480ddfa | 2012-06-01 19:15:26 -0700 | [diff] [blame] | 241 | // Helper method of Update() to calculate the update-related parameters |
| 242 | // from various sources and set the appropriate state. Please refer to |
| 243 | // Update() method for the meaning of the parametes. |
| 244 | bool CalculateUpdateParams(const std::string& app_version, |
| 245 | const std::string& omaha_url, |
| 246 | bool obey_proxies, |
| 247 | bool interactive, |
Gilad Arnold | b92f0df | 2013-01-10 16:32:45 -0800 | [diff] [blame] | 248 | bool is_test); |
Jay Srinivasan | 480ddfa | 2012-06-01 19:15:26 -0700 | [diff] [blame] | 249 | |
Jay Srinivasan | 21be075 | 2012-07-25 15:44:56 -0700 | [diff] [blame] | 250 | // Calculates all the scattering related parameters (such as waiting period, |
| 251 | // which type of scattering is enabled, etc.) and also updates/deletes |
| 252 | // the corresponding prefs file used in scattering. Should be called |
| 253 | // only after the device policy has been loaded and set in the system_state_. |
Gilad Arnold | b92f0df | 2013-01-10 16:32:45 -0800 | [diff] [blame] | 254 | void CalculateScatteringParams(bool is_interactive); |
Jay Srinivasan | 21be075 | 2012-07-25 15:44:56 -0700 | [diff] [blame] | 255 | |
Jay Srinivasan | ae4697c | 2013-03-18 17:08:08 -0700 | [diff] [blame^] | 256 | // Sets a random value for the waiting period to wait for before downloading |
| 257 | // an update, if one available. This value will be upperbounded by the |
| 258 | // scatter factor value specified from policy. |
Jay Srinivasan | 21be075 | 2012-07-25 15:44:56 -0700 | [diff] [blame] | 259 | void GenerateNewWaitingPeriod(); |
| 260 | |
Jay Srinivasan | 480ddfa | 2012-06-01 19:15:26 -0700 | [diff] [blame] | 261 | // Helper method of Update() to construct the sequence of actions to |
| 262 | // be performed for an update check. Please refer to |
| 263 | // Update() method for the meaning of the parametes. |
| 264 | void BuildUpdateActions(bool interactive); |
| 265 | |
| 266 | // Decrements the count in the kUpdateCheckCountFilePath. |
| 267 | // Returns True if successfully decremented, false otherwise. |
| 268 | bool DecrementUpdateCheckCount(); |
| 269 | |
Darin Petkov | 9b23057 | 2010-10-08 10:20:09 -0700 | [diff] [blame] | 270 | // Last status notification timestamp used for throttling. Use monotonic |
| 271 | // TimeTicks to ensure that notifications are sent even if the system clock is |
| 272 | // set back in the middle of an update. |
Darin Petkov | af18305 | 2010-08-23 12:07:13 -0700 | [diff] [blame] | 273 | base::TimeTicks last_notify_time_; |
Andrew de los Reyes | 63b96d7 | 2010-05-10 13:08:54 -0700 | [diff] [blame] | 274 | |
Andrew de los Reyes | 4e9b9f4 | 2010-04-26 15:06:43 -0700 | [diff] [blame] | 275 | std::vector<std::tr1::shared_ptr<AbstractAction> > actions_; |
Darin Petkov | f42cc1c | 2010-09-01 09:03:02 -0700 | [diff] [blame] | 276 | scoped_ptr<ActionProcessor> processor_; |
Darin Petkov | a4a8a8c | 2010-07-15 22:21:12 -0700 | [diff] [blame] | 277 | |
Jay Srinivasan | 6f6ea00 | 2012-12-14 11:26:28 -0800 | [diff] [blame] | 278 | // External state of the system outside the update_engine process |
| 279 | // carved out separately to mock out easily in unit tests. |
| 280 | SystemState* system_state_; |
| 281 | |
Andrew de los Reyes | 63b96d7 | 2010-05-10 13:08:54 -0700 | [diff] [blame] | 282 | // If non-null, this UpdateAttempter will send status updates over this |
| 283 | // dbus service. |
| 284 | UpdateEngineService* dbus_service_; |
Andrew de los Reyes | 4e9b9f4 | 2010-04-26 15:06:43 -0700 | [diff] [blame] | 285 | |
Darin Petkov | 9b23057 | 2010-10-08 10:20:09 -0700 | [diff] [blame] | 286 | // Pointer to the OmahaResponseHandlerAction in the actions_ vector. |
Andrew de los Reyes | 4e9b9f4 | 2010-04-26 15:06:43 -0700 | [diff] [blame] | 287 | std::tr1::shared_ptr<OmahaResponseHandlerAction> response_handler_action_; |
Andrew de los Reyes | 63b96d7 | 2010-05-10 13:08:54 -0700 | [diff] [blame] | 288 | |
Darin Petkov | 9b23057 | 2010-10-08 10:20:09 -0700 | [diff] [blame] | 289 | // Pointer to the DownloadAction in the actions_ vector. |
| 290 | std::tr1::shared_ptr<DownloadAction> download_action_; |
| 291 | |
Jay Srinivasan | 6f6ea00 | 2012-12-14 11:26:28 -0800 | [diff] [blame] | 292 | // Pointer to the preferences store interface. This is just a cached |
| 293 | // copy of system_state->prefs() because it's used in many methods and |
| 294 | // is convenient this way. |
Darin Petkov | 1cbd78f | 2010-07-29 12:38:34 -0700 | [diff] [blame] | 295 | PrefsInterface* prefs_; |
| 296 | |
Darin Petkov | 1023a60 | 2010-08-30 13:47:51 -0700 | [diff] [blame] | 297 | // The current UpdateCheckScheduler to notify of state transitions. |
| 298 | UpdateCheckScheduler* update_check_scheduler_; |
| 299 | |
Darin Petkov | 09f96c3 | 2010-07-20 09:24:57 -0700 | [diff] [blame] | 300 | // Pending error event, if any. |
| 301 | scoped_ptr<OmahaEvent> error_event_; |
| 302 | |
Andrew de los Reyes | c1d5c93 | 2011-04-20 17:15:47 -0700 | [diff] [blame] | 303 | // If we should request a reboot even tho we failed the update |
| 304 | bool fake_update_success_; |
| 305 | |
Darin Petkov | 85ced13 | 2010-09-01 10:20:56 -0700 | [diff] [blame] | 306 | // HTTP server response code from the last HTTP request action. |
Darin Petkov | 1023a60 | 2010-08-30 13:47:51 -0700 | [diff] [blame] | 307 | int http_response_code_; |
| 308 | |
Chris Sosa | 4f8ee27 | 2012-11-30 13:01:54 -0800 | [diff] [blame] | 309 | // Current cpu shares. |
| 310 | utils::CpuShares shares_; |
Darin Petkov | c6c135c | 2010-08-11 13:36:18 -0700 | [diff] [blame] | 311 | |
Chris Sosa | 4f8ee27 | 2012-11-30 13:01:54 -0800 | [diff] [blame] | 312 | // The cpu shares management timeout source. |
| 313 | GSource* manage_shares_source_; |
Darin Petkov | c6c135c | 2010-08-11 13:36:18 -0700 | [diff] [blame] | 314 | |
Darin Petkov | 9d911fa | 2010-08-19 09:36:08 -0700 | [diff] [blame] | 315 | // Set to true if an update download is active (and BytesReceived |
| 316 | // will be called), set to false otherwise. |
| 317 | bool download_active_; |
| 318 | |
Andrew de los Reyes | 63b96d7 | 2010-05-10 13:08:54 -0700 | [diff] [blame] | 319 | // For status: |
| 320 | UpdateStatus status_; |
| 321 | double download_progress_; |
| 322 | int64_t last_checked_time_; |
| 323 | std::string new_version_; |
Jay Srinivasan | 51dcf26 | 2012-09-13 17:24:32 -0700 | [diff] [blame] | 324 | int64_t new_payload_size_; |
Andrew de los Reyes | 63b96d7 | 2010-05-10 13:08:54 -0700 | [diff] [blame] | 325 | |
Jay Srinivasan | ae4697c | 2013-03-18 17:08:08 -0700 | [diff] [blame^] | 326 | // Common parameters for all Omaha requests. |
| 327 | OmahaRequestParams* omaha_request_params_; |
Darin Petkov | a4a8a8c | 2010-07-15 22:21:12 -0700 | [diff] [blame] | 328 | |
Andrew de los Reyes | 4516810 | 2010-11-22 11:13:50 -0800 | [diff] [blame] | 329 | // Number of consecutive manual update checks we've had where we obeyed |
| 330 | // Chrome's proxy settings. |
| 331 | int proxy_manual_checks_; |
| 332 | |
| 333 | // If true, this update cycle we are obeying proxies |
| 334 | bool obeying_proxies_; |
| 335 | |
| 336 | // Our two proxy resolvers |
| 337 | DirectProxyResolver direct_proxy_resolver_; |
Andrew de los Reyes | 000d895 | 2011-03-02 15:21:14 -0800 | [diff] [blame] | 338 | ChromeBrowserProxyResolver chrome_proxy_resolver_; |
Andrew de los Reyes | 4516810 | 2010-11-22 11:13:50 -0800 | [diff] [blame] | 339 | |
Darin Petkov | 58dd134 | 2011-05-06 12:05:13 -0700 | [diff] [blame] | 340 | // Originally, both of these flags are false. Once UpdateBootFlags is called, |
| 341 | // |update_boot_flags_running_| is set to true. As soon as UpdateBootFlags |
| 342 | // completes its asynchronous run, |update_boot_flags_running_| is reset to |
| 343 | // false and |updated_boot_flags_| is set to true. From that point on there |
| 344 | // will be no more changes to these flags. |
| 345 | bool updated_boot_flags_; // True if UpdateBootFlags has completed. |
| 346 | bool update_boot_flags_running_; // True if UpdateBootFlags is running. |
| 347 | |
| 348 | // True if the action processor needs to be started by the boot flag updater. |
| 349 | bool start_action_processor_; |
Andrew de los Reyes | 6dbf30a | 2011-04-19 10:58:16 -0700 | [diff] [blame] | 350 | |
Patrick Dubroy | 7fbbe8a | 2011-08-01 17:28:22 +0200 | [diff] [blame] | 351 | // Used for fetching information about the device policy. |
| 352 | scoped_ptr<policy::PolicyProvider> policy_provider_; |
| 353 | |
Gilad Arnold | 1ebd813 | 2012-03-05 10:19:29 -0800 | [diff] [blame] | 354 | // A flag for indicating whether we are using a test server URL. |
| 355 | bool is_using_test_url_; |
Gilad Arnold | 28e2f39 | 2012-02-09 14:36:46 -0800 | [diff] [blame] | 356 | |
Gilad Arnold | 7c04e76 | 2012-05-23 10:54:02 -0700 | [diff] [blame] | 357 | // If true, will induce a test mode update attempt. |
| 358 | bool is_test_mode_; |
| 359 | |
Gilad Arnold | 4d740eb | 2012-05-15 08:48:13 -0700 | [diff] [blame] | 360 | // A flag indicating whether a test update cycle was already attempted. |
| 361 | bool is_test_update_attempted_; |
| 362 | |
Jay Srinivasan | 480ddfa | 2012-06-01 19:15:26 -0700 | [diff] [blame] | 363 | // The current scatter factor as found in the policy setting. |
| 364 | base::TimeDelta scatter_factor_; |
| 365 | |
Andrew de los Reyes | 4e9b9f4 | 2010-04-26 15:06:43 -0700 | [diff] [blame] | 366 | DISALLOW_COPY_AND_ASSIGN(UpdateAttempter); |
| 367 | }; |
| 368 | |
| 369 | } // namespace chromeos_update_engine |
| 370 | |
| 371 | #endif // CHROMEOS_PLATFORM_UPDATE_ENGINE_UPDATE_ATTEMPTER_H__ |