| // |
| // Copyright (C) 2016 The Android Open Source Project |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| // |
| |
| #include <xz.h> |
| |
| #include <string> |
| #include <vector> |
| |
| #include <base/command_line.h> |
| #include <base/logging.h> |
| #include <base/strings/string_split.h> |
| #include <brillo/flag_helper.h> |
| #include <brillo/message_loops/base_message_loop.h> |
| |
| #include "update_engine/common/boot_control.h" |
| #include "update_engine/common/hardware.h" |
| #include "update_engine/common/prefs.h" |
| #include "update_engine/common/terminator.h" |
| #include "update_engine/common/utils.h" |
| #include "update_engine/update_attempter_android.h" |
| |
| using std::string; |
| using std::vector; |
| |
| namespace chromeos_update_engine { |
| namespace { |
| |
| void SetupLogging() { |
| string log_file; |
| logging::LoggingSettings log_settings; |
| log_settings.lock_log = logging::DONT_LOCK_LOG_FILE; |
| log_settings.delete_old = logging::APPEND_TO_OLD_LOG_FILE; |
| log_settings.log_file = nullptr; |
| log_settings.logging_dest = logging::LOG_TO_SYSTEM_DEBUG_LOG; |
| |
| logging::InitLogging(log_settings); |
| } |
| |
| class SideloadDaemonState : public DaemonStateInterface, |
| public ServiceObserverInterface { |
| public: |
| SideloadDaemonState() { |
| // Add this class as the only observer. |
| observers_.insert(this); |
| } |
| ~SideloadDaemonState() override = default; |
| |
| // DaemonStateInterface overrides. |
| bool StartUpdater() override { return true; } |
| void AddObserver(ServiceObserverInterface* observer) override {} |
| void RemoveObserver(ServiceObserverInterface* observer) override {} |
| const std::set<ServiceObserverInterface*>& service_observers() override { |
| return observers_; |
| } |
| |
| // ServiceObserverInterface overrides. |
| void SendStatusUpdate(int64_t last_checked_time, |
| double progress, |
| update_engine::UpdateStatus status, |
| const string& new_version, |
| int64_t new_size) override { |
| status_ = status; |
| } |
| |
| void SendPayloadApplicationComplete(ErrorCode error_code) override { |
| error_code_ = error_code; |
| brillo::MessageLoop::current()->BreakLoop(); |
| } |
| |
| void SendChannelChangeUpdate(const string& tracking_channel) override {} |
| |
| // Getters. |
| update_engine::UpdateStatus status() { return status_; } |
| ErrorCode error_code() { return error_code_; } |
| |
| private: |
| std::set<ServiceObserverInterface*> observers_; |
| |
| // The last status and error code reported. |
| update_engine::UpdateStatus status_{update_engine::UpdateStatus::IDLE}; |
| ErrorCode error_code_{ErrorCode::kSuccess}; |
| }; |
| |
| // Apply an update payload directly from the given payload URI. |
| bool ApplyUpdatePayload(const string& payload, |
| int64_t payload_offset, |
| int64_t payload_size, |
| const vector<string>& headers) { |
| brillo::BaseMessageLoop loop; |
| loop.SetAsCurrent(); |
| |
| SideloadDaemonState sideload_daemon_state; |
| |
| // During the sideload we don't access the prefs persisted on disk but instead |
| // use a temporary memory storage. |
| MemoryPrefs prefs; |
| |
| std::unique_ptr<BootControlInterface> boot_control = |
| boot_control::CreateBootControl(); |
| if (!boot_control) { |
| LOG(ERROR) << "Error initializing the BootControlInterface."; |
| return false; |
| } |
| |
| std::unique_ptr<HardwareInterface> hardware = hardware::CreateHardware(); |
| if (!hardware) { |
| LOG(ERROR) << "Error initializing the HardwareInterface."; |
| return false; |
| } |
| |
| UpdateAttempterAndroid update_attempter( |
| &sideload_daemon_state, &prefs, boot_control.get(), hardware.get()); |
| update_attempter.Init(); |
| |
| TEST_AND_RETURN_FALSE(update_attempter.ApplyPayload( |
| payload, payload_offset, payload_size, headers, nullptr)); |
| |
| loop.Run(); |
| return sideload_daemon_state.status() == |
| update_engine::UpdateStatus::UPDATED_NEED_REBOOT; |
| } |
| |
| } // namespace |
| } // namespace chromeos_update_engine |
| |
| int main(int argc, char** argv) { |
| DEFINE_string(payload, |
| "file:///data/payload.bin", |
| "The URI to the update payload to use."); |
| DEFINE_int64( |
| offset, 0, "The offset in the payload where the CrAU update starts. "); |
| DEFINE_int64(size, |
| 0, |
| "The size of the CrAU part of the payload. If 0 is passed, it " |
| "will be autodetected."); |
| DEFINE_string(headers, |
| "", |
| "A list of key-value pairs, one element of the list per line."); |
| |
| chromeos_update_engine::Terminator::Init(); |
| chromeos_update_engine::SetupLogging(); |
| brillo::FlagHelper::Init(argc, argv, "Update Engine Sideload"); |
| |
| LOG(INFO) << "Update Engine Sideloading starting"; |
| |
| // xz-embedded requires to initialize its CRC-32 table once on startup. |
| xz_crc32_init(); |
| |
| vector<string> headers = base::SplitString( |
| FLAGS_headers, "\n", base::KEEP_WHITESPACE, base::SPLIT_WANT_NONEMPTY); |
| |
| if (!chromeos_update_engine::ApplyUpdatePayload( |
| FLAGS_payload, FLAGS_offset, FLAGS_size, headers)) |
| return 1; |
| |
| return 0; |
| } |