tequilaOS/platform_bootable_recovery
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Merge "recovery: Change install_package() to take std::string."

Browse source
This commit is contained in:
Tao Bao 2017-05-02 02:10:41 +00:00 committed by Gerrit Code Review
commit ba365180d3
2 changed files with 76 additions and 74 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

139
install.cpp
View file

@ -100,7 +100,7 @@ bool read_metadata_from_package(ZipArchiveHandle zip, std::string* metadata) {
} }
// Read the build.version.incremental of src/tgt from the metadata and log it to last_install. // Read the build.version.incremental of src/tgt from the metadata and log it to last_install.
static void read_source_target_build(ZipArchiveHandle zip, std::vector<std::string>& log_buffer) { static void read_source_target_build(ZipArchiveHandle zip, std::vector<std::string>* log_buffer) {
std::string metadata; std::string metadata;
if (!read_metadata_from_package(zip, &metadata)) { if (!read_metadata_from_package(zip, &metadata)) {
return; return;
@ -114,12 +114,12 @@ static void read_source_target_build(ZipArchiveHandle zip, std::vector<std::stri
if (android::base::StartsWith(str, "pre-build-incremental")) { if (android::base::StartsWith(str, "pre-build-incremental")) {
int source_build = parse_build_number(str); int source_build = parse_build_number(str);
if (source_build != -1) { if (source_build != -1) {
log_buffer.push_back(android::base::StringPrintf("source_build: %d", source_build)); log_buffer->push_back(android::base::StringPrintf("source_build: %d", source_build));
} }
} else if (android::base::StartsWith(str, "post-build-incremental")) { } else if (android::base::StartsWith(str, "post-build-incremental")) {
int target_build = parse_build_number(str); int target_build = parse_build_number(str);
if (target_build != -1) { if (target_build != -1) {
log_buffer.push_back(android::base::StringPrintf("target_build: %d", target_build)); log_buffer->push_back(android::base::StringPrintf("target_build: %d", target_build));
} }
} }
} }
@ -308,8 +308,8 @@ static void log_max_temperature(int* max_temperature) {
} }
// If the package contains an update binary, extract it and run it. // If the package contains an update binary, extract it and run it.
static int try_update_binary(const char* path, ZipArchiveHandle zip, bool* wipe_cache, static int try_update_binary(const std::string& path, ZipArchiveHandle zip, bool* wipe_cache,
std::vector<std::string>& log_buffer, int retry_count, std::vector<std::string>* log_buffer, int retry_count,
int* max_temperature) { int* max_temperature) {
read_source_target_build(zip, log_buffer); read_source_target_build(zip, log_buffer);
@ -452,7 +452,7 @@ static int try_update_binary(const char* path, ZipArchiveHandle zip, bool* wipe_
} else if (command == "log") { } else if (command == "log") {
if (!args.empty()) { if (!args.empty()) {
// Save the logging request from updater and write to last_install later. // Save the logging request from updater and write to last_install later.
log_buffer.push_back(args); log_buffer->push_back(args);
} else { } else {
LOG(ERROR) << "invalid \"log\" parameters: " << line; LOG(ERROR) << "invalid \"log\" parameters: " << line;
} }
@ -547,78 +547,81 @@ bool verify_package_compatibility(ZipArchiveHandle package_zip) {
return false; return false;
} }
static int static int really_install_package(const std::string& path, bool* wipe_cache, bool needs_mount,
really_install_package(const char *path, bool* wipe_cache, bool needs_mount, std::vector<std::string>* log_buffer, int retry_count,
std::vector<std::string>& log_buffer, int retry_count, int* max_temperature) int* max_temperature) {
{ ui->SetBackground(RecoveryUI::INSTALLING_UPDATE);
ui->SetBackground(RecoveryUI::INSTALLING_UPDATE); ui->Print("Finding update package...\n");
ui->Print("Finding update package...\n"); // Give verification half the progress bar...
// Give verification half the progress bar... ui->SetProgressType(RecoveryUI::DETERMINATE);
ui->SetProgressType(RecoveryUI::DETERMINATE); ui->ShowProgress(VERIFICATION_PROGRESS_FRACTION, VERIFICATION_PROGRESS_TIME);
ui->ShowProgress(VERIFICATION_PROGRESS_FRACTION, VERIFICATION_PROGRESS_TIME); LOG(INFO) << "Update location: " << path;
LOG(INFO) << "Update location: " << path;
// Map the update package into memory. // Map the update package into memory.
ui->Print("Opening update package...\n"); ui->Print("Opening update package...\n");
if (path && needs_mount) { if (needs_mount) {
if (path[0] == '@') { if (path[0] == '@') {
ensure_path_mounted(path+1); ensure_path_mounted(path.substr(1).c_str());
} else { } else {
ensure_path_mounted(path); ensure_path_mounted(path.c_str());
}
} }
}
MemMapping map; MemMapping map;
if (sysMapFile(path, &map) != 0) { if (sysMapFile(path.c_str(), &map) != 0) {
LOG(ERROR) << "failed to map file"; LOG(ERROR) << "failed to map file";
return INSTALL_CORRUPT; return INSTALL_CORRUPT;
} }
// Verify package. // Verify package.
if (!verify_package(map.addr, map.length)) { if (!verify_package(map.addr, map.length)) {
log_buffer.push_back(android::base::StringPrintf("error: %d", kZipVerificationFailure)); log_buffer->push_back(android::base::StringPrintf("error: %d", kZipVerificationFailure));
sysReleaseMap(&map); sysReleaseMap(&map);
return INSTALL_CORRUPT; return INSTALL_CORRUPT;
} }
// Try to open the package. // Try to open the package.
ZipArchiveHandle zip; ZipArchiveHandle zip;
int err = OpenArchiveFromMemory(map.addr, map.length, path, &zip); int err = OpenArchiveFromMemory(map.addr, map.length, path.c_str(), &zip);
if (err != 0) { if (err != 0) {
LOG(ERROR) << "Can't open " << path << " : " << ErrorCodeString(err); LOG(ERROR) << "Can't open " << path << " : " << ErrorCodeString(err);
log_buffer.push_back(android::base::StringPrintf("error: %d", kZipOpenFailure)); log_buffer->push_back(android::base::StringPrintf("error: %d", kZipOpenFailure));
sysReleaseMap(&map);
CloseArchive(zip);
return INSTALL_CORRUPT;
}
// Additionally verify the compatibility of the package.
if (!verify_package_compatibility(zip)) {
log_buffer.push_back(android::base::StringPrintf("error: %d", kPackageCompatibilityFailure));
sysReleaseMap(&map);
CloseArchive(zip);
return INSTALL_CORRUPT;
}
// Verify and install the contents of the package.
ui->Print("Installing update...\n");
if (retry_count > 0) {
ui->Print("Retry attempt: %d\n", retry_count);
}
ui->SetEnableReboot(false);
int result = try_update_binary(path, zip, wipe_cache, log_buffer, retry_count, max_temperature);
ui->SetEnableReboot(true);
ui->Print("\n");
sysReleaseMap(&map); sysReleaseMap(&map);
CloseArchive(zip); CloseArchive(zip);
return result; return INSTALL_CORRUPT;
}
// Additionally verify the compatibility of the package.
if (!verify_package_compatibility(zip)) {
log_buffer->push_back(android::base::StringPrintf("error: %d", kPackageCompatibilityFailure));
sysReleaseMap(&map);
CloseArchive(zip);
return INSTALL_CORRUPT;
}
// Verify and install the contents of the package.
ui->Print("Installing update...\n");
if (retry_count > 0) {
ui->Print("Retry attempt: %d\n", retry_count);
}
ui->SetEnableReboot(false);
int result = try_update_binary(path, zip, wipe_cache, log_buffer, retry_count, max_temperature);
ui->SetEnableReboot(true);
ui->Print("\n");
sysReleaseMap(&map);
CloseArchive(zip);
return result;
} }
int install_package(const char* path, bool* wipe_cache, const char* install_file, bool needs_mount, int install_package(const std::string& path, bool* wipe_cache, const std::string& install_file,
int retry_count) { bool needs_mount, int retry_count) {
CHECK(!path.empty());
CHECK(!install_file.empty());
CHECK(wipe_cache != nullptr);
modified_flash = true; modified_flash = true;
auto start = std::chrono::system_clock::now(); auto start = std::chrono::system_clock::now();
@ -631,7 +634,7 @@ int install_package(const char* path, bool* wipe_cache, const char* install_file
LOG(ERROR) << "failed to set up expected mounts for install; aborting"; LOG(ERROR) << "failed to set up expected mounts for install; aborting";
result = INSTALL_ERROR; result = INSTALL_ERROR;
} else { } else {
result = really_install_package(path, wipe_cache, needs_mount, log_buffer, retry_count, result = really_install_package(path, wipe_cache, needs_mount, &log_buffer, retry_count,
&max_temperature); &max_temperature);
} }

11
install.h
View file

@ -23,10 +23,9 @@
enum { INSTALL_SUCCESS, INSTALL_ERROR, INSTALL_CORRUPT, INSTALL_NONE, INSTALL_SKIPPED, enum { INSTALL_SUCCESS, INSTALL_ERROR, INSTALL_CORRUPT, INSTALL_NONE, INSTALL_SKIPPED,
INSTALL_RETRY }; INSTALL_RETRY };
// Install the package specified by root_path. If INSTALL_SUCCESS is // Installs the given update package. If INSTALL_SUCCESS is returned and *wipe_cache is true on
// returned and *wipe_cache is true on exit, caller should wipe the // exit, caller should wipe the cache partition.
// cache partition. int install_package(const std::string& package, bool* wipe_cache, const std::string& install_file,
int install_package(const char* root_path, bool* wipe_cache, const char* install_file,
bool needs_mount, int retry_count); bool needs_mount, int retry_count);
// Verify the package by ota keys. Return true if the package is verified successfully, // Verify the package by ota keys. Return true if the package is verified successfully,
@ -35,9 +34,9 @@ bool verify_package(const unsigned char* package_data, size_t package_size);
// Read meta data file of the package, write its content in the string pointed by meta_data. // Read meta data file of the package, write its content in the string pointed by meta_data.
// Return true if succeed, otherwise return false. // Return true if succeed, otherwise return false.
bool read_metadata_from_package(ZipArchiveHandle zip, std::string* meta_data); bool read_metadata_from_package(ZipArchiveHandle zip, std::string* metadata);
// Verifes the compatibility info in a Treble-compatible package. Returns true directly if the // Verifies the compatibility info in a Treble-compatible package. Returns true directly if the
// entry doesn't exist. // entry doesn't exist.
bool verify_package_compatibility(ZipArchiveHandle package_zip); bool verify_package_compatibility(ZipArchiveHandle package_zip);