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