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