| /* | 
 |  * 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" | 
 |  | 
 | enum Op { | 
 |     OP_DOWNLOAD, | 
 |     OP_COMMAND, | 
 |     OP_QUERY, | 
 |     OP_NOTICE, | 
 |     OP_DOWNLOAD_SPARSE, | 
 |     OP_WAIT_FOR_DISCONNECT, | 
 |     OP_DOWNLOAD_FD, | 
 |     OP_UPLOAD, | 
 | }; | 
 |  | 
 | struct Action { | 
 |     Action(Op op, const std::string& cmd) : op(op), cmd(cmd) {} | 
 |  | 
 |     Op op; | 
 |     std::string cmd; | 
 |     std::string msg; | 
 |  | 
 |     std::string product; | 
 |  | 
 |     void* data = nullptr; | 
 |     // The protocol only supports 32-bit sizes, so you'll have to break | 
 |     // anything larger into multiple chunks. | 
 |     uint32_t size = 0; | 
 |  | 
 |     int fd = -1; | 
 |  | 
 |     int (*func)(Action& a, int status, const char* resp) = nullptr; | 
 |  | 
 |     double start = -1; | 
 | }; | 
 |  | 
 | static std::vector<std::unique_ptr<Action>> action_list; | 
 | 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); | 
 | } | 
 |  | 
 | 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 int cb_default(Action& a, int status, const char* resp) { | 
 |     if (status) { | 
 |         fprintf(stderr,"FAILED (%s)\n", resp); | 
 |     } else { | 
 |         double split = now(); | 
 |         fprintf(stderr, "OKAY [%7.3fs]\n", (split - a.start)); | 
 |         a.start = split; | 
 |     } | 
 |     return status; | 
 | } | 
 |  | 
 | static Action& queue_action(Op op, const std::string& cmd) { | 
 |     std::unique_ptr<Action> a{new Action(op, cmd)}; | 
 |     a->func = cb_default; | 
 |  | 
 |     action_list.push_back(std::move(a)); | 
 |     return *action_list.back(); | 
 | } | 
 |  | 
 | void fb_set_active(const std::string& slot) { | 
 |     Action& a = queue_action(OP_COMMAND, FB_CMD_SET_ACTIVE ":" + slot); | 
 |     a.msg = "Setting current slot to '" + slot + "'"; | 
 | } | 
 |  | 
 | void fb_queue_erase(const std::string& partition) { | 
 |     Action& a = queue_action(OP_COMMAND, FB_CMD_ERASE ":" + partition); | 
 |     a.msg = "Erasing '" + partition + "'"; | 
 | } | 
 |  | 
 | void fb_queue_flash_fd(const std::string& partition, int fd, uint32_t sz) { | 
 |     Action& a = queue_action(OP_DOWNLOAD_FD, ""); | 
 |     a.fd = fd; | 
 |     a.size = sz; | 
 |     a.msg = android::base::StringPrintf("Sending '%s' (%u KB)", partition.c_str(), sz / 1024); | 
 |  | 
 |     Action& b = queue_action(OP_COMMAND, FB_CMD_FLASH ":" + partition); | 
 |     b.msg = "Writing '" + partition + "'"; | 
 | } | 
 |  | 
 | void fb_queue_flash(const std::string& partition, void* data, uint32_t sz) { | 
 |     Action& a = queue_action(OP_DOWNLOAD, ""); | 
 |     a.data = data; | 
 |     a.size = sz; | 
 |     a.msg = android::base::StringPrintf("Sending '%s' (%u KB)", partition.c_str(), sz / 1024); | 
 |  | 
 |     Action& b = queue_action(OP_COMMAND, FB_CMD_FLASH ":" + partition); | 
 |     b.msg = "Writing '" + partition + "'"; | 
 | } | 
 |  | 
 | void fb_queue_flash_sparse(const std::string& partition, struct sparse_file* s, uint32_t sz, | 
 |                            size_t current, size_t total) { | 
 |     Action& a = queue_action(OP_DOWNLOAD_SPARSE, ""); | 
 |     a.data = s; | 
 |     a.size = 0; | 
 |     a.msg = android::base::StringPrintf("Sending sparse '%s' %zu/%zu (%u KB)", partition.c_str(), | 
 |                                         current, total, sz / 1024); | 
 |  | 
 |     Action& b = queue_action(OP_COMMAND, FB_CMD_FLASH ":" + partition); | 
 |     b.msg = android::base::StringPrintf("Writing sparse '%s' %zu/%zu", partition.c_str(), current, | 
 |                                         total); | 
 | } | 
 |  | 
 | 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; | 
 | } | 
 |  | 
 | static int cb_check(Action& a, int status, const char* resp, int invert) { | 
 |     const char** value = reinterpret_cast<const char**>(a.data); | 
 |     unsigned count = a.size; | 
 |     unsigned n; | 
 |  | 
 |     if (status) { | 
 |         fprintf(stderr,"FAILED (%s)\n", resp); | 
 |         return status; | 
 |     } | 
 |  | 
 |     if (!a.product.empty()) { | 
 |         if (a.product != cur_product) { | 
 |             double split = now(); | 
 |             fprintf(stderr, "IGNORE, product is %s required only for %s [%7.3fs]\n", cur_product, | 
 |                     a.product.c_str(), (split - a.start)); | 
 |             a.start = split; | 
 |             return 0; | 
 |         } | 
 |     } | 
 |  | 
 |     int yes = match(resp, value, count); | 
 |     if (invert) yes = !yes; | 
 |  | 
 |     if (yes) { | 
 |         double split = now(); | 
 |         fprintf(stderr, "OKAY [%7.3fs]\n", (split - a.start)); | 
 |         a.start = split; | 
 |         return 0; | 
 |     } | 
 |  | 
 |     fprintf(stderr, "FAILED\n\n"); | 
 |     fprintf(stderr, "Device %s is '%s'.\n", a.cmd.c_str() + 7, resp); | 
 |     fprintf(stderr, "Update %s '%s'", invert ? "rejects" : "requires", value[0]); | 
 |     for (n = 1; n < count; n++) { | 
 |         fprintf(stderr, " or '%s'", value[n]); | 
 |     } | 
 |     fprintf(stderr, ".\n\n"); | 
 |     return -1; | 
 | } | 
 |  | 
 | static int cb_require(Action& a, int status, const char* resp) { | 
 |     return cb_check(a, status, resp, 0); | 
 | } | 
 |  | 
 | static int cb_reject(Action& a, int status, const char* resp) { | 
 |     return cb_check(a, status, resp, 1); | 
 | } | 
 |  | 
 | void fb_queue_require(const std::string& product, const std::string& var, bool invert, | 
 |                       size_t nvalues, const char** values) { | 
 |     Action& a = queue_action(OP_QUERY, FB_CMD_GETVAR ":" + var); | 
 |     a.product = product; | 
 |     a.data = values; | 
 |     a.size = nvalues; | 
 |     a.msg = "Checking " + var; | 
 |     a.func = invert ? cb_reject : cb_require; | 
 |     if (a.data == nullptr) die("out of memory"); | 
 | } | 
 |  | 
 | static int cb_display(Action& a, int status, const char* resp) { | 
 |     if (status) { | 
 |         fprintf(stderr, "%s FAILED (%s)\n", a.cmd.c_str(), resp); | 
 |         return status; | 
 |     } | 
 |     fprintf(stderr, "%s: %s\n", static_cast<const char*>(a.data), resp); | 
 |     free(static_cast<char*>(a.data)); | 
 |     return 0; | 
 | } | 
 |  | 
 | void fb_queue_display(const std::string& label, const std::string& var) { | 
 |     Action& a = queue_action(OP_QUERY, FB_CMD_GETVAR ":" + var); | 
 |     a.data = xstrdup(label.c_str()); | 
 |     a.func = cb_display; | 
 | } | 
 |  | 
 | static int cb_save(Action& a, int status, const char* resp) { | 
 |     if (status) { | 
 |         fprintf(stderr, "%s FAILED (%s)\n", a.cmd.c_str(), resp); | 
 |         return status; | 
 |     } | 
 |     strncpy(reinterpret_cast<char*>(a.data), resp, a.size); | 
 |     return 0; | 
 | } | 
 |  | 
 | void fb_queue_query_save(const std::string& var, char* dest, uint32_t dest_size) { | 
 |     Action& a = queue_action(OP_QUERY, FB_CMD_GETVAR ":" + var); | 
 |     a.data = dest; | 
 |     a.size = dest_size; | 
 |     a.func = cb_save; | 
 | } | 
 |  | 
 | static int cb_do_nothing(Action&, int, const char*) { | 
 |     fprintf(stderr, "\n"); | 
 |     return 0; | 
 | } | 
 |  | 
 | void fb_queue_reboot() { | 
 |     Action& a = queue_action(OP_COMMAND, FB_CMD_REBOOT); | 
 |     a.func = cb_do_nothing; | 
 |     a.msg = "Rebooting"; | 
 | } | 
 |  | 
 | void fb_queue_command(const std::string& cmd, const std::string& msg) { | 
 |     Action& a = queue_action(OP_COMMAND, cmd); | 
 |     a.msg = msg; | 
 | } | 
 |  | 
 | void fb_queue_download(const std::string& name, void* data, uint32_t size) { | 
 |     Action& a = queue_action(OP_DOWNLOAD, ""); | 
 |     a.data = data; | 
 |     a.size = size; | 
 |     a.msg = "Downloading '" + name + "'"; | 
 | } | 
 |  | 
 | void fb_queue_download_fd(const std::string& name, int fd, uint32_t sz) { | 
 |     Action& a = queue_action(OP_DOWNLOAD_FD, ""); | 
 |     a.fd = fd; | 
 |     a.size = sz; | 
 |     a.msg = android::base::StringPrintf("Sending '%s' (%u KB)", name.c_str(), sz / 1024); | 
 | } | 
 |  | 
 | void fb_queue_upload(const std::string& outfile) { | 
 |     Action& a = queue_action(OP_UPLOAD, ""); | 
 |     a.data = xstrdup(outfile.c_str()); | 
 |     a.msg = "Uploading '" + outfile + "'"; | 
 | } | 
 |  | 
 | void fb_queue_notice(const std::string& notice) { | 
 |     Action& a = queue_action(OP_NOTICE, ""); | 
 |     a.msg = notice; | 
 | } | 
 |  | 
 | void fb_queue_wait_for_disconnect() { | 
 |     queue_action(OP_WAIT_FOR_DISCONNECT, ""); | 
 | } | 
 |  | 
 | int64_t fb_execute_queue() { | 
 |     int64_t status = 0; | 
 |     for (auto& a : action_list) { | 
 |         a->start = now(); | 
 |         if (!a->msg.empty()) { | 
 |             fprintf(stderr, "%-50s ", a->msg.c_str()); | 
 |             verbose("\n"); | 
 |         } | 
 |         if (a->op == OP_DOWNLOAD) { | 
 |             char* cbuf = static_cast<char*>(a->data); | 
 |             status = fb->Download(cbuf, a->size); | 
 |             status = a->func(*a, status, status ? fb_get_error().c_str() : ""); | 
 |             if (status) break; | 
 |         } else if (a->op == OP_DOWNLOAD_FD) { | 
 |             status = fb->Download(a->fd, a->size); | 
 |             status = a->func(*a, status, status ? fb_get_error().c_str() : ""); | 
 |             if (status) break; | 
 |         } else if (a->op == OP_COMMAND) { | 
 |             status = fb->RawCommand(a->cmd); | 
 |             status = a->func(*a, status, status ? fb_get_error().c_str() : ""); | 
 |             if (status) break; | 
 |         } else if (a->op == OP_QUERY) { | 
 |             std::string resp; | 
 |             status = fb->RawCommand(a->cmd, &resp); | 
 |             status = a->func(*a, status, status ? fb_get_error().c_str() : resp.c_str()); | 
 |             if (status) break; | 
 |         } else if (a->op == OP_NOTICE) { | 
 |             // We already showed the notice because it's in `Action::msg`. | 
 |             fprintf(stderr, "\n"); | 
 |         } else if (a->op == OP_DOWNLOAD_SPARSE) { | 
 |             status = fb->Download(reinterpret_cast<sparse_file*>(a->data)); | 
 |             status = a->func(*a, status, status ? fb_get_error().c_str() : ""); | 
 |             if (status) break; | 
 |         } else if (a->op == OP_WAIT_FOR_DISCONNECT) { | 
 |             fb->WaitForDisconnect(); | 
 |         } else if (a->op == OP_UPLOAD) { | 
 |             status = fb->Upload(reinterpret_cast<const char*>(a->data)); | 
 |             status = a->func(*a, status, status ? fb_get_error().c_str() : ""); | 
 |         } else { | 
 |             die("unknown action: %d", a->op); | 
 |         } | 
 |     } | 
 |     action_list.clear(); | 
 |     return status; | 
 | } |