Improve avrf.c command argument parsing.

This commit is contained in:
Zachary T Welch 2009-10-23 02:17:17 -07:00
parent 266c423bbd
commit 7b3d54a127
1 changed files with 4 additions and 7 deletions

View File

@ -456,7 +456,6 @@ static int avrf_mass_erase(struct flash_bank_s *bank)
static int avrf_handle_mass_erase_command(struct command_context_s *cmd_ctx, char *cmd, char **args, int argc) static int avrf_handle_mass_erase_command(struct command_context_s *cmd_ctx, char *cmd, char **args, int argc)
{ {
flash_bank_t *bank;
int i; int i;
if (argc < 1) if (argc < 1)
@ -465,12 +464,10 @@ static int avrf_handle_mass_erase_command(struct command_context_s *cmd_ctx, cha
return ERROR_OK; return ERROR_OK;
} }
bank = get_flash_bank_by_num(strtoul(args[0], NULL, 0)); flash_bank_t *bank;
if (!bank) int retval = flash_command_get_bank_by_num(cmd_ctx, args[0], &bank);
{ if (ERROR_OK != retval)
command_print(cmd_ctx, "flash bank '#%s' is out of bounds", args[0]); return retval;
return ERROR_OK;
}
if (avrf_mass_erase(bank) == ERROR_OK) if (avrf_mass_erase(bank) == ERROR_OK)
{ {