xref: /openbmc/u-boot/cmd/ximg.c (revision a7ae1599)
1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * (C) Copyright 2000-2004
4  * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
5  *
6  * (C) Copyright 2003
7  * Kai-Uwe Bloem, Auerswald GmbH & Co KG, <linux-development@auerswald.de>
8  */
9 
10 
11 /*
12  * Multi Image extract
13  */
14 #include <common.h>
15 #include <command.h>
16 #include <image.h>
17 #include <mapmem.h>
18 #include <watchdog.h>
19 #if defined(CONFIG_BZIP2)
20 #include <bzlib.h>
21 #endif
22 #include <asm/byteorder.h>
23 #include <asm/io.h>
24 
25 #ifndef CONFIG_SYS_XIMG_LEN
26 /* use 8MByte as default max gunzip size */
27 #define CONFIG_SYS_XIMG_LEN	0x800000
28 #endif
29 
30 static int
31 do_imgextract(cmd_tbl_t * cmdtp, int flag, int argc, char * const argv[])
32 {
33 	ulong		addr = load_addr;
34 	ulong		dest = 0;
35 	ulong		data, len;
36 	int		verify;
37 	int		part = 0;
38 #if defined(CONFIG_IMAGE_FORMAT_LEGACY)
39 	ulong		count;
40 	image_header_t	*hdr = NULL;
41 #endif
42 #if defined(CONFIG_FIT)
43 	const char	*uname = NULL;
44 	const void*	fit_hdr;
45 	int		noffset;
46 	const void	*fit_data;
47 	size_t		fit_len;
48 #endif
49 #ifdef CONFIG_GZIP
50 	uint		unc_len = CONFIG_SYS_XIMG_LEN;
51 #endif
52 	uint8_t		comp;
53 
54 	verify = env_get_yesno("verify");
55 
56 	if (argc > 1) {
57 		addr = simple_strtoul(argv[1], NULL, 16);
58 	}
59 	if (argc > 2) {
60 		part = simple_strtoul(argv[2], NULL, 16);
61 #if defined(CONFIG_FIT)
62 		uname = argv[2];
63 #endif
64 	}
65 	if (argc > 3) {
66 		dest = simple_strtoul(argv[3], NULL, 16);
67 	}
68 
69 	switch (genimg_get_format((void *)addr)) {
70 #if defined(CONFIG_IMAGE_FORMAT_LEGACY)
71 	case IMAGE_FORMAT_LEGACY:
72 
73 		printf("## Copying part %d from legacy image "
74 			"at %08lx ...\n", part, addr);
75 
76 		hdr = (image_header_t *)addr;
77 		if (!image_check_magic(hdr)) {
78 			printf("Bad Magic Number\n");
79 			return 1;
80 		}
81 
82 		if (!image_check_hcrc(hdr)) {
83 			printf("Bad Header Checksum\n");
84 			return 1;
85 		}
86 #ifdef DEBUG
87 		image_print_contents(hdr);
88 #endif
89 
90 		if (!image_check_type(hdr, IH_TYPE_MULTI) &&
91 		    !image_check_type(hdr, IH_TYPE_SCRIPT)) {
92 			printf("Wrong Image Type for %s command\n",
93 					cmdtp->name);
94 			return 1;
95 		}
96 
97 		comp = image_get_comp(hdr);
98 		if ((comp != IH_COMP_NONE) && (argc < 4)) {
99 			printf("Must specify load address for %s command "
100 					"with compressed image\n",
101 					cmdtp->name);
102 			return 1;
103 		}
104 
105 		if (verify) {
106 			printf("   Verifying Checksum ... ");
107 			if (!image_check_dcrc(hdr)) {
108 				printf("Bad Data CRC\n");
109 				return 1;
110 			}
111 			printf("OK\n");
112 		}
113 
114 		count = image_multi_count(hdr);
115 		if (part >= count) {
116 			printf("Bad Image Part\n");
117 			return 1;
118 		}
119 
120 		image_multi_getimg(hdr, part, &data, &len);
121 		break;
122 #endif
123 #if defined(CONFIG_FIT)
124 	case IMAGE_FORMAT_FIT:
125 		if (uname == NULL) {
126 			puts("No FIT subimage unit name\n");
127 			return 1;
128 		}
129 
130 		printf("## Copying '%s' subimage from FIT image "
131 			"at %08lx ...\n", uname, addr);
132 
133 		fit_hdr = (const void *)addr;
134 		if (!fit_check_format(fit_hdr)) {
135 			puts("Bad FIT image format\n");
136 			return 1;
137 		}
138 
139 		/* get subimage node offset */
140 		noffset = fit_image_get_node(fit_hdr, uname);
141 		if (noffset < 0) {
142 			printf("Can't find '%s' FIT subimage\n", uname);
143 			return 1;
144 		}
145 
146 		if (!fit_image_check_comp(fit_hdr, noffset, IH_COMP_NONE)
147 		    && (argc < 4)) {
148 			printf("Must specify load address for %s command "
149 				"with compressed image\n",
150 				cmdtp->name);
151 			return 1;
152 		}
153 
154 		/* verify integrity */
155 		if (verify) {
156 			if (!fit_image_verify(fit_hdr, noffset)) {
157 				puts("Bad Data Hash\n");
158 				return 1;
159 			}
160 		}
161 
162 		/* get subimage data address and length */
163 		if (fit_image_get_data(fit_hdr, noffset,
164 					&fit_data, &fit_len)) {
165 			puts("Could not find script subimage data\n");
166 			return 1;
167 		}
168 
169 		if (fit_image_get_comp(fit_hdr, noffset, &comp)) {
170 			puts("Could not find script subimage "
171 				"compression type\n");
172 			return 1;
173 		}
174 
175 		data = (ulong)fit_data;
176 		len = (ulong)fit_len;
177 		break;
178 #endif
179 	default:
180 		puts("Invalid image type for imxtract\n");
181 		return 1;
182 	}
183 
184 	if (argc > 3) {
185 		switch (comp) {
186 		case IH_COMP_NONE:
187 #if defined(CONFIG_HW_WATCHDOG) || defined(CONFIG_WATCHDOG)
188 			{
189 				size_t l = len;
190 				size_t tail;
191 				void *to = (void *) dest;
192 				void *from = (void *)data;
193 
194 				printf("   Loading part %d ... ", part);
195 
196 				while (l > 0) {
197 					tail = (l > CHUNKSZ) ? CHUNKSZ : l;
198 					WATCHDOG_RESET();
199 					memmove(to, from, tail);
200 					to += tail;
201 					from += tail;
202 					l -= tail;
203 				}
204 			}
205 #else	/* !(CONFIG_HW_WATCHDOG || CONFIG_WATCHDOG) */
206 			printf("   Loading part %d ... ", part);
207 			memmove((char *) dest, (char *)data, len);
208 #endif	/* CONFIG_HW_WATCHDOG || CONFIG_WATCHDOG */
209 			break;
210 #ifdef CONFIG_GZIP
211 		case IH_COMP_GZIP:
212 			printf("   Uncompressing part %d ... ", part);
213 			if (gunzip((void *) dest, unc_len,
214 				   (uchar *) data, &len) != 0) {
215 				puts("GUNZIP ERROR - image not loaded\n");
216 				return 1;
217 			}
218 			break;
219 #endif
220 #if defined(CONFIG_BZIP2) && defined(CONFIG_IMAGE_FORMAT_LEGACY)
221 		case IH_COMP_BZIP2:
222 			{
223 				int i;
224 
225 				printf("   Uncompressing part %d ... ", part);
226 				/*
227 				 * If we've got less than 4 MB of malloc()
228 				 * space, use slower decompression algorithm
229 				 * which requires at most 2300 KB of memory.
230 				 */
231 				i = BZ2_bzBuffToBuffDecompress(
232 					map_sysmem(ntohl(hdr->ih_load), 0),
233 					&unc_len, (char *)data, len,
234 					CONFIG_SYS_MALLOC_LEN < (4096 * 1024),
235 					0);
236 				if (i != BZ_OK) {
237 					printf("BUNZIP2 ERROR %d - "
238 						"image not loaded\n", i);
239 					return 1;
240 				}
241 			}
242 			break;
243 #endif /* CONFIG_BZIP2 */
244 		default:
245 			printf("Unimplemented compression type %d\n", comp);
246 			return 1;
247 		}
248 		puts("OK\n");
249 	}
250 
251 	flush_cache(dest, ALIGN(len, ARCH_DMA_MINALIGN));
252 
253 	env_set_hex("fileaddr", data);
254 	env_set_hex("filesize", len);
255 
256 	return 0;
257 }
258 
259 #ifdef CONFIG_SYS_LONGHELP
260 static char imgextract_help_text[] =
261 	"addr part [dest]\n"
262 	"    - extract <part> from legacy image at <addr> and copy to <dest>"
263 #if defined(CONFIG_FIT)
264 	"\n"
265 	"addr uname [dest]\n"
266 	"    - extract <uname> subimage from FIT image at <addr> and copy to <dest>"
267 #endif
268 	"";
269 #endif
270 
271 U_BOOT_CMD(
272 	imxtract, 4, 1, do_imgextract,
273 	"extract a part of a multi-image", imgextract_help_text
274 );
275