xref: /rk3399_rockchip-uboot/common/image.c (revision ed0cea7c5271cbfd37089f3a6392ba383eda06f7)
1 /*
2  * (C) Copyright 2008 Semihalf
3  *
4  * (C) Copyright 2000-2006
5  * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
6  *
7  * SPDX-License-Identifier:	GPL-2.0+
8  */
9 
10 #ifndef USE_HOSTCC
11 #include <common.h>
12 #include <watchdog.h>
13 
14 #ifdef CONFIG_SHOW_BOOT_PROGRESS
15 #include <status_led.h>
16 #endif
17 
18 #ifdef CONFIG_HAS_DATAFLASH
19 #include <dataflash.h>
20 #endif
21 
22 #ifdef CONFIG_LOGBUFFER
23 #include <logbuff.h>
24 #endif
25 
26 #include <rtc.h>
27 
28 #include <environment.h>
29 #include <image.h>
30 #include <mapmem.h>
31 
32 #if IMAGE_ENABLE_FIT || IMAGE_ENABLE_OF_LIBFDT
33 #include <libfdt.h>
34 #include <fdt_support.h>
35 #endif
36 
37 #include <u-boot/md5.h>
38 #include <u-boot/sha1.h>
39 #include <asm/errno.h>
40 #include <asm/io.h>
41 
42 #ifdef CONFIG_CMD_BDI
43 extern int do_bdinfo(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[]);
44 #endif
45 
46 DECLARE_GLOBAL_DATA_PTR;
47 
48 #if defined(CONFIG_IMAGE_FORMAT_LEGACY)
49 static const image_header_t *image_get_ramdisk(ulong rd_addr, uint8_t arch,
50 						int verify);
51 #endif
52 #else
53 #include "mkimage.h"
54 #include <u-boot/md5.h>
55 #include <time.h>
56 #include <image.h>
57 
58 #ifndef __maybe_unused
59 # define __maybe_unused		/* unimplemented */
60 #endif
61 #endif /* !USE_HOSTCC*/
62 
63 #include <u-boot/crc.h>
64 
65 #ifndef CONFIG_SYS_BARGSIZE
66 #define CONFIG_SYS_BARGSIZE 512
67 #endif
68 
69 static const table_entry_t uimage_arch[] = {
70 	{	IH_ARCH_INVALID,	NULL,		"Invalid ARCH",	},
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",	},
79 	{	IH_ARCH_NIOS2,		"nios2",	"NIOS II",	},
80 	{	IH_ARCH_PPC,		"powerpc",	"PowerPC",	},
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",	},
88 	{	IH_ARCH_NDS32,		"nds32",	"NDS32",	},
89 	{	IH_ARCH_OPENRISC,	"or1k",		"OpenRISC 1000",},
90 	{	IH_ARCH_SANDBOX,	"sandbox",	"Sandbox",	},
91 	{	IH_ARCH_ARM64,		"arm64",	"AArch64",	},
92 	{	IH_ARCH_ARC,		"arc",		"ARC",		},
93 	{	IH_ARCH_X86_64,		"x86_64",	"AMD x86_64",	},
94 	{	-1,			"",		"",		},
95 };
96 
97 static const table_entry_t uimage_os[] = {
98 	{	IH_OS_INVALID,	NULL,		"Invalid OS",		},
99 	{	IH_OS_LINUX,	"linux",	"Linux",		},
100 #if defined(CONFIG_LYNXKDI) || defined(USE_HOSTCC)
101 	{	IH_OS_LYNXOS,	"lynxos",	"LynxOS",		},
102 #endif
103 	{	IH_OS_NETBSD,	"netbsd",	"NetBSD",		},
104 	{	IH_OS_OSE,	"ose",		"Enea OSE",		},
105 	{	IH_OS_PLAN9,	"plan9",	"Plan 9",		},
106 	{	IH_OS_RTEMS,	"rtems",	"RTEMS",		},
107 	{	IH_OS_U_BOOT,	"u-boot",	"U-Boot",		},
108 	{	IH_OS_VXWORKS,	"vxworks",	"VxWorks",		},
109 #if defined(CONFIG_CMD_ELF) || defined(USE_HOSTCC)
110 	{	IH_OS_QNX,	"qnx",		"QNX",			},
111 #endif
112 #if defined(CONFIG_INTEGRITY) || defined(USE_HOSTCC)
113 	{	IH_OS_INTEGRITY,"integrity",	"INTEGRITY",		},
114 #endif
115 #ifdef USE_HOSTCC
116 	{	IH_OS_4_4BSD,	"4_4bsd",	"4_4BSD",		},
117 	{	IH_OS_DELL,	"dell",		"Dell",			},
118 	{	IH_OS_ESIX,	"esix",		"Esix",			},
119 	{	IH_OS_FREEBSD,	"freebsd",	"FreeBSD",		},
120 	{	IH_OS_IRIX,	"irix",		"Irix",			},
121 	{	IH_OS_NCR,	"ncr",		"NCR",			},
122 	{	IH_OS_OPENBSD,	"openbsd",	"OpenBSD",		},
123 	{	IH_OS_PSOS,	"psos",		"pSOS",			},
124 	{	IH_OS_SCO,	"sco",		"SCO",			},
125 	{	IH_OS_SOLARIS,	"solaris",	"Solaris",		},
126 	{	IH_OS_SVR4,	"svr4",		"SVR4",			},
127 #endif
128 #if defined(CONFIG_BOOTM_OPENRTOS) || defined(USE_HOSTCC)
129 	{	IH_OS_OPENRTOS,	"openrtos",	"OpenRTOS",		},
130 #endif
131 
132 	{	-1,		"",		"",			},
133 };
134 
135 static const table_entry_t uimage_type[] = {
136 	{	IH_TYPE_AISIMAGE,   "aisimage",   "Davinci AIS image",},
137 	{	IH_TYPE_FILESYSTEM, "filesystem", "Filesystem Image",	},
138 	{	IH_TYPE_FIRMWARE,   "firmware",	  "Firmware",		},
139 	{	IH_TYPE_FLATDT,     "flat_dt",    "Flat Device Tree",	},
140 	{	IH_TYPE_GPIMAGE,    "gpimage",    "TI Keystone SPL Image",},
141 	{	IH_TYPE_KERNEL,	    "kernel",	  "Kernel Image",	},
142 	{	IH_TYPE_KERNEL_NOLOAD, "kernel_noload",  "Kernel Image (no loading done)", },
143 	{	IH_TYPE_KWBIMAGE,   "kwbimage",   "Kirkwood Boot Image",},
144 	{	IH_TYPE_IMXIMAGE,   "imximage",   "Freescale i.MX Boot Image",},
145 	{	IH_TYPE_INVALID,    NULL,	  "Invalid Image",	},
146 	{	IH_TYPE_MULTI,	    "multi",	  "Multi-File Image",	},
147 	{	IH_TYPE_OMAPIMAGE,  "omapimage",  "TI OMAP SPL With GP CH",},
148 	{	IH_TYPE_PBLIMAGE,   "pblimage",   "Freescale PBL Boot Image",},
149 	{	IH_TYPE_RAMDISK,    "ramdisk",	  "RAMDisk Image",	},
150 	{	IH_TYPE_SCRIPT,     "script",	  "Script",		},
151 	{	IH_TYPE_SOCFPGAIMAGE, "socfpgaimage", "Altera SOCFPGA preloader",},
152 	{	IH_TYPE_STANDALONE, "standalone", "Standalone Program", },
153 	{	IH_TYPE_UBLIMAGE,   "ublimage",   "Davinci UBL image",},
154 	{	IH_TYPE_MXSIMAGE,   "mxsimage",   "Freescale MXS Boot Image",},
155 	{	IH_TYPE_ATMELIMAGE, "atmelimage", "ATMEL ROM-Boot Image",},
156 	{	IH_TYPE_X86_SETUP,  "x86_setup",  "x86 setup.bin",    },
157 	{	IH_TYPE_LPC32XXIMAGE, "lpc32xximage",  "LPC32XX Boot Image", },
158 	{	IH_TYPE_RKIMAGE,    "rkimage",    "Rockchip Boot Image" },
159 	{	IH_TYPE_RKSD,       "rksd",       "Rockchip SD Boot Image" },
160 	{	IH_TYPE_RKSPI,      "rkspi",      "Rockchip SPI Boot Image" },
161 	{	IH_TYPE_ZYNQIMAGE,  "zynqimage",  "Xilinx Zynq Boot Image" },
162 	{	IH_TYPE_ZYNQMPIMAGE, "zynqmpimage", "Xilinx ZynqMP Boot Image" },
163 	{	IH_TYPE_FPGA,       "fpga",       "FPGA Image" },
164 	{	-1,		    "",		  "",			},
165 };
166 
167 static const table_entry_t uimage_comp[] = {
168 	{	IH_COMP_NONE,	"none",		"uncompressed",		},
169 	{	IH_COMP_BZIP2,	"bzip2",	"bzip2 compressed",	},
170 	{	IH_COMP_GZIP,	"gzip",		"gzip compressed",	},
171 	{	IH_COMP_LZMA,	"lzma",		"lzma compressed",	},
172 	{	IH_COMP_LZO,	"lzo",		"lzo compressed",	},
173 	{	IH_COMP_LZ4,	"lz4",		"lz4 compressed",	},
174 	{	-1,		"",		"",			},
175 };
176 
177 /*****************************************************************************/
178 /* Legacy format routines */
179 /*****************************************************************************/
180 int image_check_hcrc(const image_header_t *hdr)
181 {
182 	ulong hcrc;
183 	ulong len = image_get_header_size();
184 	image_header_t header;
185 
186 	/* Copy header so we can blank CRC field for re-calculation */
187 	memmove(&header, (char *)hdr, image_get_header_size());
188 	image_set_hcrc(&header, 0);
189 
190 	hcrc = crc32(0, (unsigned char *)&header, len);
191 
192 	return (hcrc == image_get_hcrc(hdr));
193 }
194 
195 int image_check_dcrc(const image_header_t *hdr)
196 {
197 	ulong data = image_get_data(hdr);
198 	ulong len = image_get_data_size(hdr);
199 	ulong dcrc = crc32_wd(0, (unsigned char *)data, len, CHUNKSZ_CRC32);
200 
201 	return (dcrc == image_get_dcrc(hdr));
202 }
203 
204 /**
205  * image_multi_count - get component (sub-image) count
206  * @hdr: pointer to the header of the multi component image
207  *
208  * image_multi_count() returns number of components in a multi
209  * component image.
210  *
211  * Note: no checking of the image type is done, caller must pass
212  * a valid multi component image.
213  *
214  * returns:
215  *     number of components
216  */
217 ulong image_multi_count(const image_header_t *hdr)
218 {
219 	ulong i, count = 0;
220 	uint32_t *size;
221 
222 	/* get start of the image payload, which in case of multi
223 	 * component images that points to a table of component sizes */
224 	size = (uint32_t *)image_get_data(hdr);
225 
226 	/* count non empty slots */
227 	for (i = 0; size[i]; ++i)
228 		count++;
229 
230 	return count;
231 }
232 
233 /**
234  * image_multi_getimg - get component data address and size
235  * @hdr: pointer to the header of the multi component image
236  * @idx: index of the requested component
237  * @data: pointer to a ulong variable, will hold component data address
238  * @len: pointer to a ulong variable, will hold component size
239  *
240  * image_multi_getimg() returns size and data address for the requested
241  * component in a multi component image.
242  *
243  * Note: no checking of the image type is done, caller must pass
244  * a valid multi component image.
245  *
246  * returns:
247  *     data address and size of the component, if idx is valid
248  *     0 in data and len, if idx is out of range
249  */
250 void image_multi_getimg(const image_header_t *hdr, ulong idx,
251 			ulong *data, ulong *len)
252 {
253 	int i;
254 	uint32_t *size;
255 	ulong offset, count, img_data;
256 
257 	/* get number of component */
258 	count = image_multi_count(hdr);
259 
260 	/* get start of the image payload, which in case of multi
261 	 * component images that points to a table of component sizes */
262 	size = (uint32_t *)image_get_data(hdr);
263 
264 	/* get address of the proper component data start, which means
265 	 * skipping sizes table (add 1 for last, null entry) */
266 	img_data = image_get_data(hdr) + (count + 1) * sizeof(uint32_t);
267 
268 	if (idx < count) {
269 		*len = uimage_to_cpu(size[idx]);
270 		offset = 0;
271 
272 		/* go over all indices preceding requested component idx */
273 		for (i = 0; i < idx; i++) {
274 			/* add up i-th component size, rounding up to 4 bytes */
275 			offset += (uimage_to_cpu(size[i]) + 3) & ~3 ;
276 		}
277 
278 		/* calculate idx-th component data address */
279 		*data = img_data + offset;
280 	} else {
281 		*len = 0;
282 		*data = 0;
283 	}
284 }
285 
286 static void image_print_type(const image_header_t *hdr)
287 {
288 	const char __maybe_unused *os, *arch, *type, *comp;
289 
290 	os = genimg_get_os_name(image_get_os(hdr));
291 	arch = genimg_get_arch_name(image_get_arch(hdr));
292 	type = genimg_get_type_name(image_get_type(hdr));
293 	comp = genimg_get_comp_name(image_get_comp(hdr));
294 
295 	printf("%s %s %s (%s)\n", arch, os, type, comp);
296 }
297 
298 /**
299  * image_print_contents - prints out the contents of the legacy format image
300  * @ptr: pointer to the legacy format image header
301  * @p: pointer to prefix string
302  *
303  * image_print_contents() formats a multi line legacy image contents description.
304  * The routine prints out all header fields followed by the size/offset data
305  * for MULTI/SCRIPT images.
306  *
307  * returns:
308  *     no returned results
309  */
310 void image_print_contents(const void *ptr)
311 {
312 	const image_header_t *hdr = (const image_header_t *)ptr;
313 	const char __maybe_unused *p;
314 
315 	p = IMAGE_INDENT_STRING;
316 	printf("%sImage Name:   %.*s\n", p, IH_NMLEN, image_get_name(hdr));
317 	if (IMAGE_ENABLE_TIMESTAMP) {
318 		printf("%sCreated:      ", p);
319 		genimg_print_time((time_t)image_get_time(hdr));
320 	}
321 	printf("%sImage Type:   ", p);
322 	image_print_type(hdr);
323 	printf("%sData Size:    ", p);
324 	genimg_print_size(image_get_data_size(hdr));
325 	printf("%sLoad Address: %08x\n", p, image_get_load(hdr));
326 	printf("%sEntry Point:  %08x\n", p, image_get_ep(hdr));
327 
328 	if (image_check_type(hdr, IH_TYPE_MULTI) ||
329 			image_check_type(hdr, IH_TYPE_SCRIPT)) {
330 		int i;
331 		ulong data, len;
332 		ulong count = image_multi_count(hdr);
333 
334 		printf("%sContents:\n", p);
335 		for (i = 0; i < count; i++) {
336 			image_multi_getimg(hdr, i, &data, &len);
337 
338 			printf("%s   Image %d: ", p, i);
339 			genimg_print_size(len);
340 
341 			if (image_check_type(hdr, IH_TYPE_SCRIPT) && i > 0) {
342 				/*
343 				 * the user may need to know offsets
344 				 * if planning to do something with
345 				 * multiple files
346 				 */
347 				printf("%s    Offset = 0x%08lx\n", p, data);
348 			}
349 		}
350 	}
351 }
352 
353 
354 #ifndef USE_HOSTCC
355 #if defined(CONFIG_IMAGE_FORMAT_LEGACY)
356 /**
357  * image_get_ramdisk - get and verify ramdisk image
358  * @rd_addr: ramdisk image start address
359  * @arch: expected ramdisk architecture
360  * @verify: checksum verification flag
361  *
362  * image_get_ramdisk() returns a pointer to the verified ramdisk image
363  * header. Routine receives image start address and expected architecture
364  * flag. Verification done covers data and header integrity and os/type/arch
365  * fields checking.
366  *
367  * If dataflash support is enabled routine checks for dataflash addresses
368  * and handles required dataflash reads.
369  *
370  * returns:
371  *     pointer to a ramdisk image header, if image was found and valid
372  *     otherwise, return NULL
373  */
374 static const image_header_t *image_get_ramdisk(ulong rd_addr, uint8_t arch,
375 						int verify)
376 {
377 	const image_header_t *rd_hdr = (const image_header_t *)rd_addr;
378 
379 	if (!image_check_magic(rd_hdr)) {
380 		puts("Bad Magic Number\n");
381 		bootstage_error(BOOTSTAGE_ID_RD_MAGIC);
382 		return NULL;
383 	}
384 
385 	if (!image_check_hcrc(rd_hdr)) {
386 		puts("Bad Header Checksum\n");
387 		bootstage_error(BOOTSTAGE_ID_RD_HDR_CHECKSUM);
388 		return NULL;
389 	}
390 
391 	bootstage_mark(BOOTSTAGE_ID_RD_MAGIC);
392 	image_print_contents(rd_hdr);
393 
394 	if (verify) {
395 		puts("   Verifying Checksum ... ");
396 		if (!image_check_dcrc(rd_hdr)) {
397 			puts("Bad Data CRC\n");
398 			bootstage_error(BOOTSTAGE_ID_RD_CHECKSUM);
399 			return NULL;
400 		}
401 		puts("OK\n");
402 	}
403 
404 	bootstage_mark(BOOTSTAGE_ID_RD_HDR_CHECKSUM);
405 
406 	if (!image_check_os(rd_hdr, IH_OS_LINUX) ||
407 	    !image_check_arch(rd_hdr, arch) ||
408 	    !image_check_type(rd_hdr, IH_TYPE_RAMDISK)) {
409 		printf("No Linux %s Ramdisk Image\n",
410 				genimg_get_arch_name(arch));
411 		bootstage_error(BOOTSTAGE_ID_RAMDISK);
412 		return NULL;
413 	}
414 
415 	return rd_hdr;
416 }
417 #endif
418 #endif /* !USE_HOSTCC */
419 
420 /*****************************************************************************/
421 /* Shared dual-format routines */
422 /*****************************************************************************/
423 #ifndef USE_HOSTCC
424 ulong load_addr = CONFIG_SYS_LOAD_ADDR;	/* Default Load Address */
425 ulong save_addr;			/* Default Save Address */
426 ulong save_size;			/* Default Save Size (in bytes) */
427 
428 static int on_loadaddr(const char *name, const char *value, enum env_op op,
429 	int flags)
430 {
431 	switch (op) {
432 	case env_op_create:
433 	case env_op_overwrite:
434 		load_addr = simple_strtoul(value, NULL, 16);
435 		break;
436 	default:
437 		break;
438 	}
439 
440 	return 0;
441 }
442 U_BOOT_ENV_CALLBACK(loadaddr, on_loadaddr);
443 
444 ulong getenv_bootm_low(void)
445 {
446 	char *s = getenv("bootm_low");
447 	if (s) {
448 		ulong tmp = simple_strtoul(s, NULL, 16);
449 		return tmp;
450 	}
451 
452 #if defined(CONFIG_SYS_SDRAM_BASE)
453 	return CONFIG_SYS_SDRAM_BASE;
454 #elif defined(CONFIG_ARM)
455 	return gd->bd->bi_dram[0].start;
456 #else
457 	return 0;
458 #endif
459 }
460 
461 phys_size_t getenv_bootm_size(void)
462 {
463 	phys_size_t tmp, size;
464 	phys_addr_t start;
465 	char *s = getenv("bootm_size");
466 	if (s) {
467 		tmp = (phys_size_t)simple_strtoull(s, NULL, 16);
468 		return tmp;
469 	}
470 
471 #if defined(CONFIG_ARM) && defined(CONFIG_NR_DRAM_BANKS)
472 	start = gd->bd->bi_dram[0].start;
473 	size = gd->bd->bi_dram[0].size;
474 #else
475 	start = gd->bd->bi_memstart;
476 	size = gd->bd->bi_memsize;
477 #endif
478 
479 	s = getenv("bootm_low");
480 	if (s)
481 		tmp = (phys_size_t)simple_strtoull(s, NULL, 16);
482 	else
483 		tmp = start;
484 
485 	return size - (tmp - start);
486 }
487 
488 phys_size_t getenv_bootm_mapsize(void)
489 {
490 	phys_size_t tmp;
491 	char *s = getenv("bootm_mapsize");
492 	if (s) {
493 		tmp = (phys_size_t)simple_strtoull(s, NULL, 16);
494 		return tmp;
495 	}
496 
497 #if defined(CONFIG_SYS_BOOTMAPSZ)
498 	return CONFIG_SYS_BOOTMAPSZ;
499 #else
500 	return getenv_bootm_size();
501 #endif
502 }
503 
504 void memmove_wd(void *to, void *from, size_t len, ulong chunksz)
505 {
506 	if (to == from)
507 		return;
508 
509 #if defined(CONFIG_HW_WATCHDOG) || defined(CONFIG_WATCHDOG)
510 	if (to > from) {
511 		from += len;
512 		to += len;
513 	}
514 	while (len > 0) {
515 		size_t tail = (len > chunksz) ? chunksz : len;
516 		WATCHDOG_RESET();
517 		if (to > from) {
518 			to -= tail;
519 			from -= tail;
520 		}
521 		memmove(to, from, tail);
522 		if (to < from) {
523 			to += tail;
524 			from += tail;
525 		}
526 		len -= tail;
527 	}
528 #else	/* !(CONFIG_HW_WATCHDOG || CONFIG_WATCHDOG) */
529 	memmove(to, from, len);
530 #endif	/* CONFIG_HW_WATCHDOG || CONFIG_WATCHDOG */
531 }
532 #endif /* !USE_HOSTCC */
533 
534 void genimg_print_size(uint32_t size)
535 {
536 #ifndef USE_HOSTCC
537 	printf("%d Bytes = ", size);
538 	print_size(size, "\n");
539 #else
540 	printf("%d Bytes = %.2f kB = %.2f MB\n",
541 			size, (double)size / 1.024e3,
542 			(double)size / 1.048576e6);
543 #endif
544 }
545 
546 #if IMAGE_ENABLE_TIMESTAMP
547 void genimg_print_time(time_t timestamp)
548 {
549 #ifndef USE_HOSTCC
550 	struct rtc_time tm;
551 
552 	rtc_to_tm(timestamp, &tm);
553 	printf("%4d-%02d-%02d  %2d:%02d:%02d UTC\n",
554 			tm.tm_year, tm.tm_mon, tm.tm_mday,
555 			tm.tm_hour, tm.tm_min, tm.tm_sec);
556 #else
557 	printf("%s", ctime(&timestamp));
558 #endif
559 }
560 #endif
561 
562 const table_entry_t *get_table_entry(const table_entry_t *table, int id)
563 {
564 	for (; table->id >= 0; ++table) {
565 		if (table->id == id)
566 			return table;
567 	}
568 	return NULL;
569 }
570 
571 /**
572  * get_table_entry_name - translate entry id to long name
573  * @table: pointer to a translation table for entries of a specific type
574  * @msg: message to be returned when translation fails
575  * @id: entry id to be translated
576  *
577  * get_table_entry_name() will go over translation table trying to find
578  * entry that matches given id. If matching entry is found, its long
579  * name is returned to the caller.
580  *
581  * returns:
582  *     long entry name if translation succeeds
583  *     msg otherwise
584  */
585 char *get_table_entry_name(const table_entry_t *table, char *msg, int id)
586 {
587 	table = get_table_entry(table, id);
588 	if (!table)
589 		return msg;
590 #if defined(USE_HOSTCC) || !defined(CONFIG_NEEDS_MANUAL_RELOC)
591 	return table->lname;
592 #else
593 	return table->lname + gd->reloc_off;
594 #endif
595 }
596 
597 const char *genimg_get_os_name(uint8_t os)
598 {
599 	return (get_table_entry_name(uimage_os, "Unknown OS", os));
600 }
601 
602 const char *genimg_get_arch_name(uint8_t arch)
603 {
604 	return (get_table_entry_name(uimage_arch, "Unknown Architecture",
605 					arch));
606 }
607 
608 const char *genimg_get_type_name(uint8_t type)
609 {
610 	return (get_table_entry_name(uimage_type, "Unknown Image", type));
611 }
612 
613 static const char *genimg_get_short_name(const table_entry_t *table, int val)
614 {
615 	table = get_table_entry(table, val);
616 	if (!table)
617 		return "unknown";
618 #if defined(USE_HOSTCC) || !defined(CONFIG_NEEDS_MANUAL_RELOC)
619 	return table->sname;
620 #else
621 	return table->sname + gd->reloc_off;
622 #endif
623 }
624 
625 const char *genimg_get_type_short_name(uint8_t type)
626 {
627 	return genimg_get_short_name(uimage_type, type);
628 }
629 
630 const char *genimg_get_comp_name(uint8_t comp)
631 {
632 	return (get_table_entry_name(uimage_comp, "Unknown Compression",
633 					comp));
634 }
635 
636 const char *genimg_get_comp_short_name(uint8_t comp)
637 {
638 	return genimg_get_short_name(uimage_comp, comp);
639 }
640 
641 const char *genimg_get_os_short_name(uint8_t os)
642 {
643 	return genimg_get_short_name(uimage_os, os);
644 }
645 
646 const char *genimg_get_arch_short_name(uint8_t arch)
647 {
648 	return genimg_get_short_name(uimage_arch, arch);
649 }
650 
651 /**
652  * get_table_entry_id - translate short entry name to id
653  * @table: pointer to a translation table for entries of a specific type
654  * @table_name: to be used in case of error
655  * @name: entry short name to be translated
656  *
657  * get_table_entry_id() will go over translation table trying to find
658  * entry that matches given short name. If matching entry is found,
659  * its id returned to the caller.
660  *
661  * returns:
662  *     entry id if translation succeeds
663  *     -1 otherwise
664  */
665 int get_table_entry_id(const table_entry_t *table,
666 		const char *table_name, const char *name)
667 {
668 	const table_entry_t *t;
669 
670 	for (t = table; t->id >= 0; ++t) {
671 #ifdef CONFIG_NEEDS_MANUAL_RELOC
672 		if (t->sname && strcasecmp(t->sname + gd->reloc_off, name) == 0)
673 #else
674 		if (t->sname && strcasecmp(t->sname, name) == 0)
675 #endif
676 			return (t->id);
677 	}
678 	debug("Invalid %s Type: %s\n", table_name, name);
679 
680 	return -1;
681 }
682 
683 int genimg_get_os_id(const char *name)
684 {
685 	return (get_table_entry_id(uimage_os, "OS", name));
686 }
687 
688 int genimg_get_arch_id(const char *name)
689 {
690 	return (get_table_entry_id(uimage_arch, "CPU", name));
691 }
692 
693 int genimg_get_type_id(const char *name)
694 {
695 	return (get_table_entry_id(uimage_type, "Image", name));
696 }
697 
698 int genimg_get_comp_id(const char *name)
699 {
700 	return (get_table_entry_id(uimage_comp, "Compression", name));
701 }
702 
703 #ifndef USE_HOSTCC
704 /**
705  * genimg_get_kernel_addr_fit - get the real kernel address and return 2
706  *                              FIT strings
707  * @img_addr: a string might contain real image address
708  * @fit_uname_config: double pointer to a char, will hold pointer to a
709  *                    configuration unit name
710  * @fit_uname_kernel: double pointer to a char, will hold pointer to a subimage
711  *                    name
712  *
713  * genimg_get_kernel_addr_fit get the real kernel start address from a string
714  * which is normally the first argv of bootm/bootz
715  *
716  * returns:
717  *     kernel start address
718  */
719 ulong genimg_get_kernel_addr_fit(char * const img_addr,
720 			     const char **fit_uname_config,
721 			     const char **fit_uname_kernel)
722 {
723 	ulong kernel_addr;
724 
725 	/* find out kernel image address */
726 	if (!img_addr) {
727 		kernel_addr = load_addr;
728 		debug("*  kernel: default image load address = 0x%08lx\n",
729 		      load_addr);
730 #if CONFIG_IS_ENABLED(FIT)
731 	} else if (fit_parse_conf(img_addr, load_addr, &kernel_addr,
732 				  fit_uname_config)) {
733 		debug("*  kernel: config '%s' from image at 0x%08lx\n",
734 		      *fit_uname_config, kernel_addr);
735 	} else if (fit_parse_subimage(img_addr, load_addr, &kernel_addr,
736 				     fit_uname_kernel)) {
737 		debug("*  kernel: subimage '%s' from image at 0x%08lx\n",
738 		      *fit_uname_kernel, kernel_addr);
739 #endif
740 	} else {
741 		kernel_addr = simple_strtoul(img_addr, NULL, 16);
742 		debug("*  kernel: cmdline image address = 0x%08lx\n",
743 		      kernel_addr);
744 	}
745 
746 	return kernel_addr;
747 }
748 
749 /**
750  * genimg_get_kernel_addr() is the simple version of
751  * genimg_get_kernel_addr_fit(). It ignores those return FIT strings
752  */
753 ulong genimg_get_kernel_addr(char * const img_addr)
754 {
755 	const char *fit_uname_config = NULL;
756 	const char *fit_uname_kernel = NULL;
757 
758 	return genimg_get_kernel_addr_fit(img_addr, &fit_uname_config,
759 					  &fit_uname_kernel);
760 }
761 
762 /**
763  * genimg_get_format - get image format type
764  * @img_addr: image start address
765  *
766  * genimg_get_format() checks whether provided address points to a valid
767  * legacy or FIT image.
768  *
769  * New uImage format and FDT blob are based on a libfdt. FDT blob
770  * may be passed directly or embedded in a FIT image. In both situations
771  * genimg_get_format() must be able to dectect libfdt header.
772  *
773  * returns:
774  *     image format type or IMAGE_FORMAT_INVALID if no image is present
775  */
776 int genimg_get_format(const void *img_addr)
777 {
778 #if defined(CONFIG_IMAGE_FORMAT_LEGACY)
779 	const image_header_t *hdr;
780 
781 	hdr = (const image_header_t *)img_addr;
782 	if (image_check_magic(hdr))
783 		return IMAGE_FORMAT_LEGACY;
784 #endif
785 #if IMAGE_ENABLE_FIT || IMAGE_ENABLE_OF_LIBFDT
786 	if (fdt_check_header(img_addr) == 0)
787 		return IMAGE_FORMAT_FIT;
788 #endif
789 #ifdef CONFIG_ANDROID_BOOT_IMAGE
790 	if (android_image_check_header(img_addr) == 0)
791 		return IMAGE_FORMAT_ANDROID;
792 #endif
793 
794 	return IMAGE_FORMAT_INVALID;
795 }
796 
797 /**
798  * genimg_get_image - get image from special storage (if necessary)
799  * @img_addr: image start address
800  *
801  * genimg_get_image() checks if provided image start address is located
802  * in a dataflash storage. If so, image is moved to a system RAM memory.
803  *
804  * returns:
805  *     image start address after possible relocation from special storage
806  */
807 ulong genimg_get_image(ulong img_addr)
808 {
809 	ulong ram_addr = img_addr;
810 
811 #ifdef CONFIG_HAS_DATAFLASH
812 	ulong h_size, d_size;
813 
814 	if (addr_dataflash(img_addr)) {
815 		void *buf;
816 
817 		/* ger RAM address */
818 		ram_addr = CONFIG_SYS_LOAD_ADDR;
819 
820 		/* get header size */
821 		h_size = image_get_header_size();
822 #if IMAGE_ENABLE_FIT
823 		if (sizeof(struct fdt_header) > h_size)
824 			h_size = sizeof(struct fdt_header);
825 #endif
826 
827 		/* read in header */
828 		debug("   Reading image header from dataflash address "
829 			"%08lx to RAM address %08lx\n", img_addr, ram_addr);
830 
831 		buf = map_sysmem(ram_addr, 0);
832 		read_dataflash(img_addr, h_size, buf);
833 
834 		/* get data size */
835 		switch (genimg_get_format(buf)) {
836 #if defined(CONFIG_IMAGE_FORMAT_LEGACY)
837 		case IMAGE_FORMAT_LEGACY:
838 			d_size = image_get_data_size(buf);
839 			debug("   Legacy format image found at 0x%08lx, "
840 					"size 0x%08lx\n",
841 					ram_addr, d_size);
842 			break;
843 #endif
844 #if IMAGE_ENABLE_FIT
845 		case IMAGE_FORMAT_FIT:
846 			d_size = fit_get_size(buf) - h_size;
847 			debug("   FIT/FDT format image found at 0x%08lx, "
848 					"size 0x%08lx\n",
849 					ram_addr, d_size);
850 			break;
851 #endif
852 		default:
853 			printf("   No valid image found at 0x%08lx\n",
854 				img_addr);
855 			return ram_addr;
856 		}
857 
858 		/* read in image data */
859 		debug("   Reading image remaining data from dataflash address "
860 			"%08lx to RAM address %08lx\n", img_addr + h_size,
861 			ram_addr + h_size);
862 
863 		read_dataflash(img_addr + h_size, d_size,
864 				(char *)(buf + h_size));
865 
866 	}
867 #endif /* CONFIG_HAS_DATAFLASH */
868 
869 	return ram_addr;
870 }
871 
872 /**
873  * fit_has_config - check if there is a valid FIT configuration
874  * @images: pointer to the bootm command headers structure
875  *
876  * fit_has_config() checks if there is a FIT configuration in use
877  * (if FTI support is present).
878  *
879  * returns:
880  *     0, no FIT support or no configuration found
881  *     1, configuration found
882  */
883 int genimg_has_config(bootm_headers_t *images)
884 {
885 #if IMAGE_ENABLE_FIT
886 	if (images->fit_uname_cfg)
887 		return 1;
888 #endif
889 	return 0;
890 }
891 
892 /**
893  * boot_get_ramdisk - main ramdisk handling routine
894  * @argc: command argument count
895  * @argv: command argument list
896  * @images: pointer to the bootm images structure
897  * @arch: expected ramdisk architecture
898  * @rd_start: pointer to a ulong variable, will hold ramdisk start address
899  * @rd_end: pointer to a ulong variable, will hold ramdisk end
900  *
901  * boot_get_ramdisk() is responsible for finding a valid ramdisk image.
902  * Curently supported are the following ramdisk sources:
903  *      - multicomponent kernel/ramdisk image,
904  *      - commandline provided address of decicated ramdisk image.
905  *
906  * returns:
907  *     0, if ramdisk image was found and valid, or skiped
908  *     rd_start and rd_end are set to ramdisk start/end addresses if
909  *     ramdisk image is found and valid
910  *
911  *     1, if ramdisk image is found but corrupted, or invalid
912  *     rd_start and rd_end are set to 0 if no ramdisk exists
913  */
914 int boot_get_ramdisk(int argc, char * const argv[], bootm_headers_t *images,
915 		uint8_t arch, ulong *rd_start, ulong *rd_end)
916 {
917 	ulong rd_addr, rd_load;
918 	ulong rd_data, rd_len;
919 #if defined(CONFIG_IMAGE_FORMAT_LEGACY)
920 	const image_header_t *rd_hdr;
921 #endif
922 	void *buf;
923 #ifdef CONFIG_SUPPORT_RAW_INITRD
924 	char *end;
925 #endif
926 #if IMAGE_ENABLE_FIT
927 	const char	*fit_uname_config = images->fit_uname_cfg;
928 	const char	*fit_uname_ramdisk = NULL;
929 	ulong		default_addr;
930 	int		rd_noffset;
931 #endif
932 	const char *select = NULL;
933 
934 	*rd_start = 0;
935 	*rd_end = 0;
936 
937 #ifdef CONFIG_ANDROID_BOOT_IMAGE
938 	/*
939 	 * Look for an Android boot image.
940 	 */
941 	buf = map_sysmem(images->os.start, 0);
942 	if (buf && genimg_get_format(buf) == IMAGE_FORMAT_ANDROID)
943 		select = argv[0];
944 #endif
945 
946 	if (argc >= 2)
947 		select = argv[1];
948 
949 	/*
950 	 * Look for a '-' which indicates to ignore the
951 	 * ramdisk argument
952 	 */
953 	if (select && strcmp(select, "-") ==  0) {
954 		debug("## Skipping init Ramdisk\n");
955 		rd_len = rd_data = 0;
956 	} else if (select || genimg_has_config(images)) {
957 #if IMAGE_ENABLE_FIT
958 		if (select) {
959 			/*
960 			 * If the init ramdisk comes from the FIT image and
961 			 * the FIT image address is omitted in the command
962 			 * line argument, try to use os FIT image address or
963 			 * default load address.
964 			 */
965 			if (images->fit_uname_os)
966 				default_addr = (ulong)images->fit_hdr_os;
967 			else
968 				default_addr = load_addr;
969 
970 			if (fit_parse_conf(select, default_addr,
971 					   &rd_addr, &fit_uname_config)) {
972 				debug("*  ramdisk: config '%s' from image at "
973 						"0x%08lx\n",
974 						fit_uname_config, rd_addr);
975 			} else if (fit_parse_subimage(select, default_addr,
976 						&rd_addr, &fit_uname_ramdisk)) {
977 				debug("*  ramdisk: subimage '%s' from image at "
978 						"0x%08lx\n",
979 						fit_uname_ramdisk, rd_addr);
980 			} else
981 #endif
982 			{
983 				rd_addr = simple_strtoul(select, NULL, 16);
984 				debug("*  ramdisk: cmdline image address = "
985 						"0x%08lx\n",
986 						rd_addr);
987 			}
988 #if IMAGE_ENABLE_FIT
989 		} else {
990 			/* use FIT configuration provided in first bootm
991 			 * command argument. If the property is not defined,
992 			 * quit silently.
993 			 */
994 			rd_addr = map_to_sysmem(images->fit_hdr_os);
995 			rd_noffset = fit_get_node_from_config(images,
996 					FIT_RAMDISK_PROP, rd_addr);
997 			if (rd_noffset == -ENOLINK)
998 				return 0;
999 			else if (rd_noffset < 0)
1000 				return 1;
1001 		}
1002 #endif
1003 
1004 		/* copy from dataflash if needed */
1005 		rd_addr = genimg_get_image(rd_addr);
1006 
1007 		/*
1008 		 * Check if there is an initrd image at the
1009 		 * address provided in the second bootm argument
1010 		 * check image type, for FIT images get FIT node.
1011 		 */
1012 		buf = map_sysmem(rd_addr, 0);
1013 		switch (genimg_get_format(buf)) {
1014 #if defined(CONFIG_IMAGE_FORMAT_LEGACY)
1015 		case IMAGE_FORMAT_LEGACY:
1016 			printf("## Loading init Ramdisk from Legacy "
1017 					"Image at %08lx ...\n", rd_addr);
1018 
1019 			bootstage_mark(BOOTSTAGE_ID_CHECK_RAMDISK);
1020 			rd_hdr = image_get_ramdisk(rd_addr, arch,
1021 							images->verify);
1022 
1023 			if (rd_hdr == NULL)
1024 				return 1;
1025 
1026 			rd_data = image_get_data(rd_hdr);
1027 			rd_len = image_get_data_size(rd_hdr);
1028 			rd_load = image_get_load(rd_hdr);
1029 			break;
1030 #endif
1031 #if IMAGE_ENABLE_FIT
1032 		case IMAGE_FORMAT_FIT:
1033 			rd_noffset = fit_image_load(images,
1034 					rd_addr, &fit_uname_ramdisk,
1035 					&fit_uname_config, arch,
1036 					IH_TYPE_RAMDISK,
1037 					BOOTSTAGE_ID_FIT_RD_START,
1038 					FIT_LOAD_OPTIONAL_NON_ZERO,
1039 					&rd_data, &rd_len);
1040 			if (rd_noffset < 0)
1041 				return 1;
1042 
1043 			images->fit_hdr_rd = map_sysmem(rd_addr, 0);
1044 			images->fit_uname_rd = fit_uname_ramdisk;
1045 			images->fit_noffset_rd = rd_noffset;
1046 			break;
1047 #endif
1048 #ifdef CONFIG_ANDROID_BOOT_IMAGE
1049 		case IMAGE_FORMAT_ANDROID:
1050 			android_image_get_ramdisk((void *)images->os.start,
1051 				&rd_data, &rd_len);
1052 			break;
1053 #endif
1054 		default:
1055 #ifdef CONFIG_SUPPORT_RAW_INITRD
1056 			end = NULL;
1057 			if (select)
1058 				end = strchr(select, ':');
1059 			if (end) {
1060 				rd_len = simple_strtoul(++end, NULL, 16);
1061 				rd_data = rd_addr;
1062 			} else
1063 #endif
1064 			{
1065 				puts("Wrong Ramdisk Image Format\n");
1066 				rd_data = rd_len = rd_load = 0;
1067 				return 1;
1068 			}
1069 		}
1070 	} else if (images->legacy_hdr_valid &&
1071 			image_check_type(&images->legacy_hdr_os_copy,
1072 						IH_TYPE_MULTI)) {
1073 
1074 		/*
1075 		 * Now check if we have a legacy mult-component image,
1076 		 * get second entry data start address and len.
1077 		 */
1078 		bootstage_mark(BOOTSTAGE_ID_RAMDISK);
1079 		printf("## Loading init Ramdisk from multi component "
1080 				"Legacy Image at %08lx ...\n",
1081 				(ulong)images->legacy_hdr_os);
1082 
1083 		image_multi_getimg(images->legacy_hdr_os, 1, &rd_data, &rd_len);
1084 	} else {
1085 		/*
1086 		 * no initrd image
1087 		 */
1088 		bootstage_mark(BOOTSTAGE_ID_NO_RAMDISK);
1089 		rd_len = rd_data = 0;
1090 	}
1091 
1092 	if (!rd_data) {
1093 		debug("## No init Ramdisk\n");
1094 	} else {
1095 		*rd_start = rd_data;
1096 		*rd_end = rd_data + rd_len;
1097 	}
1098 	debug("   ramdisk start = 0x%08lx, ramdisk end = 0x%08lx\n",
1099 			*rd_start, *rd_end);
1100 
1101 	return 0;
1102 }
1103 
1104 #ifdef CONFIG_SYS_BOOT_RAMDISK_HIGH
1105 /**
1106  * boot_ramdisk_high - relocate init ramdisk
1107  * @lmb: pointer to lmb handle, will be used for memory mgmt
1108  * @rd_data: ramdisk data start address
1109  * @rd_len: ramdisk data length
1110  * @initrd_start: pointer to a ulong variable, will hold final init ramdisk
1111  *      start address (after possible relocation)
1112  * @initrd_end: pointer to a ulong variable, will hold final init ramdisk
1113  *      end address (after possible relocation)
1114  *
1115  * boot_ramdisk_high() takes a relocation hint from "initrd_high" environment
1116  * variable and if requested ramdisk data is moved to a specified location.
1117  *
1118  * Initrd_start and initrd_end are set to final (after relocation) ramdisk
1119  * start/end addresses if ramdisk image start and len were provided,
1120  * otherwise set initrd_start and initrd_end set to zeros.
1121  *
1122  * returns:
1123  *      0 - success
1124  *     -1 - failure
1125  */
1126 int boot_ramdisk_high(struct lmb *lmb, ulong rd_data, ulong rd_len,
1127 		  ulong *initrd_start, ulong *initrd_end)
1128 {
1129 	char	*s;
1130 	ulong	initrd_high;
1131 	int	initrd_copy_to_ram = 1;
1132 
1133 	if ((s = getenv("initrd_high")) != NULL) {
1134 		/* a value of "no" or a similar string will act like 0,
1135 		 * turning the "load high" feature off. This is intentional.
1136 		 */
1137 		initrd_high = simple_strtoul(s, NULL, 16);
1138 		if (initrd_high == ~0)
1139 			initrd_copy_to_ram = 0;
1140 	} else {
1141 		initrd_high = getenv_bootm_mapsize() + getenv_bootm_low();
1142 	}
1143 
1144 
1145 #ifdef CONFIG_LOGBUFFER
1146 	/* Prevent initrd from overwriting logbuffer */
1147 	lmb_reserve(lmb, logbuffer_base() - LOGBUFF_OVERHEAD, LOGBUFF_RESERVE);
1148 #endif
1149 
1150 	debug("## initrd_high = 0x%08lx, copy_to_ram = %d\n",
1151 			initrd_high, initrd_copy_to_ram);
1152 
1153 	if (rd_data) {
1154 		if (!initrd_copy_to_ram) {	/* zero-copy ramdisk support */
1155 			debug("   in-place initrd\n");
1156 			*initrd_start = rd_data;
1157 			*initrd_end = rd_data + rd_len;
1158 			lmb_reserve(lmb, rd_data, rd_len);
1159 		} else {
1160 			if (initrd_high)
1161 				*initrd_start = (ulong)lmb_alloc_base(lmb,
1162 						rd_len, 0x1000, initrd_high);
1163 			else
1164 				*initrd_start = (ulong)lmb_alloc(lmb, rd_len,
1165 								 0x1000);
1166 
1167 			if (*initrd_start == 0) {
1168 				puts("ramdisk - allocation error\n");
1169 				goto error;
1170 			}
1171 			bootstage_mark(BOOTSTAGE_ID_COPY_RAMDISK);
1172 
1173 			*initrd_end = *initrd_start + rd_len;
1174 			printf("   Loading Ramdisk to %08lx, end %08lx ... ",
1175 					*initrd_start, *initrd_end);
1176 
1177 			memmove_wd((void *)*initrd_start,
1178 					(void *)rd_data, rd_len, CHUNKSZ);
1179 
1180 #ifdef CONFIG_MP
1181 			/*
1182 			 * Ensure the image is flushed to memory to handle
1183 			 * AMP boot scenarios in which we might not be
1184 			 * HW cache coherent
1185 			 */
1186 			flush_cache((unsigned long)*initrd_start, rd_len);
1187 #endif
1188 			puts("OK\n");
1189 		}
1190 	} else {
1191 		*initrd_start = 0;
1192 		*initrd_end = 0;
1193 	}
1194 	debug("   ramdisk load start = 0x%08lx, ramdisk load end = 0x%08lx\n",
1195 			*initrd_start, *initrd_end);
1196 
1197 	return 0;
1198 
1199 error:
1200 	return -1;
1201 }
1202 #endif /* CONFIG_SYS_BOOT_RAMDISK_HIGH */
1203 
1204 int boot_get_setup(bootm_headers_t *images, uint8_t arch,
1205 		   ulong *setup_start, ulong *setup_len)
1206 {
1207 #if IMAGE_ENABLE_FIT
1208 	return boot_get_setup_fit(images, arch, setup_start, setup_len);
1209 #else
1210 	return -ENOENT;
1211 #endif
1212 }
1213 
1214 #if IMAGE_ENABLE_FIT
1215 int boot_get_loadable(int argc, char * const argv[], bootm_headers_t *images,
1216 		uint8_t arch, const ulong *ld_start, ulong * const ld_len)
1217 {
1218 	/*
1219 	 * These variables are used to hold the current image location
1220 	 * in system memory.
1221 	 */
1222 	ulong tmp_img_addr;
1223 	/*
1224 	 * These two variables are requirements for fit_image_load, but
1225 	 * their values are not used
1226 	 */
1227 	ulong img_data, img_len;
1228 	void *buf;
1229 	int loadables_index;
1230 	int conf_noffset;
1231 	int fit_img_result;
1232 	char *uname;
1233 
1234 	/* Check to see if the images struct has a FIT configuration */
1235 	if (!genimg_has_config(images)) {
1236 		debug("## FIT configuration was not specified\n");
1237 		return 0;
1238 	}
1239 
1240 	/*
1241 	 * Obtain the os FIT header from the images struct
1242 	 * copy from dataflash if needed
1243 	 */
1244 	tmp_img_addr = map_to_sysmem(images->fit_hdr_os);
1245 	tmp_img_addr = genimg_get_image(tmp_img_addr);
1246 	buf = map_sysmem(tmp_img_addr, 0);
1247 	/*
1248 	 * Check image type. For FIT images get FIT node
1249 	 * and attempt to locate a generic binary.
1250 	 */
1251 	switch (genimg_get_format(buf)) {
1252 	case IMAGE_FORMAT_FIT:
1253 		conf_noffset = fit_conf_get_node(buf, images->fit_uname_cfg);
1254 
1255 		for (loadables_index = 0;
1256 		     fdt_get_string_index(buf, conf_noffset,
1257 				FIT_LOADABLE_PROP,
1258 				loadables_index,
1259 				(const char **)&uname) == 0;
1260 		     loadables_index++)
1261 		{
1262 			fit_img_result = fit_image_load(images,
1263 				tmp_img_addr,
1264 				(const char **)&uname,
1265 				&(images->fit_uname_cfg), arch,
1266 				IH_TYPE_LOADABLE,
1267 				BOOTSTAGE_ID_FIT_LOADABLE_START,
1268 				FIT_LOAD_OPTIONAL_NON_ZERO,
1269 				&img_data, &img_len);
1270 			if (fit_img_result < 0) {
1271 				/* Something went wrong! */
1272 				return fit_img_result;
1273 			}
1274 		}
1275 		break;
1276 	default:
1277 		printf("The given image format is not supported (corrupt?)\n");
1278 		return 1;
1279 	}
1280 
1281 	return 0;
1282 }
1283 #endif
1284 
1285 #ifdef CONFIG_SYS_BOOT_GET_CMDLINE
1286 /**
1287  * boot_get_cmdline - allocate and initialize kernel cmdline
1288  * @lmb: pointer to lmb handle, will be used for memory mgmt
1289  * @cmd_start: pointer to a ulong variable, will hold cmdline start
1290  * @cmd_end: pointer to a ulong variable, will hold cmdline end
1291  *
1292  * boot_get_cmdline() allocates space for kernel command line below
1293  * BOOTMAPSZ + getenv_bootm_low() address. If "bootargs" U-Boot environemnt
1294  * variable is present its contents is copied to allocated kernel
1295  * command line.
1296  *
1297  * returns:
1298  *      0 - success
1299  *     -1 - failure
1300  */
1301 int boot_get_cmdline(struct lmb *lmb, ulong *cmd_start, ulong *cmd_end)
1302 {
1303 	char *cmdline;
1304 	char *s;
1305 
1306 	cmdline = (char *)(ulong)lmb_alloc_base(lmb, CONFIG_SYS_BARGSIZE, 0xf,
1307 				getenv_bootm_mapsize() + getenv_bootm_low());
1308 
1309 	if (cmdline == NULL)
1310 		return -1;
1311 
1312 	if ((s = getenv("bootargs")) == NULL)
1313 		s = "";
1314 
1315 	strcpy(cmdline, s);
1316 
1317 	*cmd_start = (ulong) & cmdline[0];
1318 	*cmd_end = *cmd_start + strlen(cmdline);
1319 
1320 	debug("## cmdline at 0x%08lx ... 0x%08lx\n", *cmd_start, *cmd_end);
1321 
1322 	return 0;
1323 }
1324 #endif /* CONFIG_SYS_BOOT_GET_CMDLINE */
1325 
1326 #ifdef CONFIG_SYS_BOOT_GET_KBD
1327 /**
1328  * boot_get_kbd - allocate and initialize kernel copy of board info
1329  * @lmb: pointer to lmb handle, will be used for memory mgmt
1330  * @kbd: double pointer to board info data
1331  *
1332  * boot_get_kbd() allocates space for kernel copy of board info data below
1333  * BOOTMAPSZ + getenv_bootm_low() address and kernel board info is initialized
1334  * with the current u-boot board info data.
1335  *
1336  * returns:
1337  *      0 - success
1338  *     -1 - failure
1339  */
1340 int boot_get_kbd(struct lmb *lmb, bd_t **kbd)
1341 {
1342 	*kbd = (bd_t *)(ulong)lmb_alloc_base(lmb, sizeof(bd_t), 0xf,
1343 				getenv_bootm_mapsize() + getenv_bootm_low());
1344 	if (*kbd == NULL)
1345 		return -1;
1346 
1347 	**kbd = *(gd->bd);
1348 
1349 	debug("## kernel board info at 0x%08lx\n", (ulong)*kbd);
1350 
1351 #if defined(DEBUG) && defined(CONFIG_CMD_BDI)
1352 	do_bdinfo(NULL, 0, 0, NULL);
1353 #endif
1354 
1355 	return 0;
1356 }
1357 #endif /* CONFIG_SYS_BOOT_GET_KBD */
1358 
1359 #ifdef CONFIG_LMB
1360 int image_setup_linux(bootm_headers_t *images)
1361 {
1362 	ulong of_size = images->ft_len;
1363 	char **of_flat_tree = &images->ft_addr;
1364 	ulong *initrd_start = &images->initrd_start;
1365 	ulong *initrd_end = &images->initrd_end;
1366 	struct lmb *lmb = &images->lmb;
1367 	ulong rd_len;
1368 	int ret;
1369 
1370 	if (IMAGE_ENABLE_OF_LIBFDT)
1371 		boot_fdt_add_mem_rsv_regions(lmb, *of_flat_tree);
1372 
1373 	if (IMAGE_BOOT_GET_CMDLINE) {
1374 		ret = boot_get_cmdline(lmb, &images->cmdline_start,
1375 				&images->cmdline_end);
1376 		if (ret) {
1377 			puts("ERROR with allocation of cmdline\n");
1378 			return ret;
1379 		}
1380 	}
1381 	if (IMAGE_ENABLE_RAMDISK_HIGH) {
1382 		rd_len = images->rd_end - images->rd_start;
1383 		ret = boot_ramdisk_high(lmb, images->rd_start, rd_len,
1384 				initrd_start, initrd_end);
1385 		if (ret)
1386 			return ret;
1387 	}
1388 
1389 	if (IMAGE_ENABLE_OF_LIBFDT) {
1390 		ret = boot_relocate_fdt(lmb, of_flat_tree, &of_size);
1391 		if (ret)
1392 			return ret;
1393 	}
1394 
1395 	if (IMAGE_ENABLE_OF_LIBFDT && of_size) {
1396 		ret = image_setup_libfdt(images, *of_flat_tree, of_size, lmb);
1397 		if (ret)
1398 			return ret;
1399 	}
1400 
1401 	return 0;
1402 }
1403 #endif /* CONFIG_LMB */
1404 #endif /* !USE_HOSTCC */
1405