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