Tom Rini | 10e4779 | 2018-05-06 17:58:06 -0400 | [diff] [blame] | 1 | // SPDX-License-Identifier: GPL-2.0+ |
Marian Balakowicz | 41d71ed | 2008-01-08 18:14:09 +0100 | [diff] [blame] | 2 | /* |
| 3 | * (C) Copyright 2008 Semihalf |
| 4 | * |
| 5 | * (C) Copyright 2000-2006 |
| 6 | * Wolfgang Denk, DENX Software Engineering, wd@denx.de. |
Marian Balakowicz | 41d71ed | 2008-01-08 18:14:09 +0100 | [diff] [blame] | 7 | */ |
Marian Balakowicz | 9c701e8 | 2008-01-31 13:57:17 +0100 | [diff] [blame] | 8 | |
Marian Balakowicz | 41d71ed | 2008-01-08 18:14:09 +0100 | [diff] [blame] | 9 | #ifndef USE_HOSTCC |
Marian Balakowicz | 9541850 | 2008-01-31 13:55:39 +0100 | [diff] [blame] | 10 | #include <common.h> |
Simon Glass | 0af6e2d | 2019-08-01 09:46:52 -0600 | [diff] [blame] | 11 | #include <env.h> |
Simon Glass | 2dc9c34 | 2020-05-10 11:40:01 -0600 | [diff] [blame] | 12 | #include <lmb.h> |
Simon Glass | 0f2af88 | 2020-05-10 11:40:05 -0600 | [diff] [blame] | 13 | #include <log.h> |
Simon Glass | 9bc1564 | 2020-02-03 07:36:16 -0700 | [diff] [blame] | 14 | #include <malloc.h> |
Simon Glass | 48b6c6b | 2019-11-14 12:57:16 -0700 | [diff] [blame] | 15 | #include <u-boot/crc.h> |
Marian Balakowicz | 9541850 | 2008-01-31 13:55:39 +0100 | [diff] [blame] | 16 | |
| 17 | #ifdef CONFIG_SHOW_BOOT_PROGRESS |
| 18 | #include <status_led.h> |
| 19 | #endif |
| 20 | |
Simon Glass | 85c057e | 2021-09-25 19:43:21 -0600 | [diff] [blame] | 21 | #if CONFIG_IS_ENABLED(FIT) || CONFIG_IS_ENABLED(OF_LIBFDT) |
Masahiro Yamada | 75f82d0 | 2018-03-05 01:20:11 +0900 | [diff] [blame] | 22 | #include <linux/libfdt.h> |
Marian Balakowicz | c536d6f | 2008-02-21 17:20:19 +0100 | [diff] [blame] | 23 | #include <fdt_support.h> |
Marian Balakowicz | a50d515 | 2008-03-12 10:12:37 +0100 | [diff] [blame] | 24 | #endif |
| 25 | |
Simon Glass | 3ba929a | 2020-10-30 21:38:53 -0600 | [diff] [blame] | 26 | #include <asm/global_data.h> |
Andy Fleming | 72c23be | 2008-04-02 16:19:07 -0500 | [diff] [blame] | 27 | #include <u-boot/md5.h> |
Jeroen Hofstee | bfe88fe | 2014-06-12 22:27:12 +0200 | [diff] [blame] | 28 | #include <u-boot/sha1.h> |
Masahiro Yamada | 56a931c | 2016-09-21 11:28:55 +0900 | [diff] [blame] | 29 | #include <linux/errno.h> |
Simon Glass | 4964806 | 2013-05-07 06:12:03 +0000 | [diff] [blame] | 30 | #include <asm/io.h> |
Marian Balakowicz | a50d515 | 2008-03-12 10:12:37 +0100 | [diff] [blame] | 31 | |
Marian Balakowicz | 2efc264 | 2008-01-31 13:58:13 +0100 | [diff] [blame] | 32 | #ifdef CONFIG_CMD_BDI |
Simon Glass | ed38aef | 2020-05-10 11:40:03 -0600 | [diff] [blame] | 33 | extern int do_bdinfo(struct cmd_tbl *cmdtp, int flag, int argc, |
| 34 | char *const argv[]); |
Marian Balakowicz | 2efc264 | 2008-01-31 13:58:13 +0100 | [diff] [blame] | 35 | #endif |
| 36 | |
| 37 | DECLARE_GLOBAL_DATA_PTR; |
Marian Balakowicz | 61fde55 | 2008-02-27 11:01:04 +0100 | [diff] [blame] | 38 | |
Simon Glass | fd86d01 | 2021-09-25 07:03:14 -0600 | [diff] [blame] | 39 | /* Set this if we have less than 4 MB of malloc() space */ |
| 40 | #if CONFIG_SYS_MALLOC_LEN < (4096 * 1024) |
| 41 | #define CONSERVE_MEMORY true |
Marian Balakowicz | 41d71ed | 2008-01-08 18:14:09 +0100 | [diff] [blame] | 42 | #else |
Simon Glass | fd86d01 | 2021-09-25 07:03:14 -0600 | [diff] [blame] | 43 | #define CONSERVE_MEMORY false |
| 44 | #endif |
| 45 | |
| 46 | #else /* USE_HOSTCC */ |
Marian Balakowicz | 9541850 | 2008-01-31 13:55:39 +0100 | [diff] [blame] | 47 | #include "mkimage.h" |
Andy Fleming | 72c23be | 2008-04-02 16:19:07 -0500 | [diff] [blame] | 48 | #include <u-boot/md5.h> |
Marian Balakowicz | e227fbb | 2008-02-29 21:24:06 +0100 | [diff] [blame] | 49 | #include <time.h> |
Heiko Schocher | 62cb156 | 2015-06-29 09:10:46 +0200 | [diff] [blame] | 50 | |
| 51 | #ifndef __maybe_unused |
| 52 | # define __maybe_unused /* unimplemented */ |
| 53 | #endif |
Simon Glass | fd86d01 | 2021-09-25 07:03:14 -0600 | [diff] [blame] | 54 | |
| 55 | #define CONSERVE_MEMORY false |
| 56 | |
Marian Balakowicz | e227fbb | 2008-02-29 21:24:06 +0100 | [diff] [blame] | 57 | #endif /* !USE_HOSTCC*/ |
Marian Balakowicz | 41d71ed | 2008-01-08 18:14:09 +0100 | [diff] [blame] | 58 | |
Simon Glass | fd86d01 | 2021-09-25 07:03:14 -0600 | [diff] [blame] | 59 | #include <abuf.h> |
| 60 | #include <bzlib.h> |
Simon Glass | 4a8a8a1 | 2021-09-25 07:03:17 -0600 | [diff] [blame] | 61 | #include <display_options.h> |
Simon Glass | fd86d01 | 2021-09-25 07:03:14 -0600 | [diff] [blame] | 62 | #include <gzip.h> |
Simon Glass | 4d7237b | 2021-09-25 07:03:15 -0600 | [diff] [blame] | 63 | #include <image.h> |
Breno Matheus Lima | c592c34 | 2019-09-23 18:39:47 +0000 | [diff] [blame] | 64 | #include <imximage.h> |
Simon Glass | 684ef38 | 2021-09-25 07:03:18 -0600 | [diff] [blame] | 65 | #include <relocate.h> |
Simon Glass | fd86d01 | 2021-09-25 07:03:14 -0600 | [diff] [blame] | 66 | #include <linux/lzo.h> |
| 67 | #include <linux/zstd.h> |
| 68 | #include <linux/kconfig.h> |
| 69 | #include <lzma/LzmaTypes.h> |
| 70 | #include <lzma/LzmaDec.h> |
| 71 | #include <lzma/LzmaTools.h> |
| 72 | #include <u-boot/crc.h> |
Simon Glass | caf6267 | 2021-10-09 09:28:21 -0600 | [diff] [blame] | 73 | #include <u-boot/lz4.h> |
Simon Glass | 9adb686 | 2013-02-24 17:33:25 +0000 | [diff] [blame] | 74 | |
Mike Frysinger | cf2feb1 | 2010-10-20 07:17:39 -0400 | [diff] [blame] | 75 | static const table_entry_t uimage_arch[] = { |
Simon Glass | a3ed737 | 2016-06-30 10:52:15 -0600 | [diff] [blame] | 76 | { IH_ARCH_INVALID, "invalid", "Invalid ARCH", }, |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 77 | { IH_ARCH_ALPHA, "alpha", "Alpha", }, |
| 78 | { IH_ARCH_ARM, "arm", "ARM", }, |
| 79 | { IH_ARCH_I386, "x86", "Intel x86", }, |
| 80 | { IH_ARCH_IA64, "ia64", "IA64", }, |
| 81 | { IH_ARCH_M68K, "m68k", "M68K", }, |
| 82 | { IH_ARCH_MICROBLAZE, "microblaze", "MicroBlaze", }, |
| 83 | { IH_ARCH_MIPS, "mips", "MIPS", }, |
| 84 | { IH_ARCH_MIPS64, "mips64", "MIPS 64 Bit", }, |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 85 | { IH_ARCH_NIOS2, "nios2", "NIOS II", }, |
Grant Erickson | 80f51b8 | 2008-05-04 16:45:01 -0700 | [diff] [blame] | 86 | { IH_ARCH_PPC, "powerpc", "PowerPC", }, |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 87 | { IH_ARCH_PPC, "ppc", "PowerPC", }, |
| 88 | { IH_ARCH_S390, "s390", "IBM S390", }, |
| 89 | { IH_ARCH_SH, "sh", "SuperH", }, |
| 90 | { IH_ARCH_SPARC, "sparc", "SPARC", }, |
| 91 | { IH_ARCH_SPARC64, "sparc64", "SPARC 64 Bit", }, |
| 92 | { IH_ARCH_BLACKFIN, "blackfin", "Blackfin", }, |
| 93 | { IH_ARCH_AVR32, "avr32", "AVR32", }, |
Macpaul Lin | 354b4e3 | 2011-10-19 20:41:09 +0000 | [diff] [blame] | 94 | { IH_ARCH_NDS32, "nds32", "NDS32", }, |
Stefan Kristiansson | 15c669c | 2011-11-26 19:04:50 +0000 | [diff] [blame] | 95 | { IH_ARCH_OPENRISC, "or1k", "OpenRISC 1000",}, |
Simon Glass | 4964806 | 2013-05-07 06:12:03 +0000 | [diff] [blame] | 96 | { IH_ARCH_SANDBOX, "sandbox", "Sandbox", }, |
David Feng | 85fd5f1 | 2013-12-14 11:47:35 +0800 | [diff] [blame] | 97 | { IH_ARCH_ARM64, "arm64", "AArch64", }, |
Alexey Brodkin | 78bc86f | 2014-02-04 12:56:16 +0400 | [diff] [blame] | 98 | { IH_ARCH_ARC, "arc", "ARC", }, |
Simon Glass | 9d42830 | 2014-10-10 08:21:57 -0600 | [diff] [blame] | 99 | { IH_ARCH_X86_64, "x86_64", "AMD x86_64", }, |
Chris Zankel | 41e3737 | 2016-08-10 18:36:43 +0300 | [diff] [blame] | 100 | { IH_ARCH_XTENSA, "xtensa", "Xtensa", }, |
Rick Chen | cda2070 | 2018-03-13 13:37:29 +0800 | [diff] [blame] | 101 | { IH_ARCH_RISCV, "riscv", "RISC-V", }, |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 102 | { -1, "", "", }, |
| 103 | }; |
| 104 | |
Mike Frysinger | cf2feb1 | 2010-10-20 07:17:39 -0400 | [diff] [blame] | 105 | static const table_entry_t uimage_os[] = { |
Simon Glass | a3ed737 | 2016-06-30 10:52:15 -0600 | [diff] [blame] | 106 | { IH_OS_INVALID, "invalid", "Invalid OS", }, |
Philipp Tomsich | 1bf3eb2 | 2017-09-13 21:29:29 +0200 | [diff] [blame] | 107 | { IH_OS_ARM_TRUSTED_FIRMWARE, "arm-trusted-firmware", "ARM Trusted Firmware" }, |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 108 | { IH_OS_LINUX, "linux", "Linux", }, |
Thomas Huth | d9fe9db | 2021-10-26 14:31:18 +0200 | [diff] [blame] | 109 | #if defined(USE_HOSTCC) |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 110 | { IH_OS_LYNXOS, "lynxos", "LynxOS", }, |
| 111 | #endif |
| 112 | { IH_OS_NETBSD, "netbsd", "NetBSD", }, |
Torkel Lundgren | 29a5866 | 2010-09-28 11:05:36 +0200 | [diff] [blame] | 113 | { IH_OS_OSE, "ose", "Enea OSE", }, |
Steven Stallion | 36ec8ac | 2013-03-20 06:31:35 +0000 | [diff] [blame] | 114 | { IH_OS_PLAN9, "plan9", "Plan 9", }, |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 115 | { IH_OS_RTEMS, "rtems", "RTEMS", }, |
Bryan O'Donoghue | 2f72dec | 2018-03-13 16:50:35 +0000 | [diff] [blame] | 116 | { IH_OS_TEE, "tee", "Trusted Execution Environment" }, |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 117 | { IH_OS_U_BOOT, "u-boot", "U-Boot", }, |
miao.yan@windriver.com | 8bb6424 | 2013-12-27 16:01:50 +0800 | [diff] [blame] | 118 | { IH_OS_VXWORKS, "vxworks", "VxWorks", }, |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 119 | #if defined(CONFIG_CMD_ELF) || defined(USE_HOSTCC) |
| 120 | { IH_OS_QNX, "qnx", "QNX", }, |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 121 | #endif |
Peter Tyser | 56b8dd1 | 2008-09-08 14:56:49 -0500 | [diff] [blame] | 122 | #if defined(CONFIG_INTEGRITY) || defined(USE_HOSTCC) |
| 123 | { IH_OS_INTEGRITY,"integrity", "INTEGRITY", }, |
| 124 | #endif |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 125 | #ifdef USE_HOSTCC |
| 126 | { IH_OS_4_4BSD, "4_4bsd", "4_4BSD", }, |
| 127 | { IH_OS_DELL, "dell", "Dell", }, |
| 128 | { IH_OS_ESIX, "esix", "Esix", }, |
| 129 | { IH_OS_FREEBSD, "freebsd", "FreeBSD", }, |
| 130 | { IH_OS_IRIX, "irix", "Irix", }, |
| 131 | { IH_OS_NCR, "ncr", "NCR", }, |
| 132 | { IH_OS_OPENBSD, "openbsd", "OpenBSD", }, |
| 133 | { IH_OS_PSOS, "psos", "pSOS", }, |
| 134 | { IH_OS_SCO, "sco", "SCO", }, |
| 135 | { IH_OS_SOLARIS, "solaris", "Solaris", }, |
| 136 | { IH_OS_SVR4, "svr4", "SVR4", }, |
| 137 | #endif |
Marek Vasut | 0e3b512 | 2014-12-16 14:07:21 +0100 | [diff] [blame] | 138 | #if defined(CONFIG_BOOTM_OPENRTOS) || defined(USE_HOSTCC) |
| 139 | { IH_OS_OPENRTOS, "openrtos", "OpenRTOS", }, |
| 140 | #endif |
Lukas Auer | 515b934 | 2019-08-21 21:14:44 +0200 | [diff] [blame] | 141 | { IH_OS_OPENSBI, "opensbi", "RISC-V OpenSBI", }, |
Cristian Ciocaltea | 217f964 | 2019-12-24 18:05:38 +0200 | [diff] [blame] | 142 | { IH_OS_EFI, "efi", "EFI Firmware" }, |
Marek Vasut | 0e3b512 | 2014-12-16 14:07:21 +0100 | [diff] [blame] | 143 | |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 144 | { -1, "", "", }, |
| 145 | }; |
| 146 | |
Mike Frysinger | cf2feb1 | 2010-10-20 07:17:39 -0400 | [diff] [blame] | 147 | static const table_entry_t uimage_type[] = { |
Stefano Babic | 3885551 | 2011-10-17 00:07:43 +0000 | [diff] [blame] | 148 | { IH_TYPE_AISIMAGE, "aisimage", "Davinci AIS image",}, |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 149 | { IH_TYPE_FILESYSTEM, "filesystem", "Filesystem Image", }, |
| 150 | { IH_TYPE_FIRMWARE, "firmware", "Firmware", }, |
John Rigby | 34d12a1 | 2011-07-21 09:10:30 -0400 | [diff] [blame] | 151 | { IH_TYPE_FLATDT, "flat_dt", "Flat Device Tree", }, |
Karicheri, Muralidharan | 8277793 | 2014-04-04 13:16:48 -0400 | [diff] [blame] | 152 | { IH_TYPE_GPIMAGE, "gpimage", "TI Keystone SPL Image",}, |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 153 | { IH_TYPE_KERNEL, "kernel", "Kernel Image", }, |
Stephen Warren | 8c87eba | 2011-11-10 13:17:53 -0700 | [diff] [blame] | 154 | { IH_TYPE_KERNEL_NOLOAD, "kernel_noload", "Kernel Image (no loading done)", }, |
Stefano Babic | 3885551 | 2011-10-17 00:07:43 +0000 | [diff] [blame] | 155 | { IH_TYPE_KWBIMAGE, "kwbimage", "Kirkwood Boot Image",}, |
| 156 | { IH_TYPE_IMXIMAGE, "imximage", "Freescale i.MX Boot Image",}, |
Peng Fan | 60a5607 | 2018-10-16 04:50:30 +0000 | [diff] [blame] | 157 | { IH_TYPE_IMX8IMAGE, "imx8image", "NXP i.MX8 Boot Image",}, |
Peng Fan | 31c51da | 2018-11-20 10:19:36 +0000 | [diff] [blame] | 158 | { IH_TYPE_IMX8MIMAGE, "imx8mimage", "NXP i.MX8M Boot Image",}, |
Simon Glass | a3ed737 | 2016-06-30 10:52:15 -0600 | [diff] [blame] | 159 | { IH_TYPE_INVALID, "invalid", "Invalid Image", }, |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 160 | { IH_TYPE_MULTI, "multi", "Multi-File Image", }, |
Stefano Babic | 3885551 | 2011-10-17 00:07:43 +0000 | [diff] [blame] | 161 | { IH_TYPE_OMAPIMAGE, "omapimage", "TI OMAP SPL With GP CH",}, |
Shaohui Xie | ea65fd8 | 2012-08-10 02:49:35 +0000 | [diff] [blame] | 162 | { IH_TYPE_PBLIMAGE, "pblimage", "Freescale PBL Boot Image",}, |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 163 | { IH_TYPE_RAMDISK, "ramdisk", "RAMDisk Image", }, |
| 164 | { IH_TYPE_SCRIPT, "script", "Script", }, |
Marek Vasut | 83f6928 | 2018-04-15 13:15:33 +0200 | [diff] [blame] | 165 | { IH_TYPE_SOCFPGAIMAGE, "socfpgaimage", "Altera SoCFPGA CV/AV preloader",}, |
| 166 | { IH_TYPE_SOCFPGAIMAGE_V1, "socfpgaimage_v1", "Altera SoCFPGA A10 preloader",}, |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 167 | { IH_TYPE_STANDALONE, "standalone", "Standalone Program", }, |
Heiko Schocher | 58cb84d | 2011-07-16 00:06:42 +0000 | [diff] [blame] | 168 | { IH_TYPE_UBLIMAGE, "ublimage", "Davinci UBL image",}, |
Marek Vasut | 4a0e05d | 2013-08-26 20:43:33 +0200 | [diff] [blame] | 169 | { IH_TYPE_MXSIMAGE, "mxsimage", "Freescale MXS Boot Image",}, |
Andreas Bießmann | 7c75d3e | 2014-05-19 14:23:39 +0200 | [diff] [blame] | 170 | { IH_TYPE_ATMELIMAGE, "atmelimage", "ATMEL ROM-Boot Image",}, |
Simon Glass | 0129b52 | 2014-10-19 21:11:24 -0600 | [diff] [blame] | 171 | { IH_TYPE_X86_SETUP, "x86_setup", "x86 setup.bin", }, |
Albert ARIBAUD \(3ADEV\) | ed1cb1a | 2015-03-31 11:40:49 +0200 | [diff] [blame] | 172 | { IH_TYPE_LPC32XXIMAGE, "lpc32xximage", "LPC32XX Boot Image", }, |
Simon Glass | c9c7403 | 2015-08-30 16:55:24 -0600 | [diff] [blame] | 173 | { IH_TYPE_RKIMAGE, "rkimage", "Rockchip Boot Image" }, |
Simon Glass | 3b1e507 | 2015-08-30 16:55:25 -0600 | [diff] [blame] | 174 | { IH_TYPE_RKSD, "rksd", "Rockchip SD Boot Image" }, |
Simon Glass | c17b145 | 2015-08-30 16:55:26 -0600 | [diff] [blame] | 175 | { IH_TYPE_RKSPI, "rkspi", "Rockchip SPI Boot Image" }, |
Albert ARIBAUD \(3ADEV\) | 2e9f494 | 2016-09-26 09:08:06 +0200 | [diff] [blame] | 176 | { IH_TYPE_VYBRIDIMAGE, "vybridimage", "Vybrid Boot Image", }, |
Nathan Rossi | ce1771e | 2015-11-17 22:56:56 +1000 | [diff] [blame] | 177 | { IH_TYPE_ZYNQIMAGE, "zynqimage", "Xilinx Zynq Boot Image" }, |
Michal Simek | fa12466 | 2016-04-27 14:03:29 +0200 | [diff] [blame] | 178 | { IH_TYPE_ZYNQMPIMAGE, "zynqmpimage", "Xilinx ZynqMP Boot Image" }, |
Alexander Graf | 5329d67 | 2018-04-13 14:18:52 +0200 | [diff] [blame] | 179 | { IH_TYPE_ZYNQMPBIF, "zynqmpbif", "Xilinx ZynqMP Boot Image (bif)" }, |
Michal Simek | 0a130b1 | 2016-05-17 13:58:44 +0200 | [diff] [blame] | 180 | { IH_TYPE_FPGA, "fpga", "FPGA Image" }, |
Andrew F. Davis | 378bcff | 2016-11-29 16:33:21 -0600 | [diff] [blame] | 181 | { IH_TYPE_TEE, "tee", "Trusted Execution Environment Image",}, |
Sven Ebenfeld | 59697a2 | 2016-11-06 16:37:56 +0100 | [diff] [blame] | 182 | { IH_TYPE_FIRMWARE_IVT, "firmware_ivt", "Firmware with HABv4 IVT" }, |
Andrew F. Davis | 30f9d56 | 2017-07-31 10:58:20 -0500 | [diff] [blame] | 183 | { IH_TYPE_PMMC, "pmmc", "TI Power Management Micro-Controller Firmware",}, |
Patrick Delaunay | e6db5df | 2018-03-12 10:46:04 +0100 | [diff] [blame] | 184 | { IH_TYPE_STM32IMAGE, "stm32image", "STMicroelectronics STM32 Image" }, |
developer | e1de5d4 | 2018-11-15 10:07:49 +0800 | [diff] [blame] | 185 | { IH_TYPE_MTKIMAGE, "mtk_image", "MediaTek BootROM loadable Image" }, |
Patrick Delaunay | 42594db | 2019-08-02 15:07:19 +0200 | [diff] [blame] | 186 | { IH_TYPE_COPRO, "copro", "Coprocessor Image"}, |
Andre Przywara | e53f9d9 | 2018-12-20 01:15:18 +0000 | [diff] [blame] | 187 | { IH_TYPE_SUNXI_EGON, "sunxi_egon", "Allwinner eGON Boot Image" }, |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 188 | { -1, "", "", }, |
| 189 | }; |
| 190 | |
Mike Frysinger | cf2feb1 | 2010-10-20 07:17:39 -0400 | [diff] [blame] | 191 | static const table_entry_t uimage_comp[] = { |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 192 | { IH_COMP_NONE, "none", "uncompressed", }, |
| 193 | { IH_COMP_BZIP2, "bzip2", "bzip2 compressed", }, |
| 194 | { IH_COMP_GZIP, "gzip", "gzip compressed", }, |
Luigi 'Comio' Mantellini | 35afc06 | 2008-09-08 02:46:13 +0200 | [diff] [blame] | 195 | { IH_COMP_LZMA, "lzma", "lzma compressed", }, |
Peter Korsgaard | f7440cd | 2009-11-19 11:37:51 +0100 | [diff] [blame] | 196 | { IH_COMP_LZO, "lzo", "lzo compressed", }, |
Julius Werner | f41a3ca | 2015-10-06 20:03:53 -0700 | [diff] [blame] | 197 | { IH_COMP_LZ4, "lz4", "lz4 compressed", }, |
Robert Marko | d1c2484 | 2020-04-25 19:37:21 +0200 | [diff] [blame] | 198 | { IH_COMP_ZSTD, "zstd", "zstd compressed", }, |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 199 | { -1, "", "", }, |
| 200 | }; |
| 201 | |
Simon Glass | 7da41d5 | 2016-06-30 10:52:14 -0600 | [diff] [blame] | 202 | struct table_info { |
| 203 | const char *desc; |
| 204 | int count; |
| 205 | const table_entry_t *table; |
| 206 | }; |
| 207 | |
Atish Patra | 04e8cd8 | 2020-03-05 16:24:22 -0800 | [diff] [blame] | 208 | static const struct comp_magic_map image_comp[] = { |
| 209 | { IH_COMP_BZIP2, "bzip2", {0x42, 0x5a},}, |
| 210 | { IH_COMP_GZIP, "gzip", {0x1f, 0x8b},}, |
| 211 | { IH_COMP_LZMA, "lzma", {0x5d, 0x00},}, |
| 212 | { IH_COMP_LZO, "lzo", {0x89, 0x4c},}, |
Artem Lapkin | c81801d | 2021-08-31 18:22:18 +0800 | [diff] [blame] | 213 | { IH_COMP_LZ4, "lz4", {0x04, 0x22},}, |
| 214 | { IH_COMP_ZSTD, "zstd", {0x28, 0xb5},}, |
Atish Patra | 04e8cd8 | 2020-03-05 16:24:22 -0800 | [diff] [blame] | 215 | { IH_COMP_NONE, "none", {}, }, |
| 216 | }; |
| 217 | |
Simon Glass | 7da41d5 | 2016-06-30 10:52:14 -0600 | [diff] [blame] | 218 | static const struct table_info table_info[IH_COUNT] = { |
| 219 | { "architecture", IH_ARCH_COUNT, uimage_arch }, |
| 220 | { "compression", IH_COMP_COUNT, uimage_comp }, |
| 221 | { "operating system", IH_OS_COUNT, uimage_os }, |
| 222 | { "image type", IH_TYPE_COUNT, uimage_type }, |
| 223 | }; |
| 224 | |
Marian Balakowicz | d7c88a4 | 2008-02-29 14:58:34 +0100 | [diff] [blame] | 225 | /*****************************************************************************/ |
| 226 | /* Legacy format routines */ |
| 227 | /*****************************************************************************/ |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 228 | int image_check_hcrc(const image_header_t *hdr) |
Marian Balakowicz | 41d71ed | 2008-01-08 18:14:09 +0100 | [diff] [blame] | 229 | { |
| 230 | ulong hcrc; |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 231 | ulong len = image_get_header_size(); |
Marian Balakowicz | 41d71ed | 2008-01-08 18:14:09 +0100 | [diff] [blame] | 232 | image_header_t header; |
| 233 | |
| 234 | /* Copy header so we can blank CRC field for re-calculation */ |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 235 | memmove(&header, (char *)hdr, image_get_header_size()); |
| 236 | image_set_hcrc(&header, 0); |
Marian Balakowicz | 41d71ed | 2008-01-08 18:14:09 +0100 | [diff] [blame] | 237 | |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 238 | hcrc = crc32(0, (unsigned char *)&header, len); |
Marian Balakowicz | 41d71ed | 2008-01-08 18:14:09 +0100 | [diff] [blame] | 239 | |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 240 | return (hcrc == image_get_hcrc(hdr)); |
Marian Balakowicz | 41d71ed | 2008-01-08 18:14:09 +0100 | [diff] [blame] | 241 | } |
| 242 | |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 243 | int image_check_dcrc(const image_header_t *hdr) |
Marian Balakowicz | 41d71ed | 2008-01-08 18:14:09 +0100 | [diff] [blame] | 244 | { |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 245 | ulong data = image_get_data(hdr); |
| 246 | ulong len = image_get_data_size(hdr); |
| 247 | ulong dcrc = crc32_wd(0, (unsigned char *)data, len, CHUNKSZ_CRC32); |
Marian Balakowicz | 41d71ed | 2008-01-08 18:14:09 +0100 | [diff] [blame] | 248 | |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 249 | return (dcrc == image_get_dcrc(hdr)); |
Marian Balakowicz | 41d71ed | 2008-01-08 18:14:09 +0100 | [diff] [blame] | 250 | } |
| 251 | |
Marian Balakowicz | b4a12a9 | 2008-01-08 18:12:17 +0100 | [diff] [blame] | 252 | /** |
| 253 | * image_multi_count - get component (sub-image) count |
| 254 | * @hdr: pointer to the header of the multi component image |
| 255 | * |
| 256 | * image_multi_count() returns number of components in a multi |
| 257 | * component image. |
| 258 | * |
| 259 | * Note: no checking of the image type is done, caller must pass |
| 260 | * a valid multi component image. |
| 261 | * |
| 262 | * returns: |
| 263 | * number of components |
| 264 | */ |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 265 | ulong image_multi_count(const image_header_t *hdr) |
Marian Balakowicz | b4a12a9 | 2008-01-08 18:12:17 +0100 | [diff] [blame] | 266 | { |
| 267 | ulong i, count = 0; |
Marian Balakowicz | a147421 | 2008-02-29 16:00:06 +0100 | [diff] [blame] | 268 | uint32_t *size; |
Marian Balakowicz | b4a12a9 | 2008-01-08 18:12:17 +0100 | [diff] [blame] | 269 | |
| 270 | /* get start of the image payload, which in case of multi |
| 271 | * component images that points to a table of component sizes */ |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 272 | size = (uint32_t *)image_get_data(hdr); |
Marian Balakowicz | b4a12a9 | 2008-01-08 18:12:17 +0100 | [diff] [blame] | 273 | |
| 274 | /* count non empty slots */ |
| 275 | for (i = 0; size[i]; ++i) |
| 276 | count++; |
| 277 | |
| 278 | return count; |
| 279 | } |
| 280 | |
| 281 | /** |
| 282 | * image_multi_getimg - get component data address and size |
| 283 | * @hdr: pointer to the header of the multi component image |
| 284 | * @idx: index of the requested component |
| 285 | * @data: pointer to a ulong variable, will hold component data address |
| 286 | * @len: pointer to a ulong variable, will hold component size |
| 287 | * |
| 288 | * image_multi_getimg() returns size and data address for the requested |
| 289 | * component in a multi component image. |
| 290 | * |
| 291 | * Note: no checking of the image type is done, caller must pass |
| 292 | * a valid multi component image. |
| 293 | * |
| 294 | * returns: |
| 295 | * data address and size of the component, if idx is valid |
| 296 | * 0 in data and len, if idx is out of range |
| 297 | */ |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 298 | void image_multi_getimg(const image_header_t *hdr, ulong idx, |
Marian Balakowicz | b4a12a9 | 2008-01-08 18:12:17 +0100 | [diff] [blame] | 299 | ulong *data, ulong *len) |
| 300 | { |
| 301 | int i; |
Marian Balakowicz | a147421 | 2008-02-29 16:00:06 +0100 | [diff] [blame] | 302 | uint32_t *size; |
Nick Spence | 143a159 | 2008-05-10 14:02:04 -0700 | [diff] [blame] | 303 | ulong offset, count, img_data; |
Marian Balakowicz | b4a12a9 | 2008-01-08 18:12:17 +0100 | [diff] [blame] | 304 | |
| 305 | /* get number of component */ |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 306 | count = image_multi_count(hdr); |
Marian Balakowicz | b4a12a9 | 2008-01-08 18:12:17 +0100 | [diff] [blame] | 307 | |
| 308 | /* get start of the image payload, which in case of multi |
| 309 | * component images that points to a table of component sizes */ |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 310 | size = (uint32_t *)image_get_data(hdr); |
Marian Balakowicz | b4a12a9 | 2008-01-08 18:12:17 +0100 | [diff] [blame] | 311 | |
| 312 | /* get address of the proper component data start, which means |
| 313 | * skipping sizes table (add 1 for last, null entry) */ |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 314 | img_data = image_get_data(hdr) + (count + 1) * sizeof(uint32_t); |
Marian Balakowicz | b4a12a9 | 2008-01-08 18:12:17 +0100 | [diff] [blame] | 315 | |
| 316 | if (idx < count) { |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 317 | *len = uimage_to_cpu(size[idx]); |
Marian Balakowicz | b4a12a9 | 2008-01-08 18:12:17 +0100 | [diff] [blame] | 318 | offset = 0; |
Marian Balakowicz | b4a12a9 | 2008-01-08 18:12:17 +0100 | [diff] [blame] | 319 | |
| 320 | /* go over all indices preceding requested component idx */ |
| 321 | for (i = 0; i < idx; i++) { |
Nick Spence | 143a159 | 2008-05-10 14:02:04 -0700 | [diff] [blame] | 322 | /* add up i-th component size, rounding up to 4 bytes */ |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 323 | offset += (uimage_to_cpu(size[i]) + 3) & ~3 ; |
Marian Balakowicz | b4a12a9 | 2008-01-08 18:12:17 +0100 | [diff] [blame] | 324 | } |
| 325 | |
| 326 | /* calculate idx-th component data address */ |
Nick Spence | 143a159 | 2008-05-10 14:02:04 -0700 | [diff] [blame] | 327 | *data = img_data + offset; |
Marian Balakowicz | b4a12a9 | 2008-01-08 18:12:17 +0100 | [diff] [blame] | 328 | } else { |
| 329 | *len = 0; |
| 330 | *data = 0; |
| 331 | } |
| 332 | } |
Marian Balakowicz | 05c1d3f | 2008-01-31 13:20:07 +0100 | [diff] [blame] | 333 | |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 334 | static void image_print_type(const image_header_t *hdr) |
Marian Balakowicz | d7c88a4 | 2008-02-29 14:58:34 +0100 | [diff] [blame] | 335 | { |
Heiko Schocher | 62cb156 | 2015-06-29 09:10:46 +0200 | [diff] [blame] | 336 | const char __maybe_unused *os, *arch, *type, *comp; |
Marian Balakowicz | d7c88a4 | 2008-02-29 14:58:34 +0100 | [diff] [blame] | 337 | |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 338 | os = genimg_get_os_name(image_get_os(hdr)); |
| 339 | arch = genimg_get_arch_name(image_get_arch(hdr)); |
| 340 | type = genimg_get_type_name(image_get_type(hdr)); |
| 341 | comp = genimg_get_comp_name(image_get_comp(hdr)); |
Marian Balakowicz | d7c88a4 | 2008-02-29 14:58:34 +0100 | [diff] [blame] | 342 | |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 343 | printf("%s %s %s (%s)\n", arch, os, type, comp); |
Marian Balakowicz | d7c88a4 | 2008-02-29 14:58:34 +0100 | [diff] [blame] | 344 | } |
| 345 | |
Marian Balakowicz | e227fbb | 2008-02-29 21:24:06 +0100 | [diff] [blame] | 346 | /** |
Bartlomiej Sieka | 4786859 | 2008-04-18 12:39:23 +0200 | [diff] [blame] | 347 | * image_print_contents - prints out the contents of the legacy format image |
Wolfgang Denk | 3b50677 | 2009-08-19 11:42:56 +0200 | [diff] [blame] | 348 | * @ptr: pointer to the legacy format image header |
Marian Balakowicz | e227fbb | 2008-02-29 21:24:06 +0100 | [diff] [blame] | 349 | * @p: pointer to prefix string |
| 350 | * |
Bartlomiej Sieka | 4786859 | 2008-04-18 12:39:23 +0200 | [diff] [blame] | 351 | * image_print_contents() formats a multi line legacy image contents description. |
Marian Balakowicz | e227fbb | 2008-02-29 21:24:06 +0100 | [diff] [blame] | 352 | * The routine prints out all header fields followed by the size/offset data |
| 353 | * for MULTI/SCRIPT images. |
| 354 | * |
| 355 | * returns: |
| 356 | * no returned results |
| 357 | */ |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 358 | void image_print_contents(const void *ptr) |
Marian Balakowicz | d7c88a4 | 2008-02-29 14:58:34 +0100 | [diff] [blame] | 359 | { |
Wolfgang Denk | 3b50677 | 2009-08-19 11:42:56 +0200 | [diff] [blame] | 360 | const image_header_t *hdr = (const image_header_t *)ptr; |
Heiko Schocher | 62cb156 | 2015-06-29 09:10:46 +0200 | [diff] [blame] | 361 | const char __maybe_unused *p; |
Bartlomiej Sieka | 4786859 | 2008-04-18 12:39:23 +0200 | [diff] [blame] | 362 | |
Simon Glass | 1030f16 | 2013-05-08 08:05:58 +0000 | [diff] [blame] | 363 | p = IMAGE_INDENT_STRING; |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 364 | printf("%sImage Name: %.*s\n", p, IH_NMLEN, image_get_name(hdr)); |
Simon Glass | af0ec0d | 2013-05-07 06:11:51 +0000 | [diff] [blame] | 365 | if (IMAGE_ENABLE_TIMESTAMP) { |
| 366 | printf("%sCreated: ", p); |
| 367 | genimg_print_time((time_t)image_get_time(hdr)); |
| 368 | } |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 369 | printf("%sImage Type: ", p); |
| 370 | image_print_type(hdr); |
| 371 | printf("%sData Size: ", p); |
| 372 | genimg_print_size(image_get_data_size(hdr)); |
| 373 | printf("%sLoad Address: %08x\n", p, image_get_load(hdr)); |
| 374 | printf("%sEntry Point: %08x\n", p, image_get_ep(hdr)); |
Marian Balakowicz | d7c88a4 | 2008-02-29 14:58:34 +0100 | [diff] [blame] | 375 | |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 376 | if (image_check_type(hdr, IH_TYPE_MULTI) || |
| 377 | image_check_type(hdr, IH_TYPE_SCRIPT)) { |
Marian Balakowicz | d7c88a4 | 2008-02-29 14:58:34 +0100 | [diff] [blame] | 378 | int i; |
| 379 | ulong data, len; |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 380 | ulong count = image_multi_count(hdr); |
Marian Balakowicz | d7c88a4 | 2008-02-29 14:58:34 +0100 | [diff] [blame] | 381 | |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 382 | printf("%sContents:\n", p); |
Marian Balakowicz | d7c88a4 | 2008-02-29 14:58:34 +0100 | [diff] [blame] | 383 | for (i = 0; i < count; i++) { |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 384 | image_multi_getimg(hdr, i, &data, &len); |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 385 | |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 386 | printf("%s Image %d: ", p, i); |
| 387 | genimg_print_size(len); |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 388 | |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 389 | if (image_check_type(hdr, IH_TYPE_SCRIPT) && i > 0) { |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 390 | /* |
| 391 | * the user may need to know offsets |
| 392 | * if planning to do something with |
| 393 | * multiple files |
| 394 | */ |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 395 | printf("%s Offset = 0x%08lx\n", p, data); |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 396 | } |
Marian Balakowicz | d7c88a4 | 2008-02-29 14:58:34 +0100 | [diff] [blame] | 397 | } |
Sven Ebenfeld | 59697a2 | 2016-11-06 16:37:56 +0100 | [diff] [blame] | 398 | } else if (image_check_type(hdr, IH_TYPE_FIRMWARE_IVT)) { |
| 399 | printf("HAB Blocks: 0x%08x 0x0000 0x%08x\n", |
Breno Matheus Lima | c592c34 | 2019-09-23 18:39:47 +0000 | [diff] [blame] | 400 | image_get_load(hdr) - image_get_header_size(), |
| 401 | (int)(image_get_size(hdr) + image_get_header_size() |
| 402 | + sizeof(flash_header_v2_t) - 0x2060)); |
Marian Balakowicz | d7c88a4 | 2008-02-29 14:58:34 +0100 | [diff] [blame] | 403 | } |
| 404 | } |
| 405 | |
Julius Werner | 47ef986 | 2019-07-24 19:37:54 -0700 | [diff] [blame] | 406 | /** |
| 407 | * print_decomp_msg() - Print a suitable decompression/loading message |
| 408 | * |
| 409 | * @type: OS type (IH_OS_...) |
| 410 | * @comp_type: Compression type being used (IH_COMP_...) |
| 411 | * @is_xip: true if the load address matches the image start |
| 412 | */ |
| 413 | static void print_decomp_msg(int comp_type, int type, bool is_xip) |
| 414 | { |
| 415 | const char *name = genimg_get_type_name(type); |
| 416 | |
| 417 | if (comp_type == IH_COMP_NONE) |
| 418 | printf(" %s %s\n", is_xip ? "XIP" : "Loading", name); |
| 419 | else |
| 420 | printf(" Uncompressing %s\n", name); |
| 421 | } |
| 422 | |
Atish Patra | 04e8cd8 | 2020-03-05 16:24:22 -0800 | [diff] [blame] | 423 | int image_decomp_type(const unsigned char *buf, ulong len) |
| 424 | { |
| 425 | const struct comp_magic_map *cmagic = image_comp; |
| 426 | |
| 427 | if (len < 2) |
| 428 | return -EINVAL; |
| 429 | |
| 430 | for (; cmagic->comp_id > 0; cmagic++) { |
| 431 | if (!memcmp(buf, cmagic->magic, 2)) |
| 432 | break; |
| 433 | } |
| 434 | |
| 435 | return cmagic->comp_id; |
| 436 | } |
| 437 | |
Julius Werner | 47ef986 | 2019-07-24 19:37:54 -0700 | [diff] [blame] | 438 | int image_decomp(int comp, ulong load, ulong image_start, int type, |
| 439 | void *load_buf, void *image_buf, ulong image_len, |
| 440 | uint unc_len, ulong *load_end) |
| 441 | { |
Simon Glass | eefec3d | 2021-09-25 07:03:11 -0600 | [diff] [blame] | 442 | int ret = -ENOSYS; |
Julius Werner | 47ef986 | 2019-07-24 19:37:54 -0700 | [diff] [blame] | 443 | |
| 444 | *load_end = load; |
| 445 | print_decomp_msg(comp, type, load == image_start); |
| 446 | |
| 447 | /* |
| 448 | * Load the image to the right place, decompressing if needed. After |
| 449 | * this, image_len will be set to the number of uncompressed bytes |
| 450 | * loaded, ret will be non-zero on error. |
| 451 | */ |
| 452 | switch (comp) { |
| 453 | case IH_COMP_NONE: |
Simon Glass | eefec3d | 2021-09-25 07:03:11 -0600 | [diff] [blame] | 454 | ret = 0; |
Julius Werner | 47ef986 | 2019-07-24 19:37:54 -0700 | [diff] [blame] | 455 | if (load == image_start) |
| 456 | break; |
| 457 | if (image_len <= unc_len) |
| 458 | memmove_wd(load_buf, image_buf, image_len, CHUNKSZ); |
| 459 | else |
| 460 | ret = -ENOSPC; |
| 461 | break; |
Simon Glass | fd86d01 | 2021-09-25 07:03:14 -0600 | [diff] [blame] | 462 | case IH_COMP_GZIP: |
Simon Glass | db503fb | 2021-09-25 19:43:14 -0600 | [diff] [blame] | 463 | if (!tools_build() && CONFIG_IS_ENABLED(GZIP)) |
Simon Glass | fd86d01 | 2021-09-25 07:03:14 -0600 | [diff] [blame] | 464 | ret = gunzip(load_buf, unc_len, image_buf, &image_len); |
Julius Werner | 47ef986 | 2019-07-24 19:37:54 -0700 | [diff] [blame] | 465 | break; |
Simon Glass | fd86d01 | 2021-09-25 07:03:14 -0600 | [diff] [blame] | 466 | case IH_COMP_BZIP2: |
Simon Glass | db503fb | 2021-09-25 19:43:14 -0600 | [diff] [blame] | 467 | if (!tools_build() && CONFIG_IS_ENABLED(BZIP2)) { |
Simon Glass | fd86d01 | 2021-09-25 07:03:14 -0600 | [diff] [blame] | 468 | uint size = unc_len; |
Julius Werner | 47ef986 | 2019-07-24 19:37:54 -0700 | [diff] [blame] | 469 | |
Simon Glass | fd86d01 | 2021-09-25 07:03:14 -0600 | [diff] [blame] | 470 | /* |
| 471 | * If we've got less than 4 MB of malloc() space, |
| 472 | * use slower decompression algorithm which requires |
| 473 | * at most 2300 KB of memory. |
| 474 | */ |
| 475 | ret = BZ2_bzBuffToBuffDecompress(load_buf, &size, |
| 476 | image_buf, image_len, CONSERVE_MEMORY, 0); |
| 477 | image_len = size; |
| 478 | } |
Julius Werner | 47ef986 | 2019-07-24 19:37:54 -0700 | [diff] [blame] | 479 | break; |
Simon Glass | fd86d01 | 2021-09-25 07:03:14 -0600 | [diff] [blame] | 480 | case IH_COMP_LZMA: |
Simon Glass | db503fb | 2021-09-25 19:43:14 -0600 | [diff] [blame] | 481 | if (!tools_build() && CONFIG_IS_ENABLED(LZMA)) { |
Simon Glass | fd86d01 | 2021-09-25 07:03:14 -0600 | [diff] [blame] | 482 | SizeT lzma_len = unc_len; |
Julius Werner | 47ef986 | 2019-07-24 19:37:54 -0700 | [diff] [blame] | 483 | |
Simon Glass | fd86d01 | 2021-09-25 07:03:14 -0600 | [diff] [blame] | 484 | ret = lzmaBuffToBuffDecompress(load_buf, &lzma_len, |
| 485 | image_buf, image_len); |
| 486 | image_len = lzma_len; |
| 487 | } |
Julius Werner | 47ef986 | 2019-07-24 19:37:54 -0700 | [diff] [blame] | 488 | break; |
Simon Glass | fd86d01 | 2021-09-25 07:03:14 -0600 | [diff] [blame] | 489 | case IH_COMP_LZO: |
Simon Glass | db503fb | 2021-09-25 19:43:14 -0600 | [diff] [blame] | 490 | if (!tools_build() && CONFIG_IS_ENABLED(LZO)) { |
Simon Glass | fd86d01 | 2021-09-25 07:03:14 -0600 | [diff] [blame] | 491 | size_t size = unc_len; |
Julius Werner | 47ef986 | 2019-07-24 19:37:54 -0700 | [diff] [blame] | 492 | |
Simon Glass | fd86d01 | 2021-09-25 07:03:14 -0600 | [diff] [blame] | 493 | ret = lzop_decompress(image_buf, image_len, load_buf, &size); |
| 494 | image_len = size; |
| 495 | } |
Julius Werner | 47ef986 | 2019-07-24 19:37:54 -0700 | [diff] [blame] | 496 | break; |
Simon Glass | fd86d01 | 2021-09-25 07:03:14 -0600 | [diff] [blame] | 497 | case IH_COMP_LZ4: |
Simon Glass | db503fb | 2021-09-25 19:43:14 -0600 | [diff] [blame] | 498 | if (!tools_build() && CONFIG_IS_ENABLED(LZ4)) { |
Simon Glass | fd86d01 | 2021-09-25 07:03:14 -0600 | [diff] [blame] | 499 | size_t size = unc_len; |
Julius Werner | 47ef986 | 2019-07-24 19:37:54 -0700 | [diff] [blame] | 500 | |
Simon Glass | fd86d01 | 2021-09-25 07:03:14 -0600 | [diff] [blame] | 501 | ret = ulz4fn(image_buf, image_len, load_buf, &size); |
| 502 | image_len = size; |
| 503 | } |
Julius Werner | 47ef986 | 2019-07-24 19:37:54 -0700 | [diff] [blame] | 504 | break; |
Simon Glass | fd86d01 | 2021-09-25 07:03:14 -0600 | [diff] [blame] | 505 | case IH_COMP_ZSTD: |
Simon Glass | db503fb | 2021-09-25 19:43:14 -0600 | [diff] [blame] | 506 | if (!tools_build() && CONFIG_IS_ENABLED(ZSTD)) { |
Simon Glass | fd86d01 | 2021-09-25 07:03:14 -0600 | [diff] [blame] | 507 | struct abuf in, out; |
Robert Marko | d1c2484 | 2020-04-25 19:37:21 +0200 | [diff] [blame] | 508 | |
Simon Glass | fd86d01 | 2021-09-25 07:03:14 -0600 | [diff] [blame] | 509 | abuf_init_set(&in, image_buf, image_len); |
| 510 | abuf_init_set(&in, load_buf, unc_len); |
| 511 | ret = zstd_decompress(&in, &out); |
| 512 | if (ret >= 0) { |
| 513 | image_len = ret; |
| 514 | ret = 0; |
| 515 | } |
Robert Marko | d1c2484 | 2020-04-25 19:37:21 +0200 | [diff] [blame] | 516 | } |
Robert Marko | d1c2484 | 2020-04-25 19:37:21 +0200 | [diff] [blame] | 517 | break; |
| 518 | } |
Simon Glass | eefec3d | 2021-09-25 07:03:11 -0600 | [diff] [blame] | 519 | if (ret == -ENOSYS) { |
Julius Werner | 47ef986 | 2019-07-24 19:37:54 -0700 | [diff] [blame] | 520 | printf("Unimplemented compression type %d\n", comp); |
Simon Glass | eefec3d | 2021-09-25 07:03:11 -0600 | [diff] [blame] | 521 | return ret; |
Julius Werner | 47ef986 | 2019-07-24 19:37:54 -0700 | [diff] [blame] | 522 | } |
Simon Glass | eefec3d | 2021-09-25 07:03:11 -0600 | [diff] [blame] | 523 | if (ret) |
| 524 | return ret; |
Julius Werner | 47ef986 | 2019-07-24 19:37:54 -0700 | [diff] [blame] | 525 | |
| 526 | *load_end = load + image_len; |
| 527 | |
Simon Glass | eefec3d | 2021-09-25 07:03:11 -0600 | [diff] [blame] | 528 | return 0; |
Julius Werner | 47ef986 | 2019-07-24 19:37:54 -0700 | [diff] [blame] | 529 | } |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 530 | |
Simon Glass | 434a31b | 2015-06-23 15:38:25 -0600 | [diff] [blame] | 531 | const table_entry_t *get_table_entry(const table_entry_t *table, int id) |
| 532 | { |
| 533 | for (; table->id >= 0; ++table) { |
| 534 | if (table->id == id) |
| 535 | return table; |
| 536 | } |
| 537 | return NULL; |
| 538 | } |
| 539 | |
Simon Glass | 1986433 | 2016-06-30 10:52:16 -0600 | [diff] [blame] | 540 | static const char *unknown_msg(enum ih_category category) |
| 541 | { |
Simon Glass | ddb2607 | 2016-10-31 10:21:09 -0600 | [diff] [blame] | 542 | static const char unknown_str[] = "Unknown "; |
Simon Glass | 1986433 | 2016-06-30 10:52:16 -0600 | [diff] [blame] | 543 | static char msg[30]; |
| 544 | |
Simon Glass | ddb2607 | 2016-10-31 10:21:09 -0600 | [diff] [blame] | 545 | strcpy(msg, unknown_str); |
| 546 | strncat(msg, table_info[category].desc, |
| 547 | sizeof(msg) - sizeof(unknown_str)); |
Simon Glass | 1986433 | 2016-06-30 10:52:16 -0600 | [diff] [blame] | 548 | |
| 549 | return msg; |
| 550 | } |
| 551 | |
| 552 | /** |
Naoki Hayama | 9f1622f | 2020-10-07 11:22:24 +0900 | [diff] [blame] | 553 | * genimg_get_cat_name - translate entry id to long name |
Simon Glass | 1986433 | 2016-06-30 10:52:16 -0600 | [diff] [blame] | 554 | * @category: category to look up (enum ih_category) |
| 555 | * @id: entry id to be translated |
| 556 | * |
| 557 | * This will scan the translation table trying to find the entry that matches |
| 558 | * the given id. |
| 559 | * |
Naoki Hayama | 9f1622f | 2020-10-07 11:22:24 +0900 | [diff] [blame] | 560 | * @return long entry name if translation succeeds; error string on failure |
Simon Glass | 1986433 | 2016-06-30 10:52:16 -0600 | [diff] [blame] | 561 | */ |
| 562 | const char *genimg_get_cat_name(enum ih_category category, uint id) |
| 563 | { |
| 564 | const table_entry_t *entry; |
| 565 | |
| 566 | entry = get_table_entry(table_info[category].table, id); |
| 567 | if (!entry) |
| 568 | return unknown_msg(category); |
Simon Glass | 684ef38 | 2021-09-25 07:03:18 -0600 | [diff] [blame] | 569 | return manual_reloc(entry->lname); |
Simon Glass | 1986433 | 2016-06-30 10:52:16 -0600 | [diff] [blame] | 570 | } |
| 571 | |
| 572 | /** |
Naoki Hayama | 9f1622f | 2020-10-07 11:22:24 +0900 | [diff] [blame] | 573 | * genimg_get_cat_short_name - translate entry id to short name |
Simon Glass | 1986433 | 2016-06-30 10:52:16 -0600 | [diff] [blame] | 574 | * @category: category to look up (enum ih_category) |
| 575 | * @id: entry id to be translated |
| 576 | * |
| 577 | * This will scan the translation table trying to find the entry that matches |
| 578 | * the given id. |
| 579 | * |
Naoki Hayama | 9f1622f | 2020-10-07 11:22:24 +0900 | [diff] [blame] | 580 | * @return short entry name if translation succeeds; error string on failure |
Simon Glass | 1986433 | 2016-06-30 10:52:16 -0600 | [diff] [blame] | 581 | */ |
| 582 | const char *genimg_get_cat_short_name(enum ih_category category, uint id) |
| 583 | { |
| 584 | const table_entry_t *entry; |
| 585 | |
| 586 | entry = get_table_entry(table_info[category].table, id); |
| 587 | if (!entry) |
| 588 | return unknown_msg(category); |
Simon Glass | 684ef38 | 2021-09-25 07:03:18 -0600 | [diff] [blame] | 589 | return manual_reloc(entry->sname); |
Simon Glass | 1986433 | 2016-06-30 10:52:16 -0600 | [diff] [blame] | 590 | } |
| 591 | |
| 592 | int genimg_get_cat_count(enum ih_category category) |
| 593 | { |
| 594 | return table_info[category].count; |
| 595 | } |
| 596 | |
| 597 | const char *genimg_get_cat_desc(enum ih_category category) |
| 598 | { |
| 599 | return table_info[category].desc; |
| 600 | } |
| 601 | |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 602 | /** |
Naoki Hayama | d7e8291 | 2020-10-07 11:21:25 +0900 | [diff] [blame] | 603 | * genimg_cat_has_id - check whether category has entry id |
| 604 | * @category: category to look up (enum ih_category) |
| 605 | * @id: entry id to be checked |
| 606 | * |
| 607 | * This will scan the translation table trying to find the entry that matches |
| 608 | * the given id. |
| 609 | * |
| 610 | * @return true if category has entry id; false if not |
| 611 | */ |
| 612 | bool genimg_cat_has_id(enum ih_category category, uint id) |
| 613 | { |
| 614 | if (get_table_entry(table_info[category].table, id)) |
| 615 | return true; |
| 616 | |
| 617 | return false; |
| 618 | } |
| 619 | |
| 620 | /** |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 621 | * get_table_entry_name - translate entry id to long name |
| 622 | * @table: pointer to a translation table for entries of a specific type |
| 623 | * @msg: message to be returned when translation fails |
| 624 | * @id: entry id to be translated |
| 625 | * |
| 626 | * get_table_entry_name() will go over translation table trying to find |
| 627 | * entry that matches given id. If matching entry is found, its long |
| 628 | * name is returned to the caller. |
| 629 | * |
| 630 | * returns: |
| 631 | * long entry name if translation succeeds |
| 632 | * msg otherwise |
| 633 | */ |
Mike Frysinger | cf2feb1 | 2010-10-20 07:17:39 -0400 | [diff] [blame] | 634 | char *get_table_entry_name(const table_entry_t *table, char *msg, int id) |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 635 | { |
Simon Glass | 434a31b | 2015-06-23 15:38:25 -0600 | [diff] [blame] | 636 | table = get_table_entry(table, id); |
| 637 | if (!table) |
| 638 | return msg; |
Simon Glass | 684ef38 | 2021-09-25 07:03:18 -0600 | [diff] [blame] | 639 | return manual_reloc(table->lname); |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 640 | } |
Marian Balakowicz | 05c1d3f | 2008-01-31 13:20:07 +0100 | [diff] [blame] | 641 | |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 642 | const char *genimg_get_os_name(uint8_t os) |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 643 | { |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 644 | return (get_table_entry_name(uimage_os, "Unknown OS", os)); |
Marian Balakowicz | 05c1d3f | 2008-01-31 13:20:07 +0100 | [diff] [blame] | 645 | } |
| 646 | |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 647 | const char *genimg_get_arch_name(uint8_t arch) |
Marian Balakowicz | 05c1d3f | 2008-01-31 13:20:07 +0100 | [diff] [blame] | 648 | { |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 649 | return (get_table_entry_name(uimage_arch, "Unknown Architecture", |
| 650 | arch)); |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 651 | } |
Marian Balakowicz | 05c1d3f | 2008-01-31 13:20:07 +0100 | [diff] [blame] | 652 | |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 653 | const char *genimg_get_type_name(uint8_t type) |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 654 | { |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 655 | return (get_table_entry_name(uimage_type, "Unknown Image", type)); |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 656 | } |
Marian Balakowicz | 05c1d3f | 2008-01-31 13:20:07 +0100 | [diff] [blame] | 657 | |
Naoki Hayama | 9f1622f | 2020-10-07 11:22:24 +0900 | [diff] [blame] | 658 | const char *genimg_get_comp_name(uint8_t comp) |
| 659 | { |
| 660 | return (get_table_entry_name(uimage_comp, "Unknown Compression", |
| 661 | comp)); |
| 662 | } |
| 663 | |
Simon Glass | 86c362d | 2016-02-22 22:55:50 -0700 | [diff] [blame] | 664 | static const char *genimg_get_short_name(const table_entry_t *table, int val) |
Simon Glass | 434a31b | 2015-06-23 15:38:25 -0600 | [diff] [blame] | 665 | { |
Simon Glass | 86c362d | 2016-02-22 22:55:50 -0700 | [diff] [blame] | 666 | table = get_table_entry(table, val); |
Simon Glass | 434a31b | 2015-06-23 15:38:25 -0600 | [diff] [blame] | 667 | if (!table) |
| 668 | return "unknown"; |
Simon Glass | 684ef38 | 2021-09-25 07:03:18 -0600 | [diff] [blame] | 669 | return manual_reloc(table->sname); |
Simon Glass | 434a31b | 2015-06-23 15:38:25 -0600 | [diff] [blame] | 670 | } |
| 671 | |
Simon Glass | 86c362d | 2016-02-22 22:55:50 -0700 | [diff] [blame] | 672 | const char *genimg_get_type_short_name(uint8_t type) |
| 673 | { |
| 674 | return genimg_get_short_name(uimage_type, type); |
| 675 | } |
| 676 | |
Simon Glass | 86c362d | 2016-02-22 22:55:50 -0700 | [diff] [blame] | 677 | const char *genimg_get_comp_short_name(uint8_t comp) |
| 678 | { |
| 679 | return genimg_get_short_name(uimage_comp, comp); |
| 680 | } |
| 681 | |
| 682 | const char *genimg_get_os_short_name(uint8_t os) |
| 683 | { |
| 684 | return genimg_get_short_name(uimage_os, os); |
| 685 | } |
| 686 | |
| 687 | const char *genimg_get_arch_short_name(uint8_t arch) |
| 688 | { |
| 689 | return genimg_get_short_name(uimage_arch, arch); |
| 690 | } |
| 691 | |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 692 | /** |
| 693 | * get_table_entry_id - translate short entry name to id |
| 694 | * @table: pointer to a translation table for entries of a specific type |
| 695 | * @table_name: to be used in case of error |
| 696 | * @name: entry short name to be translated |
| 697 | * |
| 698 | * get_table_entry_id() will go over translation table trying to find |
| 699 | * entry that matches given short name. If matching entry is found, |
| 700 | * its id returned to the caller. |
| 701 | * |
| 702 | * returns: |
| 703 | * entry id if translation succeeds |
| 704 | * -1 otherwise |
| 705 | */ |
Mike Frysinger | cf2feb1 | 2010-10-20 07:17:39 -0400 | [diff] [blame] | 706 | int get_table_entry_id(const table_entry_t *table, |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 707 | const char *table_name, const char *name) |
Marian Balakowicz | 05c1d3f | 2008-01-31 13:20:07 +0100 | [diff] [blame] | 708 | { |
Mike Frysinger | cf2feb1 | 2010-10-20 07:17:39 -0400 | [diff] [blame] | 709 | const table_entry_t *t; |
Marian Balakowicz | 05c1d3f | 2008-01-31 13:20:07 +0100 | [diff] [blame] | 710 | |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 711 | for (t = table; t->id >= 0; ++t) { |
Simon Glass | 684ef38 | 2021-09-25 07:03:18 -0600 | [diff] [blame] | 712 | if (t->sname && !strcasecmp(manual_reloc(t->sname), name)) |
| 713 | return t->id; |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 714 | } |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 715 | debug("Invalid %s Type: %s\n", table_name, name); |
Simon Glass | 434a31b | 2015-06-23 15:38:25 -0600 | [diff] [blame] | 716 | |
| 717 | return -1; |
Marian Balakowicz | 05c1d3f | 2008-01-31 13:20:07 +0100 | [diff] [blame] | 718 | } |
| 719 | |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 720 | int genimg_get_os_id(const char *name) |
Marian Balakowicz | 05c1d3f | 2008-01-31 13:20:07 +0100 | [diff] [blame] | 721 | { |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 722 | return (get_table_entry_id(uimage_os, "OS", name)); |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 723 | } |
Marian Balakowicz | 05c1d3f | 2008-01-31 13:20:07 +0100 | [diff] [blame] | 724 | |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 725 | int genimg_get_arch_id(const char *name) |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 726 | { |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 727 | return (get_table_entry_id(uimage_arch, "CPU", name)); |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 728 | } |
Marian Balakowicz | 05c1d3f | 2008-01-31 13:20:07 +0100 | [diff] [blame] | 729 | |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 730 | int genimg_get_type_id(const char *name) |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 731 | { |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 732 | return (get_table_entry_id(uimage_type, "Image", name)); |
Marian Balakowicz | 05c1d3f | 2008-01-31 13:20:07 +0100 | [diff] [blame] | 733 | } |
Marian Balakowicz | 9541850 | 2008-01-31 13:55:39 +0100 | [diff] [blame] | 734 | |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 735 | int genimg_get_comp_id(const char *name) |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 736 | { |
Stephen Warren | 6904b6e | 2011-10-18 11:11:49 +0000 | [diff] [blame] | 737 | return (get_table_entry_id(uimage_comp, "Compression", name)); |
Marian Balakowicz | c3c7eb2 | 2008-02-29 15:59:59 +0100 | [diff] [blame] | 738 | } |