From 27114b30bca48b51131f697d3211e8e9f3d08e1b Mon Sep 17 00:00:00 2001 From: Sascha Hauer Date: Tue, 9 Dec 2014 18:26:52 +0100 Subject: scripts: Add mxs-usb-loader tool This is directly taken from the rockbox projects sbloader tool, just renamed to mxs-usb-loader to avoid confusion with bareboxes several different image tools. Signed-off-by: Sascha Hauer --- arch/arm/mach-mxs/Kconfig | 7 ++ scripts/Makefile | 3 + scripts/mxs-usb-loader.c | 236 ++++++++++++++++++++++++++++++++++++++++++++++ 3 files changed, 246 insertions(+) create mode 100644 scripts/mxs-usb-loader.c diff --git a/arch/arm/mach-mxs/Kconfig b/arch/arm/mach-mxs/Kconfig index 214f9404df..997e326474 100644 --- a/arch/arm/mach-mxs/Kconfig +++ b/arch/arm/mach-mxs/Kconfig @@ -134,4 +134,11 @@ endif endmenu +config ARCH_MXS_USBLOADER + bool "compile mxs-usb-loader" + help + mxs-usb-loader is a tool to upload and start mxs bootstream images to an + i.MX SoC in ROM boot mode. It requires libusb, so make sure you have the libusb + devel package installed on your machine. + endif diff --git a/scripts/Makefile b/scripts/Makefile index 442eba5db4..74c22136db 100644 --- a/scripts/Makefile +++ b/scripts/Makefile @@ -21,6 +21,9 @@ hostprogs-$(CONFIG_ARCH_ZYNQ) += zynq_mkimage hostprogs-$(CONFIG_ARCH_SOCFPGA) += socfpga_mkimage hostprogs-$(CONFIG_ARCH_MXS) += mxsimage mxsboot HOSTLOADLIBES_mxsimage = `pkg-config --libs openssl` +HOSTCFLAGS_mxs-usb-loader.o = `pkg-config --cflags libusb-1.0` +HOSTLOADLIBES_mxs-usb-loader = `pkg-config --libs libusb-1.0` +hostprogs-$(CONFIG_ARCH_MXS_USBLOADER) += mxs-usb-loader subdir-y += mod subdir-$(CONFIG_OMAP4_USBBOOT) += omap4_usbboot diff --git a/scripts/mxs-usb-loader.c b/scripts/mxs-usb-loader.c new file mode 100644 index 0000000000..8529274d6e --- /dev/null +++ b/scripts/mxs-usb-loader.c @@ -0,0 +1,236 @@ +/*************************************************************************** + * __________ __ ___. + * Open \______ \ ____ ____ | | _\_ |__ _______ ___ + * Source | _// _ \_/ ___\| |/ /| __ \ / _ \ \/ / + * Jukebox | | ( <_> ) \___| < | \_\ ( <_> > < < + * Firmware |____|_ /\____/ \___ >__|_ \|___ /\____/__/\_ \ + * \/ \/ \/ \/ \/ + * $Id$ + * + * Copyright (C) 2010 Amaury Pouly + * + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU General Public License + * as published by the Free Software Foundation; either version 2 + * of the License, or (at your option) any later version. + * + * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY + * KIND, either express or implied. + * + ****************************************************************************/ +#include +#include +#include +#include +#include + +#ifndef MIN +#define MIN(a,b) ((a) < (b) ? (a) : (b)) +#endif + +void put32le(uint8_t * buf, uint32_t i) +{ + *buf++ = i & 0xff; + *buf++ = (i >> 8) & 0xff; + *buf++ = (i >> 16) & 0xff; + *buf++ = (i >> 24) & 0xff; +} + +void put32be(uint8_t * buf, uint32_t i) +{ + *buf++ = (i >> 24) & 0xff; + *buf++ = (i >> 16) & 0xff; + *buf++ = (i >> 8) & 0xff; + *buf++ = i & 0xff; +} + +enum dev_type_t { + HID_DEVICE, + RECOVERY_DEVICE, +}; + +struct dev_info_t { + uint16_t vendor_id; + uint16_t product_id; + unsigned xfer_size; + enum dev_type_t dev_type; +}; + +struct dev_info_t g_dev_info[] = { + {0x066f, 0x3780, 1024, HID_DEVICE}, /* i.MX233 / STMP3780 */ + {0x066f, 0x3770, 48, HID_DEVICE}, /* STMP3770 */ + {0x15A2, 0x004F, 1024, HID_DEVICE}, /* i.MX28 */ + {0x066f, 0x3600, 4096, RECOVERY_DEVICE}, /* STMP36xx */ +}; + +int send_hid(libusb_device_handle * dev, int xfer_size, uint8_t * data, + int size, int nr_xfers) +{ + int i; + + libusb_detach_kernel_driver(dev, 0); + libusb_detach_kernel_driver(dev, 4); + + libusb_claim_interface(dev, 0); + libusb_claim_interface(dev, 4); + + uint8_t *xfer_buf = malloc(1 + xfer_size); + uint8_t *p = xfer_buf; + + *p++ = 0x01; /* Report id */ + + /* Command block wrapper */ + *p++ = 'B'; /* Signature */ + *p++ = 'L'; + *p++ = 'T'; + *p++ = 'C'; + put32le(p, 0x1); /* Tag */ + p += 4; + put32le(p, size); /* Payload size */ + p += 4; + *p++ = 0; /* Flags (host to device) */ + p += 2; /* Reserved */ + + /* Command descriptor block */ + *p++ = 0x02; /* Firmware download */ + put32be(p, size); /* Download size */ + + int ret = libusb_control_transfer(dev, + LIBUSB_REQUEST_TYPE_CLASS | + LIBUSB_RECIPIENT_INTERFACE, 0x9, + 0x201, 0, + xfer_buf, xfer_size + 1, 1000); + if (ret < 0) { + printf("transfer error at init step\n"); + return 1; + } + + for (i = 0; i < nr_xfers; i++) { + xfer_buf[0] = 0x2; + memcpy(&xfer_buf[1], &data[i * xfer_size], xfer_size); + + ret = libusb_control_transfer(dev, + LIBUSB_REQUEST_TYPE_CLASS | + LIBUSB_RECIPIENT_INTERFACE, 0x9, + 0x202, 0, xfer_buf, xfer_size + 1, + 1000); + if (ret < 0) { + printf("transfer error at send step %d\n", i); + return 1; + } + } + + int recv_size; + ret = + libusb_interrupt_transfer(dev, 0x81, xfer_buf, xfer_size, + &recv_size, 1000); + if (ret < 0) { + printf("transfer error at final stage\n"); + return 1; + } + + return ret; +} + +int send_recovery(libusb_device_handle * dev, int xfer_size, uint8_t * data, + int size, int nr_xfers) +{ + (void)nr_xfers; + // there should be no kernel driver attached but in doubt... + libusb_detach_kernel_driver(dev, 0); + libusb_claim_interface(dev, 0); + + int sent = 0; + while (sent < size) { + int xfered; + int len = MIN(size - sent, xfer_size); + int ret = + libusb_bulk_transfer(dev, 1, data + sent, len, &xfered, + 1000); + if (ret < 0) { + printf("transfer error at send offset %d\n", sent); + return 1; + } + if (xfered == 0) { + printf("empty transfer at step offset %d\n", sent); + return 2; + } + sent += xfered; + } + return 0; +} + +int main(int argc, char **argv) +{ + if (argc != 3) { + printf("usage: %s \n", argv[0]); + printf + ("If is set to zero, the preferred one is used.\n"); + return 1; + } + + char *end; + int xfer_size = strtol(argv[1], &end, 0); + if (end != (argv[1] + strlen(argv[1]))) { + printf("Invalid transfer size !\n"); + return 1; + } + + libusb_device_handle *dev; + + libusb_init(NULL); + + libusb_set_debug(NULL, 3); + + unsigned i; + for (i = 0; i < sizeof(g_dev_info) / sizeof(g_dev_info[0]); i++) { + dev = libusb_open_device_with_vid_pid(NULL, + g_dev_info[i].vendor_id, + g_dev_info[i].product_id); + if (dev == NULL) + continue; + if (xfer_size == 0) + xfer_size = g_dev_info[i].xfer_size; + printf("Found a match for %04x:%04x\n", + g_dev_info[i].vendor_id, g_dev_info[i].product_id); + break; + } + if (dev == NULL) { + printf("Cannot open device\n"); + return 1; + } + + FILE *f = fopen(argv[2], "r"); + if (f == NULL) { + perror("cannot open file"); + return 1; + } + fseek(f, 0, SEEK_END); + size_t size = ftell(f); + fseek(f, 0, SEEK_SET); + + printf("Transfer size: %d\n", xfer_size); + int nr_xfers = (size + xfer_size - 1) / xfer_size; + uint8_t *file_buf = malloc(nr_xfers * xfer_size); + memset(file_buf, 0xff, nr_xfers * xfer_size); // pad with 0xff + if (fread(file_buf, size, 1, f) != 1) { + perror("read error"); + fclose(f); + return 1; + } + fclose(f); + + switch (g_dev_info[i].dev_type) { + case HID_DEVICE: + send_hid(dev, xfer_size, file_buf, size, nr_xfers); + break; + case RECOVERY_DEVICE: + send_recovery(dev, xfer_size, file_buf, size, nr_xfers); + break; + default: + printf("unknown device type\n"); + break; + } + + return 0; +} -- cgit v1.2.3