common/cmd_bootm.c

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