134 lines
3.1 KiB
C
134 lines
3.1 KiB
C
/*
|
|
* cmd_dfu.c -- dfu command
|
|
*
|
|
* Copyright (C) 2015
|
|
* Lukasz Majewski <l.majewski@majess.pl>
|
|
*
|
|
* Copyright (C) 2012 Samsung Electronics
|
|
* authors: Andrzej Pietrasiewicz <andrzej.p@samsung.com>
|
|
* Lukasz Majewski <l.majewski@samsung.com>
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0+
|
|
*/
|
|
|
|
#include <common.h>
|
|
#include <watchdog.h>
|
|
#include <dfu.h>
|
|
#include <console.h>
|
|
#include <g_dnl.h>
|
|
#include <usb.h>
|
|
#include <net.h>
|
|
|
|
static int do_dfu(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
|
|
{
|
|
bool dfu_reset = false;
|
|
|
|
if (argc < 4)
|
|
return CMD_RET_USAGE;
|
|
|
|
char *usb_controller = argv[1];
|
|
char *interface = argv[2];
|
|
char *devstring = argv[3];
|
|
|
|
int ret, i = 0;
|
|
#ifdef CONFIG_DFU_TFTP
|
|
unsigned long addr = 0;
|
|
if (!strcmp(argv[1], "tftp")) {
|
|
if (argc == 5)
|
|
addr = simple_strtoul(argv[4], NULL, 0);
|
|
|
|
return update_tftp(addr, interface, devstring);
|
|
}
|
|
#endif
|
|
|
|
ret = dfu_init_env_entities(interface, devstring);
|
|
if (ret)
|
|
goto done;
|
|
|
|
ret = CMD_RET_SUCCESS;
|
|
if (argc > 4 && strcmp(argv[4], "list") == 0) {
|
|
dfu_show_entities();
|
|
goto done;
|
|
}
|
|
|
|
int controller_index = simple_strtoul(usb_controller, NULL, 0);
|
|
board_usb_init(controller_index, USB_INIT_DEVICE);
|
|
g_dnl_clear_detach();
|
|
g_dnl_register("usb_dnl_dfu");
|
|
while (1) {
|
|
if (g_dnl_detach()) {
|
|
/*
|
|
* Check if USB bus reset is performed after detach,
|
|
* which indicates that -R switch has been passed to
|
|
* dfu-util. In this case reboot the device
|
|
*/
|
|
if (dfu_usb_get_reset()) {
|
|
dfu_reset = true;
|
|
goto exit;
|
|
}
|
|
|
|
/*
|
|
* This extra number of usb_gadget_handle_interrupts()
|
|
* calls is necessary to assure correct transmission
|
|
* completion with dfu-util
|
|
*/
|
|
if (++i == 10000)
|
|
goto exit;
|
|
}
|
|
|
|
if (ctrlc())
|
|
goto exit;
|
|
|
|
if (dfu_get_defer_flush()) {
|
|
/*
|
|
* Call to usb_gadget_handle_interrupts() is necessary
|
|
* to act on ZLP OUT transaction from HOST PC after
|
|
* transmitting the whole file.
|
|
*
|
|
* If this ZLP OUT packet is NAK'ed, the HOST libusb
|
|
* function fails after timeout (by default it is set to
|
|
* 5 seconds). In such situation the dfu-util program
|
|
* exits with error message.
|
|
*/
|
|
usb_gadget_handle_interrupts(controller_index);
|
|
ret = dfu_flush(dfu_get_defer_flush(), NULL, 0, 0);
|
|
dfu_set_defer_flush(NULL);
|
|
if (ret) {
|
|
error("Deferred dfu_flush() failed!");
|
|
goto exit;
|
|
}
|
|
}
|
|
|
|
WATCHDOG_RESET();
|
|
usb_gadget_handle_interrupts(controller_index);
|
|
}
|
|
exit:
|
|
g_dnl_unregister();
|
|
board_usb_cleanup(controller_index, USB_INIT_DEVICE);
|
|
done:
|
|
dfu_free_entities();
|
|
|
|
if (dfu_reset)
|
|
run_command("reset", 0);
|
|
|
|
g_dnl_clear_detach();
|
|
|
|
return ret;
|
|
}
|
|
|
|
U_BOOT_CMD(dfu, CONFIG_SYS_MAXARGS, 1, do_dfu,
|
|
"Device Firmware Upgrade",
|
|
"<USB_controller> <interface> <dev> [list]\n"
|
|
" - device firmware upgrade via <USB_controller>\n"
|
|
" on device <dev>, attached to interface\n"
|
|
" <interface>\n"
|
|
" [list] - list available alt settings\n"
|
|
#ifdef CONFIG_DFU_TFTP
|
|
"dfu tftp <interface> <dev> [<addr>]\n"
|
|
" - device firmware upgrade via TFTP\n"
|
|
" on device <dev>, attached to interface\n"
|
|
" <interface>\n"
|
|
" [<addr>] - address where FIT image has been stored\n"
|
|
#endif
|
|
);
|