]>
Commit | Line | Data |
---|---|---|
83d290c5 | 1 | // SPDX-License-Identifier: GPL-2.0+ |
b97a2a0a MB |
2 | /* |
3 | * (C) Copyright 2008 Semihalf | |
4 | * | |
5 | * (C) Copyright 2000-2006 | |
6 | * Wolfgang Denk, DENX Software Engineering, [email protected]. | |
b97a2a0a | 7 | */ |
ceaed2b1 | 8 | |
b97a2a0a | 9 | #ifndef USE_HOSTCC |
7b51b576 | 10 | #include <env.h> |
4e4bf944 | 11 | #include <display_options.h> |
4b32531b | 12 | #include <init.h> |
4d72caa5 | 13 | #include <lmb.h> |
f7ae49fc | 14 | #include <log.h> |
336d4615 | 15 | #include <malloc.h> |
3db71108 | 16 | #include <u-boot/crc.h> |
5ad03eb3 MB |
17 | |
18 | #ifdef CONFIG_SHOW_BOOT_PROGRESS | |
19 | #include <status_led.h> | |
20 | #endif | |
21 | ||
0c303f9a | 22 | #if CONFIG_IS_ENABLED(FIT) || CONFIG_IS_ENABLED(OF_LIBFDT) |
b08c8c48 | 23 | #include <linux/libfdt.h> |
fff888a1 | 24 | #include <fdt_support.h> |
c8779648 MB |
25 | #endif |
26 | ||
401d1c4f | 27 | #include <asm/global_data.h> |
1221ce45 | 28 | #include <linux/errno.h> |
35e7b0f1 | 29 | #include <asm/io.h> |
c8779648 | 30 | |
b6b0fe64 | 31 | DECLARE_GLOBAL_DATA_PTR; |
8a5ea3e6 | 32 | |
458b30af SG |
33 | /* Set this if we have less than 4 MB of malloc() space */ |
34 | #if CONFIG_SYS_MALLOC_LEN < (4096 * 1024) | |
35 | #define CONSERVE_MEMORY true | |
b97a2a0a | 36 | #else |
458b30af SG |
37 | #define CONSERVE_MEMORY false |
38 | #endif | |
39 | ||
40 | #else /* USE_HOSTCC */ | |
5ad03eb3 | 41 | #include "mkimage.h" |
b106961c | 42 | #include <linux/kconfig.h> |
20a14a42 | 43 | #include <u-boot/md5.h> |
5dfb5213 | 44 | #include <time.h> |
80402f34 HS |
45 | |
46 | #ifndef __maybe_unused | |
47 | # define __maybe_unused /* unimplemented */ | |
48 | #endif | |
458b30af SG |
49 | |
50 | #define CONSERVE_MEMORY false | |
51 | ||
5dfb5213 | 52 | #endif /* !USE_HOSTCC*/ |
b97a2a0a | 53 | |
458b30af SG |
54 | #include <abuf.h> |
55 | #include <bzlib.h> | |
5d3248a6 | 56 | #include <display_options.h> |
458b30af | 57 | #include <gzip.h> |
41506ff5 | 58 | #include <image.h> |
5b20d141 | 59 | #include <imximage.h> |
2ac00c05 | 60 | #include <relocate.h> |
458b30af SG |
61 | #include <linux/lzo.h> |
62 | #include <linux/zstd.h> | |
458b30af SG |
63 | #include <lzma/LzmaTypes.h> |
64 | #include <lzma/LzmaDec.h> | |
65 | #include <lzma/LzmaTools.h> | |
66 | #include <u-boot/crc.h> | |
2a2d8e94 | 67 | #include <u-boot/lz4.h> |
0ccff500 | 68 | |
7edb186f | 69 | static const table_entry_t uimage_arch[] = { |
30495bff | 70 | { IH_ARCH_INVALID, "invalid", "Invalid ARCH", }, |
570abb0a MB |
71 | { IH_ARCH_ALPHA, "alpha", "Alpha", }, |
72 | { IH_ARCH_ARM, "arm", "ARM", }, | |
73 | { IH_ARCH_I386, "x86", "Intel x86", }, | |
74 | { IH_ARCH_IA64, "ia64", "IA64", }, | |
75 | { IH_ARCH_M68K, "m68k", "M68K", }, | |
76 | { IH_ARCH_MICROBLAZE, "microblaze", "MicroBlaze", }, | |
77 | { IH_ARCH_MIPS, "mips", "MIPS", }, | |
78 | { IH_ARCH_MIPS64, "mips64", "MIPS 64 Bit", }, | |
570abb0a | 79 | { IH_ARCH_NIOS2, "nios2", "NIOS II", }, |
e419e12d | 80 | { IH_ARCH_PPC, "powerpc", "PowerPC", }, |
570abb0a MB |
81 | { IH_ARCH_PPC, "ppc", "PowerPC", }, |
82 | { IH_ARCH_S390, "s390", "IBM S390", }, | |
83 | { IH_ARCH_SH, "sh", "SuperH", }, | |
84 | { IH_ARCH_SPARC, "sparc", "SPARC", }, | |
85 | { IH_ARCH_SPARC64, "sparc64", "SPARC 64 Bit", }, | |
86 | { IH_ARCH_BLACKFIN, "blackfin", "Blackfin", }, | |
87 | { IH_ARCH_AVR32, "avr32", "AVR32", }, | |
64d61461 | 88 | { IH_ARCH_NDS32, "nds32", "NDS32", }, |
3ddcaccd | 89 | { IH_ARCH_OPENRISC, "or1k", "OpenRISC 1000",}, |
35e7b0f1 | 90 | { IH_ARCH_SANDBOX, "sandbox", "Sandbox", }, |
0ae76531 | 91 | { IH_ARCH_ARM64, "arm64", "AArch64", }, |
bc5d5428 | 92 | { IH_ARCH_ARC, "arc", "ARC", }, |
5bda35cf | 93 | { IH_ARCH_X86_64, "x86_64", "AMD x86_64", }, |
de5e5cea | 94 | { IH_ARCH_XTENSA, "xtensa", "Xtensa", }, |
86aa65a0 | 95 | { IH_ARCH_RISCV, "riscv", "RISC-V", }, |
570abb0a MB |
96 | { -1, "", "", }, |
97 | }; | |
98 | ||
7edb186f | 99 | static const table_entry_t uimage_os[] = { |
30495bff | 100 | { IH_OS_INVALID, "invalid", "Invalid OS", }, |
4914af12 | 101 | { IH_OS_ARM_TRUSTED_FIRMWARE, "arm-trusted-firmware", "ARM Trusted Firmware" }, |
570abb0a | 102 | { IH_OS_LINUX, "linux", "Linux", }, |
570abb0a | 103 | { IH_OS_NETBSD, "netbsd", "NetBSD", }, |
3df61957 | 104 | { IH_OS_OSE, "ose", "Enea OSE", }, |
04d41409 | 105 | { IH_OS_PLAN9, "plan9", "Plan 9", }, |
570abb0a | 106 | { IH_OS_RTEMS, "rtems", "RTEMS", }, |
45b55712 | 107 | { IH_OS_TEE, "tee", "Trusted Execution Environment" }, |
570abb0a | 108 | { IH_OS_U_BOOT, "u-boot", "U-Boot", }, |
68b15e83 | 109 | { IH_OS_VXWORKS, "vxworks", "VxWorks", }, |
570abb0a MB |
110 | #if defined(CONFIG_CMD_ELF) || defined(USE_HOSTCC) |
111 | { IH_OS_QNX, "qnx", "QNX", }, | |
570abb0a | 112 | #endif |
f5ed9e39 PT |
113 | #if defined(CONFIG_INTEGRITY) || defined(USE_HOSTCC) |
114 | { IH_OS_INTEGRITY,"integrity", "INTEGRITY", }, | |
115 | #endif | |
570abb0a MB |
116 | #ifdef USE_HOSTCC |
117 | { IH_OS_4_4BSD, "4_4bsd", "4_4BSD", }, | |
118 | { IH_OS_DELL, "dell", "Dell", }, | |
119 | { IH_OS_ESIX, "esix", "Esix", }, | |
120 | { IH_OS_FREEBSD, "freebsd", "FreeBSD", }, | |
121 | { IH_OS_IRIX, "irix", "Irix", }, | |
122 | { IH_OS_NCR, "ncr", "NCR", }, | |
123 | { IH_OS_OPENBSD, "openbsd", "OpenBSD", }, | |
124 | { IH_OS_PSOS, "psos", "pSOS", }, | |
125 | { IH_OS_SCO, "sco", "SCO", }, | |
126 | { IH_OS_SOLARIS, "solaris", "Solaris", }, | |
127 | { IH_OS_SVR4, "svr4", "SVR4", }, | |
128 | #endif | |
67ddd955 MV |
129 | #if defined(CONFIG_BOOTM_OPENRTOS) || defined(USE_HOSTCC) |
130 | { IH_OS_OPENRTOS, "openrtos", "OpenRTOS", }, | |
131 | #endif | |
5e30e45c | 132 | { IH_OS_OPENSBI, "opensbi", "RISC-V OpenSBI", }, |
a031b03f | 133 | { IH_OS_EFI, "efi", "EFI Firmware" }, |
67ddd955 | 134 | |
570abb0a MB |
135 | { -1, "", "", }, |
136 | }; | |
137 | ||
7edb186f | 138 | static const table_entry_t uimage_type[] = { |
4962e38e | 139 | { IH_TYPE_AISIMAGE, "aisimage", "Davinci AIS image",}, |
570abb0a MB |
140 | { IH_TYPE_FILESYSTEM, "filesystem", "Filesystem Image", }, |
141 | { IH_TYPE_FIRMWARE, "firmware", "Firmware", }, | |
3decb14a | 142 | { IH_TYPE_FLATDT, "flat_dt", "Flat Device Tree", }, |
bf411ea9 | 143 | { IH_TYPE_GPIMAGE, "gpimage", "TI Keystone SPL Image",}, |
570abb0a | 144 | { IH_TYPE_KERNEL, "kernel", "Kernel Image", }, |
b9b50e89 | 145 | { IH_TYPE_KERNEL_NOLOAD, "kernel_noload", "Kernel Image (no loading done)", }, |
4962e38e SB |
146 | { IH_TYPE_KWBIMAGE, "kwbimage", "Kirkwood Boot Image",}, |
147 | { IH_TYPE_IMXIMAGE, "imximage", "Freescale i.MX Boot Image",}, | |
a2b96ece | 148 | { IH_TYPE_IMX8IMAGE, "imx8image", "NXP i.MX8 Boot Image",}, |
6609c266 | 149 | { IH_TYPE_IMX8MIMAGE, "imx8mimage", "NXP i.MX8M Boot Image",}, |
30495bff | 150 | { IH_TYPE_INVALID, "invalid", "Invalid Image", }, |
570abb0a | 151 | { IH_TYPE_MULTI, "multi", "Multi-File Image", }, |
4962e38e | 152 | { IH_TYPE_OMAPIMAGE, "omapimage", "TI OMAP SPL With GP CH",}, |
5d898a00 | 153 | { IH_TYPE_PBLIMAGE, "pblimage", "Freescale PBL Boot Image",}, |
570abb0a MB |
154 | { IH_TYPE_RAMDISK, "ramdisk", "RAMDisk Image", }, |
155 | { IH_TYPE_SCRIPT, "script", "Script", }, | |
662abc4f MV |
156 | { IH_TYPE_SOCFPGAIMAGE, "socfpgaimage", "Altera SoCFPGA CV/AV preloader",}, |
157 | { IH_TYPE_SOCFPGAIMAGE_V1, "socfpgaimage_v1", "Altera SoCFPGA A10 preloader",}, | |
570abb0a | 158 | { IH_TYPE_STANDALONE, "standalone", "Standalone Program", }, |
7816f2cf | 159 | { IH_TYPE_UBLIMAGE, "ublimage", "Davinci UBL image",}, |
bce88370 | 160 | { IH_TYPE_MXSIMAGE, "mxsimage", "Freescale MXS Boot Image",}, |
7b1a4117 | 161 | { IH_TYPE_ATMELIMAGE, "atmelimage", "ATMEL ROM-Boot Image",}, |
90268b87 | 162 | { IH_TYPE_X86_SETUP, "x86_setup", "x86 setup.bin", }, |
39f520bb | 163 | { IH_TYPE_LPC32XXIMAGE, "lpc32xximage", "LPC32XX Boot Image", }, |
a131c1f4 | 164 | { IH_TYPE_RKIMAGE, "rkimage", "Rockchip Boot Image" }, |
f9a3c278 | 165 | { IH_TYPE_RKSD, "rksd", "Rockchip SD Boot Image" }, |
10b84fe1 | 166 | { IH_TYPE_RKSPI, "rkspi", "Rockchip SPI Boot Image" }, |
ed0c2c0a | 167 | { IH_TYPE_VYBRIDIMAGE, "vybridimage", "Vybrid Boot Image", }, |
66eef1e7 | 168 | { IH_TYPE_ZYNQIMAGE, "zynqimage", "Xilinx Zynq Boot Image" }, |
d9b58b30 | 169 | { IH_TYPE_ZYNQMPIMAGE, "zynqmpimage", "Xilinx ZynqMP Boot Image" }, |
6915dcf3 | 170 | { IH_TYPE_ZYNQMPBIF, "zynqmpbif", "Xilinx ZynqMP Boot Image (bif)" }, |
ed0cea7c | 171 | { IH_TYPE_FPGA, "fpga", "FPGA Image" }, |
7e719ee7 | 172 | { IH_TYPE_TEE, "tee", "Trusted Execution Environment Image",}, |
d21bd69b | 173 | { IH_TYPE_FIRMWARE_IVT, "firmware_ivt", "Firmware with HABv4 IVT" }, |
6442c964 | 174 | { IH_TYPE_PMMC, "pmmc", "TI Power Management Micro-Controller Firmware",}, |
81260e33 | 175 | { IH_TYPE_STM32IMAGE, "stm32image", "STMicroelectronics STM32 Image" }, |
3b975a14 | 176 | { IH_TYPE_MTKIMAGE, "mtk_image", "MediaTek BootROM loadable Image" }, |
e7fabe75 | 177 | { IH_TYPE_COPRO, "copro", "Coprocessor Image"}, |
6d295099 | 178 | { IH_TYPE_SUNXI_EGON, "sunxi_egon", "Allwinner eGON Boot Image" }, |
e9e87ec4 | 179 | { IH_TYPE_SUNXI_TOC0, "sunxi_toc0", "Allwinner TOC0 Boot Image" }, |
28f924f2 | 180 | { IH_TYPE_FDT_LEGACY, "fdt_legacy", "legacy Image with Flat Device Tree ", }, |
afdfcb11 | 181 | { IH_TYPE_RENESAS_SPKG, "spkgimage", "Renesas SPKG Image" }, |
64fd30d3 | 182 | { IH_TYPE_STARFIVE_SPL, "sfspl", "StarFive SPL Image" }, |
570abb0a MB |
183 | { -1, "", "", }, |
184 | }; | |
185 | ||
7edb186f | 186 | static const table_entry_t uimage_comp[] = { |
570abb0a MB |
187 | { IH_COMP_NONE, "none", "uncompressed", }, |
188 | { IH_COMP_BZIP2, "bzip2", "bzip2 compressed", }, | |
189 | { IH_COMP_GZIP, "gzip", "gzip compressed", }, | |
fc9c1727 | 190 | { IH_COMP_LZMA, "lzma", "lzma compressed", }, |
20dde48b | 191 | { IH_COMP_LZO, "lzo", "lzo compressed", }, |
027b728d | 192 | { IH_COMP_LZ4, "lz4", "lz4 compressed", }, |
26073f9e | 193 | { IH_COMP_ZSTD, "zstd", "zstd compressed", }, |
570abb0a MB |
194 | { -1, "", "", }, |
195 | }; | |
196 | ||
44ad35a0 SG |
197 | static const table_entry_t uimage_phase[] = { |
198 | { IH_PHASE_NONE, "none", "any", }, | |
199 | { IH_PHASE_U_BOOT, "u-boot", "U-Boot phase", }, | |
200 | { IH_PHASE_SPL, "spl", "SPL Phase", }, | |
201 | { -1, "", "", }, | |
202 | }; | |
203 | ||
56d7ab74 SG |
204 | struct table_info { |
205 | const char *desc; | |
206 | int count; | |
207 | const table_entry_t *table; | |
208 | }; | |
209 | ||
155d6a35 AP |
210 | static const struct comp_magic_map image_comp[] = { |
211 | { IH_COMP_BZIP2, "bzip2", {0x42, 0x5a},}, | |
212 | { IH_COMP_GZIP, "gzip", {0x1f, 0x8b},}, | |
213 | { IH_COMP_LZMA, "lzma", {0x5d, 0x00},}, | |
214 | { IH_COMP_LZO, "lzo", {0x89, 0x4c},}, | |
2ddb8fcd AL |
215 | { IH_COMP_LZ4, "lz4", {0x04, 0x22},}, |
216 | { IH_COMP_ZSTD, "zstd", {0x28, 0xb5},}, | |
155d6a35 AP |
217 | { IH_COMP_NONE, "none", {}, }, |
218 | }; | |
219 | ||
56d7ab74 SG |
220 | static const struct table_info table_info[IH_COUNT] = { |
221 | { "architecture", IH_ARCH_COUNT, uimage_arch }, | |
222 | { "compression", IH_COMP_COUNT, uimage_comp }, | |
223 | { "operating system", IH_OS_COUNT, uimage_os }, | |
224 | { "image type", IH_TYPE_COUNT, uimage_type }, | |
44ad35a0 | 225 | { "phase", IH_PHASE_COUNT, uimage_phase }, |
56d7ab74 SG |
226 | }; |
227 | ||
9a4daad0 MB |
228 | /*****************************************************************************/ |
229 | /* Legacy format routines */ | |
230 | /*****************************************************************************/ | |
f3543e69 | 231 | int image_check_hcrc(const struct legacy_img_hdr *hdr) |
b97a2a0a MB |
232 | { |
233 | ulong hcrc; | |
712fbcf3 | 234 | ulong len = image_get_header_size(); |
f3543e69 | 235 | struct legacy_img_hdr header; |
b97a2a0a MB |
236 | |
237 | /* Copy header so we can blank CRC field for re-calculation */ | |
712fbcf3 SW |
238 | memmove(&header, (char *)hdr, image_get_header_size()); |
239 | image_set_hcrc(&header, 0); | |
b97a2a0a | 240 | |
712fbcf3 | 241 | hcrc = crc32(0, (unsigned char *)&header, len); |
b97a2a0a | 242 | |
712fbcf3 | 243 | return (hcrc == image_get_hcrc(hdr)); |
b97a2a0a MB |
244 | } |
245 | ||
f3543e69 | 246 | int image_check_dcrc(const struct legacy_img_hdr *hdr) |
b97a2a0a | 247 | { |
712fbcf3 SW |
248 | ulong data = image_get_data(hdr); |
249 | ulong len = image_get_data_size(hdr); | |
250 | ulong dcrc = crc32_wd(0, (unsigned char *)data, len, CHUNKSZ_CRC32); | |
b97a2a0a | 251 | |
712fbcf3 | 252 | return (dcrc == image_get_dcrc(hdr)); |
b97a2a0a MB |
253 | } |
254 | ||
f13e7b2e MB |
255 | /** |
256 | * image_multi_count - get component (sub-image) count | |
257 | * @hdr: pointer to the header of the multi component image | |
258 | * | |
259 | * image_multi_count() returns number of components in a multi | |
260 | * component image. | |
261 | * | |
262 | * Note: no checking of the image type is done, caller must pass | |
263 | * a valid multi component image. | |
264 | * | |
265 | * returns: | |
266 | * number of components | |
267 | */ | |
f3543e69 | 268 | ulong image_multi_count(const struct legacy_img_hdr *hdr) |
f13e7b2e MB |
269 | { |
270 | ulong i, count = 0; | |
df6f1b89 | 271 | uint32_t *size; |
f13e7b2e MB |
272 | |
273 | /* get start of the image payload, which in case of multi | |
274 | * component images that points to a table of component sizes */ | |
712fbcf3 | 275 | size = (uint32_t *)image_get_data(hdr); |
f13e7b2e MB |
276 | |
277 | /* count non empty slots */ | |
278 | for (i = 0; size[i]; ++i) | |
279 | count++; | |
280 | ||
281 | return count; | |
282 | } | |
283 | ||
284 | /** | |
285 | * image_multi_getimg - get component data address and size | |
286 | * @hdr: pointer to the header of the multi component image | |
287 | * @idx: index of the requested component | |
288 | * @data: pointer to a ulong variable, will hold component data address | |
289 | * @len: pointer to a ulong variable, will hold component size | |
290 | * | |
291 | * image_multi_getimg() returns size and data address for the requested | |
292 | * component in a multi component image. | |
293 | * | |
294 | * Note: no checking of the image type is done, caller must pass | |
295 | * a valid multi component image. | |
296 | * | |
297 | * returns: | |
298 | * data address and size of the component, if idx is valid | |
299 | * 0 in data and len, if idx is out of range | |
300 | */ | |
f3543e69 | 301 | void image_multi_getimg(const struct legacy_img_hdr *hdr, ulong idx, |
f13e7b2e MB |
302 | ulong *data, ulong *len) |
303 | { | |
304 | int i; | |
df6f1b89 | 305 | uint32_t *size; |
02b9b224 | 306 | ulong offset, count, img_data; |
f13e7b2e MB |
307 | |
308 | /* get number of component */ | |
712fbcf3 | 309 | count = image_multi_count(hdr); |
f13e7b2e MB |
310 | |
311 | /* get start of the image payload, which in case of multi | |
312 | * component images that points to a table of component sizes */ | |
712fbcf3 | 313 | size = (uint32_t *)image_get_data(hdr); |
f13e7b2e MB |
314 | |
315 | /* get address of the proper component data start, which means | |
316 | * skipping sizes table (add 1 for last, null entry) */ | |
712fbcf3 | 317 | img_data = image_get_data(hdr) + (count + 1) * sizeof(uint32_t); |
f13e7b2e MB |
318 | |
319 | if (idx < count) { | |
712fbcf3 | 320 | *len = uimage_to_cpu(size[idx]); |
f13e7b2e | 321 | offset = 0; |
f13e7b2e MB |
322 | |
323 | /* go over all indices preceding requested component idx */ | |
324 | for (i = 0; i < idx; i++) { | |
02b9b224 | 325 | /* add up i-th component size, rounding up to 4 bytes */ |
712fbcf3 | 326 | offset += (uimage_to_cpu(size[i]) + 3) & ~3 ; |
f13e7b2e MB |
327 | } |
328 | ||
329 | /* calculate idx-th component data address */ | |
02b9b224 | 330 | *data = img_data + offset; |
f13e7b2e MB |
331 | } else { |
332 | *len = 0; | |
333 | *data = 0; | |
334 | } | |
335 | } | |
42b73e8e | 336 | |
f3543e69 | 337 | static void image_print_type(const struct legacy_img_hdr *hdr) |
9a4daad0 | 338 | { |
80402f34 | 339 | const char __maybe_unused *os, *arch, *type, *comp; |
9a4daad0 | 340 | |
712fbcf3 SW |
341 | os = genimg_get_os_name(image_get_os(hdr)); |
342 | arch = genimg_get_arch_name(image_get_arch(hdr)); | |
343 | type = genimg_get_type_name(image_get_type(hdr)); | |
344 | comp = genimg_get_comp_name(image_get_comp(hdr)); | |
9a4daad0 | 345 | |
712fbcf3 | 346 | printf("%s %s %s (%s)\n", arch, os, type, comp); |
9a4daad0 MB |
347 | } |
348 | ||
5dfb5213 | 349 | /** |
edbed247 | 350 | * image_print_contents - prints out the contents of the legacy format image |
3a2003f6 | 351 | * @ptr: pointer to the legacy format image header |
5dfb5213 MB |
352 | * @p: pointer to prefix string |
353 | * | |
edbed247 | 354 | * image_print_contents() formats a multi line legacy image contents description. |
5dfb5213 MB |
355 | * The routine prints out all header fields followed by the size/offset data |
356 | * for MULTI/SCRIPT images. | |
357 | * | |
358 | * returns: | |
359 | * no returned results | |
360 | */ | |
712fbcf3 | 361 | void image_print_contents(const void *ptr) |
9a4daad0 | 362 | { |
f3543e69 | 363 | const struct legacy_img_hdr *hdr = (const struct legacy_img_hdr *)ptr; |
80402f34 | 364 | const char __maybe_unused *p; |
edbed247 | 365 | |
1fe7d938 | 366 | p = IMAGE_INDENT_STRING; |
712fbcf3 | 367 | printf("%sImage Name: %.*s\n", p, IH_NMLEN, image_get_name(hdr)); |
859e92b7 SG |
368 | if (IMAGE_ENABLE_TIMESTAMP) { |
369 | printf("%sCreated: ", p); | |
370 | genimg_print_time((time_t)image_get_time(hdr)); | |
371 | } | |
712fbcf3 SW |
372 | printf("%sImage Type: ", p); |
373 | image_print_type(hdr); | |
374 | printf("%sData Size: ", p); | |
375 | genimg_print_size(image_get_data_size(hdr)); | |
376 | printf("%sLoad Address: %08x\n", p, image_get_load(hdr)); | |
377 | printf("%sEntry Point: %08x\n", p, image_get_ep(hdr)); | |
378 | ||
379 | if (image_check_type(hdr, IH_TYPE_MULTI) || | |
380 | image_check_type(hdr, IH_TYPE_SCRIPT)) { | |
9a4daad0 MB |
381 | int i; |
382 | ulong data, len; | |
712fbcf3 | 383 | ulong count = image_multi_count(hdr); |
9a4daad0 | 384 | |
712fbcf3 | 385 | printf("%sContents:\n", p); |
9a4daad0 | 386 | for (i = 0; i < count; i++) { |
712fbcf3 | 387 | image_multi_getimg(hdr, i, &data, &len); |
570abb0a | 388 | |
712fbcf3 SW |
389 | printf("%s Image %d: ", p, i); |
390 | genimg_print_size(len); | |
570abb0a | 391 | |
712fbcf3 | 392 | if (image_check_type(hdr, IH_TYPE_SCRIPT) && i > 0) { |
570abb0a MB |
393 | /* |
394 | * the user may need to know offsets | |
395 | * if planning to do something with | |
396 | * multiple files | |
397 | */ | |
712fbcf3 | 398 | printf("%s Offset = 0x%08lx\n", p, data); |
570abb0a | 399 | } |
9a4daad0 | 400 | } |
d21bd69b SE |
401 | } else if (image_check_type(hdr, IH_TYPE_FIRMWARE_IVT)) { |
402 | printf("HAB Blocks: 0x%08x 0x0000 0x%08x\n", | |
5b20d141 BML |
403 | image_get_load(hdr) - image_get_header_size(), |
404 | (int)(image_get_size(hdr) + image_get_header_size() | |
405 | + sizeof(flash_header_v2_t) - 0x2060)); | |
9a4daad0 MB |
406 | } |
407 | } | |
408 | ||
2090854c JW |
409 | /** |
410 | * print_decomp_msg() - Print a suitable decompression/loading message | |
411 | * | |
412 | * @type: OS type (IH_OS_...) | |
413 | * @comp_type: Compression type being used (IH_COMP_...) | |
414 | * @is_xip: true if the load address matches the image start | |
30ad6366 | 415 | * @load: Load address for printing |
2090854c | 416 | */ |
30ad6366 SG |
417 | static void print_decomp_msg(int comp_type, int type, bool is_xip, |
418 | ulong load) | |
2090854c JW |
419 | { |
420 | const char *name = genimg_get_type_name(type); | |
421 | ||
30ad6366 | 422 | /* Shows "Loading Kernel Image" for example */ |
2090854c | 423 | if (comp_type == IH_COMP_NONE) |
30ad6366 | 424 | printf(" %s %s", is_xip ? "XIP" : "Loading", name); |
2090854c | 425 | else |
30ad6366 SG |
426 | printf(" Uncompressing %s", name); |
427 | ||
428 | printf(" to %lx\n", load); | |
2090854c JW |
429 | } |
430 | ||
155d6a35 AP |
431 | int image_decomp_type(const unsigned char *buf, ulong len) |
432 | { | |
433 | const struct comp_magic_map *cmagic = image_comp; | |
434 | ||
435 | if (len < 2) | |
436 | return -EINVAL; | |
437 | ||
438 | for (; cmagic->comp_id > 0; cmagic++) { | |
439 | if (!memcmp(buf, cmagic->magic, 2)) | |
440 | break; | |
441 | } | |
442 | ||
443 | return cmagic->comp_id; | |
444 | } | |
445 | ||
2090854c JW |
446 | int image_decomp(int comp, ulong load, ulong image_start, int type, |
447 | void *load_buf, void *image_buf, ulong image_len, | |
448 | uint unc_len, ulong *load_end) | |
449 | { | |
b876eb87 | 450 | int ret = -ENOSYS; |
2090854c JW |
451 | |
452 | *load_end = load; | |
30ad6366 | 453 | print_decomp_msg(comp, type, load == image_start, load); |
2090854c JW |
454 | |
455 | /* | |
456 | * Load the image to the right place, decompressing if needed. After | |
457 | * this, image_len will be set to the number of uncompressed bytes | |
458 | * loaded, ret will be non-zero on error. | |
459 | */ | |
460 | switch (comp) { | |
461 | case IH_COMP_NONE: | |
b876eb87 | 462 | ret = 0; |
2090854c JW |
463 | if (load == image_start) |
464 | break; | |
465 | if (image_len <= unc_len) | |
466 | memmove_wd(load_buf, image_buf, image_len, CHUNKSZ); | |
467 | else | |
468 | ret = -ENOSPC; | |
469 | break; | |
458b30af | 470 | case IH_COMP_GZIP: |
c9d6b5b5 | 471 | if (!tools_build() && CONFIG_IS_ENABLED(GZIP)) |
458b30af | 472 | ret = gunzip(load_buf, unc_len, image_buf, &image_len); |
2090854c | 473 | break; |
458b30af | 474 | case IH_COMP_BZIP2: |
c9d6b5b5 | 475 | if (!tools_build() && CONFIG_IS_ENABLED(BZIP2)) { |
458b30af SG |
476 | uint size = unc_len; |
477 | ||
478 | /* | |
479 | * If we've got less than 4 MB of malloc() space, | |
480 | * use slower decompression algorithm which requires | |
481 | * at most 2300 KB of memory. | |
482 | */ | |
483 | ret = BZ2_bzBuffToBuffDecompress(load_buf, &size, | |
484 | image_buf, image_len, CONSERVE_MEMORY, 0); | |
485 | image_len = size; | |
486 | } | |
2090854c | 487 | break; |
458b30af | 488 | case IH_COMP_LZMA: |
c9d6b5b5 | 489 | if (!tools_build() && CONFIG_IS_ENABLED(LZMA)) { |
458b30af | 490 | SizeT lzma_len = unc_len; |
2090854c | 491 | |
458b30af SG |
492 | ret = lzmaBuffToBuffDecompress(load_buf, &lzma_len, |
493 | image_buf, image_len); | |
494 | image_len = lzma_len; | |
495 | } | |
2090854c | 496 | break; |
458b30af | 497 | case IH_COMP_LZO: |
c9d6b5b5 | 498 | if (!tools_build() && CONFIG_IS_ENABLED(LZO)) { |
458b30af | 499 | size_t size = unc_len; |
2090854c | 500 | |
458b30af SG |
501 | ret = lzop_decompress(image_buf, image_len, load_buf, &size); |
502 | image_len = size; | |
503 | } | |
2090854c | 504 | break; |
458b30af | 505 | case IH_COMP_LZ4: |
c9d6b5b5 | 506 | if (!tools_build() && CONFIG_IS_ENABLED(LZ4)) { |
458b30af | 507 | size_t size = unc_len; |
2090854c | 508 | |
458b30af SG |
509 | ret = ulz4fn(image_buf, image_len, load_buf, &size); |
510 | image_len = size; | |
511 | } | |
2090854c | 512 | break; |
458b30af | 513 | case IH_COMP_ZSTD: |
c9d6b5b5 | 514 | if (!tools_build() && CONFIG_IS_ENABLED(ZSTD)) { |
458b30af SG |
515 | struct abuf in, out; |
516 | ||
517 | abuf_init_set(&in, image_buf, image_len); | |
408e2d5a | 518 | abuf_init_set(&out, load_buf, unc_len); |
458b30af SG |
519 | ret = zstd_decompress(&in, &out); |
520 | if (ret >= 0) { | |
521 | image_len = ret; | |
522 | ret = 0; | |
523 | } | |
26073f9e | 524 | } |
26073f9e RM |
525 | break; |
526 | } | |
b876eb87 | 527 | if (ret == -ENOSYS) { |
2090854c | 528 | printf("Unimplemented compression type %d\n", comp); |
b876eb87 | 529 | return ret; |
2090854c JW |
530 | } |
531 | ||
532 | *load_end = load + image_len; | |
30787661 MK |
533 | if (ret) |
534 | return ret; | |
2090854c | 535 | |
b876eb87 | 536 | return 0; |
2090854c JW |
537 | } |
538 | ||
5b9d44df SG |
539 | const table_entry_t *get_table_entry(const table_entry_t *table, int id) |
540 | { | |
541 | for (; table->id >= 0; ++table) { | |
542 | if (table->id == id) | |
543 | return table; | |
544 | } | |
545 | return NULL; | |
546 | } | |
547 | ||
1426220b SG |
548 | static const char *unknown_msg(enum ih_category category) |
549 | { | |
ae3de0d8 | 550 | static const char unknown_str[] = "Unknown "; |
1426220b SG |
551 | static char msg[30]; |
552 | ||
ae3de0d8 SG |
553 | strcpy(msg, unknown_str); |
554 | strncat(msg, table_info[category].desc, | |
555 | sizeof(msg) - sizeof(unknown_str)); | |
1426220b SG |
556 | |
557 | return msg; | |
558 | } | |
559 | ||
560 | /** | |
898a0849 | 561 | * genimg_get_cat_name - translate entry id to long name |
1426220b SG |
562 | * @category: category to look up (enum ih_category) |
563 | * @id: entry id to be translated | |
564 | * | |
565 | * This will scan the translation table trying to find the entry that matches | |
566 | * the given id. | |
567 | * | |
185f812c | 568 | * Return: long entry name if translation succeeds; error string on failure |
1426220b SG |
569 | */ |
570 | const char *genimg_get_cat_name(enum ih_category category, uint id) | |
571 | { | |
572 | const table_entry_t *entry; | |
573 | ||
574 | entry = get_table_entry(table_info[category].table, id); | |
575 | if (!entry) | |
576 | return unknown_msg(category); | |
ba377a6c | 577 | return entry->lname; |
1426220b SG |
578 | } |
579 | ||
580 | /** | |
898a0849 | 581 | * genimg_get_cat_short_name - translate entry id to short name |
1426220b SG |
582 | * @category: category to look up (enum ih_category) |
583 | * @id: entry id to be translated | |
584 | * | |
585 | * This will scan the translation table trying to find the entry that matches | |
586 | * the given id. | |
587 | * | |
185f812c | 588 | * Return: short entry name if translation succeeds; error string on failure |
1426220b SG |
589 | */ |
590 | const char *genimg_get_cat_short_name(enum ih_category category, uint id) | |
591 | { | |
592 | const table_entry_t *entry; | |
593 | ||
594 | entry = get_table_entry(table_info[category].table, id); | |
595 | if (!entry) | |
596 | return unknown_msg(category); | |
ba377a6c | 597 | return entry->sname; |
1426220b SG |
598 | } |
599 | ||
600 | int genimg_get_cat_count(enum ih_category category) | |
601 | { | |
602 | return table_info[category].count; | |
603 | } | |
604 | ||
605 | const char *genimg_get_cat_desc(enum ih_category category) | |
606 | { | |
607 | return table_info[category].desc; | |
608 | } | |
609 | ||
02d41b01 NH |
610 | /** |
611 | * genimg_cat_has_id - check whether category has entry id | |
612 | * @category: category to look up (enum ih_category) | |
613 | * @id: entry id to be checked | |
614 | * | |
615 | * This will scan the translation table trying to find the entry that matches | |
616 | * the given id. | |
617 | * | |
185f812c | 618 | * Return: true if category has entry id; false if not |
02d41b01 NH |
619 | */ |
620 | bool genimg_cat_has_id(enum ih_category category, uint id) | |
621 | { | |
622 | if (get_table_entry(table_info[category].table, id)) | |
623 | return true; | |
624 | ||
625 | return false; | |
626 | } | |
627 | ||
570abb0a MB |
628 | /** |
629 | * get_table_entry_name - translate entry id to long name | |
630 | * @table: pointer to a translation table for entries of a specific type | |
631 | * @msg: message to be returned when translation fails | |
632 | * @id: entry id to be translated | |
633 | * | |
634 | * get_table_entry_name() will go over translation table trying to find | |
635 | * entry that matches given id. If matching entry is found, its long | |
636 | * name is returned to the caller. | |
637 | * | |
638 | * returns: | |
639 | * long entry name if translation succeeds | |
640 | * msg otherwise | |
641 | */ | |
7edb186f | 642 | char *get_table_entry_name(const table_entry_t *table, char *msg, int id) |
570abb0a | 643 | { |
5b9d44df SG |
644 | table = get_table_entry(table, id); |
645 | if (!table) | |
646 | return msg; | |
ba377a6c | 647 | return table->lname; |
570abb0a | 648 | } |
42b73e8e | 649 | |
712fbcf3 | 650 | const char *genimg_get_os_name(uint8_t os) |
570abb0a | 651 | { |
712fbcf3 | 652 | return (get_table_entry_name(uimage_os, "Unknown OS", os)); |
42b73e8e MB |
653 | } |
654 | ||
712fbcf3 | 655 | const char *genimg_get_arch_name(uint8_t arch) |
42b73e8e | 656 | { |
712fbcf3 SW |
657 | return (get_table_entry_name(uimage_arch, "Unknown Architecture", |
658 | arch)); | |
570abb0a | 659 | } |
42b73e8e | 660 | |
712fbcf3 | 661 | const char *genimg_get_type_name(uint8_t type) |
570abb0a | 662 | { |
712fbcf3 | 663 | return (get_table_entry_name(uimage_type, "Unknown Image", type)); |
570abb0a | 664 | } |
42b73e8e | 665 | |
898a0849 NH |
666 | const char *genimg_get_comp_name(uint8_t comp) |
667 | { | |
668 | return (get_table_entry_name(uimage_comp, "Unknown Compression", | |
669 | comp)); | |
670 | } | |
671 | ||
44ad35a0 SG |
672 | const char *genimg_get_phase_name(enum image_phase_t phase) |
673 | { | |
674 | return get_table_entry_name(uimage_phase, "Unknown Phase", phase); | |
675 | } | |
676 | ||
cef2e514 | 677 | static const char *genimg_get_short_name(const table_entry_t *table, int val) |
5b9d44df | 678 | { |
cef2e514 | 679 | table = get_table_entry(table, val); |
5b9d44df SG |
680 | if (!table) |
681 | return "unknown"; | |
ba377a6c | 682 | return table->sname; |
5b9d44df SG |
683 | } |
684 | ||
cef2e514 SG |
685 | const char *genimg_get_type_short_name(uint8_t type) |
686 | { | |
687 | return genimg_get_short_name(uimage_type, type); | |
688 | } | |
689 | ||
cef2e514 SG |
690 | const char *genimg_get_comp_short_name(uint8_t comp) |
691 | { | |
692 | return genimg_get_short_name(uimage_comp, comp); | |
693 | } | |
694 | ||
695 | const char *genimg_get_os_short_name(uint8_t os) | |
696 | { | |
697 | return genimg_get_short_name(uimage_os, os); | |
698 | } | |
699 | ||
700 | const char *genimg_get_arch_short_name(uint8_t arch) | |
701 | { | |
702 | return genimg_get_short_name(uimage_arch, arch); | |
703 | } | |
704 | ||
570abb0a MB |
705 | /** |
706 | * get_table_entry_id - translate short entry name to id | |
707 | * @table: pointer to a translation table for entries of a specific type | |
708 | * @table_name: to be used in case of error | |
709 | * @name: entry short name to be translated | |
710 | * | |
711 | * get_table_entry_id() will go over translation table trying to find | |
712 | * entry that matches given short name. If matching entry is found, | |
713 | * its id returned to the caller. | |
714 | * | |
715 | * returns: | |
716 | * entry id if translation succeeds | |
717 | * -1 otherwise | |
718 | */ | |
7edb186f | 719 | int get_table_entry_id(const table_entry_t *table, |
570abb0a | 720 | const char *table_name, const char *name) |
42b73e8e | 721 | { |
7edb186f | 722 | const table_entry_t *t; |
42b73e8e | 723 | |
570abb0a | 724 | for (t = table; t->id >= 0; ++t) { |
ba377a6c | 725 | if (t->sname && !strcasecmp(t->sname, name)) |
2ac00c05 | 726 | return t->id; |
570abb0a | 727 | } |
712fbcf3 | 728 | debug("Invalid %s Type: %s\n", table_name, name); |
5b9d44df SG |
729 | |
730 | return -1; | |
570abb0a MB |
731 | } |
732 | ||
712fbcf3 | 733 | int genimg_get_os_id(const char *name) |
570abb0a | 734 | { |
712fbcf3 | 735 | return (get_table_entry_id(uimage_os, "OS", name)); |
570abb0a MB |
736 | } |
737 | ||
712fbcf3 | 738 | int genimg_get_arch_id(const char *name) |
570abb0a | 739 | { |
712fbcf3 | 740 | return (get_table_entry_id(uimage_arch, "CPU", name)); |
42b73e8e | 741 | } |
5ad03eb3 | 742 | |
712fbcf3 | 743 | int genimg_get_type_id(const char *name) |
570abb0a | 744 | { |
712fbcf3 | 745 | return (get_table_entry_id(uimage_type, "Image", name)); |
570abb0a MB |
746 | } |
747 | ||
712fbcf3 | 748 | int genimg_get_comp_id(const char *name) |
570abb0a | 749 | { |
712fbcf3 | 750 | return (get_table_entry_id(uimage_comp, "Compression", name)); |
570abb0a | 751 | } |
44ad35a0 SG |
752 | |
753 | int genimg_get_phase_id(const char *name) | |
754 | { | |
755 | return get_table_entry_id(uimage_phase, "Phase", name); | |
756 | } |