395 lines
8.6 KiB
C
395 lines
8.6 KiB
C
/*
|
|
* Copyright (c) 2009, Google Inc.
|
|
* 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 <debug.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <kernel/thread.h>
|
|
#include <kernel/event.h>
|
|
#include <dev/udc.h>
|
|
|
|
|
|
void boot_linux(void *bootimg, unsigned sz);
|
|
|
|
/*
|
|
unsigned int c_usb_stat=0;
|
|
unsigned int fastboot_cstat(void){
|
|
return c_usb_stat;
|
|
}
|
|
void fastboot_sstat(unsigned int stat){
|
|
c_usb_stat = stat;
|
|
}
|
|
*/
|
|
/* todo: give lk strtoul and nuke this */
|
|
unsigned hex2unsigned(const char *x)
|
|
{
|
|
unsigned n = 0;
|
|
|
|
while(*x) {
|
|
switch(*x) {
|
|
case '0': case '1': case '2': case '3': case '4':
|
|
case '5': case '6': case '7': case '8': case '9':
|
|
n = (n << 4) | (*x - '0');
|
|
break;
|
|
case 'a': case 'b': case 'c':
|
|
case 'd': case 'e': case 'f':
|
|
n = (n << 4) | (*x - 'a' + 10);
|
|
break;
|
|
case 'A': case 'B': case 'C':
|
|
case 'D': case 'E': case 'F':
|
|
n = (n << 4) | (*x - 'A' + 10);
|
|
break;
|
|
default:
|
|
return n;
|
|
}
|
|
x++;
|
|
}
|
|
|
|
return n;
|
|
}
|
|
|
|
struct fastboot_cmd {
|
|
struct fastboot_cmd *next;
|
|
const char *prefix;
|
|
unsigned prefix_len;
|
|
void (*handle)(const char *arg, void *data, unsigned sz);
|
|
};
|
|
|
|
struct fastboot_var {
|
|
struct fastboot_var *next;
|
|
const char *name;
|
|
const char *value;
|
|
};
|
|
|
|
struct fastboot_cmd *cmdlist;
|
|
|
|
void fastboot_register(const char *prefix,
|
|
void (*handle)(const char *arg, void *data, unsigned sz))
|
|
{
|
|
struct fastboot_cmd *cmd;
|
|
cmd = malloc(sizeof(*cmd));
|
|
if (cmd) {
|
|
cmd->prefix = prefix;
|
|
cmd->prefix_len = strlen(prefix);
|
|
cmd->handle = handle;
|
|
cmd->next = cmdlist;
|
|
cmdlist = cmd;
|
|
}
|
|
}
|
|
|
|
struct fastboot_var *varlist;
|
|
|
|
void fastboot_publish(const char *name, const char *value)
|
|
{
|
|
struct fastboot_var *var;
|
|
var = malloc(sizeof(*var));
|
|
if (var) {
|
|
var->name = name;
|
|
var->value = value;
|
|
var->next = varlist;
|
|
varlist = var;
|
|
}
|
|
}
|
|
|
|
|
|
event_t usb_online;
|
|
event_t txn_done;
|
|
unsigned char buffer[4096];
|
|
struct udc_endpoint *in, *out;
|
|
struct udc_request *req;
|
|
int txn_status;
|
|
|
|
void *download_base;
|
|
unsigned download_max;
|
|
unsigned download_size;
|
|
|
|
#define STATE_OFFLINE 0
|
|
#define STATE_COMMAND 1
|
|
#define STATE_COMPLETE 2
|
|
#define STATE_ERROR 3
|
|
|
|
unsigned fastboot_state = STATE_OFFLINE;
|
|
|
|
void req_complete(struct udc_request *req, unsigned actual, int status)
|
|
{
|
|
txn_status = status;
|
|
req->length = actual;
|
|
event_signal(&txn_done, 0);
|
|
}
|
|
|
|
int usb_read(void *_buf, unsigned len)
|
|
{
|
|
int r;
|
|
unsigned xfer;
|
|
unsigned char *buf = _buf;
|
|
int count = 0;
|
|
|
|
if (fastboot_state == STATE_ERROR)
|
|
goto oops;
|
|
|
|
while (len > 0) {
|
|
xfer = (len > 4096) ? 4096 : len;
|
|
req->buf = buf;
|
|
req->length = xfer;
|
|
req->complete = req_complete;
|
|
r = udc_request_queue(out, req);
|
|
if (r < 0) {
|
|
//dprintf(INFO, "usb_read() queue failed\n");
|
|
goto oops;
|
|
}
|
|
event_wait(&txn_done);
|
|
|
|
if (txn_status < 0) {
|
|
//dprintf(INFO, "usb_read() transaction failed\n");
|
|
goto oops;
|
|
}
|
|
|
|
count += req->length;
|
|
buf += req->length;
|
|
len -= req->length;
|
|
|
|
/* short transfer? */
|
|
if (req->length != xfer) break;
|
|
}
|
|
|
|
return count;
|
|
|
|
oops:
|
|
//fastboot_sstat(0);
|
|
fastboot_state = STATE_ERROR;
|
|
return -1;
|
|
}
|
|
|
|
int usb_write(void *buf, unsigned len)
|
|
{
|
|
int r;
|
|
|
|
if (fastboot_state == STATE_ERROR)
|
|
goto oops;
|
|
|
|
req->buf = buf;
|
|
req->length = len;
|
|
req->complete = req_complete;
|
|
r = udc_request_queue(in, req);
|
|
if (r < 0) {
|
|
//dprintf(INFO, "usb_write() queue failed\n");
|
|
goto oops;
|
|
}
|
|
event_wait(&txn_done);
|
|
if (txn_status < 0) {
|
|
//dprintf(INFO, "usb_write() transaction failed\n");
|
|
goto oops;
|
|
}
|
|
return req->length;
|
|
|
|
oops:
|
|
//fastboot_sstat(0);
|
|
fastboot_state = STATE_ERROR;
|
|
return -1;
|
|
}
|
|
|
|
void fastboot_ack(const char *code, const char *reason)
|
|
{
|
|
char response[64];
|
|
|
|
if (fastboot_state != STATE_COMMAND)
|
|
return;
|
|
|
|
if (reason == 0)
|
|
reason = "";
|
|
|
|
snprintf(response, 64, "%s%s", code, reason);
|
|
fastboot_state = STATE_COMPLETE;
|
|
|
|
usb_write(response, strlen(response));
|
|
|
|
}
|
|
|
|
void fastboot_fail(const char *reason)
|
|
{
|
|
fastboot_ack("FAIL", reason);
|
|
}
|
|
|
|
void fastboot_okay(const char *info)
|
|
{
|
|
fastboot_ack("OKAY", info);
|
|
}
|
|
|
|
int fastboot_write(void *buf, unsigned len)
|
|
{
|
|
return usb_write(buf, len);
|
|
}
|
|
|
|
void cmd_getvar(const char *arg, void *data, unsigned sz)
|
|
{
|
|
struct fastboot_var *var;
|
|
|
|
for (var = varlist; var; var = var->next) {
|
|
if (!strcmp(var->name, arg)) {
|
|
fastboot_okay(var->value);
|
|
return;
|
|
}
|
|
}
|
|
fastboot_okay("");
|
|
}
|
|
|
|
void cmd_download(const char *arg, void *data, unsigned sz)
|
|
{
|
|
char response[64];
|
|
unsigned len = hex2unsigned(arg);
|
|
int r;
|
|
|
|
download_size = 0;
|
|
if (len > download_max) {
|
|
fastboot_fail("data too large");
|
|
return;
|
|
}
|
|
|
|
sprintf(response,"DATA%08x", len);
|
|
if (usb_write(response, strlen(response)) < 0)
|
|
return;
|
|
|
|
r = usb_read(download_base, len);
|
|
if ((r < 0) || ((unsigned)r != len)) {
|
|
fastboot_state = STATE_ERROR;
|
|
return;
|
|
}
|
|
download_size = len;
|
|
fastboot_okay("");
|
|
}
|
|
|
|
void fastboot_command_loop(void)
|
|
{
|
|
struct fastboot_cmd *cmd;
|
|
int r;
|
|
//dprintf(INFO,"fastboot: processing commands\n");
|
|
//fastboot_sstat(1);
|
|
again:
|
|
while (fastboot_state != STATE_ERROR) {
|
|
r = usb_read(buffer, 64);
|
|
if (r < 0) break;
|
|
buffer[r] = 0;
|
|
dprintf(INFO,"fastboot: %s\n", buffer);
|
|
|
|
for (cmd = cmdlist; cmd; cmd = cmd->next) {
|
|
if (memcmp(buffer, cmd->prefix, cmd->prefix_len))
|
|
continue;
|
|
fastboot_state = STATE_COMMAND;
|
|
cmd->handle((const char*) buffer + cmd->prefix_len,
|
|
(void*) download_base, download_size);
|
|
if (fastboot_state == STATE_COMMAND)
|
|
fastboot_fail("unknown reason");
|
|
goto again;
|
|
}
|
|
|
|
fastboot_fail("unknown command");
|
|
|
|
}
|
|
fastboot_state = STATE_OFFLINE;
|
|
//dprintf(INFO,"fastboot: oops!\n");
|
|
//fastboot_sstat(0);
|
|
}
|
|
|
|
int fastboot_handler(void *arg)
|
|
{
|
|
for (;;) {
|
|
event_wait(&usb_online);
|
|
fastboot_command_loop();
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
void fastboot_notify(struct udc_gadget *gadget, unsigned event)
|
|
{
|
|
if (event == UDC_EVENT_ONLINE) {
|
|
//c_usb_conn = 1;
|
|
//fastboot_sstat(1);
|
|
event_signal(&usb_online, 0);
|
|
} else if (event == UDC_EVENT_OFFLINE) {
|
|
//c_usb_conn = 0;
|
|
//fastboot_sstat(0);
|
|
}
|
|
}
|
|
|
|
struct udc_endpoint *fastboot_endpoints[2];
|
|
|
|
struct udc_gadget fastboot_gadget = {
|
|
.notify = fastboot_notify,
|
|
.ifc_class = 0xff,
|
|
.ifc_subclass = 0x42,
|
|
.ifc_protocol = 0x03,
|
|
.ifc_endpoints = 2,
|
|
.ifc_string = "fastboot",
|
|
.ept = fastboot_endpoints,
|
|
};
|
|
|
|
int fastboot_init(void *base, unsigned size)
|
|
{
|
|
thread_t *thr;
|
|
//dprintf(INFO, "fastboot_init()\n");
|
|
|
|
download_base = base;
|
|
download_max = size;
|
|
|
|
event_init(&usb_online, 0, EVENT_FLAG_AUTOUNSIGNAL);
|
|
event_init(&txn_done, 0, EVENT_FLAG_AUTOUNSIGNAL);
|
|
|
|
in = udc_endpoint_alloc(UDC_TYPE_BULK_IN, 512);
|
|
if (!in)
|
|
goto fail_alloc_in;
|
|
out = udc_endpoint_alloc(UDC_TYPE_BULK_OUT, 512);
|
|
if (!out)
|
|
goto fail_alloc_out;
|
|
|
|
fastboot_endpoints[0] = in;
|
|
fastboot_endpoints[1] = out;
|
|
|
|
req = udc_request_alloc();
|
|
if (!req)
|
|
goto fail_alloc_req;
|
|
|
|
if (udc_register_gadget(&fastboot_gadget))
|
|
goto fail_udc_register;
|
|
|
|
fastboot_register("getvar:", cmd_getvar);
|
|
fastboot_register("download:", cmd_download);
|
|
|
|
thr = thread_create("fastboot", fastboot_handler, 0, DEFAULT_PRIORITY, 4096);
|
|
thread_resume(thr);
|
|
return 0;
|
|
|
|
fail_udc_register:
|
|
udc_request_free(req);
|
|
fail_alloc_req:
|
|
udc_endpoint_free(out);
|
|
fail_alloc_out:
|
|
udc_endpoint_free(in);
|
|
fail_alloc_in:
|
|
return -1;
|
|
}
|