diff options
Diffstat (limited to 'common/cmd_autoscript.c')
| -rw-r--r-- | common/cmd_autoscript.c | 174 |
1 files changed, 174 insertions, 0 deletions
diff --git a/common/cmd_autoscript.c b/common/cmd_autoscript.c new file mode 100644 index 0000000..49bd463 --- /dev/null +++ b/common/cmd_autoscript.c @@ -0,0 +1,174 @@ +/* + * (C) Copyright 2001 + * Kyle Harris, kharris@nexus-tech.net + * + * See file CREDITS for list of people who contributed to this + * project. + * + * 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 + */ + +/* + * autoscript allows a remote host to download a command file and, + * optionally, binary data for automatically updating the target. For + * example, you create a new kernel image and want the user to be + * able to simply download the image and the machine does the rest. + * The kernel image is postprocessed with mkimage, which creates an + * image with a script file prepended. If enabled, autoscript will + * verify the script and contents of the download and execute the + * script portion. This would be responsible for erasing flash, + * copying the new image, and rebooting the machine. + */ + +/* #define DEBUG */ + +#include <common.h> +#include <command.h> +#include <image.h> +#include <malloc.h> +#include <asm/byteorder.h> +#include <cmd_boot.h> +#include <cmd_autoscript.h> +#if defined(CONFIG_8xx) +#include <mpc8xx.h> +#endif +#ifdef CFG_HUSH_PARSER +#include <hush.h> +#endif + +#if defined(CONFIG_AUTOSCRIPT) || \ + (CONFIG_COMMANDS & CFG_CMD_AUTOSCRIPT) + +extern image_header_t header; /* from cmd_bootm.c */ +int +autoscript (ulong addr) +{ + ulong crc, data, len; + image_header_t *hdr = &header; + ulong *len_ptr; + char *cmd; + int rcode = 0; + int verify; + + cmd = getenv ("verify"); + verify = (cmd && (*cmd == 'n')) ? 0 : 1; + + + memmove (hdr, (char *)addr, sizeof(image_header_t)); + + if (ntohl(hdr->ih_magic) != IH_MAGIC) { + printf ("Bad magic number\n"); + return 1; + } + + crc = ntohl(hdr->ih_hcrc); + hdr->ih_hcrc = 0; + len = sizeof (image_header_t); + data = (ulong)hdr; + if (crc32(0, (char *)data, len) != crc) { + printf ("Bad header crc\n"); + return 1; + } + + data = addr + sizeof(image_header_t); + len = ntohl(hdr->ih_size); + + if (verify) { + if (crc32(0, (char *)data, len) != ntohl(hdr->ih_dcrc)) { + printf ("Bad data crc\n"); + return 1; + } + } + + if (hdr->ih_type != IH_TYPE_SCRIPT) { + printf ("Bad image type\n"); + return 1; + } + + /* get length of script */ + len_ptr = (ulong *)data; + + if ((len = ntohl(*len_ptr)) == 0) { + printf ("Empty Script\n"); + return 1; + } + + debug ("** Script length: %d\n", len); + + if ((cmd = malloc (len + 1)) == NULL) { + return 1; + } + + while (*len_ptr++); + + /* make sure cmd is null terminated */ + memmove (cmd, (char *)len_ptr, len); + *(cmd + len) = 0; + +#ifdef CFG_HUSH_PARSER + rcode = parse_string_outer (cmd, FLAG_PARSE_SEMICOLON); +#else + { + char *line = cmd; + char *next = cmd; + + /* + * break into individual lines, + * and execute each line; + * terminate on error. + */ + while (*next) { + if (*next == '\n') { + *next = '\0'; + /* run only non-empty commands */ + if ((next - line) > 1) { + debug ("** exec: \"%s\"\n", + line); + if (run_command (line, 0) < 0) { + rcode = 1; + break; + } + } + line = next + 1; + } + ++next; + } + } +#endif + free (cmd); + return rcode; +} + +#endif /* CONFIG_AUTOSCRIPT || CFG_CMD_AUTOSCRIPT */ + +#if (CONFIG_COMMANDS & CFG_CMD_AUTOSCRIPT) +int +do_autoscript (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[]) +{ + ulong addr; + int rcode; + + if (argc < 2) { + addr = CFG_LOAD_ADDR; + } else { + addr = simple_strtoul (argv[1],0,16); + } + + printf ("## Executing script at %08lx\n",addr); + rcode = autoscript (addr); + return rcode; +} +#endif /* CFG_CMD_AUTOSCRIPT */ |