mirror of
https://github.com/AsahiLinux/u-boot
synced 2024-11-18 02:38:56 +00:00
df93cd9c6f
Fix obvious crash when not enough arguments are given to the dfu command. Signed-off-by: Pantelis Antoniou <panto@antoniou-consulting.com>
84 lines
2.1 KiB
C
84 lines
2.1 KiB
C
/*
|
|
* cmd_dfu.c -- dfu command
|
|
*
|
|
* Copyright (C) 2012 Samsung Electronics
|
|
* authors: Andrzej Pietrasiewicz <andrzej.p@samsung.com>
|
|
* Lukasz Majewski <l.majewski@samsung.com>
|
|
*
|
|
* 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 program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
*/
|
|
|
|
#include <common.h>
|
|
#include <command.h>
|
|
#include <malloc.h>
|
|
#include <dfu.h>
|
|
#include <asm/errno.h>
|
|
#include <g_dnl.h>
|
|
|
|
static int do_dfu(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
|
|
{
|
|
const char *str_env;
|
|
char *s = "dfu";
|
|
char *env_bkp;
|
|
int ret;
|
|
|
|
if (argc < 3)
|
|
return CMD_RET_USAGE;
|
|
|
|
str_env = getenv("dfu_alt_info");
|
|
if (str_env == NULL) {
|
|
printf("%s: \"dfu_alt_info\" env variable not defined!\n",
|
|
__func__);
|
|
return CMD_RET_FAILURE;
|
|
}
|
|
|
|
env_bkp = strdup(str_env);
|
|
ret = dfu_config_entities(env_bkp, argv[1],
|
|
(int)simple_strtoul(argv[2], NULL, 10));
|
|
if (ret)
|
|
return CMD_RET_FAILURE;
|
|
|
|
if (argc > 3 && strcmp(argv[3], "list") == 0) {
|
|
dfu_show_entities();
|
|
goto done;
|
|
}
|
|
|
|
#ifdef CONFIG_TRATS
|
|
board_usb_init();
|
|
#endif
|
|
|
|
g_dnl_register(s);
|
|
while (1) {
|
|
if (ctrlc())
|
|
goto exit;
|
|
|
|
usb_gadget_handle_interrupts();
|
|
}
|
|
exit:
|
|
g_dnl_unregister();
|
|
done:
|
|
dfu_free_entities();
|
|
free(env_bkp);
|
|
|
|
return CMD_RET_SUCCESS;
|
|
}
|
|
|
|
U_BOOT_CMD(dfu, CONFIG_SYS_MAXARGS, 1, do_dfu,
|
|
"Device Firmware Upgrade",
|
|
"<interface> <dev> [list]\n"
|
|
" - device firmware upgrade on a device <dev>\n"
|
|
" attached to interface <interface>\n"
|
|
" [list] - list available alt settings"
|
|
);
|