/*
|
* (C) Copyright 2022 Rockchip Electronics Co., Ltd.
|
*
|
* SPDX-License-Identifier: GPL-2.0+
|
*/
|
|
#include <common.h>
|
#include <malloc.h>
|
#include <dm/device.h>
|
|
/*
|
* Script file example:
|
**************************
|
* > # script file start
|
* > echo "hello world"
|
* > % script file end
|
**************************
|
*/
|
#define SCRIPT_FILE_MAX_SIZE (12 * 1024)
|
#define SCRIPT_FILE_COMMENT '#'
|
#define SCRIPT_FILE_END '%'
|
#define IS_COMMENT(x) (SCRIPT_FILE_COMMENT == (x))
|
#define IS_FILE_END(x) (SCRIPT_FILE_END == (x))
|
#define IS_LINE_END(x) ('\r' == (x) || '\n' == (x))
|
#define MAX_LINE_SIZE 8000
|
|
static char *script_next_line(char **line_buf_ptr)
|
{
|
char *line_buf = (*line_buf_ptr);
|
char *next_line;
|
int i = 0;
|
|
/* strip '\r', '\n' and comment */
|
while (1) {
|
/* strip '\r' & '\n' */
|
if (IS_LINE_END(line_buf[0])) {
|
line_buf++;
|
/* strip comment */
|
} else if (IS_COMMENT(line_buf[0])) {
|
for (i = 0; !IS_LINE_END(line_buf[0]) && i <= MAX_LINE_SIZE; i++)
|
line_buf++;
|
|
if (i > MAX_LINE_SIZE) {
|
line_buf[0] = SCRIPT_FILE_END;
|
printf("Error: max line length is %d!!!\n", MAX_LINE_SIZE);
|
break;
|
}
|
} else {
|
break;
|
}
|
}
|
|
/* get next line */
|
if (IS_FILE_END(line_buf[0])) {
|
next_line = NULL;
|
} else {
|
next_line = line_buf;
|
for (i = 0; !IS_LINE_END(line_buf[0]) && i <= MAX_LINE_SIZE; i++)
|
line_buf++;
|
|
if (i > MAX_LINE_SIZE) {
|
next_line = NULL;
|
printf("Error: max line length is %d!!!\n", MAX_LINE_SIZE);
|
} else {
|
line_buf[0] = '\0';
|
*line_buf_ptr = line_buf + 1;
|
}
|
}
|
|
return next_line;
|
}
|
|
static int do_script(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
|
{
|
char *next_line, *script, *buf;
|
ulong addr;
|
int ret = CMD_RET_SUCCESS;
|
|
#ifdef CONFIG_FIT_SIGNATURE
|
printf("Verify-boot: forbit no-signed script\n");
|
return CMD_RET_FAILURE;
|
#endif
|
|
if (argc != 2 || !argv[1])
|
return CMD_RET_USAGE;
|
|
addr = simple_strtoul(argv[1], NULL, 16);
|
if (!addr)
|
return CMD_RET_USAGE;
|
|
buf = calloc(SCRIPT_FILE_MAX_SIZE, 1);
|
if (!buf)
|
return CMD_RET_FAILURE;
|
|
script = buf;
|
memcpy(buf, (char *)addr, SCRIPT_FILE_MAX_SIZE);
|
while ((next_line = script_next_line(&script)) != NULL) {
|
printf("\n$ %s\n", next_line);
|
ret = run_command(next_line, 0);
|
if (ret)
|
break; /* fail */
|
}
|
free(buf);
|
|
return ret;
|
}
|
|
static int do_sd_update(cmd_tbl_t *cmdtp, int flag,
|
int argc, char * const argv[])
|
{
|
struct blk_desc *desc;
|
int part_type;
|
char cmd[128];
|
char *buf;
|
int ret;
|
|
printf("## retrieving sd_update.txt ...\n");
|
|
desc = blk_get_devnum_by_type(IF_TYPE_MMC, 1);
|
if (!desc)
|
return CMD_RET_FAILURE;
|
|
buf = memalign(ARCH_DMA_MINALIGN, SCRIPT_FILE_MAX_SIZE * 2);
|
if (!buf)
|
return CMD_RET_FAILURE;
|
|
part_type = desc->part_type;
|
desc->part_type = PART_TYPE_DOS;
|
|
snprintf(cmd, sizeof(cmd),
|
"fatload mmc 1 0x%08lx sd_update.txt", (ulong)buf);
|
ret = run_command(cmd, 0);
|
if (!ret) {
|
snprintf(cmd, sizeof(cmd), "script 0x%08lx", (ulong)buf);
|
ret = run_command(cmd, 0);
|
}
|
free(buf);
|
|
desc->part_type = part_type;
|
|
return ret;
|
}
|
|
static int do_usb_update(cmd_tbl_t *cmdtp, int flag,
|
int argc, char * const argv[])
|
{
|
struct blk_desc *desc;
|
struct udevice *dev;
|
int devnum = -1;
|
int part_type;
|
char cmd[128];
|
char *buf;
|
int ret;
|
|
printf("## retrieving usb_update.txt ...\n");
|
|
if (run_command("usb reset", 0))
|
return CMD_RET_FAILURE;
|
|
for (blk_first_device(IF_TYPE_USB, &dev);
|
dev;
|
blk_next_device(&dev)) {
|
desc = dev_get_uclass_platdata(dev);
|
if (desc->type == DEV_TYPE_UNKNOWN)
|
continue;
|
|
if (desc->lba > 0L && desc->blksz > 0L) {
|
devnum = desc->devnum;
|
break;
|
}
|
}
|
if (devnum < 0) {
|
printf("No available udisk\n");
|
return CMD_RET_FAILURE;
|
}
|
|
desc = blk_get_devnum_by_type(IF_TYPE_USB, devnum);
|
if (!desc) {
|
printf("No usb %d found\n", devnum);
|
return CMD_RET_FAILURE;
|
}
|
|
buf = memalign(ARCH_DMA_MINALIGN, SCRIPT_FILE_MAX_SIZE * 2);
|
if (!buf)
|
return CMD_RET_FAILURE;
|
|
part_type = desc->part_type;
|
desc->part_type = PART_TYPE_DOS;
|
|
printf("## scanning usb %d\n", devnum);
|
snprintf(cmd, sizeof(cmd), "fatload usb %d 0x%08lx usb_update.txt",
|
devnum, (ulong)buf);
|
ret = run_command(cmd, 0);
|
if (!ret) {
|
snprintf(cmd, sizeof(cmd), "script 0x%08lx", (ulong)buf);
|
ret = run_command(cmd, 0);
|
}
|
free(buf);
|
desc->part_type = part_type;
|
|
return ret;
|
}
|
|
static int do_tftp_update(cmd_tbl_t *cmdtp, int flag,
|
int argc, char * const argv[])
|
{
|
char cmd[128];
|
char *buf;
|
int dhcp = 0;
|
int ret;
|
|
printf("## retrieving tftp_update.txt ...\n");
|
|
if ((argc > 1) && !strcmp(argv[1], "-d"))
|
dhcp = 1;
|
|
buf = memalign(ARCH_DMA_MINALIGN, SCRIPT_FILE_MAX_SIZE * 2);
|
if (!buf)
|
return CMD_RET_FAILURE;
|
|
if (dhcp)
|
run_command("dhcp", 0);
|
|
snprintf(cmd, sizeof(cmd), "tftp 0x%08lx tftp_update.txt", (ulong)buf);
|
ret = run_command(cmd, 0);
|
if (!ret) {
|
snprintf(cmd, sizeof(cmd), "script 0x%08lx", (ulong)buf);
|
ret = run_command(cmd, 0);
|
}
|
free(buf);
|
|
return ret;
|
}
|
|
U_BOOT_CMD(
|
script, 2, 1, do_script,
|
"Run a script", "[file addr]"
|
);
|
|
U_BOOT_CMD(
|
sd_update, 1, 1, do_sd_update,
|
"sdcard auto upgrade", ""
|
);
|
|
U_BOOT_CMD(
|
usb_update, 1, 1, do_usb_update,
|
"usb auto upgrade", ""
|
);
|
|
U_BOOT_CMD(
|
tftp_update, 2, 1, do_tftp_update,
|
"tftp auto upgrade", "[-d]"
|
);
|
