| /* |
| * 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 "fastboot.h" |
| #include "fs.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> |
| |
| #define OP_DOWNLOAD 1 |
| #define OP_COMMAND 2 |
| #define OP_QUERY 3 |
| #define OP_NOTICE 4 |
| #define OP_DOWNLOAD_SPARSE 5 |
| #define OP_WAIT_FOR_DISCONNECT 6 |
| #define OP_DOWNLOAD_FD 7 |
| #define OP_UPLOAD 8 |
| |
| typedef struct Action Action; |
| |
| #define CMD_SIZE 64 |
| |
| struct Action { |
| unsigned op; |
| Action* next; |
| |
| char cmd[CMD_SIZE]; |
| const char* prod; |
| void* data; |
| int fd; |
| |
| // The protocol only supports 32-bit sizes, so you'll have to break |
| // anything larger into chunks. |
| uint32_t size; |
| |
| const char *msg; |
| int (*func)(Action* a, int status, const char* resp); |
| |
| double start; |
| }; |
| |
| static Action *action_list = 0; |
| static Action *action_last = 0; |
| |
| |
| |
| |
| bool fb_getvar(Transport* transport, const std::string& key, std::string* value) { |
| std::string cmd = "getvar:"; |
| cmd += key; |
| |
| char buf[FB_RESPONSE_SZ + 1]; |
| memset(buf, 0, sizeof(buf)); |
| if (fb_command_response(transport, cmd.c_str(), buf)) { |
| return false; |
| } |
| *value = buf; |
| return true; |
| } |
| |
| 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(unsigned op, const char *fmt, ...) |
| { |
| va_list ap; |
| size_t cmdsize; |
| |
| Action* a = reinterpret_cast<Action*>(calloc(1, sizeof(Action))); |
| if (a == nullptr) die("out of memory"); |
| |
| va_start(ap, fmt); |
| cmdsize = vsnprintf(a->cmd, sizeof(a->cmd), fmt, ap); |
| va_end(ap); |
| |
| if (cmdsize >= sizeof(a->cmd)) { |
| free(a); |
| die("Command length (%d) exceeds maximum size (%d)", cmdsize, sizeof(a->cmd)); |
| } |
| |
| if (action_last) { |
| action_last->next = a; |
| } else { |
| action_list = a; |
| } |
| action_last = a; |
| a->op = op; |
| a->func = cb_default; |
| |
| a->start = -1; |
| |
| return a; |
| } |
| |
| void fb_set_active(const char *slot) |
| { |
| Action *a; |
| a = queue_action(OP_COMMAND, "set_active:%s", slot); |
| a->msg = mkmsg("Setting current slot to '%s'", slot); |
| } |
| |
| void fb_queue_erase(const char *ptn) |
| { |
| Action *a; |
| a = queue_action(OP_COMMAND, "erase:%s", ptn); |
| a->msg = mkmsg("erasing '%s'", ptn); |
| } |
| |
| void fb_queue_flash_fd(const char *ptn, int fd, uint32_t sz) |
| { |
| Action *a; |
| |
| a = queue_action(OP_DOWNLOAD_FD, ""); |
| a->fd = fd; |
| a->size = sz; |
| a->msg = mkmsg("sending '%s' (%d KB)", ptn, sz / 1024); |
| |
| a = queue_action(OP_COMMAND, "flash:%s", ptn); |
| a->msg = mkmsg("writing '%s'", ptn); |
| } |
| |
| void fb_queue_flash(const char *ptn, void *data, uint32_t sz) |
| { |
| Action *a; |
| |
| a = queue_action(OP_DOWNLOAD, ""); |
| a->data = data; |
| a->size = sz; |
| a->msg = mkmsg("sending '%s' (%d KB)", ptn, sz / 1024); |
| |
| a = queue_action(OP_COMMAND, "flash:%s", ptn); |
| a->msg = mkmsg("writing '%s'", ptn); |
| } |
| |
| void fb_queue_flash_sparse(const char* ptn, struct sparse_file* s, uint32_t sz, size_t current, |
| size_t total) { |
| Action *a; |
| |
| a = queue_action(OP_DOWNLOAD_SPARSE, ""); |
| a->data = s; |
| a->size = 0; |
| a->msg = mkmsg("sending sparse '%s' %zu/%zu (%d KB)", ptn, current, total, sz / 1024); |
| |
| a = queue_action(OP_COMMAND, "flash:%s", ptn); |
| a->msg = mkmsg("writing '%s' %zu/%zu", ptn, 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; |
| int yes; |
| |
| if (status) { |
| fprintf(stderr,"FAILED (%s)\n", resp); |
| return status; |
| } |
| |
| if (a->prod) { |
| if (strcmp(a->prod, cur_product) != 0) { |
| double split = now(); |
| fprintf(stderr,"IGNORE, product is %s required only for %s [%7.3fs]\n", |
| cur_product, a->prod, (split - a->start)); |
| a->start = split; |
| return 0; |
| } |
| } |
| |
| 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 + 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); |
| } |
| |
| static char* xstrdup(const char* s) { |
| char* result = strdup(s); |
| if (!result) die("out of memory"); |
| return result; |
| } |
| |
| void fb_queue_require(const char *prod, const char *var, |
| bool invert, size_t nvalues, const char **value) |
| { |
| Action *a; |
| a = queue_action(OP_QUERY, "getvar:%s", var); |
| a->prod = prod; |
| a->data = value; |
| a->size = nvalues; |
| a->msg = mkmsg("checking %s", 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, 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 char* var, const char* prettyname) { |
| Action* a = queue_action(OP_QUERY, "getvar:%s", var); |
| a->data = xstrdup(prettyname); |
| 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, resp); |
| return status; |
| } |
| strncpy(reinterpret_cast<char*>(a->data), resp, a->size); |
| return 0; |
| } |
| |
| void fb_queue_query_save(const char* var, char* dest, uint32_t dest_size) { |
| Action* a = queue_action(OP_QUERY, "getvar:%s", 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(void) |
| { |
| Action *a = queue_action(OP_COMMAND, "reboot"); |
| a->func = cb_do_nothing; |
| a->msg = "rebooting"; |
| } |
| |
| void fb_queue_command(const char *cmd, const char *msg) |
| { |
| Action *a = queue_action(OP_COMMAND, cmd); |
| a->msg = msg; |
| } |
| |
| void fb_queue_download(const char *name, void *data, uint32_t size) |
| { |
| Action *a = queue_action(OP_DOWNLOAD, ""); |
| a->data = data; |
| a->size = size; |
| a->msg = mkmsg("downloading '%s'", name); |
| } |
| |
| void fb_queue_download_fd(const char *name, int fd, uint32_t sz) |
| { |
| Action *a; |
| a = queue_action(OP_DOWNLOAD_FD, ""); |
| a->fd = fd; |
| a->size = sz; |
| a->msg = mkmsg("sending '%s' (%d KB)", name, sz / 1024); |
| } |
| |
| void fb_queue_upload(const char* outfile) { |
| Action* a = queue_action(OP_UPLOAD, ""); |
| a->data = xstrdup(outfile); |
| a->msg = mkmsg("uploading '%s'", outfile); |
| } |
| |
| void fb_queue_notice(const char* notice) { |
| Action *a = queue_action(OP_NOTICE, ""); |
| a->data = (void*) notice; |
| } |
| |
| void fb_queue_wait_for_disconnect(void) |
| { |
| queue_action(OP_WAIT_FOR_DISCONNECT, ""); |
| } |
| |
| int64_t fb_execute_queue(Transport* transport) |
| { |
| Action *a; |
| char resp[FB_RESPONSE_SZ+1]; |
| int64_t status = 0; |
| |
| a = action_list; |
| if (!a) |
| return status; |
| resp[FB_RESPONSE_SZ] = 0; |
| |
| double start = -1; |
| for (a = action_list; a; a = a->next) { |
| a->start = now(); |
| if (start < 0) start = a->start; |
| if (a->msg) { |
| // fprintf(stderr,"%30s... ",a->msg); |
| fprintf(stderr,"%s...\n",a->msg); |
| } |
| if (a->op == OP_DOWNLOAD) { |
| status = fb_download_data(transport, a->data, 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_data_fd(transport, 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_command(transport, a->cmd); |
| status = a->func(a, status, status ? fb_get_error().c_str() : ""); |
| if (status) break; |
| } else if (a->op == OP_QUERY) { |
| status = fb_command_response(transport, a->cmd, resp); |
| status = a->func(a, status, status ? fb_get_error().c_str() : resp); |
| if (status) break; |
| } else if (a->op == OP_NOTICE) { |
| fprintf(stderr,"%s\n",(char*)a->data); |
| } else if (a->op == OP_DOWNLOAD_SPARSE) { |
| status = fb_download_data_sparse(transport, 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) { |
| transport->WaitForDisconnect(); |
| } else if (a->op == OP_UPLOAD) { |
| status = fb_upload_data(transport, reinterpret_cast<char*>(a->data)); |
| status = a->func(a, status, status ? fb_get_error().c_str() : ""); |
| } else { |
| die("bogus action"); |
| } |
| } |
| |
| fprintf(stderr,"finished. total time: %.3fs\n", (now() - start)); |
| return status; |
| } |