cmd: fastboot: Validate user input
In case when user provides '-' as USB controller index, like this:
=> fastboot -
data abort occurs in strcmp() function in do_fastboot(), here:
if (!strcmp(argv[1], "udp"))
(tested on BeagleBone Black).
That's because argv[1] is NULL when user types in the '-', and null
pointer dereference occurs in strcmp() (which is ok according to C
standard specification). So we must validate user input to prevent such
behavior.
While at it, check also the result of strtoul() function and handle
error cases properly.
Signed-off-by: Sam Protsenko <semen.protsenko@linaro.org>
Reviewed-by: Simon Glass <sjg@chromium.org>
Reviewed-by: Lukasz Majewski <lukma@denx.de>
This commit is contained in:
parent
9f8cf76be2
commit
aa51579f9a
|
|
@ -38,13 +38,18 @@ static int do_fastboot_usb(int argc, char *const argv[],
|
|||
#if CONFIG_IS_ENABLED(USB_FUNCTION_FASTBOOT)
|
||||
int controller_index;
|
||||
char *usb_controller;
|
||||
char *endp;
|
||||
int ret;
|
||||
|
||||
if (argc < 2)
|
||||
return CMD_RET_USAGE;
|
||||
|
||||
usb_controller = argv[1];
|
||||
controller_index = simple_strtoul(usb_controller, NULL, 0);
|
||||
controller_index = simple_strtoul(usb_controller, &endp, 0);
|
||||
if (*endp != '\0') {
|
||||
pr_err("Error: Wrong USB controller index format\n");
|
||||
return CMD_RET_FAILURE;
|
||||
}
|
||||
|
||||
ret = board_usb_init(controller_index, USB_INIT_DEVICE);
|
||||
if (ret) {
|
||||
|
|
@ -120,6 +125,12 @@ NXTARG:
|
|||
;
|
||||
}
|
||||
|
||||
/* Handle case when USB controller param is just '-' */
|
||||
if (argc == 1) {
|
||||
pr_err("Error: Incorrect USB controller index\n");
|
||||
return CMD_RET_USAGE;
|
||||
}
|
||||
|
||||
fastboot_init((void *)buf_addr, buf_size);
|
||||
|
||||
if (!strcmp(argv[1], "udp"))
|
||||
|
|
|
|||
Loading…
Reference in New Issue