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