common/cmd_bootm.c

   1 /*
   2  * (C) Copyright 2000-2009
   3  * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
   4  *
   5  * SPDX-License-Identifier:     GPL-2.0+
   6  */
   7
   8 /*
   9  * Boot support
  10  */
  11 #include <common.h>
  12 #include <bootm.h>
  13 #include <command.h>
  14 #include <environment.h>
  15 #include <image.h>
  16 #include <lmb.h>
  17 #include <malloc.h>
  18 #include <nand.h>
  19 #include <asm/byteorder.h>
  20 #include <linux/compiler.h>
  21 #include <linux/ctype.h>
  22 #include <linux/err.h>
  23 #include <u-boot/zlib.h>
  24
  25 DECLARE_GLOBAL_DATA_PTR;
  26
  27 #if defined(CONFIG_CMD_IMI)
  28 static int image_info(unsigned long addr);
  29 #endif
  30
  31 #if defined(CONFIG_CMD_IMLS)
  32 #include <flash.h>
  33 #include <mtd/cfi_flash.h>
  34 extern flash_info_t flash_info[]; /* info for FLASH chips */
  35 #endif
  36
  37 #if defined(CONFIG_CMD_IMLS) || defined(CONFIG_CMD_IMLS_NAND)
  38 static int do_imls(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[]);
  39 #endif
  40
  41 bootm_headers_t images;         /* pointers to os/initrd/fdt images */
  42
  43 /* we overload the cmd field with our state machine info instead of a
  44  * function pointer */
  45 static cmd_tbl_t cmd_bootm_sub[] = {
  46         U_BOOT_CMD_MKENT(start, 0, 1, (void *)BOOTM_STATE_START, "", ""),
  47         U_BOOT_CMD_MKENT(loados, 0, 1, (void *)BOOTM_STATE_LOADOS, "", ""),
  48 #ifdef CONFIG_SYS_BOOT_RAMDISK_HIGH
  49         U_BOOT_CMD_MKENT(ramdisk, 0, 1, (void *)BOOTM_STATE_RAMDISK, "", ""),
  50 #endif
  51 #ifdef CONFIG_OF_LIBFDT
  52         U_BOOT_CMD_MKENT(fdt, 0, 1, (void *)BOOTM_STATE_FDT, "", ""),
  53 #endif
  54         U_BOOT_CMD_MKENT(cmdline, 0, 1, (void *)BOOTM_STATE_OS_CMDLINE, "", ""),
  55         U_BOOT_CMD_MKENT(bdt, 0, 1, (void *)BOOTM_STATE_OS_BD_T, "", ""),
  56         U_BOOT_CMD_MKENT(prep, 0, 1, (void *)BOOTM_STATE_OS_PREP, "", ""),
  57         U_BOOT_CMD_MKENT(fake, 0, 1, (void *)BOOTM_STATE_OS_FAKE_GO, "", ""),
  58         U_BOOT_CMD_MKENT(go, 0, 1, (void *)BOOTM_STATE_OS_GO, "", ""),
  59 };
  60
  61 static int do_bootm_subcommand(cmd_tbl_t *cmdtp, int flag, int argc,
  62                         char * const argv[])
  63 {
  64         int ret = 0;
  65         long state;
  66         cmd_tbl_t *c;
  67
  68         c = find_cmd_tbl(argv[0], &cmd_bootm_sub[0], ARRAY_SIZE(cmd_bootm_sub));
  69         argc--; argv++;
  70
  71         if (c) {
  72                 state = (long)c->cmd;
  73                 if (state == BOOTM_STATE_START)
  74                         state |= BOOTM_STATE_FINDOS | BOOTM_STATE_FINDOTHER;
  75         } else {
  76                 /* Unrecognized command */
  77                 return CMD_RET_USAGE;
  78         }
  79
  80         if (state != BOOTM_STATE_START && images.state >= state) {
  81                 printf("Trying to execute a command out of order\n");
  82                 return CMD_RET_USAGE;
  83         }
  84
  85         ret = do_bootm_states(cmdtp, flag, argc, argv, state, &images, 0);
  86
  87         return ret;
  88 }
  89
  90 /*******************************************************************/
  91 /* bootm - boot application image from image in memory */
  92 /*******************************************************************/
  93
  94 int do_bootm(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
  95 {
  96 #ifdef CONFIG_NEEDS_MANUAL_RELOC
  97         static int relocated = 0;
  98
  99         if (!relocated) {
 100                 int i;
 101
 102                 /* relocate names of sub-command table */
 103                 for (i = 0; i < ARRAY_SIZE(cmd_bootm_sub); i++)
 104                         cmd_bootm_sub[i].name += gd->reloc_off;
 105
 106                 relocated = 1;
 107         }
 108 #endif
 109
 110         /* determine if we have a sub command */
 111         argc--; argv++;
 112         if (argc > 0) {
 113                 char *endp;
 114
 115                 simple_strtoul(argv[0], &endp, 16);
 116                 /* endp pointing to NULL means that argv[0] was just a
 117                  * valid number, pass it along to the normal bootm processing
 118                  *
 119                  * If endp is ':' or '#' assume a FIT identifier so pass
 120                  * along for normal processing.
 121                  *
 122                  * Right now we assume the first arg should never be '-'
 123                  */
 124                 if ((*endp != 0) && (*endp != ':') && (*endp != '#'))
 125                         return do_bootm_subcommand(cmdtp, flag, argc, argv);
 126         }
 127
 128         return do_bootm_states(cmdtp, flag, argc, argv, BOOTM_STATE_START |
 129                 BOOTM_STATE_FINDOS | BOOTM_STATE_FINDOTHER |
 130                 BOOTM_STATE_LOADOS |
 131 #if defined(CONFIG_PPC) || defined(CONFIG_MIPS)
 132                 BOOTM_STATE_OS_CMDLINE |
 133 #endif
 134                 BOOTM_STATE_OS_PREP | BOOTM_STATE_OS_FAKE_GO |
 135                 BOOTM_STATE_OS_GO, &images, 1);
 136 }
 137
 138 int bootm_maybe_autostart(cmd_tbl_t *cmdtp, const char *cmd)
 139 {
 140         const char *ep = getenv("autostart");
 141
 142         if (ep && !strcmp(ep, "yes")) {
 143                 char *local_args[2];
 144                 local_args[0] = (char *)cmd;
 145                 local_args[1] = NULL;
 146                 printf("Automatic boot of image at addr 0x%08lX ...\n", load_addr);
 147                 return do_bootm(cmdtp, 0, 1, local_args);
 148         }
 149
 150         return 0;
 151 }
 152
 153 #ifdef CONFIG_SYS_LONGHELP
 154 static char bootm_help_text[] =
 155         "[addr [arg ...]]\n    - boot application image stored in memory\n"
 156         "\tpassing arguments 'arg ...'; when booting a Linux kernel,\n"
 157         "\t'arg' can be the address of an initrd image\n"
 158 #if defined(CONFIG_OF_LIBFDT)
 159         "\tWhen booting a Linux kernel which requires a flat device-tree\n"
 160         "\ta third argument is required which is the address of the\n"
 161         "\tdevice-tree blob. To boot that kernel without an initrd image,\n"
 162         "\tuse a '-' for the second argument. If you do not pass a third\n"
 163         "\ta bd_info struct will be passed instead\n"
 164 #endif
 165 #if defined(CONFIG_FIT)
 166         "\t\nFor the new multi component uImage format (FIT) addresses\n"
 167         "\tmust be extened to include component or configuration unit name:\n"
 168         "\taddr:<subimg_uname> - direct component image specification\n"
 169         "\taddr#<conf_uname>   - configuration specification\n"
 170         "\tUse iminfo command to get the list of existing component\n"
 171         "\timages and configurations.\n"
 172 #endif
 173         "\nSub-commands to do part of the bootm sequence.  The sub-commands "
 174         "must be\n"
 175         "issued in the order below (it's ok to not issue all sub-commands):\n"
 176         "\tstart [addr [arg ...]]\n"
 177         "\tloados  - load OS image\n"
 178 #if defined(CONFIG_SYS_BOOT_RAMDISK_HIGH)
 179         "\tramdisk - relocate initrd, set env initrd_start/initrd_end\n"
 180 #endif
 181 #if defined(CONFIG_OF_LIBFDT)
 182         "\tfdt     - relocate flat device tree\n"
 183 #endif
 184         "\tcmdline - OS specific command line processing/setup\n"
 185         "\tbdt     - OS specific bd_t processing\n"
 186         "\tprep    - OS specific prep before relocation or go\n"
 187         "\tgo      - start OS";
 188 #endif
 189
 190 U_BOOT_CMD(
 191         bootm,  CONFIG_SYS_MAXARGS,     1,      do_bootm,
 192         "boot application image from memory", bootm_help_text
 193 );
 194
 195 /*******************************************************************/
 196 /* bootd - boot default image */
 197 /*******************************************************************/
 198 #if defined(CONFIG_CMD_BOOTD)
 199 int do_bootd(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
 200 {
 201         return run_command(getenv("bootcmd"), flag);
 202 }
 203
 204 U_BOOT_CMD(
 205         boot,   1,      1,      do_bootd,
 206         "boot default, i.e., run 'bootcmd'",
 207         ""
 208 );
 209
 210 /* keep old command name "bootd" for backward compatibility */
 211 U_BOOT_CMD(
 212         bootd, 1,       1,      do_bootd,
 213         "boot default, i.e., run 'bootcmd'",
 214         ""
 215 );
 216
 217 #endif
 218
 219
 220 /*******************************************************************/
 221 /* iminfo - print header info for a requested image */
 222 /*******************************************************************/
 223 #if defined(CONFIG_CMD_IMI)
 224 static int do_iminfo(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
 225 {
 226         int     arg;
 227         ulong   addr;
 228         int     rcode = 0;
 229
 230         if (argc < 2) {
 231                 return image_info(load_addr);
 232         }
 233
 234         for (arg = 1; arg < argc; ++arg) {
 235                 addr = simple_strtoul(argv[arg], NULL, 16);
 236                 if (image_info(addr) != 0)
 237                         rcode = 1;
 238         }
 239         return rcode;
 240 }
 241
 242 static int image_info(ulong addr)
 243 {
 244         void *hdr = (void *)addr;
 245
 246         printf("\n## Checking Image at %08lx ...\n", addr);
 247
 248         switch (genimg_get_format(hdr)) {
 249 #if defined(CONFIG_IMAGE_FORMAT_LEGACY)
 250         case IMAGE_FORMAT_LEGACY:
 251                 puts("   Legacy image found\n");
 252                 if (!image_check_magic(hdr)) {
 253                         puts("   Bad Magic Number\n");
 254                         return 1;
 255                 }
 256
 257                 if (!image_check_hcrc(hdr)) {
 258                         puts("   Bad Header Checksum\n");
 259                         return 1;
 260                 }
 261
 262                 image_print_contents(hdr);
 263
 264                 puts("   Verifying Checksum ... ");
 265                 if (!image_check_dcrc(hdr)) {
 266                         puts("   Bad Data CRC\n");
 267                         return 1;
 268                 }
 269                 puts("OK\n");
 270                 return 0;
 271 #endif
 272 #if defined(CONFIG_FIT)
 273         case IMAGE_FORMAT_FIT:
 274                 puts("   FIT image found\n");
 275
 276                 if (!fit_check_format(hdr)) {
 277                         puts("Bad FIT image format!\n");
 278                         return 1;
 279                 }
 280
 281                 fit_print_contents(hdr);
 282
 283                 if (!fit_all_image_verify(hdr)) {
 284                         puts("Bad hash in FIT image!\n");
 285                         return 1;
 286                 }
 287
 288                 return 0;
 289 #endif
 290         default:
 291                 puts("Unknown image format!\n");
 292                 break;
 293         }
 294
 295         return 1;
 296 }
 297
 298 U_BOOT_CMD(
 299         iminfo, CONFIG_SYS_MAXARGS,     1,      do_iminfo,
 300         "print header information for application image",
 301         "addr [addr ...]\n"
 302         "    - print header information for application image starting at\n"
 303         "      address 'addr' in memory; this includes verification of the\n"
 304         "      image contents (magic number, header and payload checksums)"
 305 );
 306 #endif
 307
 308
 309 /*******************************************************************/
 310 /* imls - list all images found in flash */
 311 /*******************************************************************/
 312 #if defined(CONFIG_CMD_IMLS)
 313 static int do_imls_nor(void)
 314 {
 315         flash_info_t *info;
 316         int i, j;
 317         void *hdr;
 318
 319         for (i = 0, info = &flash_info[0];
 320                 i < CONFIG_SYS_MAX_FLASH_BANKS; ++i, ++info) {
 321
 322                 if (info->flash_id == FLASH_UNKNOWN)
 323                         goto next_bank;
 324                 for (j = 0; j < info->sector_count; ++j) {
 325
 326                         hdr = (void *)info->start[j];
 327                         if (!hdr)
 328                                 goto next_sector;
 329
 330                         switch (genimg_get_format(hdr)) {
 331 #if defined(CONFIG_IMAGE_FORMAT_LEGACY)
 332                         case IMAGE_FORMAT_LEGACY:
 333                                 if (!image_check_hcrc(hdr))
 334                                         goto next_sector;
 335
 336                                 printf("Legacy Image at %08lX:\n", (ulong)hdr);
 337                                 image_print_contents(hdr);
 338
 339                                 puts("   Verifying Checksum ... ");
 340                                 if (!image_check_dcrc(hdr)) {
 341                                         puts("Bad Data CRC\n");
 342                                 } else {
 343                                         puts("OK\n");
 344                                 }
 345                                 break;
 346 #endif
 347 #if defined(CONFIG_FIT)
 348                         case IMAGE_FORMAT_FIT:
 349                                 if (!fit_check_format(hdr))
 350                                         goto next_sector;
 351
 352                                 printf("FIT Image at %08lX:\n", (ulong)hdr);
 353                                 fit_print_contents(hdr);
 354                                 break;
 355 #endif
 356                         default:
 357                                 goto next_sector;
 358                         }
 359
 360 next_sector:            ;
 361                 }
 362 next_bank:      ;
 363         }
 364         return 0;
 365 }
 366 #endif
 367
 368 #if defined(CONFIG_CMD_IMLS_NAND)
 369 static int nand_imls_legacyimage(nand_info_t *nand, int nand_dev, loff_t off,
 370                 size_t len)
 371 {
 372         void *imgdata;
 373         int ret;
 374
 375         imgdata = malloc(len);
 376         if (!imgdata) {
 377                 printf("May be a Legacy Image at NAND device %d offset %08llX:\n",
 378                                 nand_dev, off);
 379                 printf("   Low memory(cannot allocate memory for image)\n");
 380                 return -ENOMEM;
 381         }
 382
 383         ret = nand_read_skip_bad(nand, off, &len,
 384                         imgdata);
 385         if (ret < 0 && ret != -EUCLEAN) {
 386                 free(imgdata);
 387                 return ret;
 388         }
 389
 390         if (!image_check_hcrc(imgdata)) {
 391                 free(imgdata);
 392                 return 0;
 393         }
 394
 395         printf("Legacy Image at NAND device %d offset %08llX:\n",
 396                         nand_dev, off);
 397         image_print_contents(imgdata);
 398
 399         puts("   Verifying Checksum ... ");
 400         if (!image_check_dcrc(imgdata))
 401                 puts("Bad Data CRC\n");
 402         else
 403                 puts("OK\n");
 404
 405         free(imgdata);
 406
 407         return 0;
 408 }
 409
 410 static int nand_imls_fitimage(nand_info_t *nand, int nand_dev, loff_t off,
 411                 size_t len)
 412 {
 413         void *imgdata;
 414         int ret;
 415
 416         imgdata = malloc(len);
 417         if (!imgdata) {
 418                 printf("May be a FIT Image at NAND device %d offset %08llX:\n",
 419                                 nand_dev, off);
 420                 printf("   Low memory(cannot allocate memory for image)\n");
 421                 return -ENOMEM;
 422         }
 423
 424         ret = nand_read_skip_bad(nand, off, &len,
 425                         imgdata);
 426         if (ret < 0 && ret != -EUCLEAN) {
 427                 free(imgdata);
 428                 return ret;
 429         }
 430
 431         if (!fit_check_format(imgdata)) {
 432                 free(imgdata);
 433                 return 0;
 434         }
 435
 436         printf("FIT Image at NAND device %d offset %08llX:\n", nand_dev, off);
 437
 438         fit_print_contents(imgdata);
 439         free(imgdata);
 440
 441         return 0;
 442 }
 443
 444 static int do_imls_nand(void)
 445 {
 446         nand_info_t *nand;
 447         int nand_dev = nand_curr_device;
 448         size_t len;
 449         loff_t off;
 450         u32 buffer[16];
 451
 452         if (nand_dev < 0 || nand_dev >= CONFIG_SYS_MAX_NAND_DEVICE) {
 453                 puts("\nNo NAND devices available\n");
 454                 return -ENODEV;
 455         }
 456
 457         printf("\n");
 458
 459         for (nand_dev = 0; nand_dev < CONFIG_SYS_MAX_NAND_DEVICE; nand_dev++) {
 460                 nand = &nand_info[nand_dev];
 461                 if (!nand->name || !nand->size)
 462                         continue;
 463
 464                 for (off = 0; off < nand->size; off += nand->erasesize) {
 465                         const image_header_t *header;
 466                         int ret;
 467
 468                         if (nand_block_isbad(nand, off))
 469                                 continue;
 470
 471                         len = sizeof(buffer);
 472
 473                         ret = nand_read(nand, off, &len, (u8 *)buffer);
 474                         if (ret < 0 && ret != -EUCLEAN) {
 475                                 printf("NAND read error %d at offset %08llX\n",
 476                                                 ret, off);
 477                                 continue;
 478                         }
 479
 480                         switch (genimg_get_format(buffer)) {
 481 #if defined(CONFIG_IMAGE_FORMAT_LEGACY)
 482                         case IMAGE_FORMAT_LEGACY:
 483                                 header = (const image_header_t *)buffer;
 484
 485                                 len = image_get_image_size(header);
 486                                 nand_imls_legacyimage(nand, nand_dev, off, len);
 487                                 break;
 488 #endif
 489 #if defined(CONFIG_FIT)
 490                         case IMAGE_FORMAT_FIT:
 491                                 len = fit_get_size(buffer);
 492                                 nand_imls_fitimage(nand, nand_dev, off, len);
 493                                 break;
 494 #endif
 495                         }
 496                 }
 497         }
 498
 499         return 0;
 500 }
 501 #endif
 502
 503 #if defined(CONFIG_CMD_IMLS) || defined(CONFIG_CMD_IMLS_NAND)
 504 static int do_imls(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
 505 {
 506         int ret_nor = 0, ret_nand = 0;
 507
 508 #if defined(CONFIG_CMD_IMLS)
 509         ret_nor = do_imls_nor();
 510 #endif
 511
 512 #if defined(CONFIG_CMD_IMLS_NAND)
 513         ret_nand = do_imls_nand();
 514 #endif
 515
 516         if (ret_nor)
 517                 return ret_nor;
 518
 519         if (ret_nand)
 520                 return ret_nand;
 521
 522         return (0);
 523 }
 524
 525 U_BOOT_CMD(
 526         imls,   1,              1,      do_imls,
 527         "list all images found in flash",
 528         "\n"
 529         "    - Prints information about all images found at sector/block\n"
 530         "      boundaries in nor/nand flash."
 531 );
 532 #endif
 533
 534 #ifdef CONFIG_CMD_BOOTZ
 535
 536 int __weak bootz_setup(ulong image, ulong *start, ulong *end)
 537 {
 538         /* Please define bootz_setup() for your platform */
 539
 540         puts("Your platform's zImage format isn't supported yet!\n");
 541         return -1;
 542 }
 543
 544 /*
 545  * zImage booting support
 546  */
 547 static int bootz_start(cmd_tbl_t *cmdtp, int flag, int argc,
 548                         char * const argv[], bootm_headers_t *images)
 549 {
 550         int ret;
 551         ulong zi_start, zi_end;
 552
 553         ret = do_bootm_states(cmdtp, flag, argc, argv, BOOTM_STATE_START,
 554                               images, 1);
 555
 556         /* Setup Linux kernel zImage entry point */
 557         if (!argc) {
 558                 images->ep = load_addr;
 559                 debug("*  kernel: default image load address = 0x%08lx\n",
 560                                 load_addr);
 561         } else {
 562                 images->ep = simple_strtoul(argv[0], NULL, 16);
 563                 debug("*  kernel: cmdline image address = 0x%08lx\n",
 564                         images->ep);
 565         }
 566
 567         ret = bootz_setup(images->ep, &zi_start, &zi_end);
 568         if (ret != 0)
 569                 return 1;
 570
 571         lmb_reserve(&images->lmb, images->ep, zi_end - zi_start);
 572
 573         /*
 574          * Handle the BOOTM_STATE_FINDOTHER state ourselves as we do not
 575          * have a header that provide this informaiton.
 576          */
 577         if (bootm_find_ramdisk_fdt(flag, argc, argv))
 578                 return 1;
 579
 580         return 0;
 581 }
 582
 583 int do_bootz(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
 584 {
 585         int ret;
 586
 587         /* Consume 'bootz' */
 588         argc--; argv++;
 589
 590         if (bootz_start(cmdtp, flag, argc, argv, &images))
 591                 return 1;
 592
 593         /*
 594          * We are doing the BOOTM_STATE_LOADOS state ourselves, so must
 595          * disable interrupts ourselves
 596          */
 597         bootm_disable_interrupts();
 598
 599         images.os.os = IH_OS_LINUX;
 600         ret = do_bootm_states(cmdtp, flag, argc, argv,
 601                               BOOTM_STATE_OS_PREP | BOOTM_STATE_OS_FAKE_GO |
 602                               BOOTM_STATE_OS_GO,
 603                               &images, 1);
 604
 605         return ret;
 606 }
 607
 608 #ifdef CONFIG_SYS_LONGHELP
 609 static char bootz_help_text[] =
 610         "[addr [initrd[:size]] [fdt]]\n"
 611         "    - boot Linux zImage stored in memory\n"
 612         "\tThe argument 'initrd' is optional and specifies the address\n"
 613         "\tof the initrd in memory. The optional argument ':size' allows\n"
 614         "\tspecifying the size of RAW initrd.\n"
 615 #if defined(CONFIG_OF_LIBFDT)
 616         "\tWhen booting a Linux kernel which requires a flat device-tree\n"
 617         "\ta third argument is required which is the address of the\n"
 618         "\tdevice-tree blob. To boot that kernel without an initrd image,\n"
 619         "\tuse a '-' for the second argument. If you do not pass a third\n"
 620         "\ta bd_info struct will be passed instead\n"
 621 #endif
 622         "";
 623 #endif
 624
 625 U_BOOT_CMD(
 626         bootz,  CONFIG_SYS_MAXARGS,     1,      do_bootz,
 627         "boot Linux zImage image from memory", bootz_help_text
 628 );
 629 #endif  /* CONFIG_CMD_BOOTZ */
 630
 631 #ifdef CONFIG_CMD_BOOTI
 632 /* See Documentation/arm64/booting.txt in the Linux kernel */
 633 struct Image_header {
 634         uint32_t        code0;          /* Executable code */
 635         uint32_t        code1;          /* Executable code */
 636         uint64_t        text_offset;    /* Image load offset, LE */
 637         uint64_t        image_size;     /* Effective Image size, LE */
 638         uint64_t        res1;           /* reserved */
 639         uint64_t        res2;           /* reserved */
 640         uint64_t        res3;           /* reserved */
 641         uint64_t        res4;           /* reserved */
 642         uint32_t        magic;          /* Magic number */
 643         uint32_t        res5;
 644 };
 645
 646 #define LINUX_ARM64_IMAGE_MAGIC 0x644d5241
 647
 648 static int booti_setup(bootm_headers_t *images)
 649 {
 650         struct Image_header *ih;
 651         uint64_t dst;
 652
 653         ih = (struct Image_header *)map_sysmem(images->ep, 0);
 654
 655         if (ih->magic != le32_to_cpu(LINUX_ARM64_IMAGE_MAGIC)) {
 656                 puts("Bad Linux ARM64 Image magic!\n");
 657                 return 1;
 658         }
 659
 660         if (ih->image_size == 0) {
 661                 puts("Image lacks image_size field, assuming 16MiB\n");
 662                 ih->image_size = (16 << 20);
 663         }
 664
 665         /*
 666          * If we are not at the correct run-time location, set the new
 667          * correct location and then move the image there.
 668          */
 669         dst = gd->bd->bi_dram[0].start + le32_to_cpu(ih->text_offset);
 670         if (images->ep != dst) {
 671                 void *src;
 672
 673                 debug("Moving Image from 0x%lx to 0x%llx\n", images->ep, dst);
 674
 675                 src = (void *)images->ep;
 676                 images->ep = dst;
 677                 memmove((void *)dst, src, le32_to_cpu(ih->image_size));
 678         }
 679
 680         return 0;
 681 }
 682
 683 /*
 684  * Image booting support
 685  */
 686 static int booti_start(cmd_tbl_t *cmdtp, int flag, int argc,
 687                         char * const argv[], bootm_headers_t *images)
 688 {
 689         int ret;
 690         struct Image_header *ih;
 691
 692         ret = do_bootm_states(cmdtp, flag, argc, argv, BOOTM_STATE_START,
 693                               images, 1);
 694
 695         /* Setup Linux kernel Image entry point */
 696         if (!argc) {
 697                 images->ep = load_addr;
 698                 debug("*  kernel: default image load address = 0x%08lx\n",
 699                                 load_addr);
 700         } else {
 701                 images->ep = simple_strtoul(argv[0], NULL, 16);
 702                 debug("*  kernel: cmdline image address = 0x%08lx\n",
 703                         images->ep);
 704         }
 705
 706         ret = booti_setup(images);
 707         if (ret != 0)
 708                 return 1;
 709
 710         ih = (struct Image_header *)map_sysmem(images->ep, 0);
 711
 712         lmb_reserve(&images->lmb, images->ep, le32_to_cpu(ih->image_size));
 713
 714         /*
 715          * Handle the BOOTM_STATE_FINDOTHER state ourselves as we do not
 716          * have a header that provide this informaiton.
 717          */
 718         if (bootm_find_ramdisk_fdt(flag, argc, argv))
 719                 return 1;
 720
 721         return 0;
 722 }
 723
 724 int do_booti(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
 725 {
 726         int ret;
 727
 728         /* Consume 'booti' */
 729         argc--; argv++;
 730
 731         if (booti_start(cmdtp, flag, argc, argv, &images))
 732                 return 1;
 733
 734         /*
 735          * We are doing the BOOTM_STATE_LOADOS state ourselves, so must
 736          * disable interrupts ourselves
 737          */
 738         bootm_disable_interrupts();
 739
 740         images.os.os = IH_OS_LINUX;
 741         ret = do_bootm_states(cmdtp, flag, argc, argv,
 742                               BOOTM_STATE_OS_PREP | BOOTM_STATE_OS_FAKE_GO |
 743                               BOOTM_STATE_OS_GO,
 744                               &images, 1);
 745
 746         return ret;
 747 }
 748
 749 #ifdef CONFIG_SYS_LONGHELP
 750 static char booti_help_text[] =
 751         "[addr [initrd[:size]] [fdt]]\n"
 752         "    - boot Linux Image stored in memory\n"
 753         "\tThe argument 'initrd' is optional and specifies the address\n"
 754         "\tof the initrd in memory. The optional argument ':size' allows\n"
 755         "\tspecifying the size of RAW initrd.\n"
 756 #if defined(CONFIG_OF_LIBFDT)
 757         "\tSince booting a Linux kernelrequires a flat device-tree\n"
 758         "\ta third argument is required which is the address of the\n"
 759         "\tdevice-tree blob. To boot that kernel without an initrd image,\n"
 760         "\tuse a '-' for the second argument.\n"
 761 #endif
 762         "";
 763 #endif
 764
 765 U_BOOT_CMD(
 766         booti,  CONFIG_SYS_MAXARGS,     1,      do_booti,
 767         "boot arm64 Linux Image image from memory", booti_help_text
 768 );
 769 #endif  /* CONFIG_CMD_BOOTI */