11f12099cc
There is little advantage and increasingly much disadvantage to queueing up fastboot commands before executing them. This change removes the queue in the most simple way possible to enable further clean up. Test: fastboot works Change-Id: I9abab05df07ed167dbe6a42e1eb9eab8f8d4f157
272 lines
8 KiB
C++
272 lines
8 KiB
C++
/*
|
|
* Copyright (C) 2008 The Android Open Source Project
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions
|
|
* are met:
|
|
* * Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
* * Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimer in
|
|
* the documentation and/or other materials provided with the
|
|
* distribution.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
|
|
* FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
|
|
* COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
|
|
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
|
|
* BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
|
|
* OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
|
|
* AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
|
|
* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
|
|
* OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
|
* SUCH DAMAGE.
|
|
*/
|
|
#include "engine.h"
|
|
|
|
#include <errno.h>
|
|
#include <stdarg.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/types.h>
|
|
#include <unistd.h>
|
|
|
|
#include <memory>
|
|
#include <vector>
|
|
|
|
#include <android-base/stringprintf.h>
|
|
|
|
#include "constants.h"
|
|
#include "transport.h"
|
|
|
|
using android::base::StringPrintf;
|
|
|
|
static fastboot::FastBootDriver* fb = nullptr;
|
|
|
|
void fb_init(fastboot::FastBootDriver& fbi) {
|
|
fb = &fbi;
|
|
auto cb = [](std::string& info) { fprintf(stderr, "(bootloader) %s\n", info.c_str()); };
|
|
fb->SetInfoCallback(cb);
|
|
}
|
|
|
|
void fb_reinit(Transport* transport) {
|
|
if (Transport* old_transport = fb->set_transport(transport)) {
|
|
delete old_transport;
|
|
}
|
|
}
|
|
|
|
const std::string fb_get_error() {
|
|
return fb->Error();
|
|
}
|
|
|
|
bool fb_getvar(const std::string& key, std::string* value) {
|
|
return !fb->GetVar(key, value);
|
|
}
|
|
|
|
static void HandleResult(double start, int status) {
|
|
if (status) {
|
|
fprintf(stderr, "FAILED (%s)\n", fb->Error().c_str());
|
|
die("Command failed");
|
|
} else {
|
|
double split = now();
|
|
fprintf(stderr, "OKAY [%7.3fs]\n", (split - start));
|
|
}
|
|
}
|
|
|
|
#define RUN_COMMAND(command) \
|
|
{ \
|
|
double start = now(); \
|
|
auto status = (command); \
|
|
HandleResult(start, status); \
|
|
}
|
|
|
|
void fb_set_active(const std::string& slot) {
|
|
Status("Setting current slot to '" + slot + "'");
|
|
RUN_COMMAND(fb->SetActive(slot));
|
|
}
|
|
|
|
void fb_erase(const std::string& partition) {
|
|
Status("Erasing '" + partition + "'");
|
|
RUN_COMMAND(fb->Erase(partition));
|
|
}
|
|
|
|
void fb_flash_fd(const std::string& partition, int fd, uint32_t sz) {
|
|
Status(StringPrintf("Sending '%s' (%u KB)", partition.c_str(), sz / 1024));
|
|
RUN_COMMAND(fb->Download(fd, sz));
|
|
|
|
Status("Writing '" + partition + "'");
|
|
RUN_COMMAND(fb->Flash(partition));
|
|
}
|
|
|
|
void fb_flash(const std::string& partition, void* data, uint32_t sz) {
|
|
Status(StringPrintf("Sending '%s' (%u KB)", partition.c_str(), sz / 1024));
|
|
RUN_COMMAND(fb->Download(static_cast<char*>(data), sz));
|
|
|
|
Status("Writing '" + partition + "'");
|
|
RUN_COMMAND(fb->Flash(partition));
|
|
}
|
|
|
|
void fb_flash_sparse(const std::string& partition, struct sparse_file* s, uint32_t sz,
|
|
size_t current, size_t total) {
|
|
Status(StringPrintf("Sending sparse '%s' %zu/%zu (%u KB)", partition.c_str(), current, total,
|
|
sz / 1024));
|
|
RUN_COMMAND(fb->Download(s));
|
|
|
|
Status(StringPrintf("Writing sparse '%s' %zu/%zu", partition.c_str(), current, total));
|
|
RUN_COMMAND(fb->Flash(partition));
|
|
}
|
|
|
|
void fb_create_partition(const std::string& partition, const std::string& size) {
|
|
Status("Creating '" + partition + "'");
|
|
RUN_COMMAND(fb->RawCommand(FB_CMD_CREATE_PARTITION ":" + partition + ":" + size));
|
|
}
|
|
|
|
void fb_delete_partition(const std::string& partition) {
|
|
Status("Deleting '" + partition + "'");
|
|
RUN_COMMAND(fb->RawCommand(FB_CMD_DELETE_PARTITION ":" + partition));
|
|
}
|
|
|
|
void fb_resize_partition(const std::string& partition, const std::string& size) {
|
|
Status("Resizing '" + partition + "'");
|
|
RUN_COMMAND(fb->RawCommand(FB_CMD_RESIZE_PARTITION ":" + partition + ":" + size));
|
|
}
|
|
|
|
static int match(const char* str, const char** value, unsigned count) {
|
|
unsigned n;
|
|
|
|
for (n = 0; n < count; n++) {
|
|
const char *val = value[n];
|
|
int len = strlen(val);
|
|
int match;
|
|
|
|
if ((len > 1) && (val[len-1] == '*')) {
|
|
len--;
|
|
match = !strncmp(val, str, len);
|
|
} else {
|
|
match = !strcmp(val, str);
|
|
}
|
|
|
|
if (match) return 1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
void fb_require(const std::string& product, const std::string& var, bool invert, size_t count,
|
|
const char** values) {
|
|
Status("Checking '" + var + "'");
|
|
|
|
double start = now();
|
|
|
|
std::string var_value;
|
|
auto status = fb->GetVar(var, &var_value);
|
|
|
|
if (status) {
|
|
fprintf(stderr, "getvar:%s FAILED (%s)\n", var.c_str(), fb->Error().c_str());
|
|
die("requirements not met!");
|
|
}
|
|
|
|
if (!product.empty()) {
|
|
if (product != cur_product) {
|
|
double split = now();
|
|
fprintf(stderr, "IGNORE, product is %s required only for %s [%7.3fs]\n", cur_product,
|
|
product.c_str(), (split - start));
|
|
return;
|
|
}
|
|
}
|
|
|
|
int yes = match(var_value.c_str(), values, count);
|
|
if (invert) yes = !yes;
|
|
|
|
if (yes) {
|
|
double split = now();
|
|
fprintf(stderr, "OKAY [%7.3fs]\n", (split - start));
|
|
return;
|
|
}
|
|
|
|
fprintf(stderr, "FAILED\n\n");
|
|
fprintf(stderr, "Device %s is '%s'.\n", var.c_str(), var_value.c_str());
|
|
fprintf(stderr, "Update %s '%s'", invert ? "rejects" : "requires", values[0]);
|
|
for (size_t n = 1; n < count; n++) {
|
|
fprintf(stderr, " or '%s'", values[n]);
|
|
}
|
|
fprintf(stderr, ".\n\n");
|
|
die("requirements not met!");
|
|
}
|
|
|
|
void fb_display(const std::string& label, const std::string& var) {
|
|
std::string value;
|
|
auto status = fb->GetVar(var, &value);
|
|
|
|
if (status) {
|
|
fprintf(stderr, "getvar:%s FAILED (%s)\n", var.c_str(), fb->Error().c_str());
|
|
return;
|
|
}
|
|
fprintf(stderr, "%s: %s\n", label.c_str(), value.c_str());
|
|
}
|
|
|
|
void fb_query_save(const std::string& var, char* dest, uint32_t dest_size) {
|
|
std::string value;
|
|
auto status = fb->GetVar(var, &value);
|
|
|
|
if (status) {
|
|
fprintf(stderr, "getvar:%s FAILED (%s)\n", var.c_str(), fb->Error().c_str());
|
|
return;
|
|
}
|
|
|
|
strncpy(dest, value.c_str(), dest_size);
|
|
}
|
|
|
|
void fb_reboot() {
|
|
fprintf(stderr, "Rebooting");
|
|
fb->Reboot();
|
|
fprintf(stderr, "\n");
|
|
}
|
|
|
|
void fb_command(const std::string& cmd, const std::string& msg) {
|
|
Status(msg);
|
|
RUN_COMMAND(fb->RawCommand(cmd));
|
|
}
|
|
|
|
void fb_download(const std::string& name, void* data, uint32_t size) {
|
|
Status("Downloading '" + name + "'");
|
|
RUN_COMMAND(fb->Download(static_cast<char*>(data), size));
|
|
}
|
|
|
|
void fb_download_fd(const std::string& name, int fd, uint32_t sz) {
|
|
Status(StringPrintf("Sending '%s' (%u KB)", name.c_str(), sz / 1024));
|
|
RUN_COMMAND(fb->Download(fd, sz));
|
|
}
|
|
|
|
void fb_upload(const std::string& outfile) {
|
|
Status("Uploading '" + outfile + "'");
|
|
RUN_COMMAND(fb->Upload(outfile));
|
|
}
|
|
|
|
void fb_notice(const std::string& notice) {
|
|
Status(notice);
|
|
fprintf(stderr, "\n");
|
|
}
|
|
|
|
void fb_wait_for_disconnect() {
|
|
fb->WaitForDisconnect();
|
|
}
|
|
|
|
bool fb_reboot_to_userspace() {
|
|
Status("Rebooting to userspace fastboot");
|
|
verbose("\n");
|
|
|
|
if (fb->RebootTo("fastboot") != fastboot::RetCode::SUCCESS) {
|
|
fprintf(stderr, "FAILED (%s)\n", fb->Error().c_str());
|
|
return false;
|
|
}
|
|
fprintf(stderr, "OKAY\n");
|
|
|
|
fb_reinit(nullptr);
|
|
return true;
|
|
}
|