blob: 177f86bb54d946e9f32ab4e57460ae3403c778a6 [file] [log] [blame]
wdenkd0dd1072002-10-20 17:17:16 +00001/*
2 * (C) Copyright 2001
3 * Kyle Harris, kharris@nexus-tech.net
4 *
Wolfgang Denk1a459662013-07-08 09:37:19 +02005 * SPDX-License-Identifier: GPL-2.0+
wdenkd0dd1072002-10-20 17:17:16 +00006 */
7
8/*
Wolfgang Denk74de7ae2009-04-01 23:34:12 +02009 * The "source" command allows to define "script images", i. e. files
10 * that contain command sequences that can be executed by the command
11 * interpreter. It returns the exit status of the last command
12 * executed from the script. This is very similar to running a shell
13 * script in a UNIX shell, hence the name for the command.
wdenkd0dd1072002-10-20 17:17:16 +000014 */
15
16/* #define DEBUG */
17
18#include <common.h>
19#include <command.h>
20#include <image.h>
21#include <malloc.h>
Joe Hershberger0eb25b62015-03-22 17:08:59 -050022#include <mapmem.h>
wdenkd0dd1072002-10-20 17:17:16 +000023#include <asm/byteorder.h>
Simon Glass4ca30d62013-04-20 08:42:49 +000024#include <asm/io.h>
wdenkd0dd1072002-10-20 17:17:16 +000025
wdenkd0dd1072002-10-20 17:17:16 +000026int
Wolfgang Denk74de7ae2009-04-01 23:34:12 +020027source (ulong addr, const char *fit_uname)
wdenkd0dd1072002-10-20 17:17:16 +000028{
Wolfgang Denk53677ef2008-05-20 16:00:29 +020029 ulong len;
Heiko Schocher21d29f72014-05-28 11:33:33 +020030#if defined(CONFIG_IMAGE_FORMAT_LEGACY)
Simon Glass4ca30d62013-04-20 08:42:49 +000031 const image_header_t *hdr;
Heiko Schocher21d29f72014-05-28 11:33:33 +020032#endif
Gong Qianyu210fbee2015-07-30 14:00:01 +080033 u32 *data;
Marian Balakowicz424c4ab2008-03-12 10:33:00 +010034 int verify;
Simon Glass4ca30d62013-04-20 08:42:49 +000035 void *buf;
Marian Balakowicz424c4ab2008-03-12 10:33:00 +010036#if defined(CONFIG_FIT)
37 const void* fit_hdr;
38 int noffset;
39 const void *fit_data;
40 size_t fit_len;
41#endif
wdenkd0dd1072002-10-20 17:17:16 +000042
Bartlomiej Siekaedbed242008-04-18 12:39:23 +020043 verify = getenv_yesno ("verify");
wdenkd0dd1072002-10-20 17:17:16 +000044
Simon Glass4ca30d62013-04-20 08:42:49 +000045 buf = map_sysmem(addr, 0);
46 switch (genimg_get_format(buf)) {
Heiko Schocher21d29f72014-05-28 11:33:33 +020047#if defined(CONFIG_IMAGE_FORMAT_LEGACY)
Marian Balakowiczd5934ad2008-02-04 08:28:09 +010048 case IMAGE_FORMAT_LEGACY:
Simon Glass4ca30d62013-04-20 08:42:49 +000049 hdr = buf;
wdenkd0dd1072002-10-20 17:17:16 +000050
Marian Balakowiczd5934ad2008-02-04 08:28:09 +010051 if (!image_check_magic (hdr)) {
52 puts ("Bad magic number\n");
wdenkd0dd1072002-10-20 17:17:16 +000053 return 1;
54 }
wdenkd0dd1072002-10-20 17:17:16 +000055
Marian Balakowiczd5934ad2008-02-04 08:28:09 +010056 if (!image_check_hcrc (hdr)) {
57 puts ("Bad header crc\n");
58 return 1;
59 }
60
61 if (verify) {
62 if (!image_check_dcrc (hdr)) {
63 puts ("Bad data crc\n");
64 return 1;
65 }
66 }
67
68 if (!image_check_type (hdr, IH_TYPE_SCRIPT)) {
69 puts ("Bad image type\n");
70 return 1;
71 }
72
73 /* get length of script */
Gong Qianyu210fbee2015-07-30 14:00:01 +080074 data = (u32 *)image_get_data (hdr);
Marian Balakowiczd5934ad2008-02-04 08:28:09 +010075
Marian Balakowicz9a4daad2008-02-29 14:58:34 +010076 if ((len = uimage_to_cpu (*data)) == 0) {
Marian Balakowiczd5934ad2008-02-04 08:28:09 +010077 puts ("Empty Script\n");
78 return 1;
79 }
Bartlomiej Sieka36cc8cb2008-03-20 23:10:19 +010080
81 /*
82 * scripts are just multi-image files with one component, seek
83 * past the zero-terminated sequence of image lengths to get
84 * to the actual image data
85 */
86 while (*data++);
Marian Balakowiczd5934ad2008-02-04 08:28:09 +010087 break;
Heiko Schocher21d29f72014-05-28 11:33:33 +020088#endif
Marian Balakowiczd5934ad2008-02-04 08:28:09 +010089#if defined(CONFIG_FIT)
90 case IMAGE_FORMAT_FIT:
Marian Balakowicz424c4ab2008-03-12 10:33:00 +010091 if (fit_uname == NULL) {
92 puts ("No FIT subimage unit name\n");
93 return 1;
94 }
95
Simon Glass4ca30d62013-04-20 08:42:49 +000096 fit_hdr = buf;
Marian Balakowicz424c4ab2008-03-12 10:33:00 +010097 if (!fit_check_format (fit_hdr)) {
98 puts ("Bad FIT image format\n");
99 return 1;
100 }
101
102 /* get script component image node offset */
103 noffset = fit_image_get_node (fit_hdr, fit_uname);
104 if (noffset < 0) {
105 printf ("Can't find '%s' FIT subimage\n", fit_uname);
106 return 1;
107 }
108
109 if (!fit_image_check_type (fit_hdr, noffset, IH_TYPE_SCRIPT)) {
110 puts ("Not a image image\n");
111 return 1;
112 }
113
114 /* verify integrity */
115 if (verify) {
Simon Glassb8da8362013-05-07 06:11:57 +0000116 if (!fit_image_verify(fit_hdr, noffset)) {
Marian Balakowicz424c4ab2008-03-12 10:33:00 +0100117 puts ("Bad Data Hash\n");
118 return 1;
119 }
120 }
121
122 /* get script subimage data address and length */
123 if (fit_image_get_data (fit_hdr, noffset, &fit_data, &fit_len)) {
124 puts ("Could not find script subimage data\n");
125 return 1;
126 }
127
Gong Qianyu210fbee2015-07-30 14:00:01 +0800128 data = (u32 *)fit_data;
Marian Balakowicz424c4ab2008-03-12 10:33:00 +0100129 len = (ulong)fit_len;
130 break;
Marian Balakowiczd5934ad2008-02-04 08:28:09 +0100131#endif
132 default:
Wolfgang Denk74de7ae2009-04-01 23:34:12 +0200133 puts ("Wrong image format for \"source\" command\n");
wdenkd0dd1072002-10-20 17:17:16 +0000134 return 1;
135 }
136
wdenk699b13a2002-11-03 18:03:52 +0000137 debug ("** Script length: %ld\n", len);
Simon Glassd51004a2012-03-30 21:30:55 +0000138 return run_command_list((char *)data, len, 0);
wdenkd0dd1072002-10-20 17:17:16 +0000139}
140
wdenk8bde7f72003-06-27 21:31:46 +0000141/**************************************************/
Wolfgang Denk74de7ae2009-04-01 23:34:12 +0200142#if defined(CONFIG_CMD_SOURCE)
Jeroen Hofstee0e350f82014-06-23 00:22:08 +0200143static int do_source(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
wdenkd0dd1072002-10-20 17:17:16 +0000144{
145 ulong addr;
146 int rcode;
Marian Balakowicz424c4ab2008-03-12 10:33:00 +0100147 const char *fit_uname = NULL;
wdenkd0dd1072002-10-20 17:17:16 +0000148
Marian Balakowicz424c4ab2008-03-12 10:33:00 +0100149 /* Find script image */
wdenkd0dd1072002-10-20 17:17:16 +0000150 if (argc < 2) {
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +0200151 addr = CONFIG_SYS_LOAD_ADDR;
Wolfgang Denk74de7ae2009-04-01 23:34:12 +0200152 debug ("* source: default load address = 0x%08lx\n", addr);
Marian Balakowicz424c4ab2008-03-12 10:33:00 +0100153#if defined(CONFIG_FIT)
154 } else if (fit_parse_subimage (argv[1], load_addr, &addr, &fit_uname)) {
Wolfgang Denk74de7ae2009-04-01 23:34:12 +0200155 debug ("* source: subimage '%s' from FIT image at 0x%08lx\n",
Marian Balakowicz424c4ab2008-03-12 10:33:00 +0100156 fit_uname, addr);
157#endif
wdenkd0dd1072002-10-20 17:17:16 +0000158 } else {
Marian Balakowicz424c4ab2008-03-12 10:33:00 +0100159 addr = simple_strtoul(argv[1], NULL, 16);
Wolfgang Denk74de7ae2009-04-01 23:34:12 +0200160 debug ("* source: cmdline image address = 0x%08lx\n", addr);
wdenkd0dd1072002-10-20 17:17:16 +0000161 }
162
Marian Balakowicz424c4ab2008-03-12 10:33:00 +0100163 printf ("## Executing script at %08lx\n", addr);
Wolfgang Denk74de7ae2009-04-01 23:34:12 +0200164 rcode = source (addr, fit_uname);
wdenkd0dd1072002-10-20 17:17:16 +0000165 return rcode;
166}
wdenk8bde7f72003-06-27 21:31:46 +0000167
Kim Phillips088f1b12012-10-29 13:34:31 +0000168#ifdef CONFIG_SYS_LONGHELP
169static char source_help_text[] =
Wolfgang Denk74de7ae2009-04-01 23:34:12 +0200170 "[addr]\n"
171 "\t- run script starting at addr\n"
Wolfgang Denka89c33d2009-05-24 17:06:54 +0200172 "\t- A valid image header must be present"
Marian Balakowicz424c4ab2008-03-12 10:33:00 +0100173#if defined(CONFIG_FIT)
Wolfgang Denka89c33d2009-05-24 17:06:54 +0200174 "\n"
Marian Balakowicz424c4ab2008-03-12 10:33:00 +0100175 "For FIT format uImage addr must include subimage\n"
Wolfgang Denka89c33d2009-05-24 17:06:54 +0200176 "unit name in the form of addr:<subimg_uname>"
Jon Loeliger90253172007-07-10 11:02:44 -0500177#endif
Kim Phillips088f1b12012-10-29 13:34:31 +0000178 "";
179#endif
180
181U_BOOT_CMD(
182 source, 2, 0, do_source,
183 "run script from memory", source_help_text
Marian Balakowicz424c4ab2008-03-12 10:33:00 +0100184);
Jon Loeliger90253172007-07-10 11:02:44 -0500185#endif