]> git.kernelconcepts.de Git - karo-tx-uboot.git/blob - common/cmd_bootm.c
449bb363f554a622d11834dbc149b2951185c6c1
[karo-tx-uboot.git] / common / cmd_bootm.c
1 /*
2  * (C) Copyright 2000-2009
3  * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
4  *
5  * SPDX-License-Identifier:     GPL-2.0+
6  */
7
8
9 /*
10  * Boot support
11  */
12 #include <common.h>
13 #include <watchdog.h>
14 #include <command.h>
15 #include <image.h>
16 #include <malloc.h>
17 #include <u-boot/zlib.h>
18 #include <bzlib.h>
19 #include <environment.h>
20 #include <lmb.h>
21 #include <linux/ctype.h>
22 #include <asm/byteorder.h>
23 #include <asm/io.h>
24 #include <linux/compiler.h>
25
26 #if defined(CONFIG_BOOTM_VXWORKS) && \
27         (defined(CONFIG_PPC) || defined(CONFIG_ARM))
28 #include <vxworks.h>
29 #endif
30
31 #if defined(CONFIG_CMD_USB)
32 #include <usb.h>
33 #endif
34
35 #if defined(CONFIG_OF_LIBFDT)
36 #include <libfdt.h>
37 #include <fdt_support.h>
38 #endif
39
40 #ifdef CONFIG_LZMA
41 #include <lzma/LzmaTypes.h>
42 #include <lzma/LzmaDec.h>
43 #include <lzma/LzmaTools.h>
44 #endif /* CONFIG_LZMA */
45
46 #ifdef CONFIG_LZO
47 #include <linux/lzo.h>
48 #endif /* CONFIG_LZO */
49
50 DECLARE_GLOBAL_DATA_PTR;
51
52 #ifndef CONFIG_SYS_BOOTM_LEN
53 #define CONFIG_SYS_BOOTM_LEN    0x800000        /* use 8MByte as default max gunzip size */
54 #endif
55
56 #ifdef CONFIG_BZIP2
57 extern void bz_internal_error(int);
58 #endif
59
60 #if defined(CONFIG_CMD_IMI)
61 static int image_info(unsigned long addr);
62 #endif
63
64 #if defined(CONFIG_CMD_IMLS)
65 #include <flash.h>
66 #include <mtd/cfi_flash.h>
67 extern flash_info_t flash_info[]; /* info for FLASH chips */
68 #endif
69
70 #if defined(CONFIG_CMD_IMLS) || defined(CONFIG_CMD_IMLS_NAND)
71 static int do_imls(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[]);
72 #endif
73
74 #include <linux/err.h>
75 #include <nand.h>
76
77 #if defined(CONFIG_SILENT_CONSOLE) && !defined(CONFIG_SILENT_U_BOOT_ONLY)
78 static void fixup_silent_linux(void);
79 #endif
80
81 static int do_bootm_standalone(int flag, int argc, char * const argv[],
82                                bootm_headers_t *images);
83
84 static const void *boot_get_kernel(cmd_tbl_t *cmdtp, int flag, int argc,
85                                 char * const argv[], bootm_headers_t *images,
86                                 ulong *os_data, ulong *os_len);
87
88 /*
89  *  Continue booting an OS image; caller already has:
90  *  - copied image header to global variable `header'
91  *  - checked header magic number, checksums (both header & image),
92  *  - verified image architecture (PPC) and type (KERNEL or MULTI),
93  *  - loaded (first part of) image to header load address,
94  *  - disabled interrupts.
95  *
96  * @flag: Flags indicating what to do (BOOTM_STATE_...)
97  * @argc: Number of arguments. Note that the arguments are shifted down
98  *       so that 0 is the first argument not processed by U-Boot, and
99  *       argc is adjusted accordingly. This avoids confusion as to how
100  *       many arguments are available for the OS.
101  * @images: Pointers to os/initrd/fdt
102  * @return 1 on error. On success the OS boots so this function does
103  * not return.
104  */
105 typedef int boot_os_fn(int flag, int argc, char * const argv[],
106                         bootm_headers_t *images);
107
108 #ifdef CONFIG_BOOTM_LINUX
109 extern boot_os_fn do_bootm_linux;
110 #endif
111 #ifdef CONFIG_BOOTM_NETBSD
112 static boot_os_fn do_bootm_netbsd;
113 #endif
114 #if defined(CONFIG_LYNXKDI)
115 static boot_os_fn do_bootm_lynxkdi;
116 extern void lynxkdi_boot(image_header_t *);
117 #endif
118 #ifdef CONFIG_BOOTM_RTEMS
119 static boot_os_fn do_bootm_rtems;
120 #endif
121 #if defined(CONFIG_BOOTM_OSE)
122 static boot_os_fn do_bootm_ose;
123 #endif
124 #if defined(CONFIG_BOOTM_PLAN9)
125 static boot_os_fn do_bootm_plan9;
126 #endif
127 #if defined(CONFIG_BOOTM_VXWORKS) && \
128         (defined(CONFIG_PPC) || defined(CONFIG_ARM))
129 static boot_os_fn do_bootm_vxworks;
130 #endif
131 #if defined(CONFIG_CMD_ELF)
132 static boot_os_fn do_bootm_qnxelf;
133 int do_bootvx(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[]);
134 int do_bootelf(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[]);
135 #endif
136 #if defined(CONFIG_INTEGRITY)
137 static boot_os_fn do_bootm_integrity;
138 #endif
139
140 static boot_os_fn *boot_os[] = {
141         [IH_OS_U_BOOT] = do_bootm_standalone,
142 #ifdef CONFIG_BOOTM_LINUX
143         [IH_OS_LINUX] = do_bootm_linux,
144 #endif
145 #ifdef CONFIG_BOOTM_NETBSD
146         [IH_OS_NETBSD] = do_bootm_netbsd,
147 #endif
148 #ifdef CONFIG_LYNXKDI
149         [IH_OS_LYNXOS] = do_bootm_lynxkdi,
150 #endif
151 #ifdef CONFIG_BOOTM_RTEMS
152         [IH_OS_RTEMS] = do_bootm_rtems,
153 #endif
154 #if defined(CONFIG_BOOTM_OSE)
155         [IH_OS_OSE] = do_bootm_ose,
156 #endif
157 #if defined(CONFIG_BOOTM_PLAN9)
158         [IH_OS_PLAN9] = do_bootm_plan9,
159 #endif
160 #if defined(CONFIG_BOOTM_VXWORKS) && \
161         (defined(CONFIG_PPC) || defined(CONFIG_ARM))
162         [IH_OS_VXWORKS] = do_bootm_vxworks,
163 #endif
164 #if defined(CONFIG_CMD_ELF)
165         [IH_OS_QNX] = do_bootm_qnxelf,
166 #endif
167 #ifdef CONFIG_INTEGRITY
168         [IH_OS_INTEGRITY] = do_bootm_integrity,
169 #endif
170 };
171
172 bootm_headers_t images;         /* pointers to os/initrd/fdt images */
173
174 /* Allow for arch specific config before we boot */
175 static void __arch_preboot_os(void)
176 {
177         /* please define platform specific arch_preboot_os() */
178 }
179 void arch_preboot_os(void) __attribute__((weak, alias("__arch_preboot_os")));
180
181 #define IH_INITRD_ARCH IH_ARCH_DEFAULT
182
183 #ifdef CONFIG_LMB
184 static void boot_start_lmb(bootm_headers_t *images)
185 {
186         ulong           mem_start;
187         phys_size_t     mem_size;
188
189         lmb_init(&images->lmb);
190
191         mem_start = getenv_bootm_low();
192         mem_size = getenv_bootm_size();
193
194         lmb_add(&images->lmb, (phys_addr_t)mem_start, mem_size);
195
196         arch_lmb_reserve(&images->lmb);
197         board_lmb_reserve(&images->lmb);
198 }
199 #else
200 #define lmb_reserve(lmb, base, size)
201 static inline void boot_start_lmb(bootm_headers_t *images) { }
202 #endif
203
204 static int bootm_start(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
205 {
206         memset((void *)&images, 0, sizeof(images));
207         images.verify = getenv_yesno("verify");
208
209         boot_start_lmb(&images);
210
211         bootstage_mark_name(BOOTSTAGE_ID_BOOTM_START, "bootm_start");
212         images.state = BOOTM_STATE_START;
213
214         return 0;
215 }
216
217 static int bootm_find_os(cmd_tbl_t *cmdtp, int flag, int argc,
218                          char * const argv[])
219 {
220         const void *os_hdr;
221         bool ep_found = false;
222
223         /* get kernel image header, start address and length */
224         os_hdr = boot_get_kernel(cmdtp, flag, argc, argv,
225                         &images, &images.os.image_start, &images.os.image_len);
226         if (images.os.image_len == 0) {
227                 puts("ERROR: can't get kernel image!\n");
228                 return 1;
229         }
230
231         /* get image parameters */
232         switch (genimg_get_format(os_hdr)) {
233         case IMAGE_FORMAT_LEGACY:
234                 images.os.type = image_get_type(os_hdr);
235                 images.os.comp = image_get_comp(os_hdr);
236                 images.os.os = image_get_os(os_hdr);
237
238                 images.os.end = image_get_image_end(os_hdr);
239                 images.os.load = image_get_load(os_hdr);
240                 break;
241 #if defined(CONFIG_FIT)
242         case IMAGE_FORMAT_FIT:
243                 if (fit_image_get_type(images.fit_hdr_os,
244                                         images.fit_noffset_os, &images.os.type)) {
245                         puts("Can't get image type!\n");
246                         bootstage_error(BOOTSTAGE_ID_FIT_TYPE);
247                         return 1;
248                 }
249
250                 if (fit_image_get_comp(images.fit_hdr_os,
251                                         images.fit_noffset_os, &images.os.comp)) {
252                         puts("Can't get image compression!\n");
253                         bootstage_error(BOOTSTAGE_ID_FIT_COMPRESSION);
254                         return 1;
255                 }
256
257                 if (fit_image_get_os(images.fit_hdr_os,
258                                         images.fit_noffset_os, &images.os.os)) {
259                         puts("Can't get image OS!\n");
260                         bootstage_error(BOOTSTAGE_ID_FIT_OS);
261                         return 1;
262                 }
263
264                 images.os.end = fit_get_end(images.fit_hdr_os);
265
266                 if (fit_image_get_load(images.fit_hdr_os, images.fit_noffset_os,
267                                         &images.os.load)) {
268                         puts("Can't get image load address!\n");
269                         bootstage_error(BOOTSTAGE_ID_FIT_LOADADDR);
270                         return 1;
271                 }
272                 break;
273 #endif
274 #ifdef CONFIG_ANDROID_BOOT_IMAGE
275         case IMAGE_FORMAT_ANDROID:
276                 images.os.type = IH_TYPE_KERNEL;
277                 images.os.comp = IH_COMP_NONE;
278                 images.os.os = IH_OS_LINUX;
279                 images.ep = images.os.load;
280                 ep_found = true;
281
282                 images.os.end = android_image_get_end(os_hdr);
283                 images.os.load = android_image_get_kload(os_hdr);
284                 break;
285 #endif
286         default:
287                 puts("ERROR: unknown image format type!\n");
288                 return 1;
289         }
290
291         /* find kernel entry point */
292         if (images.legacy_hdr_valid) {
293                 images.ep = image_get_ep(&images.legacy_hdr_os_copy);
294 #if defined(CONFIG_FIT)
295         } else if (images.fit_uname_os) {
296                 int ret;
297
298                 ret = fit_image_get_entry(images.fit_hdr_os,
299                                           images.fit_noffset_os, &images.ep);
300                 if (ret) {
301                         puts("Can't get entry point property!\n");
302                         return 1;
303                 }
304 #endif
305         } else if (!ep_found) {
306                 puts("Could not find kernel entry point!\n");
307                 return 1;
308         }
309
310         if (images.os.type == IH_TYPE_KERNEL_NOLOAD) {
311                 images.os.load = images.os.image_start;
312                 images.ep += images.os.load;
313         }
314
315         images.os.start = (ulong)os_hdr;
316
317         return 0;
318 }
319
320 static int bootm_find_ramdisk(int flag, int argc, char * const argv[])
321 {
322         int ret;
323
324         /* find ramdisk */
325         ret = boot_get_ramdisk(argc, argv, &images, IH_INITRD_ARCH,
326                                &images.rd_start, &images.rd_end);
327         if (ret) {
328                 puts("Ramdisk image is corrupt or invalid\n");
329                 return 1;
330         }
331
332         return 0;
333 }
334
335 #if defined(CONFIG_OF_LIBFDT)
336 static int bootm_find_fdt(int flag, int argc, char * const argv[])
337 {
338         int ret;
339
340         /* find flattened device tree */
341         ret = boot_get_fdt(flag, argc, argv, IH_ARCH_DEFAULT, &images,
342                            &images.ft_addr, &images.ft_len);
343         if (ret) {
344                 puts("Could not find a valid device tree\n");
345                 return 1;
346         }
347
348         set_working_fdt_addr(images.ft_addr);
349
350         return 0;
351 }
352 #endif
353
354 static int bootm_find_other(cmd_tbl_t *cmdtp, int flag, int argc,
355                             char * const argv[])
356 {
357         if (((images.os.type == IH_TYPE_KERNEL) ||
358              (images.os.type == IH_TYPE_KERNEL_NOLOAD) ||
359              (images.os.type == IH_TYPE_MULTI)) &&
360             (images.os.os == IH_OS_LINUX ||
361                  images.os.os == IH_OS_VXWORKS)) {
362                 if (bootm_find_ramdisk(flag, argc, argv))
363                         return 1;
364
365 #if defined(CONFIG_OF_LIBFDT)
366                 if (bootm_find_fdt(flag, argc, argv))
367                         return 1;
368 #endif
369         }
370
371         return 0;
372 }
373
374 #define BOOTM_ERR_RESET         -1
375 #define BOOTM_ERR_OVERLAP       -2
376 #define BOOTM_ERR_UNIMPLEMENTED -3
377 static int bootm_load_os(bootm_headers_t *images, unsigned long *load_end,
378                 int boot_progress)
379 {
380         image_info_t os = images->os;
381         uint8_t comp = os.comp;
382         ulong load = os.load;
383         ulong blob_start = os.start;
384         ulong blob_end = os.end;
385         ulong image_start = os.image_start;
386         ulong image_len = os.image_len;
387         __maybe_unused uint unc_len = CONFIG_SYS_BOOTM_LEN;
388         int no_overlap = 0;
389         void *load_buf, *image_buf;
390 #if defined(CONFIG_LZMA) || defined(CONFIG_LZO)
391         int ret;
392 #endif /* defined(CONFIG_LZMA) || defined(CONFIG_LZO) */
393
394         const char *type_name = genimg_get_type_name(os.type);
395
396         load_buf = map_sysmem(load, unc_len);
397         image_buf = map_sysmem(image_start, image_len);
398         switch (comp) {
399         case IH_COMP_NONE:
400                 if (load == image_start) {
401                         printf("   XIP %s ... ", type_name);
402                         no_overlap = 1;
403                 } else {
404                         printf("   Loading %s ... ", type_name);
405                         memmove_wd(load_buf, image_buf, image_len, CHUNKSZ);
406                 }
407                 *load_end = load + image_len;
408                 break;
409 #ifdef CONFIG_GZIP
410         case IH_COMP_GZIP:
411                 printf("   Uncompressing %s ... ", type_name);
412                 if (gunzip(load_buf, unc_len, image_buf, &image_len) != 0) {
413                         puts("GUNZIP: uncompress, out-of-mem or overwrite "
414                                 "error - must RESET board to recover\n");
415                         if (boot_progress)
416                                 bootstage_error(BOOTSTAGE_ID_DECOMP_IMAGE);
417                         return BOOTM_ERR_RESET;
418                 }
419
420                 *load_end = load + image_len;
421                 break;
422 #endif /* CONFIG_GZIP */
423 #ifdef CONFIG_BZIP2
424         case IH_COMP_BZIP2:
425                 printf("   Uncompressing %s ... ", type_name);
426                 /*
427                  * If we've got less than 4 MB of malloc() space,
428                  * use slower decompression algorithm which requires
429                  * at most 2300 KB of memory.
430                  */
431                 int i = BZ2_bzBuffToBuffDecompress(load_buf, &unc_len,
432                         image_buf, image_len,
433                         CONFIG_SYS_MALLOC_LEN < (4096 * 1024), 0);
434                 if (i != BZ_OK) {
435                         printf("BUNZIP2: uncompress or overwrite error %d "
436                                 "- must RESET board to recover\n", i);
437                         if (boot_progress)
438                                 bootstage_error(BOOTSTAGE_ID_DECOMP_IMAGE);
439                         return BOOTM_ERR_RESET;
440                 }
441
442                 *load_end = load + unc_len;
443                 break;
444 #endif /* CONFIG_BZIP2 */
445 #ifdef CONFIG_LZMA
446         case IH_COMP_LZMA: {
447                 SizeT lzma_len = unc_len;
448                 printf("   Uncompressing %s ... ", type_name);
449
450                 ret = lzmaBuffToBuffDecompress(load_buf, &lzma_len,
451                                                image_buf, image_len);
452                 unc_len = lzma_len;
453                 if (ret != SZ_OK) {
454                         printf("LZMA: uncompress or overwrite error %d "
455                                 "- must RESET board to recover\n", ret);
456                         bootstage_error(BOOTSTAGE_ID_DECOMP_IMAGE);
457                         return BOOTM_ERR_RESET;
458                 }
459                 *load_end = load + unc_len;
460                 break;
461         }
462 #endif /* CONFIG_LZMA */
463 #ifdef CONFIG_LZO
464         case IH_COMP_LZO: {
465                 size_t size = unc_len;
466
467                 printf("   Uncompressing %s ... ", type_name);
468
469                 ret = lzop_decompress(image_buf, image_len, load_buf, &size);
470                 if (ret != LZO_E_OK) {
471                         printf("LZO: uncompress or overwrite error %d "
472                               "- must RESET board to recover\n", ret);
473                         if (boot_progress)
474                                 bootstage_error(BOOTSTAGE_ID_DECOMP_IMAGE);
475                         return BOOTM_ERR_RESET;
476                 }
477
478                 *load_end = load + size;
479                 break;
480         }
481 #endif /* CONFIG_LZO */
482         default:
483                 printf("Unimplemented compression type %d\n", comp);
484                 return BOOTM_ERR_UNIMPLEMENTED;
485         }
486
487         flush_cache(load, (*load_end - load) * sizeof(ulong));
488
489         puts("OK\n");
490         debug("   kernel loaded at 0x%08lx, end = 0x%08lx\n", load, *load_end);
491         bootstage_mark(BOOTSTAGE_ID_KERNEL_LOADED);
492
493         if (!no_overlap && (load < blob_end) && (*load_end > blob_start)) {
494                 debug("images.os.start = 0x%lX, images.os.end = 0x%lx\n",
495                         blob_start, blob_end);
496                 debug("images.os.load = 0x%lx, load_end = 0x%lx\n", load,
497                         *load_end);
498
499                 /* Check what type of image this is. */
500                 if (images->legacy_hdr_valid) {
501                         if (image_get_type(&images->legacy_hdr_os_copy)
502                                         == IH_TYPE_MULTI)
503                                 puts("WARNING: legacy format multi component image overwritten\n");
504                         return BOOTM_ERR_OVERLAP;
505                 } else {
506                         puts("ERROR: new format image overwritten - must RESET the board to recover\n");
507                         bootstage_error(BOOTSTAGE_ID_OVERWRITTEN);
508                         return BOOTM_ERR_RESET;
509                 }
510         }
511
512         return 0;
513 }
514
515 static int do_bootm_standalone(int flag, int argc, char * const argv[],
516                                bootm_headers_t *images)
517 {
518         char  *s;
519         int   (*appl)(int, char * const []);
520
521         /* Don't start if "autostart" is set to "no" */
522         if (((s = getenv("autostart")) != NULL) && (strcmp(s, "no") == 0)) {
523                 setenv_hex("filesize", images->os.image_len);
524                 return 0;
525         }
526         appl = (int (*)(int, char * const []))images->ep;
527         appl(argc, argv);
528         return 0;
529 }
530
531 /* we overload the cmd field with our state machine info instead of a
532  * function pointer */
533 static cmd_tbl_t cmd_bootm_sub[] = {
534         U_BOOT_CMD_MKENT(start, 0, 1, (void *)BOOTM_STATE_START, "", ""),
535         U_BOOT_CMD_MKENT(loados, 0, 1, (void *)BOOTM_STATE_LOADOS, "", ""),
536 #ifdef CONFIG_SYS_BOOT_RAMDISK_HIGH
537         U_BOOT_CMD_MKENT(ramdisk, 0, 1, (void *)BOOTM_STATE_RAMDISK, "", ""),
538 #endif
539 #ifdef CONFIG_OF_LIBFDT
540         U_BOOT_CMD_MKENT(fdt, 0, 1, (void *)BOOTM_STATE_FDT, "", ""),
541 #endif
542         U_BOOT_CMD_MKENT(cmdline, 0, 1, (void *)BOOTM_STATE_OS_CMDLINE, "", ""),
543         U_BOOT_CMD_MKENT(bdt, 0, 1, (void *)BOOTM_STATE_OS_BD_T, "", ""),
544         U_BOOT_CMD_MKENT(prep, 0, 1, (void *)BOOTM_STATE_OS_PREP, "", ""),
545         U_BOOT_CMD_MKENT(fake, 0, 1, (void *)BOOTM_STATE_OS_FAKE_GO, "", ""),
546         U_BOOT_CMD_MKENT(go, 0, 1, (void *)BOOTM_STATE_OS_GO, "", ""),
547 };
548
549 static int boot_selected_os(int argc, char * const argv[], int state,
550                 bootm_headers_t *images, boot_os_fn *boot_fn)
551 {
552         arch_preboot_os();
553         boot_fn(state, argc, argv, images);
554
555         /* Stand-alone may return when 'autostart' is 'no' */
556         if (images->os.type == IH_TYPE_STANDALONE ||
557             state == BOOTM_STATE_OS_FAKE_GO) /* We expect to return */
558                 return 0;
559         bootstage_error(BOOTSTAGE_ID_BOOT_OS_RETURNED);
560 #ifdef DEBUG
561         puts("\n## Control returned to monitor - resetting...\n");
562 #endif
563         return BOOTM_ERR_RESET;
564 }
565
566 /**
567  * bootm_disable_interrupts() - Disable interrupts in preparation for load/boot
568  *
569  * @return interrupt flag (0 if interrupts were disabled, non-zero if they were
570  *      enabled)
571  */
572 static ulong bootm_disable_interrupts(void)
573 {
574         ulong iflag;
575
576         /*
577          * We have reached the point of no return: we are going to
578          * overwrite all exception vector code, so we cannot easily
579          * recover from any failures any more...
580          */
581         iflag = disable_interrupts();
582 #ifdef CONFIG_NETCONSOLE
583         /* Stop the ethernet stack if NetConsole could have left it up */
584         eth_halt();
585         eth_unregister(eth_get_dev());
586 #endif
587
588 #if defined(CONFIG_CMD_USB)
589         /*
590          * turn off USB to prevent the host controller from writing to the
591          * SDRAM while Linux is booting. This could happen (at least for OHCI
592          * controller), because the HCCA (Host Controller Communication Area)
593          * lies within the SDRAM and the host controller writes continously to
594          * this area (as busmaster!). The HccaFrameNumber is for example
595          * updated every 1 ms within the HCCA structure in SDRAM! For more
596          * details see the OpenHCI specification.
597          */
598         usb_stop();
599 #endif
600         return iflag;
601 }
602
603 /**
604  * Execute selected states of the bootm command.
605  *
606  * Note the arguments to this state must be the first argument, Any 'bootm'
607  * or sub-command arguments must have already been taken.
608  *
609  * Note that if states contains more than one flag it MUST contain
610  * BOOTM_STATE_START, since this handles and consumes the command line args.
611  *
612  * Also note that aside from boot_os_fn functions and bootm_load_os no other
613  * functions we store the return value of in 'ret' may use a negative return
614  * value, without special handling.
615  *
616  * @param cmdtp         Pointer to bootm command table entry
617  * @param flag          Command flags (CMD_FLAG_...)
618  * @param argc          Number of subcommand arguments (0 = no arguments)
619  * @param argv          Arguments
620  * @param states        Mask containing states to run (BOOTM_STATE_...)
621  * @param images        Image header information
622  * @param boot_progress 1 to show boot progress, 0 to not do this
623  * @return 0 if ok, something else on error. Some errors will cause this
624  *      function to perform a reboot! If states contains BOOTM_STATE_OS_GO
625  *      then the intent is to boot an OS, so this function will not return
626  *      unless the image type is standalone.
627  */
628 static int do_bootm_states(cmd_tbl_t *cmdtp, int flag, int argc,
629                 char * const argv[], int states, bootm_headers_t *images,
630                 int boot_progress)
631 {
632         boot_os_fn *boot_fn;
633         ulong iflag = 0;
634         int ret = 0, need_boot_fn;
635
636         images->state |= states;
637
638         /*
639          * Work through the states and see how far we get. We stop on
640          * any error.
641          */
642         if (states & BOOTM_STATE_START)
643                 ret = bootm_start(cmdtp, flag, argc, argv);
644
645         if (!ret && (states & BOOTM_STATE_FINDOS))
646                 ret = bootm_find_os(cmdtp, flag, argc, argv);
647
648         if (!ret && (states & BOOTM_STATE_FINDOTHER)) {
649                 ret = bootm_find_other(cmdtp, flag, argc, argv);
650                 argc = 0;       /* consume the args */
651         }
652
653         /* Load the OS */
654         if (!ret && (states & BOOTM_STATE_LOADOS)) {
655                 ulong load_end;
656
657                 iflag = bootm_disable_interrupts();
658                 ret = bootm_load_os(images, &load_end, 0);
659                 if (ret == 0)
660                         lmb_reserve(&images->lmb, images->os.load,
661                                     (load_end - images->os.load));
662                 else if (ret && ret != BOOTM_ERR_OVERLAP)
663                         goto err;
664                 else if (ret == BOOTM_ERR_OVERLAP)
665                         ret = 0;
666 #if defined(CONFIG_SILENT_CONSOLE) && !defined(CONFIG_SILENT_U_BOOT_ONLY)
667                 if (images->os.os == IH_OS_LINUX)
668                         fixup_silent_linux();
669 #endif
670         }
671
672         /* Relocate the ramdisk */
673 #ifdef CONFIG_SYS_BOOT_RAMDISK_HIGH
674         if (!ret && (states & BOOTM_STATE_RAMDISK)) {
675                 ulong rd_len = images->rd_end - images->rd_start;
676
677                 ret = boot_ramdisk_high(&images->lmb, images->rd_start,
678                         rd_len, &images->initrd_start, &images->initrd_end);
679                 if (!ret) {
680                         setenv_hex("initrd_start", images->initrd_start);
681                         setenv_hex("initrd_end", images->initrd_end);
682                 }
683         }
684 #endif
685 #if defined(CONFIG_OF_LIBFDT) && defined(CONFIG_LMB)
686         if (!ret && (states & BOOTM_STATE_FDT)) {
687                 boot_fdt_add_mem_rsv_regions(&images->lmb, images->ft_addr);
688                 ret = boot_relocate_fdt(&images->lmb, &images->ft_addr,
689                                         &images->ft_len);
690         }
691 #endif
692
693         /* From now on, we need the OS boot function */
694         if (ret)
695                 return ret;
696         boot_fn = boot_os[images->os.os];
697         need_boot_fn = states & (BOOTM_STATE_OS_CMDLINE |
698                         BOOTM_STATE_OS_BD_T | BOOTM_STATE_OS_PREP |
699                         BOOTM_STATE_OS_FAKE_GO | BOOTM_STATE_OS_GO);
700         if (boot_fn == NULL && need_boot_fn) {
701                 if (iflag)
702                         enable_interrupts();
703                 printf("ERROR: booting os '%s' (%d) is not supported\n",
704                        genimg_get_os_name(images->os.os), images->os.os);
705                 bootstage_error(BOOTSTAGE_ID_CHECK_BOOT_OS);
706                 return 1;
707         }
708
709         /* Call various other states that are not generally used */
710         if (!ret && (states & BOOTM_STATE_OS_CMDLINE))
711                 ret = boot_fn(BOOTM_STATE_OS_CMDLINE, argc, argv, images);
712         if (!ret && (states & BOOTM_STATE_OS_BD_T))
713                 ret = boot_fn(BOOTM_STATE_OS_BD_T, argc, argv, images);
714         if (!ret && (states & BOOTM_STATE_OS_PREP))
715                 ret = boot_fn(BOOTM_STATE_OS_PREP, argc, argv, images);
716
717 #ifdef CONFIG_TRACE
718         /* Pretend to run the OS, then run a user command */
719         if (!ret && (states & BOOTM_STATE_OS_FAKE_GO)) {
720                 char *cmd_list = getenv("fakegocmd");
721
722                 ret = boot_selected_os(argc, argv, BOOTM_STATE_OS_FAKE_GO,
723                                 images, boot_fn);
724                 if (!ret && cmd_list)
725                         ret = run_command_list(cmd_list, -1, flag);
726         }
727 #endif
728
729         /* Check for unsupported subcommand. */
730         if (ret) {
731                 puts("subcommand not supported\n");
732                 return ret;
733         }
734
735         /* Now run the OS! We hope this doesn't return */
736         if (!ret && (states & BOOTM_STATE_OS_GO))
737                 ret = boot_selected_os(argc, argv, BOOTM_STATE_OS_GO,
738                                 images, boot_fn);
739
740         /* Deal with any fallout */
741 err:
742         if (iflag)
743                 enable_interrupts();
744
745         if (ret == BOOTM_ERR_UNIMPLEMENTED)
746                 bootstage_error(BOOTSTAGE_ID_DECOMP_UNIMPL);
747         else if (ret == BOOTM_ERR_RESET)
748                 do_reset(cmdtp, flag, argc, argv);
749
750         return ret;
751 }
752
753 static int do_bootm_subcommand(cmd_tbl_t *cmdtp, int flag, int argc,
754                         char * const argv[])
755 {
756         int ret = 0;
757         long state;
758         cmd_tbl_t *c;
759
760         c = find_cmd_tbl(argv[0], &cmd_bootm_sub[0], ARRAY_SIZE(cmd_bootm_sub));
761         argc--; argv++;
762
763         if (c) {
764                 state = (long)c->cmd;
765                 if (state == BOOTM_STATE_START)
766                         state |= BOOTM_STATE_FINDOS | BOOTM_STATE_FINDOTHER;
767         } else {
768                 /* Unrecognized command */
769                 return CMD_RET_USAGE;
770         }
771
772         if (state != BOOTM_STATE_START && images.state >= state) {
773                 printf("Trying to execute a command out of order\n");
774                 return CMD_RET_USAGE;
775         }
776
777         ret = do_bootm_states(cmdtp, flag, argc, argv, state, &images, 0);
778
779         return ret;
780 }
781
782 /*******************************************************************/
783 /* bootm - boot application image from image in memory */
784 /*******************************************************************/
785
786 int do_bootm(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
787 {
788 #ifdef CONFIG_NEEDS_MANUAL_RELOC
789         static int relocated = 0;
790
791         if (!relocated) {
792                 int i;
793
794                 /* relocate boot function table */
795                 for (i = 0; i < ARRAY_SIZE(boot_os); i++)
796                         if (boot_os[i] != NULL)
797                                 boot_os[i] += gd->reloc_off;
798
799                 /* relocate names of sub-command table */
800                 for (i = 0; i < ARRAY_SIZE(cmd_bootm_sub); i++)
801                         cmd_bootm_sub[i].name += gd->reloc_off;
802
803                 relocated = 1;
804         }
805 #endif
806
807         /* determine if we have a sub command */
808         argc--; argv++;
809         if (argc > 0) {
810                 char *endp;
811
812                 simple_strtoul(argv[0], &endp, 16);
813                 /* endp pointing to NULL means that argv[0] was just a
814                  * valid number, pass it along to the normal bootm processing
815                  *
816                  * If endp is ':' or '#' assume a FIT identifier so pass
817                  * along for normal processing.
818                  *
819                  * Right now we assume the first arg should never be '-'
820                  */
821                 if ((*endp != 0) && (*endp != ':') && (*endp != '#'))
822                         return do_bootm_subcommand(cmdtp, flag, argc, argv);
823         }
824
825         return do_bootm_states(cmdtp, flag, argc, argv, BOOTM_STATE_START |
826                 BOOTM_STATE_FINDOS | BOOTM_STATE_FINDOTHER |
827                 BOOTM_STATE_LOADOS |
828 #if defined(CONFIG_PPC) || defined(CONFIG_MIPS)
829                 BOOTM_STATE_OS_CMDLINE |
830 #endif
831                 BOOTM_STATE_OS_PREP | BOOTM_STATE_OS_FAKE_GO |
832                 BOOTM_STATE_OS_GO, &images, 1);
833 }
834
835 int bootm_maybe_autostart(cmd_tbl_t *cmdtp, const char *cmd)
836 {
837         const char *ep = getenv("autostart");
838
839         if (ep && !strcmp(ep, "yes")) {
840                 char *local_args[2];
841                 local_args[0] = (char *)cmd;
842                 local_args[1] = NULL;
843                 printf("Automatic boot of image at addr 0x%08lX ...\n", load_addr);
844                 return do_bootm(cmdtp, 0, 1, local_args);
845         }
846
847         return 0;
848 }
849
850 /**
851  * image_get_kernel - verify legacy format kernel image
852  * @img_addr: in RAM address of the legacy format image to be verified
853  * @verify: data CRC verification flag
854  *
855  * image_get_kernel() verifies legacy image integrity and returns pointer to
856  * legacy image header if image verification was completed successfully.
857  *
858  * returns:
859  *     pointer to a legacy image header if valid image was found
860  *     otherwise return NULL
861  */
862 static image_header_t *image_get_kernel(ulong img_addr, int verify)
863 {
864         image_header_t *hdr = (image_header_t *)img_addr;
865
866         if (!image_check_magic(hdr)) {
867                 puts("Bad Magic Number\n");
868                 bootstage_error(BOOTSTAGE_ID_CHECK_MAGIC);
869                 return NULL;
870         }
871         bootstage_mark(BOOTSTAGE_ID_CHECK_HEADER);
872
873         if (!image_check_hcrc(hdr)) {
874                 puts("Bad Header Checksum\n");
875                 bootstage_error(BOOTSTAGE_ID_CHECK_HEADER);
876                 return NULL;
877         }
878
879         bootstage_mark(BOOTSTAGE_ID_CHECK_CHECKSUM);
880         image_print_contents(hdr);
881
882         if (verify) {
883                 puts("   Verifying Checksum ... ");
884                 if (!image_check_dcrc(hdr)) {
885                         printf("Bad Data CRC\n");
886                         bootstage_error(BOOTSTAGE_ID_CHECK_CHECKSUM);
887                         return NULL;
888                 }
889                 puts("OK\n");
890         }
891         bootstage_mark(BOOTSTAGE_ID_CHECK_ARCH);
892
893         if (!image_check_target_arch(hdr)) {
894                 printf("Unsupported Architecture 0x%x\n", image_get_arch(hdr));
895                 bootstage_error(BOOTSTAGE_ID_CHECK_ARCH);
896                 return NULL;
897         }
898         return hdr;
899 }
900
901 /**
902  * boot_get_kernel - find kernel image
903  * @os_data: pointer to a ulong variable, will hold os data start address
904  * @os_len: pointer to a ulong variable, will hold os data length
905  *
906  * boot_get_kernel() tries to find a kernel image, verifies its integrity
907  * and locates kernel data.
908  *
909  * returns:
910  *     pointer to image header if valid image was found, plus kernel start
911  *     address and length, otherwise NULL
912  */
913 static const void *boot_get_kernel(cmd_tbl_t *cmdtp, int flag, int argc,
914                 char * const argv[], bootm_headers_t *images, ulong *os_data,
915                 ulong *os_len)
916 {
917         image_header_t  *hdr;
918         ulong           img_addr;
919         const void *buf;
920 #if defined(CONFIG_FIT)
921         const char      *fit_uname_config = NULL;
922         const char      *fit_uname_kernel = NULL;
923         int             os_noffset;
924 #endif
925
926         /* find out kernel image address */
927         if (argc < 1) {
928                 img_addr = load_addr;
929                 debug("*  kernel: default image load address = 0x%08lx\n",
930                                 load_addr);
931 #if defined(CONFIG_FIT)
932         } else if (fit_parse_conf(argv[0], load_addr, &img_addr,
933                                                         &fit_uname_config)) {
934                 debug("*  kernel: config '%s' from image at 0x%08lx\n",
935                                 fit_uname_config, img_addr);
936         } else if (fit_parse_subimage(argv[0], load_addr, &img_addr,
937                                                         &fit_uname_kernel)) {
938                 debug("*  kernel: subimage '%s' from image at 0x%08lx\n",
939                                 fit_uname_kernel, img_addr);
940 #endif
941         } else {
942                 img_addr = simple_strtoul(argv[0], NULL, 16);
943                 debug("*  kernel: cmdline image address = 0x%08lx\n", img_addr);
944         }
945
946         bootstage_mark(BOOTSTAGE_ID_CHECK_MAGIC);
947
948         /* copy from dataflash if needed */
949         img_addr = genimg_get_image(img_addr);
950
951         /* check image type, for FIT images get FIT kernel node */
952         *os_data = *os_len = 0;
953         buf = map_sysmem(img_addr, 0);
954         switch (genimg_get_format(buf)) {
955         case IMAGE_FORMAT_LEGACY:
956                 printf("## Booting kernel from Legacy Image at %08lx ...\n",
957                                 img_addr);
958                 hdr = image_get_kernel(img_addr, images->verify);
959                 if (!hdr)
960                         return NULL;
961                 bootstage_mark(BOOTSTAGE_ID_CHECK_IMAGETYPE);
962
963                 /* get os_data and os_len */
964                 switch (image_get_type(hdr)) {
965                 case IH_TYPE_KERNEL:
966                 case IH_TYPE_KERNEL_NOLOAD:
967                         *os_data = image_get_data(hdr);
968                         *os_len = image_get_data_size(hdr);
969                         break;
970                 case IH_TYPE_MULTI:
971                         image_multi_getimg(hdr, 0, os_data, os_len);
972                         break;
973                 case IH_TYPE_STANDALONE:
974                         *os_data = image_get_data(hdr);
975                         *os_len = image_get_data_size(hdr);
976                         break;
977                 default:
978                         printf("Wrong Image Type for %s command\n",
979                                 cmdtp->name);
980                         bootstage_error(BOOTSTAGE_ID_CHECK_IMAGETYPE);
981                         return NULL;
982                 }
983
984                 /*
985                  * copy image header to allow for image overwrites during
986                  * kernel decompression.
987                  */
988                 memmove(&images->legacy_hdr_os_copy, hdr,
989                         sizeof(image_header_t));
990
991                 /* save pointer to image header */
992                 images->legacy_hdr_os = hdr;
993
994                 images->legacy_hdr_valid = 1;
995                 bootstage_mark(BOOTSTAGE_ID_DECOMP_IMAGE);
996                 break;
997 #if defined(CONFIG_FIT)
998         case IMAGE_FORMAT_FIT:
999                 os_noffset = fit_image_load(images, FIT_KERNEL_PROP,
1000                                 img_addr,
1001                                 &fit_uname_kernel, &fit_uname_config,
1002                                 IH_ARCH_DEFAULT, IH_TYPE_KERNEL,
1003                                 BOOTSTAGE_ID_FIT_KERNEL_START,
1004                                 FIT_LOAD_IGNORED, os_data, os_len);
1005                 if (os_noffset < 0)
1006                         return NULL;
1007
1008                 images->fit_hdr_os = map_sysmem(img_addr, 0);
1009                 images->fit_uname_os = fit_uname_kernel;
1010                 images->fit_uname_cfg = fit_uname_config;
1011                 images->fit_noffset_os = os_noffset;
1012                 break;
1013 #endif
1014 #ifdef CONFIG_ANDROID_BOOT_IMAGE
1015         case IMAGE_FORMAT_ANDROID:
1016                 printf("## Booting Android Image at 0x%08lx ...\n", img_addr);
1017                 if (android_image_get_kernel((void *)img_addr, images->verify,
1018                                              os_data, os_len))
1019                         return NULL;
1020                 break;
1021 #endif
1022         default:
1023                 printf("Wrong Image Format for %s command\n", cmdtp->name);
1024                 bootstage_error(BOOTSTAGE_ID_FIT_KERNEL_INFO);
1025                 return NULL;
1026         }
1027
1028         debug("   kernel data at 0x%08lx, len = 0x%08lx (%ld)\n",
1029                         *os_data, *os_len, *os_len);
1030
1031         return buf;
1032 }
1033
1034 #ifdef CONFIG_SYS_LONGHELP
1035 static char bootm_help_text[] =
1036         "[addr [arg ...]]\n    - boot application image stored in memory\n"
1037         "\tpassing arguments 'arg ...'; when booting a Linux kernel,\n"
1038         "\t'arg' can be the address of an initrd image\n"
1039 #if defined(CONFIG_OF_LIBFDT)
1040         "\tWhen booting a Linux kernel which requires a flat device-tree\n"
1041         "\ta third argument is required which is the address of the\n"
1042         "\tdevice-tree blob. To boot that kernel without an initrd image,\n"
1043         "\tuse a '-' for the second argument. If you do not pass a third\n"
1044         "\ta bd_info struct will be passed instead\n"
1045 #endif
1046 #if defined(CONFIG_FIT)
1047         "\t\nFor the new multi component uImage format (FIT) addresses\n"
1048         "\tmust be extened to include component or configuration unit name:\n"
1049         "\taddr:<subimg_uname> - direct component image specification\n"
1050         "\taddr#<conf_uname>   - configuration specification\n"
1051         "\tUse iminfo command to get the list of existing component\n"
1052         "\timages and configurations.\n"
1053 #endif
1054         "\nSub-commands to do part of the bootm sequence.  The sub-commands "
1055         "must be\n"
1056         "issued in the order below (it's ok to not issue all sub-commands):\n"
1057         "\tstart [addr [arg ...]]\n"
1058         "\tloados  - load OS image\n"
1059 #if defined(CONFIG_SYS_BOOT_RAMDISK_HIGH)
1060         "\tramdisk - relocate initrd, set env initrd_start/initrd_end\n"
1061 #endif
1062 #if defined(CONFIG_OF_LIBFDT)
1063         "\tfdt     - relocate flat device tree\n"
1064 #endif
1065         "\tcmdline - OS specific command line processing/setup\n"
1066         "\tbdt     - OS specific bd_t processing\n"
1067         "\tprep    - OS specific prep before relocation or go\n"
1068         "\tgo      - start OS";
1069 #endif
1070
1071 U_BOOT_CMD(
1072         bootm,  CONFIG_SYS_MAXARGS,     1,      do_bootm,
1073         "boot application image from memory", bootm_help_text
1074 );
1075
1076 /*******************************************************************/
1077 /* bootd - boot default image */
1078 /*******************************************************************/
1079 #if defined(CONFIG_CMD_BOOTD)
1080 int do_bootd(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
1081 {
1082         int rcode = 0;
1083
1084         if (run_command(getenv("bootcmd"), flag) < 0)
1085                 rcode = 1;
1086         return rcode;
1087 }
1088
1089 U_BOOT_CMD(
1090         boot,   1,      1,      do_bootd,
1091         "boot default, i.e., run 'bootcmd'",
1092         ""
1093 );
1094
1095 /* keep old command name "bootd" for backward compatibility */
1096 U_BOOT_CMD(
1097         bootd, 1,       1,      do_bootd,
1098         "boot default, i.e., run 'bootcmd'",
1099         ""
1100 );
1101
1102 #endif
1103
1104
1105 /*******************************************************************/
1106 /* iminfo - print header info for a requested image */
1107 /*******************************************************************/
1108 #if defined(CONFIG_CMD_IMI)
1109 static int do_iminfo(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
1110 {
1111         int     arg;
1112         ulong   addr;
1113         int     rcode = 0;
1114
1115         if (argc < 2) {
1116                 return image_info(load_addr);
1117         }
1118
1119         for (arg = 1; arg < argc; ++arg) {
1120                 addr = simple_strtoul(argv[arg], NULL, 16);
1121                 if (image_info(addr) != 0)
1122                         rcode = 1;
1123         }
1124         return rcode;
1125 }
1126
1127 static int image_info(ulong addr)
1128 {
1129         void *hdr = (void *)addr;
1130
1131         printf("\n## Checking Image at %08lx ...\n", addr);
1132
1133         switch (genimg_get_format(hdr)) {
1134         case IMAGE_FORMAT_LEGACY:
1135                 puts("   Legacy image found\n");
1136                 if (!image_check_magic(hdr)) {
1137                         puts("   Bad Magic Number\n");
1138                         return 1;
1139                 }
1140
1141                 if (!image_check_hcrc(hdr)) {
1142                         puts("   Bad Header Checksum\n");
1143                         return 1;
1144                 }
1145
1146                 image_print_contents(hdr);
1147
1148                 puts("   Verifying Checksum ... ");
1149                 if (!image_check_dcrc(hdr)) {
1150                         puts("   Bad Data CRC\n");
1151                         return 1;
1152                 }
1153                 puts("OK\n");
1154                 return 0;
1155 #if defined(CONFIG_FIT)
1156         case IMAGE_FORMAT_FIT:
1157                 puts("   FIT image found\n");
1158
1159                 if (!fit_check_format(hdr)) {
1160                         puts("Bad FIT image format!\n");
1161                         return 1;
1162                 }
1163
1164                 fit_print_contents(hdr);
1165
1166                 if (!fit_all_image_verify(hdr)) {
1167                         puts("Bad hash in FIT image!\n");
1168                         return 1;
1169                 }
1170
1171                 return 0;
1172 #endif
1173         default:
1174                 puts("Unknown image format!\n");
1175                 break;
1176         }
1177
1178         return 1;
1179 }
1180
1181 U_BOOT_CMD(
1182         iminfo, CONFIG_SYS_MAXARGS,     1,      do_iminfo,
1183         "print header information for application image",
1184         "addr [addr ...]\n"
1185         "    - print header information for application image starting at\n"
1186         "      address 'addr' in memory; this includes verification of the\n"
1187         "      image contents (magic number, header and payload checksums)"
1188 );
1189 #endif
1190
1191
1192 /*******************************************************************/
1193 /* imls - list all images found in flash */
1194 /*******************************************************************/
1195 #if defined(CONFIG_CMD_IMLS)
1196 static int do_imls_nor(void)
1197 {
1198         flash_info_t *info;
1199         int i, j;
1200         void *hdr;
1201
1202         for (i = 0, info = &flash_info[0];
1203                 i < CONFIG_SYS_MAX_FLASH_BANKS; ++i, ++info) {
1204
1205                 if (info->flash_id == FLASH_UNKNOWN)
1206                         goto next_bank;
1207                 for (j = 0; j < info->sector_count; ++j) {
1208
1209                         hdr = (void *)info->start[j];
1210                         if (!hdr)
1211                                 goto next_sector;
1212
1213                         switch (genimg_get_format(hdr)) {
1214                         case IMAGE_FORMAT_LEGACY:
1215                                 if (!image_check_hcrc(hdr))
1216                                         goto next_sector;
1217
1218                                 printf("Legacy Image at %08lX:\n", (ulong)hdr);
1219                                 image_print_contents(hdr);
1220
1221                                 puts("   Verifying Checksum ... ");
1222                                 if (!image_check_dcrc(hdr)) {
1223                                         puts("Bad Data CRC\n");
1224                                 } else {
1225                                         puts("OK\n");
1226                                 }
1227                                 break;
1228 #if defined(CONFIG_FIT)
1229                         case IMAGE_FORMAT_FIT:
1230                                 if (!fit_check_format(hdr))
1231                                         goto next_sector;
1232
1233                                 printf("FIT Image at %08lX:\n", (ulong)hdr);
1234                                 fit_print_contents(hdr);
1235                                 break;
1236 #endif
1237                         default:
1238                                 goto next_sector;
1239                         }
1240
1241 next_sector:            ;
1242                 }
1243 next_bank:      ;
1244         }
1245         return 0;
1246 }
1247 #endif
1248
1249 #if defined(CONFIG_CMD_IMLS_NAND)
1250 static int nand_imls_legacyimage(nand_info_t *nand, int nand_dev, loff_t off,
1251                 size_t len)
1252 {
1253         void *imgdata;
1254         int ret;
1255
1256         imgdata = malloc(len);
1257         if (!imgdata) {
1258                 printf("May be a Legacy Image at NAND device %d offset %08llX:\n",
1259                                 nand_dev, off);
1260                 printf("   Low memory(cannot allocate memory for image)\n");
1261                 return -ENOMEM;
1262         }
1263
1264         ret = nand_read_skip_bad(nand, off, &len,
1265                         imgdata);
1266         if (ret < 0 && ret != -EUCLEAN) {
1267                 free(imgdata);
1268                 return ret;
1269         }
1270
1271         if (!image_check_hcrc(imgdata)) {
1272                 free(imgdata);
1273                 return 0;
1274         }
1275
1276         printf("Legacy Image at NAND device %d offset %08llX:\n",
1277                         nand_dev, off);
1278         image_print_contents(imgdata);
1279
1280         puts("   Verifying Checksum ... ");
1281         if (!image_check_dcrc(imgdata))
1282                 puts("Bad Data CRC\n");
1283         else
1284                 puts("OK\n");
1285
1286         free(imgdata);
1287
1288         return 0;
1289 }
1290
1291 static int nand_imls_fitimage(nand_info_t *nand, int nand_dev, loff_t off,
1292                 size_t len)
1293 {
1294         void *imgdata;
1295         int ret;
1296
1297         imgdata = malloc(len);
1298         if (!imgdata) {
1299                 printf("May be a FIT Image at NAND device %d offset %08llX:\n",
1300                                 nand_dev, off);
1301                 printf("   Low memory(cannot allocate memory for image)\n");
1302                 return -ENOMEM;
1303         }
1304
1305         ret = nand_read_skip_bad(nand, off, &len,
1306                         imgdata);
1307         if (ret < 0 && ret != -EUCLEAN) {
1308                 free(imgdata);
1309                 return ret;
1310         }
1311
1312         if (!fit_check_format(imgdata)) {
1313                 free(imgdata);
1314                 return 0;
1315         }
1316
1317         printf("FIT Image at NAND device %d offset %08llX:\n", nand_dev, off);
1318
1319         fit_print_contents(imgdata);
1320         free(imgdata);
1321
1322         return 0;
1323 }
1324
1325 static int do_imls_nand(void)
1326 {
1327         nand_info_t *nand;
1328         int nand_dev = nand_curr_device;
1329         size_t len;
1330         loff_t off;
1331         u32 buffer[16];
1332
1333         if (nand_dev < 0 || nand_dev >= CONFIG_SYS_MAX_NAND_DEVICE) {
1334                 puts("\nNo NAND devices available\n");
1335                 return -ENODEV;
1336         }
1337
1338         printf("\n");
1339
1340         for (nand_dev = 0; nand_dev < CONFIG_SYS_MAX_NAND_DEVICE; nand_dev++) {
1341                 nand = &nand_info[nand_dev];
1342                 if (!nand->name || !nand->size)
1343                         continue;
1344
1345                 for (off = 0; off < nand->size; off += nand->erasesize) {
1346                         const image_header_t *header;
1347                         int ret;
1348
1349                         if (nand_block_isbad(nand, off))
1350                                 continue;
1351
1352                         len = sizeof(buffer);
1353
1354                         ret = nand_read(nand, off, &len, (u8 *)buffer);
1355                         if (ret < 0 && ret != -EUCLEAN) {
1356                                 printf("NAND read error %d at offset %08llX\n",
1357                                                 ret, off);
1358                                 continue;
1359                         }
1360
1361                         switch (genimg_get_format(buffer)) {
1362                         case IMAGE_FORMAT_LEGACY:
1363                                 header = (const image_header_t *)buffer;
1364
1365                                 len = image_get_image_size(header);
1366                                 nand_imls_legacyimage(nand, nand_dev, off, len);
1367                                 break;
1368 #if defined(CONFIG_FIT)
1369                         case IMAGE_FORMAT_FIT:
1370                                 len = fit_get_size(buffer);
1371                                 nand_imls_fitimage(nand, nand_dev, off, len);
1372                                 break;
1373 #endif
1374                         }
1375                 }
1376         }
1377
1378         return 0;
1379 }
1380 #endif
1381
1382 #if defined(CONFIG_CMD_IMLS) || defined(CONFIG_CMD_IMLS_NAND)
1383 static int do_imls(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
1384 {
1385         int ret_nor = 0, ret_nand = 0;
1386
1387 #if defined(CONFIG_CMD_IMLS)
1388         ret_nor = do_imls_nor();
1389 #endif
1390
1391 #if defined(CONFIG_CMD_IMLS_NAND)
1392         ret_nand = do_imls_nand();
1393 #endif
1394
1395         if (ret_nor)
1396                 return ret_nor;
1397
1398         if (ret_nand)
1399                 return ret_nand;
1400
1401         return (0);
1402 }
1403
1404 U_BOOT_CMD(
1405         imls,   1,              1,      do_imls,
1406         "list all images found in flash",
1407         "\n"
1408         "    - Prints information about all images found at sector/block\n"
1409         "      boundaries in nor/nand flash."
1410 );
1411 #endif
1412
1413 /*******************************************************************/
1414 /* helper routines */
1415 /*******************************************************************/
1416 #if defined(CONFIG_SILENT_CONSOLE) && !defined(CONFIG_SILENT_U_BOOT_ONLY)
1417
1418 #define CONSOLE_ARG     "console="
1419 #define CONSOLE_ARG_LEN (sizeof(CONSOLE_ARG) - 1)
1420
1421 static void fixup_silent_linux(void)
1422 {
1423         char *buf;
1424         const char *env_val;
1425         char *cmdline = getenv("bootargs");
1426         int want_silent;
1427
1428         /*
1429          * Only fix cmdline when requested. The environment variable can be:
1430          *
1431          *      no - we never fixup
1432          *      yes - we always fixup
1433          *      unset - we rely on the console silent flag
1434          */
1435         want_silent = getenv_yesno("silent_linux");
1436         if (want_silent == 0)
1437                 return;
1438         else if (want_silent == -1 && !(gd->flags & GD_FLG_SILENT))
1439                 return;
1440
1441         debug("before silent fix-up: %s\n", cmdline);
1442         if (cmdline && (cmdline[0] != '\0')) {
1443                 char *start = strstr(cmdline, CONSOLE_ARG);
1444
1445                 /* Allocate space for maximum possible new command line */
1446                 buf = malloc(strlen(cmdline) + 1 + CONSOLE_ARG_LEN + 1);
1447                 if (!buf) {
1448                         debug("%s: out of memory\n", __func__);
1449                         return;
1450                 }
1451
1452                 if (start) {
1453                         char *end = strchr(start, ' ');
1454                         int num_start_bytes = start - cmdline + CONSOLE_ARG_LEN;
1455
1456                         strncpy(buf, cmdline, num_start_bytes);
1457                         if (end)
1458                                 strcpy(buf + num_start_bytes, end);
1459                         else
1460                                 buf[num_start_bytes] = '\0';
1461                 } else {
1462                         sprintf(buf, "%s %s", cmdline, CONSOLE_ARG);
1463                 }
1464                 env_val = buf;
1465         } else {
1466                 buf = NULL;
1467                 env_val = CONSOLE_ARG;
1468         }
1469
1470         setenv("bootargs", env_val);
1471         debug("after silent fix-up: %s\n", env_val);
1472         free(buf);
1473 }
1474 #endif /* CONFIG_SILENT_CONSOLE */
1475
1476 #if defined(CONFIG_BOOTM_NETBSD) || defined(CONFIG_BOOTM_PLAN9)
1477 static void copy_args(char *dest, int argc, char * const argv[], char delim)
1478 {
1479         int i;
1480
1481         for (i = 0; i < argc; i++) {
1482                 if (i > 0)
1483                         *dest++ = delim;
1484                 strcpy(dest, argv[i]);
1485                 dest += strlen(argv[i]);
1486         }
1487 }
1488 #endif
1489
1490 /*******************************************************************/
1491 /* OS booting routines */
1492 /*******************************************************************/
1493
1494 #ifdef CONFIG_BOOTM_NETBSD
1495 static int do_bootm_netbsd(int flag, int argc, char * const argv[],
1496                             bootm_headers_t *images)
1497 {
1498         void (*loader)(bd_t *, image_header_t *, char *, char *);
1499         image_header_t *os_hdr, *hdr;
1500         ulong kernel_data, kernel_len;
1501         char *consdev;
1502         char *cmdline;
1503
1504         if (flag != BOOTM_STATE_OS_GO)
1505                 return 0;
1506
1507 #if defined(CONFIG_FIT)
1508         if (!images->legacy_hdr_valid) {
1509                 fit_unsupported_reset("NetBSD");
1510                 return 1;
1511         }
1512 #endif
1513         hdr = images->legacy_hdr_os;
1514
1515         /*
1516          * Booting a (NetBSD) kernel image
1517          *
1518          * This process is pretty similar to a standalone application:
1519          * The (first part of an multi-) image must be a stage-2 loader,
1520          * which in turn is responsible for loading & invoking the actual
1521          * kernel.  The only differences are the parameters being passed:
1522          * besides the board info strucure, the loader expects a command
1523          * line, the name of the console device, and (optionally) the
1524          * address of the original image header.
1525          */
1526         os_hdr = NULL;
1527         if (image_check_type(&images->legacy_hdr_os_copy, IH_TYPE_MULTI)) {
1528                 image_multi_getimg(hdr, 1, &kernel_data, &kernel_len);
1529                 if (kernel_len)
1530                         os_hdr = hdr;
1531         }
1532
1533         consdev = "";
1534 #if   defined(CONFIG_8xx_CONS_SMC1)
1535         consdev = "smc1";
1536 #elif defined(CONFIG_8xx_CONS_SMC2)
1537         consdev = "smc2";
1538 #elif defined(CONFIG_8xx_CONS_SCC2)
1539         consdev = "scc2";
1540 #elif defined(CONFIG_8xx_CONS_SCC3)
1541         consdev = "scc3";
1542 #endif
1543
1544         if (argc > 0) {
1545                 ulong len;
1546                 int   i;
1547
1548                 for (i = 0, len = 0; i < argc; i += 1)
1549                         len += strlen(argv[i]) + 1;
1550                 cmdline = malloc(len);
1551                 copy_args(cmdline, argc, argv, ' ');
1552         } else if ((cmdline = getenv("bootargs")) == NULL) {
1553                 cmdline = "";
1554         }
1555
1556         loader = (void (*)(bd_t *, image_header_t *, char *, char *))images->ep;
1557
1558         printf("## Transferring control to NetBSD stage-2 loader "
1559                 "(at address %08lx) ...\n",
1560                 (ulong)loader);
1561
1562         bootstage_mark(BOOTSTAGE_ID_RUN_OS);
1563
1564         /*
1565          * NetBSD Stage-2 Loader Parameters:
1566          *   arg[0]: pointer to board info data
1567          *   arg[1]: image load address
1568          *   arg[2]: char pointer to the console device to use
1569          *   arg[3]: char pointer to the boot arguments
1570          */
1571         (*loader)(gd->bd, os_hdr, consdev, cmdline);
1572
1573         return 1;
1574 }
1575 #endif /* CONFIG_BOOTM_NETBSD*/
1576
1577 #ifdef CONFIG_LYNXKDI
1578 static int do_bootm_lynxkdi(int flag, int argc, char * const argv[],
1579                              bootm_headers_t *images)
1580 {
1581         image_header_t *hdr = &images->legacy_hdr_os_copy;
1582
1583         if (flag != BOOTM_STATE_OS_GO)
1584                 return 0;
1585
1586 #if defined(CONFIG_FIT)
1587         if (!images->legacy_hdr_valid) {
1588                 fit_unsupported_reset("Lynx");
1589                 return 1;
1590         }
1591 #endif
1592
1593         lynxkdi_boot((image_header_t *)hdr);
1594
1595         return 1;
1596 }
1597 #endif /* CONFIG_LYNXKDI */
1598
1599 #ifdef CONFIG_BOOTM_RTEMS
1600 static int do_bootm_rtems(int flag, int argc, char * const argv[],
1601                            bootm_headers_t *images)
1602 {
1603         void (*entry_point)(bd_t *);
1604
1605         if (flag != BOOTM_STATE_OS_GO)
1606                 return 0;
1607
1608 #if defined(CONFIG_FIT)
1609         if (!images->legacy_hdr_valid) {
1610                 fit_unsupported_reset("RTEMS");
1611                 return 1;
1612         }
1613 #endif
1614
1615         entry_point = (void (*)(bd_t *))images->ep;
1616
1617         printf("## Transferring control to RTEMS (at address %08lx) ...\n",
1618                 (ulong)entry_point);
1619
1620         bootstage_mark(BOOTSTAGE_ID_RUN_OS);
1621
1622         /*
1623          * RTEMS Parameters:
1624          *   r3: ptr to board info data
1625          */
1626         (*entry_point)(gd->bd);
1627
1628         return 1;
1629 }
1630 #endif /* CONFIG_BOOTM_RTEMS */
1631
1632 #if defined(CONFIG_BOOTM_OSE)
1633 static int do_bootm_ose(int flag, int argc, char * const argv[],
1634                            bootm_headers_t *images)
1635 {
1636         void (*entry_point)(void);
1637
1638         if (flag != BOOTM_STATE_OS_GO)
1639                 return 0;
1640
1641 #if defined(CONFIG_FIT)
1642         if (!images->legacy_hdr_valid) {
1643                 fit_unsupported_reset("OSE");
1644                 return 1;
1645         }
1646 #endif
1647
1648         entry_point = (void (*)(void))images->ep;
1649
1650         printf("## Transferring control to OSE (at address %08lx) ...\n",
1651                 (ulong)entry_point);
1652
1653         bootstage_mark(BOOTSTAGE_ID_RUN_OS);
1654
1655         /*
1656          * OSE Parameters:
1657          *   None
1658          */
1659         (*entry_point)();
1660
1661         return 1;
1662 }
1663 #endif /* CONFIG_BOOTM_OSE */
1664
1665 #if defined(CONFIG_BOOTM_PLAN9)
1666 static int do_bootm_plan9(int flag, int argc, char * const argv[],
1667                            bootm_headers_t *images)
1668 {
1669         void (*entry_point)(void);
1670         char *s;
1671
1672         if (flag != BOOTM_STATE_OS_GO)
1673                 return 0;
1674
1675 #if defined(CONFIG_FIT)
1676         if (!images->legacy_hdr_valid) {
1677                 fit_unsupported_reset("Plan 9");
1678                 return 1;
1679         }
1680 #endif
1681
1682         /* See README.plan9 */
1683         s = getenv("confaddr");
1684         if (s != NULL) {
1685                 char *confaddr = (char *)simple_strtoul(s, NULL, 16);
1686
1687                 if (argc > 0) {
1688                         copy_args(confaddr, argc, argv, '\n');
1689                 } else {
1690                         s = getenv("bootargs");
1691                         if (s != NULL)
1692                                 strcpy(confaddr, s);
1693                 }
1694         }
1695
1696         entry_point = (void (*)(void))images->ep;
1697
1698         printf("## Transferring control to Plan 9 (at address %08lx) ...\n",
1699                 (ulong)entry_point);
1700
1701         bootstage_mark(BOOTSTAGE_ID_RUN_OS);
1702
1703         /*
1704          * Plan 9 Parameters:
1705          *   None
1706          */
1707         (*entry_point)();
1708
1709         return 1;
1710 }
1711 #endif /* CONFIG_BOOTM_PLAN9 */
1712
1713 #if defined(CONFIG_BOOTM_VXWORKS) && \
1714         (defined(CONFIG_PPC) || defined(CONFIG_ARM))
1715
1716 void do_bootvx_fdt(bootm_headers_t *images)
1717 {
1718 #if defined(CONFIG_OF_LIBFDT)
1719         int ret;
1720         char *bootline;
1721         ulong of_size = images->ft_len;
1722         char **of_flat_tree = &images->ft_addr;
1723         struct lmb *lmb = &images->lmb;
1724
1725         if (*of_flat_tree) {
1726                 boot_fdt_add_mem_rsv_regions(lmb, *of_flat_tree);
1727
1728                 ret = boot_relocate_fdt(lmb, of_flat_tree, &of_size);
1729                 if (ret)
1730                         return;
1731
1732                 ret = fdt_add_subnode(*of_flat_tree, 0, "chosen");
1733                 if ((ret >= 0 || ret == -FDT_ERR_EXISTS)) {
1734                         bootline = getenv("bootargs");
1735                         if (bootline) {
1736                                 ret = fdt_find_and_setprop(*of_flat_tree,
1737                                                 "/chosen", "bootargs",
1738                                                 bootline,
1739                                                 strlen(bootline) + 1, 1);
1740                                 if (ret < 0) {
1741                                         printf("## ERROR: %s : %s\n", __func__,
1742                                                fdt_strerror(ret));
1743                                         return;
1744                                 }
1745                         }
1746                 } else {
1747                         printf("## ERROR: %s : %s\n", __func__,
1748                                fdt_strerror(ret));
1749                         return;
1750                 }
1751         }
1752 #endif
1753
1754         boot_prep_vxworks(images);
1755
1756         bootstage_mark(BOOTSTAGE_ID_RUN_OS);
1757
1758 #if defined(CONFIG_OF_LIBFDT)
1759         printf("## Starting vxWorks at 0x%08lx, device tree at 0x%08lx ...\n",
1760                (ulong)images->ep, (ulong)*of_flat_tree);
1761 #else
1762         printf("## Starting vxWorks at 0x%08lx\n", (ulong)images->ep);
1763 #endif
1764
1765         boot_jump_vxworks(images);
1766
1767         puts("## vxWorks terminated\n");
1768 }
1769
1770 static int do_bootm_vxworks(int flag, int argc, char * const argv[],
1771                              bootm_headers_t *images)
1772 {
1773         if (flag != BOOTM_STATE_OS_GO)
1774                 return 0;
1775
1776 #if defined(CONFIG_FIT)
1777         if (!images->legacy_hdr_valid) {
1778                 fit_unsupported_reset("VxWorks");
1779                 return 1;
1780         }
1781 #endif
1782
1783         do_bootvx_fdt(images);
1784
1785         return 1;
1786 }
1787 #endif
1788
1789 #if defined(CONFIG_CMD_ELF)
1790 static int do_bootm_qnxelf(int flag, int argc, char * const argv[],
1791                             bootm_headers_t *images)
1792 {
1793         char *local_args[2];
1794         char str[16];
1795
1796         if (flag != BOOTM_STATE_OS_GO)
1797                 return 0;
1798
1799 #if defined(CONFIG_FIT)
1800         if (!images->legacy_hdr_valid) {
1801                 fit_unsupported_reset("QNX");
1802                 return 1;
1803         }
1804 #endif
1805
1806         sprintf(str, "%lx", images->ep); /* write entry-point into string */
1807         local_args[0] = argv[0];
1808         local_args[1] = str;    /* and provide it via the arguments */
1809         do_bootelf(NULL, 0, 2, local_args);
1810
1811         return 1;
1812 }
1813 #endif
1814
1815 #ifdef CONFIG_INTEGRITY
1816 static int do_bootm_integrity(int flag, int argc, char * const argv[],
1817                            bootm_headers_t *images)
1818 {
1819         void (*entry_point)(void);
1820
1821         if (flag != BOOTM_STATE_OS_GO)
1822                 return 0;
1823
1824 #if defined(CONFIG_FIT)
1825         if (!images->legacy_hdr_valid) {
1826                 fit_unsupported_reset("INTEGRITY");
1827                 return 1;
1828         }
1829 #endif
1830
1831         entry_point = (void (*)(void))images->ep;
1832
1833         printf("## Transferring control to INTEGRITY (at address %08lx) ...\n",
1834                 (ulong)entry_point);
1835
1836         bootstage_mark(BOOTSTAGE_ID_RUN_OS);
1837
1838         /*
1839          * INTEGRITY Parameters:
1840          *   None
1841          */
1842         (*entry_point)();
1843
1844         return 1;
1845 }
1846 #endif
1847
1848 #ifdef CONFIG_CMD_BOOTZ
1849
1850 int __weak bootz_setup(ulong image, ulong *start, ulong *end)
1851 {
1852         /* Please define bootz_setup() for your platform */
1853
1854         puts("Your platform's zImage format isn't supported yet!\n");
1855         return -1;
1856 }
1857
1858 /*
1859  * zImage booting support
1860  */
1861 static int bootz_start(cmd_tbl_t *cmdtp, int flag, int argc,
1862                         char * const argv[], bootm_headers_t *images)
1863 {
1864         int ret;
1865         ulong zi_start, zi_end;
1866
1867         ret = do_bootm_states(cmdtp, flag, argc, argv, BOOTM_STATE_START,
1868                               images, 1);
1869
1870         /* Setup Linux kernel zImage entry point */
1871         if (!argc) {
1872                 images->ep = load_addr;
1873                 debug("*  kernel: default image load address = 0x%08lx\n",
1874                                 load_addr);
1875         } else {
1876                 images->ep = simple_strtoul(argv[0], NULL, 16);
1877                 debug("*  kernel: cmdline image address = 0x%08lx\n",
1878                         images->ep);
1879         }
1880
1881         ret = bootz_setup(images->ep, &zi_start, &zi_end);
1882         if (ret != 0)
1883                 return 1;
1884
1885         lmb_reserve(&images->lmb, images->ep, zi_end - zi_start);
1886
1887         /*
1888          * Handle the BOOTM_STATE_FINDOTHER state ourselves as we do not
1889          * have a header that provide this informaiton.
1890          */
1891         if (bootm_find_ramdisk(flag, argc, argv))
1892                 return 1;
1893
1894 #if defined(CONFIG_OF_LIBFDT)
1895         if (bootm_find_fdt(flag, argc, argv))
1896                 return 1;
1897 #endif
1898
1899         return 0;
1900 }
1901
1902 int do_bootz(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
1903 {
1904         int ret;
1905
1906         /* Consume 'bootz' */
1907         argc--; argv++;
1908
1909         if (bootz_start(cmdtp, flag, argc, argv, &images))
1910                 return 1;
1911
1912         /*
1913          * We are doing the BOOTM_STATE_LOADOS state ourselves, so must
1914          * disable interrupts ourselves
1915          */
1916         bootm_disable_interrupts();
1917
1918         images.os.os = IH_OS_LINUX;
1919         ret = do_bootm_states(cmdtp, flag, argc, argv,
1920                               BOOTM_STATE_OS_PREP | BOOTM_STATE_OS_FAKE_GO |
1921                               BOOTM_STATE_OS_GO,
1922                               &images, 1);
1923
1924         return ret;
1925 }
1926
1927 #ifdef CONFIG_SYS_LONGHELP
1928 static char bootz_help_text[] =
1929         "[addr [initrd[:size]] [fdt]]\n"
1930         "    - boot Linux zImage stored in memory\n"
1931         "\tThe argument 'initrd' is optional and specifies the address\n"
1932         "\tof the initrd in memory. The optional argument ':size' allows\n"
1933         "\tspecifying the size of RAW initrd.\n"
1934 #if defined(CONFIG_OF_LIBFDT)
1935         "\tWhen booting a Linux kernel which requires a flat device-tree\n"
1936         "\ta third argument is required which is the address of the\n"
1937         "\tdevice-tree blob. To boot that kernel without an initrd image,\n"
1938         "\tuse a '-' for the second argument. If you do not pass a third\n"
1939         "\ta bd_info struct will be passed instead\n"
1940 #endif
1941         "";
1942 #endif
1943
1944 U_BOOT_CMD(
1945         bootz,  CONFIG_SYS_MAXARGS,     1,      do_bootz,
1946         "boot Linux zImage image from memory", bootz_help_text
1947 );
1948 #endif  /* CONFIG_CMD_BOOTZ */