common/cmd_autoscript.c

   1 /*
   2  * (C) Copyright 2001
   3  * Kyle Harris, kharris@nexus-tech.net
   4  *
   5  * See file CREDITS for list of people who contributed to this
   6  * project.
   7  *
   8  * This program is free software; you can redistribute it and/or
   9  * modify it under the terms of the GNU General Public License as
  10  * published by the Free Software Foundation; either version 2 of
  11  * the License, or (at your option) any later version.
  12  *
  13  * This program is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16  * GNU General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * along with this program; if not, write to the Free Software
  20  * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
  21  * MA 02111-1307 USA
  22  */
  23
  24 /*
  25  * autoscript allows a remote host to download a command file and,
  26  * optionally, binary data for automatically updating the target. For
  27  * example, you create a new kernel image and want the user to be
  28  * able to simply download the image and the machine does the rest.
  29  * The kernel image is postprocessed with mkimage, which creates an
  30  * image with a script file prepended. If enabled, autoscript will
  31  * verify the script and contents of the download and execute the
  32  * script portion. This would be responsible for erasing flash,
  33  * copying the new image, and rebooting the machine.
  34  */
  35
  36 /* #define DEBUG */
  37
  38 #include <common.h>
  39 #include <command.h>
  40 #include <image.h>
  41 #include <malloc.h>
  42 #include <asm/byteorder.h>
  43 #if defined(CONFIG_8xx)
  44 #include <mpc8xx.h>
  45 #endif
  46 #ifdef CFG_HUSH_PARSER
  47 #include <hush.h>
  48 #endif
  49
  50 #if defined(CONFIG_AUTOSCRIPT) || \
  51          (CONFIG_COMMANDS & CFG_CMD_AUTOSCRIPT )
  52
  53 extern image_header_t header;           /* from cmd_bootm.c */
  54 int
  55 autoscript (ulong addr)
  56 {
  57         ulong crc, data, len;
  58         image_header_t *hdr = &header;
  59         ulong *len_ptr;
  60         char *cmd;
  61         int rcode = 0;
  62         int verify;
  63
  64         cmd = getenv ("verify");
  65         verify = (cmd && (*cmd == 'n')) ? 0 : 1;
  66
  67
  68         memmove (hdr, (char *)addr, sizeof(image_header_t));
  69
  70         if (ntohl(hdr->ih_magic) != IH_MAGIC) {
  71                 puts ("Bad magic number\n");
  72                 return 1;
  73         }
  74
  75         crc = ntohl(hdr->ih_hcrc);
  76         hdr->ih_hcrc = 0;
  77         len = sizeof (image_header_t);
  78         data = (ulong)hdr;
  79         if (crc32(0, (uchar *)data, len) != crc) {
  80                 puts ("Bad header crc\n");
  81                 return 1;
  82         }
  83
  84         data = addr + sizeof(image_header_t);
  85         len = ntohl(hdr->ih_size);
  86
  87         if (verify) {
  88                 if (crc32(0, (uchar *)data, len) != ntohl(hdr->ih_dcrc)) {
  89                         puts ("Bad data crc\n");
  90                         return 1;
  91                 }
  92         }
  93
  94         if (hdr->ih_type != IH_TYPE_SCRIPT) {
  95                 puts ("Bad image type\n");
  96                 return 1;
  97         }
  98
  99         /* get length of script */
 100         len_ptr = (ulong *)data;
 101
 102         if ((len = ntohl(*len_ptr)) == 0) {
 103                 puts ("Empty Script\n");
 104                 return 1;
 105         }
 106
 107         debug ("** Script length: %ld\n", len);
 108
 109         if ((cmd = malloc (len + 1)) == NULL) {
 110                 return 1;
 111         }
 112
 113         while (*len_ptr++);
 114
 115         /* make sure cmd is null terminated */
 116         memmove (cmd, (char *)len_ptr, len);
 117         *(cmd + len) = 0;
 118
 119 #ifdef CFG_HUSH_PARSER /*?? */
 120         rcode = parse_string_outer (cmd, FLAG_PARSE_SEMICOLON);
 121 #else
 122         {
 123                 char *line = cmd;
 124                 char *next = cmd;
 125
 126                 /*
 127                  * break into individual lines,
 128                  * and execute each line;
 129                  * terminate on error.
 130                  */
 131                 while (*next) {
 132                         if (*next == '\n') {
 133                                 *next = '\0';
 134                                 /* run only non-empty commands */
 135                                 if ((next - line) > 1) {
 136                                         debug ("** exec: \"%s\"\n",
 137                                                 line);
 138                                         if (run_command (line, 0) < 0) {
 139                                                 rcode = 1;
 140                                                 break;
 141                                         }
 142                                 }
 143                                 line = next + 1;
 144                         }
 145                         ++next;
 146                 }
 147         }
 148 #endif
 149         free (cmd);
 150         return rcode;
 151 }
 152
 153 #endif  /* CONFIG_AUTOSCRIPT || CFG_CMD_AUTOSCRIPT */
 154 /**************************************************/
 155 #if (CONFIG_COMMANDS & CFG_CMD_AUTOSCRIPT)
 156 int
 157 do_autoscript (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
 158 {
 159         ulong addr;
 160         int rcode;
 161
 162         if (argc < 2) {
 163                 addr = CFG_LOAD_ADDR;
 164         } else {
 165                 addr = simple_strtoul (argv[1],0,16);
 166         }
 167
 168         printf ("## Executing script at %08lx\n",addr);
 169         rcode = autoscript (addr);
 170         return rcode;
 171 }
 172
 173 #if (CONFIG_COMMANDS & CFG_CMD_AUTOSCRIPT)
 174 U_BOOT_CMD(
 175         autoscr, 2, 0,  do_autoscript,
 176         "autoscr - run script from memory\n",
 177         "[addr] - run script starting at addr"
 178         " - A valid autoscr header must be present\n"
 179 );
 180 #endif /* CFG_CMD_AUTOSCRIPT */
 181
 182 #endif /* CONFIG_AUTOSCRIPT || CFG_CMD_AUTOSCRIPT */