225 lines
9.2 KiB
C++
225 lines
9.2 KiB
C++
//
|
|
// Copyright (C) 2011 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 "update_engine/omaha_response_handler_action.h"
|
|
|
|
#include <string>
|
|
|
|
#include <base/logging.h>
|
|
#include <base/strings/string_number_conversions.h>
|
|
#include <base/strings/string_util.h>
|
|
#include <policy/device_policy.h>
|
|
|
|
#include "update_engine/common/constants.h"
|
|
#include "update_engine/common/hardware_interface.h"
|
|
#include "update_engine/common/prefs_interface.h"
|
|
#include "update_engine/common/utils.h"
|
|
#include "update_engine/connection_manager_interface.h"
|
|
#include "update_engine/omaha_request_params.h"
|
|
#include "update_engine/payload_consumer/delta_performer.h"
|
|
#include "update_engine/payload_state_interface.h"
|
|
|
|
using std::string;
|
|
|
|
namespace chromeos_update_engine {
|
|
|
|
OmahaResponseHandlerAction::OmahaResponseHandlerAction(
|
|
SystemState* system_state)
|
|
: OmahaResponseHandlerAction(system_state,
|
|
constants::kOmahaResponseDeadlineFile) {}
|
|
|
|
OmahaResponseHandlerAction::OmahaResponseHandlerAction(
|
|
SystemState* system_state, const string& deadline_file)
|
|
: system_state_(system_state),
|
|
got_no_update_response_(false),
|
|
key_path_(constants::kUpdatePayloadPublicKeyPath),
|
|
deadline_file_(deadline_file) {}
|
|
|
|
void OmahaResponseHandlerAction::PerformAction() {
|
|
CHECK(HasInputObject());
|
|
ScopedActionCompleter completer(processor_, this);
|
|
const OmahaResponse& response = GetInputObject();
|
|
if (!response.update_exists) {
|
|
got_no_update_response_ = true;
|
|
LOG(INFO) << "There are no updates. Aborting.";
|
|
return;
|
|
}
|
|
|
|
// All decisions as to which URL should be used have already been done. So,
|
|
// make the current URL as the download URL.
|
|
string current_url = system_state_->payload_state()->GetCurrentUrl();
|
|
if (current_url.empty()) {
|
|
// This shouldn't happen as we should always supply the HTTPS backup URL.
|
|
// Handling this anyway, just in case.
|
|
LOG(ERROR) << "There are no suitable URLs in the response to use.";
|
|
completer.set_code(ErrorCode::kOmahaResponseInvalid);
|
|
return;
|
|
}
|
|
|
|
install_plan_.download_url = current_url;
|
|
install_plan_.version = response.version;
|
|
|
|
OmahaRequestParams* const params = system_state_->request_params();
|
|
PayloadStateInterface* const payload_state = system_state_->payload_state();
|
|
|
|
// If we're using p2p to download and there is a local peer, use it.
|
|
if (payload_state->GetUsingP2PForDownloading() &&
|
|
!payload_state->GetP2PUrl().empty()) {
|
|
LOG(INFO) << "Replacing URL " << install_plan_.download_url
|
|
<< " with local URL " << payload_state->GetP2PUrl()
|
|
<< " since p2p is enabled.";
|
|
install_plan_.download_url = payload_state->GetP2PUrl();
|
|
payload_state->SetUsingP2PForDownloading(true);
|
|
}
|
|
|
|
// Fill up the other properties based on the response.
|
|
string update_check_response_hash;
|
|
for (const auto& package : response.packages) {
|
|
brillo::Blob raw_hash;
|
|
if (!base::HexStringToBytes(package.hash, &raw_hash)) {
|
|
LOG(ERROR) << "Failed to convert payload hash from hex string to bytes: "
|
|
<< package.hash;
|
|
completer.set_code(ErrorCode::kOmahaResponseInvalid);
|
|
return;
|
|
}
|
|
install_plan_.payloads.push_back(
|
|
{.size = package.size,
|
|
.metadata_size = package.metadata_size,
|
|
.metadata_signature = package.metadata_signature,
|
|
.hash = raw_hash,
|
|
.type = package.is_delta ? InstallPayloadType::kDelta
|
|
: InstallPayloadType::kFull});
|
|
update_check_response_hash += package.hash + ":";
|
|
}
|
|
install_plan_.public_key_rsa = response.public_key_rsa;
|
|
install_plan_.hash_checks_mandatory = AreHashChecksMandatory(response);
|
|
install_plan_.is_resume = DeltaPerformer::CanResumeUpdate(
|
|
system_state_->prefs(), update_check_response_hash);
|
|
if (install_plan_.is_resume) {
|
|
payload_state->UpdateResumed();
|
|
} else {
|
|
payload_state->UpdateRestarted();
|
|
LOG_IF(WARNING,
|
|
!DeltaPerformer::ResetUpdateProgress(system_state_->prefs(), false))
|
|
<< "Unable to reset the update progress.";
|
|
LOG_IF(WARNING,
|
|
!system_state_->prefs()->SetString(kPrefsUpdateCheckResponseHash,
|
|
update_check_response_hash))
|
|
<< "Unable to save the update check response hash.";
|
|
}
|
|
|
|
install_plan_.source_slot = system_state_->boot_control()->GetCurrentSlot();
|
|
install_plan_.target_slot = install_plan_.source_slot == 0 ? 1 : 0;
|
|
|
|
// The Omaha response doesn't include the channel name for this image, so we
|
|
// use the download_channel we used during the request to tag the target slot.
|
|
// This will be used in the next boot to know the channel the image was
|
|
// downloaded from.
|
|
string current_channel_key =
|
|
kPrefsChannelOnSlotPrefix + std::to_string(install_plan_.target_slot);
|
|
system_state_->prefs()->SetString(current_channel_key,
|
|
params->download_channel());
|
|
|
|
if (params->to_more_stable_channel() && params->is_powerwash_allowed())
|
|
install_plan_.powerwash_required = true;
|
|
|
|
TEST_AND_RETURN(HasOutputPipe());
|
|
if (HasOutputPipe())
|
|
SetOutputObject(install_plan_);
|
|
LOG(INFO) << "Using this install plan:";
|
|
install_plan_.Dump();
|
|
|
|
// Send the deadline data (if any) to Chrome through a file. This is a pretty
|
|
// hacky solution but should be OK for now.
|
|
//
|
|
// TODO(petkov): Re-architect this to avoid communication through a
|
|
// file. Ideally, we would include this information in D-Bus's GetStatus
|
|
// method and UpdateStatus signal. A potential issue is that update_engine may
|
|
// be unresponsive during an update download.
|
|
if (!deadline_file_.empty()) {
|
|
utils::WriteFile(deadline_file_.c_str(),
|
|
response.deadline.data(),
|
|
response.deadline.size());
|
|
chmod(deadline_file_.c_str(), S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH);
|
|
}
|
|
|
|
completer.set_code(ErrorCode::kSuccess);
|
|
}
|
|
|
|
bool OmahaResponseHandlerAction::AreHashChecksMandatory(
|
|
const OmahaResponse& response) {
|
|
// We sometimes need to waive the hash checks in order to download from
|
|
// sources that don't provide hashes, such as dev server.
|
|
// At this point UpdateAttempter::IsAnyUpdateSourceAllowed() has already been
|
|
// checked, so an unofficial update URL won't get this far unless it's OK to
|
|
// use without a hash. Additionally, we want to always waive hash checks on
|
|
// unofficial builds (i.e. dev/test images).
|
|
// The end result is this:
|
|
// * Base image:
|
|
// - Official URLs require a hash.
|
|
// - Unofficial URLs only get this far if the IsAnyUpdateSourceAllowed()
|
|
// devmode/debugd checks pass, in which case the hash is waived.
|
|
// * Dev/test image:
|
|
// - Any URL is allowed through with no hash checking.
|
|
if (!system_state_->request_params()->IsUpdateUrlOfficial() ||
|
|
!system_state_->hardware()->IsOfficialBuild()) {
|
|
// Still do a hash check if a public key is included.
|
|
if (!response.public_key_rsa.empty()) {
|
|
// The autoupdate_CatchBadSignatures test checks for this string
|
|
// in log-files. Keep in sync.
|
|
LOG(INFO) << "Mandating payload hash checks since Omaha Response "
|
|
<< "for unofficial build includes public RSA key.";
|
|
return true;
|
|
} else {
|
|
LOG(INFO) << "Waiving payload hash checks for unofficial update URL.";
|
|
return false;
|
|
}
|
|
}
|
|
|
|
// If we're using p2p, |install_plan_.download_url| may contain a
|
|
// HTTP URL even if |response.payload_urls| contain only HTTPS URLs.
|
|
if (!base::StartsWith(install_plan_.download_url, "https://",
|
|
base::CompareCase::INSENSITIVE_ASCII)) {
|
|
LOG(INFO) << "Mandating hash checks since download_url is not HTTPS.";
|
|
return true;
|
|
}
|
|
|
|
// TODO(jaysri): VALIDATION: For official builds, we currently waive hash
|
|
// checks for HTTPS until we have rolled out at least once and are confident
|
|
// nothing breaks. chromium-os:37082 tracks turning this on for HTTPS
|
|
// eventually.
|
|
|
|
// Even if there's a single non-HTTPS URL, make the hash checks as
|
|
// mandatory because we could be downloading the payload from any URL later
|
|
// on. It's really hard to do book-keeping based on each byte being
|
|
// downloaded to see whether we only used HTTPS throughout.
|
|
for (const auto& package : response.packages) {
|
|
for (const string& payload_url : package.payload_urls) {
|
|
if (!base::StartsWith(
|
|
payload_url, "https://", base::CompareCase::INSENSITIVE_ASCII)) {
|
|
LOG(INFO) << "Mandating payload hash checks since Omaha response "
|
|
<< "contains non-HTTPS URL(s)";
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
|
|
LOG(INFO) << "Waiving payload hash checks since Omaha response "
|
|
<< "only has HTTPS URL(s)";
|
|
return false;
|
|
}
|
|
|
|
} // namespace chromeos_update_engine
|