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 #ifndef USE_HOSTCC 9 #include <common.h> 10 #include <bootstage.h> 11 #include <bzlib.h> 12 #include <errno.h> 13 #include <fdt_support.h> 14 #include <lmb.h> 15 #include <malloc.h> 16 #include <asm/io.h> 17 #include <linux/lzo.h> 18 #include <lzma/LzmaTypes.h> 19 #include <lzma/LzmaDec.h> 20 #include <lzma/LzmaTools.h> 21 #if defined(CONFIG_CMD_USB) 22 #include <usb.h> 23 #endif 24 #else 25 #include "mkimage.h" 26 #endif 27 28 #include <command.h> 29 #include <bootm.h> 30 #include <image.h> 31 32 #ifndef CONFIG_SYS_BOOTM_LEN 33 /* use 8MByte as default max gunzip size */ 34 #define CONFIG_SYS_BOOTM_LEN 0x800000 35 #endif 36 37 #define IH_INITRD_ARCH IH_ARCH_DEFAULT 38 39 #ifndef USE_HOSTCC 40 41 DECLARE_GLOBAL_DATA_PTR; 42 43 static const void *boot_get_kernel(cmd_tbl_t *cmdtp, int flag, int argc, 44 char * const argv[], bootm_headers_t *images, 45 ulong *os_data, ulong *os_len); 46 47 #ifdef CONFIG_LMB 48 static void boot_start_lmb(bootm_headers_t *images) 49 { 50 ulong mem_start; 51 phys_size_t mem_size; 52 53 lmb_init(&images->lmb); 54 55 mem_start = getenv_bootm_low(); 56 mem_size = getenv_bootm_size(); 57 58 lmb_add(&images->lmb, (phys_addr_t)mem_start, mem_size); 59 60 arch_lmb_reserve(&images->lmb); 61 board_lmb_reserve(&images->lmb); 62 } 63 #else 64 #define lmb_reserve(lmb, base, size) 65 static inline void boot_start_lmb(bootm_headers_t *images) { } 66 #endif 67 68 static int bootm_start(cmd_tbl_t *cmdtp, int flag, int argc, 69 char * const argv[]) 70 { 71 memset((void *)&images, 0, sizeof(images)); 72 images.verify = getenv_yesno("verify"); 73 74 boot_start_lmb(&images); 75 76 bootstage_mark_name(BOOTSTAGE_ID_BOOTM_START, "bootm_start"); 77 images.state = BOOTM_STATE_START; 78 79 return 0; 80 } 81 82 static int bootm_find_os(cmd_tbl_t *cmdtp, int flag, int argc, 83 char * const argv[]) 84 { 85 const void *os_hdr; 86 bool ep_found = false; 87 int ret; 88 89 /* get kernel image header, start address and length */ 90 os_hdr = boot_get_kernel(cmdtp, flag, argc, argv, 91 &images, &images.os.image_start, &images.os.image_len); 92 if (images.os.image_len == 0) { 93 puts("ERROR: can't get kernel image!\n"); 94 return 1; 95 } 96 97 /* get image parameters */ 98 switch (genimg_get_format(os_hdr)) { 99 #if defined(CONFIG_IMAGE_FORMAT_LEGACY) 100 case IMAGE_FORMAT_LEGACY: 101 images.os.type = image_get_type(os_hdr); 102 images.os.comp = image_get_comp(os_hdr); 103 images.os.os = image_get_os(os_hdr); 104 105 images.os.end = image_get_image_end(os_hdr); 106 images.os.load = image_get_load(os_hdr); 107 images.os.arch = image_get_arch(os_hdr); 108 break; 109 #endif 110 #if defined(CONFIG_FIT) 111 case IMAGE_FORMAT_FIT: 112 if (fit_image_get_type(images.fit_hdr_os, 113 images.fit_noffset_os, 114 &images.os.type)) { 115 puts("Can't get image type!\n"); 116 bootstage_error(BOOTSTAGE_ID_FIT_TYPE); 117 return 1; 118 } 119 120 if (fit_image_get_comp(images.fit_hdr_os, 121 images.fit_noffset_os, 122 &images.os.comp)) { 123 puts("Can't get image compression!\n"); 124 bootstage_error(BOOTSTAGE_ID_FIT_COMPRESSION); 125 return 1; 126 } 127 128 if (fit_image_get_os(images.fit_hdr_os, images.fit_noffset_os, 129 &images.os.os)) { 130 puts("Can't get image OS!\n"); 131 bootstage_error(BOOTSTAGE_ID_FIT_OS); 132 return 1; 133 } 134 135 if (fit_image_get_arch(images.fit_hdr_os, 136 images.fit_noffset_os, 137 &images.os.arch)) { 138 puts("Can't get image ARCH!\n"); 139 return 1; 140 } 141 142 images.os.end = fit_get_end(images.fit_hdr_os); 143 144 if (fit_image_get_load(images.fit_hdr_os, images.fit_noffset_os, 145 &images.os.load)) { 146 puts("Can't get image load address!\n"); 147 bootstage_error(BOOTSTAGE_ID_FIT_LOADADDR); 148 return 1; 149 } 150 break; 151 #endif 152 #ifdef CONFIG_ANDROID_BOOT_IMAGE 153 case IMAGE_FORMAT_ANDROID: 154 images.os.type = IH_TYPE_KERNEL; 155 images.os.comp = IH_COMP_NONE; 156 images.os.os = IH_OS_LINUX; 157 158 images.os.end = android_image_get_end(os_hdr); 159 images.os.load = android_image_get_kload(os_hdr); 160 images.ep = images.os.load; 161 ep_found = true; 162 break; 163 #endif 164 default: 165 puts("ERROR: unknown image format type!\n"); 166 return 1; 167 } 168 169 /* If we have a valid setup.bin, we will use that for entry (x86) */ 170 if (images.os.arch == IH_ARCH_I386 || 171 images.os.arch == IH_ARCH_X86_64) { 172 ulong len; 173 174 ret = boot_get_setup(&images, IH_ARCH_I386, &images.ep, &len); 175 if (ret < 0 && ret != -ENOENT) { 176 puts("Could not find a valid setup.bin for x86\n"); 177 return 1; 178 } 179 /* Kernel entry point is the setup.bin */ 180 } else if (images.legacy_hdr_valid) { 181 images.ep = image_get_ep(&images.legacy_hdr_os_copy); 182 #if defined(CONFIG_FIT) 183 } else if (images.fit_uname_os) { 184 int ret; 185 186 ret = fit_image_get_entry(images.fit_hdr_os, 187 images.fit_noffset_os, &images.ep); 188 if (ret) { 189 puts("Can't get entry point property!\n"); 190 return 1; 191 } 192 #endif 193 } else if (!ep_found) { 194 puts("Could not find kernel entry point!\n"); 195 return 1; 196 } 197 198 if (images.os.type == IH_TYPE_KERNEL_NOLOAD) { 199 images.os.load = images.os.image_start; 200 images.ep += images.os.load; 201 } 202 203 images.os.start = (ulong)os_hdr; 204 205 return 0; 206 } 207 208 static int bootm_find_ramdisk(int flag, int argc, char * const argv[]) 209 { 210 int ret; 211 212 /* find ramdisk */ 213 ret = boot_get_ramdisk(argc, argv, &images, IH_INITRD_ARCH, 214 &images.rd_start, &images.rd_end); 215 if (ret) { 216 puts("Ramdisk image is corrupt or invalid\n"); 217 return 1; 218 } 219 220 return 0; 221 } 222 223 #if defined(CONFIG_OF_LIBFDT) 224 static int bootm_find_fdt(int flag, int argc, char * const argv[]) 225 { 226 int ret; 227 228 /* find flattened device tree */ 229 ret = boot_get_fdt(flag, argc, argv, IH_ARCH_DEFAULT, &images, 230 &images.ft_addr, &images.ft_len); 231 if (ret) { 232 puts("Could not find a valid device tree\n"); 233 return 1; 234 } 235 236 set_working_fdt_addr(images.ft_addr); 237 238 return 0; 239 } 240 #endif 241 242 int bootm_find_ramdisk_fdt(int flag, int argc, char * const argv[]) 243 { 244 if (bootm_find_ramdisk(flag, argc, argv)) 245 return 1; 246 247 #if defined(CONFIG_OF_LIBFDT) 248 if (bootm_find_fdt(flag, argc, argv)) 249 return 1; 250 #endif 251 252 return 0; 253 } 254 255 static int bootm_find_other(cmd_tbl_t *cmdtp, int flag, int argc, 256 char * const argv[]) 257 { 258 if (((images.os.type == IH_TYPE_KERNEL) || 259 (images.os.type == IH_TYPE_KERNEL_NOLOAD) || 260 (images.os.type == IH_TYPE_MULTI)) && 261 (images.os.os == IH_OS_LINUX || 262 images.os.os == IH_OS_VXWORKS)) 263 return bootm_find_ramdisk_fdt(flag, argc, argv); 264 265 return 0; 266 } 267 #endif /* USE_HOSTC */ 268 269 /** 270 * print_decomp_msg() - Print a suitable decompression/loading message 271 * 272 * @type: OS type (IH_OS_...) 273 * @comp_type: Compression type being used (IH_COMP_...) 274 * @is_xip: true if the load address matches the image start 275 */ 276 static void print_decomp_msg(int comp_type, int type, bool is_xip) 277 { 278 const char *name = genimg_get_type_name(type); 279 280 if (comp_type == IH_COMP_NONE) 281 printf(" %s %s ... ", is_xip ? "XIP" : "Loading", name); 282 else 283 printf(" Uncompressing %s ... ", name); 284 } 285 286 #if defined(CONFIG_GZIP) || defined(CONFIG_GZIP) || defined(CONFIG_BZIP2) || \ 287 defined(CONFIG_LZMA) || defined(CONFIG_LZO) 288 static int handle_decomp_error(const char *algo, size_t size, size_t unc_len, 289 int ret) 290 { 291 if (size >= unc_len) 292 puts("Image too large: increase CONFIG_SYS_BOOTM_LEN\n"); 293 else 294 printf("%s: uncompress or overwrite error %d\n", algo, ret); 295 puts("Must RESET board to recover\n"); 296 #ifndef USE_HOSTCC 297 bootstage_error(BOOTSTAGE_ID_DECOMP_IMAGE); 298 #endif 299 300 return BOOTM_ERR_RESET; 301 } 302 #endif 303 304 int bootm_decomp_image(int comp, ulong load, ulong image_start, int type, 305 void *load_buf, void *image_buf, ulong image_len, 306 uint unc_len, ulong *load_end) 307 { 308 *load_end = load; 309 print_decomp_msg(comp, type, load == image_start); 310 switch (comp) { 311 case IH_COMP_NONE: 312 if (load != image_start) 313 memmove_wd(load_buf, image_buf, image_len, CHUNKSZ); 314 *load_end = load + image_len; 315 break; 316 #ifdef CONFIG_GZIP 317 case IH_COMP_GZIP: { 318 int ret; 319 320 ret = gunzip(load_buf, unc_len, image_buf, &image_len); 321 if (ret != 0) { 322 return handle_decomp_error("GUNZIP", image_len, 323 unc_len, ret); 324 } 325 326 *load_end = load + image_len; 327 break; 328 } 329 #endif /* CONFIG_GZIP */ 330 #ifdef CONFIG_BZIP2 331 case IH_COMP_BZIP2: { 332 size_t size = unc_len; 333 334 /* 335 * If we've got less than 4 MB of malloc() space, 336 * use slower decompression algorithm which requires 337 * at most 2300 KB of memory. 338 */ 339 int i = BZ2_bzBuffToBuffDecompress(load_buf, &unc_len, 340 image_buf, image_len, 341 CONFIG_SYS_MALLOC_LEN < (4096 * 1024), 0); 342 if (i != BZ_OK) { 343 return handle_decomp_error("BUNZIP2", size, unc_len, 344 i); 345 } 346 347 *load_end = load + unc_len; 348 break; 349 } 350 #endif /* CONFIG_BZIP2 */ 351 #ifdef CONFIG_LZMA 352 case IH_COMP_LZMA: { 353 SizeT lzma_len = unc_len; 354 int ret; 355 356 ret = lzmaBuffToBuffDecompress(load_buf, &lzma_len, 357 image_buf, image_len); 358 if (ret != SZ_OK) { 359 return handle_decomp_error("LZMA", lzma_len, unc_len, 360 ret); 361 } 362 unc_len = lzma_len; 363 *load_end = load + unc_len; 364 break; 365 } 366 #endif /* CONFIG_LZMA */ 367 #ifdef CONFIG_LZO 368 case IH_COMP_LZO: { 369 size_t size = unc_len; 370 int ret; 371 372 ret = lzop_decompress(image_buf, image_len, load_buf, &size); 373 if (ret != LZO_E_OK) 374 return handle_decomp_error("LZO", size, unc_len, ret); 375 376 *load_end = load + size; 377 break; 378 } 379 #endif /* CONFIG_LZO */ 380 default: 381 printf("Unimplemented compression type %d\n", comp); 382 return BOOTM_ERR_UNIMPLEMENTED; 383 } 384 385 puts("OK\n"); 386 387 return 0; 388 } 389 390 #ifndef USE_HOSTCC 391 static int bootm_load_os(bootm_headers_t *images, unsigned long *load_end, 392 int boot_progress) 393 { 394 image_info_t os = images->os; 395 ulong load = os.load; 396 ulong blob_start = os.start; 397 ulong blob_end = os.end; 398 ulong image_start = os.image_start; 399 ulong image_len = os.image_len; 400 bool no_overlap; 401 void *load_buf, *image_buf; 402 int err; 403 404 load_buf = map_sysmem(load, 0); 405 image_buf = map_sysmem(os.image_start, image_len); 406 err = bootm_decomp_image(os.comp, load, os.image_start, os.type, 407 load_buf, image_buf, image_len, 408 CONFIG_SYS_BOOTM_LEN, load_end); 409 if (err) { 410 bootstage_error(BOOTSTAGE_ID_DECOMP_IMAGE); 411 return err; 412 } 413 flush_cache(load, (*load_end - load) * sizeof(ulong)); 414 415 debug(" kernel loaded at 0x%08lx, end = 0x%08lx\n", load, *load_end); 416 bootstage_mark(BOOTSTAGE_ID_KERNEL_LOADED); 417 418 no_overlap = (os.comp == IH_COMP_NONE && load == image_start); 419 420 if (!no_overlap && (load < blob_end) && (*load_end > blob_start)) { 421 debug("images.os.start = 0x%lX, images.os.end = 0x%lx\n", 422 blob_start, blob_end); 423 debug("images.os.load = 0x%lx, load_end = 0x%lx\n", load, 424 *load_end); 425 426 /* Check what type of image this is. */ 427 if (images->legacy_hdr_valid) { 428 if (image_get_type(&images->legacy_hdr_os_copy) 429 == IH_TYPE_MULTI) 430 puts("WARNING: legacy format multi component image overwritten\n"); 431 return BOOTM_ERR_OVERLAP; 432 } else { 433 puts("ERROR: new format image overwritten - must RESET the board to recover\n"); 434 bootstage_error(BOOTSTAGE_ID_OVERWRITTEN); 435 return BOOTM_ERR_RESET; 436 } 437 } 438 439 return 0; 440 } 441 442 /** 443 * bootm_disable_interrupts() - Disable interrupts in preparation for load/boot 444 * 445 * @return interrupt flag (0 if interrupts were disabled, non-zero if they were 446 * enabled) 447 */ 448 ulong bootm_disable_interrupts(void) 449 { 450 ulong iflag; 451 452 /* 453 * We have reached the point of no return: we are going to 454 * overwrite all exception vector code, so we cannot easily 455 * recover from any failures any more... 456 */ 457 iflag = disable_interrupts(); 458 #ifdef CONFIG_NETCONSOLE 459 /* Stop the ethernet stack if NetConsole could have left it up */ 460 eth_halt(); 461 eth_unregister(eth_get_dev()); 462 #endif 463 464 #if defined(CONFIG_CMD_USB) 465 /* 466 * turn off USB to prevent the host controller from writing to the 467 * SDRAM while Linux is booting. This could happen (at least for OHCI 468 * controller), because the HCCA (Host Controller Communication Area) 469 * lies within the SDRAM and the host controller writes continously to 470 * this area (as busmaster!). The HccaFrameNumber is for example 471 * updated every 1 ms within the HCCA structure in SDRAM! For more 472 * details see the OpenHCI specification. 473 */ 474 usb_stop(); 475 #endif 476 return iflag; 477 } 478 479 #if defined(CONFIG_SILENT_CONSOLE) && !defined(CONFIG_SILENT_U_BOOT_ONLY) 480 481 #define CONSOLE_ARG "console=" 482 #define CONSOLE_ARG_LEN (sizeof(CONSOLE_ARG) - 1) 483 484 static void fixup_silent_linux(void) 485 { 486 char *buf; 487 const char *env_val; 488 char *cmdline = getenv("bootargs"); 489 int want_silent; 490 491 /* 492 * Only fix cmdline when requested. The environment variable can be: 493 * 494 * no - we never fixup 495 * yes - we always fixup 496 * unset - we rely on the console silent flag 497 */ 498 want_silent = getenv_yesno("silent_linux"); 499 if (want_silent == 0) 500 return; 501 else if (want_silent == -1 && !(gd->flags & GD_FLG_SILENT)) 502 return; 503 504 debug("before silent fix-up: %s\n", cmdline); 505 if (cmdline && (cmdline[0] != '\0')) { 506 char *start = strstr(cmdline, CONSOLE_ARG); 507 508 /* Allocate space for maximum possible new command line */ 509 buf = malloc(strlen(cmdline) + 1 + CONSOLE_ARG_LEN + 1); 510 if (!buf) { 511 debug("%s: out of memory\n", __func__); 512 return; 513 } 514 515 if (start) { 516 char *end = strchr(start, ' '); 517 int num_start_bytes = start - cmdline + CONSOLE_ARG_LEN; 518 519 strncpy(buf, cmdline, num_start_bytes); 520 if (end) 521 strcpy(buf + num_start_bytes, end); 522 else 523 buf[num_start_bytes] = '\0'; 524 } else { 525 sprintf(buf, "%s %s", cmdline, CONSOLE_ARG); 526 } 527 env_val = buf; 528 } else { 529 buf = NULL; 530 env_val = CONSOLE_ARG; 531 } 532 533 setenv("bootargs", env_val); 534 debug("after silent fix-up: %s\n", env_val); 535 free(buf); 536 } 537 #endif /* CONFIG_SILENT_CONSOLE */ 538 539 /** 540 * Execute selected states of the bootm command. 541 * 542 * Note the arguments to this state must be the first argument, Any 'bootm' 543 * or sub-command arguments must have already been taken. 544 * 545 * Note that if states contains more than one flag it MUST contain 546 * BOOTM_STATE_START, since this handles and consumes the command line args. 547 * 548 * Also note that aside from boot_os_fn functions and bootm_load_os no other 549 * functions we store the return value of in 'ret' may use a negative return 550 * value, without special handling. 551 * 552 * @param cmdtp Pointer to bootm command table entry 553 * @param flag Command flags (CMD_FLAG_...) 554 * @param argc Number of subcommand arguments (0 = no arguments) 555 * @param argv Arguments 556 * @param states Mask containing states to run (BOOTM_STATE_...) 557 * @param images Image header information 558 * @param boot_progress 1 to show boot progress, 0 to not do this 559 * @return 0 if ok, something else on error. Some errors will cause this 560 * function to perform a reboot! If states contains BOOTM_STATE_OS_GO 561 * then the intent is to boot an OS, so this function will not return 562 * unless the image type is standalone. 563 */ 564 int do_bootm_states(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[], 565 int states, bootm_headers_t *images, int boot_progress) 566 { 567 boot_os_fn *boot_fn; 568 ulong iflag = 0; 569 int ret = 0, need_boot_fn; 570 571 images->state |= states; 572 573 /* 574 * Work through the states and see how far we get. We stop on 575 * any error. 576 */ 577 if (states & BOOTM_STATE_START) 578 ret = bootm_start(cmdtp, flag, argc, argv); 579 580 if (!ret && (states & BOOTM_STATE_FINDOS)) 581 ret = bootm_find_os(cmdtp, flag, argc, argv); 582 583 if (!ret && (states & BOOTM_STATE_FINDOTHER)) { 584 ret = bootm_find_other(cmdtp, flag, argc, argv); 585 argc = 0; /* consume the args */ 586 } 587 588 /* Load the OS */ 589 if (!ret && (states & BOOTM_STATE_LOADOS)) { 590 ulong load_end; 591 592 iflag = bootm_disable_interrupts(); 593 ret = bootm_load_os(images, &load_end, 0); 594 if (ret == 0) 595 lmb_reserve(&images->lmb, images->os.load, 596 (load_end - images->os.load)); 597 else if (ret && ret != BOOTM_ERR_OVERLAP) 598 goto err; 599 else if (ret == BOOTM_ERR_OVERLAP) 600 ret = 0; 601 #if defined(CONFIG_SILENT_CONSOLE) && !defined(CONFIG_SILENT_U_BOOT_ONLY) 602 if (images->os.os == IH_OS_LINUX) 603 fixup_silent_linux(); 604 #endif 605 } 606 607 /* Relocate the ramdisk */ 608 #ifdef CONFIG_SYS_BOOT_RAMDISK_HIGH 609 if (!ret && (states & BOOTM_STATE_RAMDISK)) { 610 ulong rd_len = images->rd_end - images->rd_start; 611 612 ret = boot_ramdisk_high(&images->lmb, images->rd_start, 613 rd_len, &images->initrd_start, &images->initrd_end); 614 if (!ret) { 615 setenv_hex("initrd_start", images->initrd_start); 616 setenv_hex("initrd_end", images->initrd_end); 617 } 618 } 619 #endif 620 #if defined(CONFIG_OF_LIBFDT) && defined(CONFIG_LMB) 621 if (!ret && (states & BOOTM_STATE_FDT)) { 622 boot_fdt_add_mem_rsv_regions(&images->lmb, images->ft_addr); 623 ret = boot_relocate_fdt(&images->lmb, &images->ft_addr, 624 &images->ft_len); 625 } 626 #endif 627 628 /* From now on, we need the OS boot function */ 629 if (ret) 630 return ret; 631 boot_fn = bootm_os_get_boot_func(images->os.os); 632 need_boot_fn = states & (BOOTM_STATE_OS_CMDLINE | 633 BOOTM_STATE_OS_BD_T | BOOTM_STATE_OS_PREP | 634 BOOTM_STATE_OS_FAKE_GO | BOOTM_STATE_OS_GO); 635 if (boot_fn == NULL && need_boot_fn) { 636 if (iflag) 637 enable_interrupts(); 638 printf("ERROR: booting os '%s' (%d) is not supported\n", 639 genimg_get_os_name(images->os.os), images->os.os); 640 bootstage_error(BOOTSTAGE_ID_CHECK_BOOT_OS); 641 return 1; 642 } 643 644 /* Call various other states that are not generally used */ 645 if (!ret && (states & BOOTM_STATE_OS_CMDLINE)) 646 ret = boot_fn(BOOTM_STATE_OS_CMDLINE, argc, argv, images); 647 if (!ret && (states & BOOTM_STATE_OS_BD_T)) 648 ret = boot_fn(BOOTM_STATE_OS_BD_T, argc, argv, images); 649 if (!ret && (states & BOOTM_STATE_OS_PREP)) 650 ret = boot_fn(BOOTM_STATE_OS_PREP, argc, argv, images); 651 652 #ifdef CONFIG_TRACE 653 /* Pretend to run the OS, then run a user command */ 654 if (!ret && (states & BOOTM_STATE_OS_FAKE_GO)) { 655 char *cmd_list = getenv("fakegocmd"); 656 657 ret = boot_selected_os(argc, argv, BOOTM_STATE_OS_FAKE_GO, 658 images, boot_fn); 659 if (!ret && cmd_list) 660 ret = run_command_list(cmd_list, -1, flag); 661 } 662 #endif 663 664 /* Check for unsupported subcommand. */ 665 if (ret) { 666 puts("subcommand not supported\n"); 667 return ret; 668 } 669 670 /* Now run the OS! We hope this doesn't return */ 671 if (!ret && (states & BOOTM_STATE_OS_GO)) 672 ret = boot_selected_os(argc, argv, BOOTM_STATE_OS_GO, 673 images, boot_fn); 674 675 /* Deal with any fallout */ 676 err: 677 if (iflag) 678 enable_interrupts(); 679 680 if (ret == BOOTM_ERR_UNIMPLEMENTED) 681 bootstage_error(BOOTSTAGE_ID_DECOMP_UNIMPL); 682 else if (ret == BOOTM_ERR_RESET) 683 do_reset(cmdtp, flag, argc, argv); 684 685 return ret; 686 } 687 688 #if defined(CONFIG_IMAGE_FORMAT_LEGACY) 689 /** 690 * image_get_kernel - verify legacy format kernel image 691 * @img_addr: in RAM address of the legacy format image to be verified 692 * @verify: data CRC verification flag 693 * 694 * image_get_kernel() verifies legacy image integrity and returns pointer to 695 * legacy image header if image verification was completed successfully. 696 * 697 * returns: 698 * pointer to a legacy image header if valid image was found 699 * otherwise return NULL 700 */ 701 static image_header_t *image_get_kernel(ulong img_addr, int verify) 702 { 703 image_header_t *hdr = (image_header_t *)img_addr; 704 705 if (!image_check_magic(hdr)) { 706 puts("Bad Magic Number\n"); 707 bootstage_error(BOOTSTAGE_ID_CHECK_MAGIC); 708 return NULL; 709 } 710 bootstage_mark(BOOTSTAGE_ID_CHECK_HEADER); 711 712 if (!image_check_hcrc(hdr)) { 713 puts("Bad Header Checksum\n"); 714 bootstage_error(BOOTSTAGE_ID_CHECK_HEADER); 715 return NULL; 716 } 717 718 bootstage_mark(BOOTSTAGE_ID_CHECK_CHECKSUM); 719 image_print_contents(hdr); 720 721 if (verify) { 722 puts(" Verifying Checksum ... "); 723 if (!image_check_dcrc(hdr)) { 724 printf("Bad Data CRC\n"); 725 bootstage_error(BOOTSTAGE_ID_CHECK_CHECKSUM); 726 return NULL; 727 } 728 puts("OK\n"); 729 } 730 bootstage_mark(BOOTSTAGE_ID_CHECK_ARCH); 731 732 if (!image_check_target_arch(hdr)) { 733 printf("Unsupported Architecture 0x%x\n", image_get_arch(hdr)); 734 bootstage_error(BOOTSTAGE_ID_CHECK_ARCH); 735 return NULL; 736 } 737 return hdr; 738 } 739 #endif 740 741 /** 742 * boot_get_kernel - find kernel image 743 * @os_data: pointer to a ulong variable, will hold os data start address 744 * @os_len: pointer to a ulong variable, will hold os data length 745 * 746 * boot_get_kernel() tries to find a kernel image, verifies its integrity 747 * and locates kernel data. 748 * 749 * returns: 750 * pointer to image header if valid image was found, plus kernel start 751 * address and length, otherwise NULL 752 */ 753 static const void *boot_get_kernel(cmd_tbl_t *cmdtp, int flag, int argc, 754 char * const argv[], bootm_headers_t *images, 755 ulong *os_data, ulong *os_len) 756 { 757 #if defined(CONFIG_IMAGE_FORMAT_LEGACY) 758 image_header_t *hdr; 759 #endif 760 ulong img_addr; 761 const void *buf; 762 const char *fit_uname_config = NULL; 763 const char *fit_uname_kernel = NULL; 764 #if defined(CONFIG_FIT) 765 int os_noffset; 766 #endif 767 768 img_addr = genimg_get_kernel_addr_fit(argc < 1 ? NULL : argv[0], 769 &fit_uname_config, 770 &fit_uname_kernel); 771 772 bootstage_mark(BOOTSTAGE_ID_CHECK_MAGIC); 773 774 /* copy from dataflash if needed */ 775 img_addr = genimg_get_image(img_addr); 776 777 /* check image type, for FIT images get FIT kernel node */ 778 *os_data = *os_len = 0; 779 buf = map_sysmem(img_addr, 0); 780 switch (genimg_get_format(buf)) { 781 #if defined(CONFIG_IMAGE_FORMAT_LEGACY) 782 case IMAGE_FORMAT_LEGACY: 783 printf("## Booting kernel from Legacy Image at %08lx ...\n", 784 img_addr); 785 hdr = image_get_kernel(img_addr, images->verify); 786 if (!hdr) 787 return NULL; 788 bootstage_mark(BOOTSTAGE_ID_CHECK_IMAGETYPE); 789 790 /* get os_data and os_len */ 791 switch (image_get_type(hdr)) { 792 case IH_TYPE_KERNEL: 793 case IH_TYPE_KERNEL_NOLOAD: 794 *os_data = image_get_data(hdr); 795 *os_len = image_get_data_size(hdr); 796 break; 797 case IH_TYPE_MULTI: 798 image_multi_getimg(hdr, 0, os_data, os_len); 799 break; 800 case IH_TYPE_STANDALONE: 801 *os_data = image_get_data(hdr); 802 *os_len = image_get_data_size(hdr); 803 break; 804 default: 805 printf("Wrong Image Type for %s command\n", 806 cmdtp->name); 807 bootstage_error(BOOTSTAGE_ID_CHECK_IMAGETYPE); 808 return NULL; 809 } 810 811 /* 812 * copy image header to allow for image overwrites during 813 * kernel decompression. 814 */ 815 memmove(&images->legacy_hdr_os_copy, hdr, 816 sizeof(image_header_t)); 817 818 /* save pointer to image header */ 819 images->legacy_hdr_os = hdr; 820 821 images->legacy_hdr_valid = 1; 822 bootstage_mark(BOOTSTAGE_ID_DECOMP_IMAGE); 823 break; 824 #endif 825 #if defined(CONFIG_FIT) 826 case IMAGE_FORMAT_FIT: 827 os_noffset = fit_image_load(images, img_addr, 828 &fit_uname_kernel, &fit_uname_config, 829 IH_ARCH_DEFAULT, IH_TYPE_KERNEL, 830 BOOTSTAGE_ID_FIT_KERNEL_START, 831 FIT_LOAD_IGNORED, os_data, os_len); 832 if (os_noffset < 0) 833 return NULL; 834 835 images->fit_hdr_os = map_sysmem(img_addr, 0); 836 images->fit_uname_os = fit_uname_kernel; 837 images->fit_uname_cfg = fit_uname_config; 838 images->fit_noffset_os = os_noffset; 839 break; 840 #endif 841 #ifdef CONFIG_ANDROID_BOOT_IMAGE 842 case IMAGE_FORMAT_ANDROID: 843 printf("## Booting Android Image at 0x%08lx ...\n", img_addr); 844 if (android_image_get_kernel(buf, images->verify, 845 os_data, os_len)) 846 return NULL; 847 break; 848 #endif 849 default: 850 printf("Wrong Image Format for %s command\n", cmdtp->name); 851 bootstage_error(BOOTSTAGE_ID_FIT_KERNEL_INFO); 852 return NULL; 853 } 854 855 debug(" kernel data at 0x%08lx, len = 0x%08lx (%ld)\n", 856 *os_data, *os_len, *os_len); 857 858 return buf; 859 } 860 #else /* USE_HOSTCC */ 861 862 void memmove_wd(void *to, void *from, size_t len, ulong chunksz) 863 { 864 memmove(to, from, len); 865 } 866 867 static int bootm_host_load_image(const void *fit, int req_image_type) 868 { 869 const char *fit_uname_config = NULL; 870 ulong data, len; 871 bootm_headers_t images; 872 int noffset; 873 ulong load_end; 874 uint8_t image_type; 875 uint8_t imape_comp; 876 void *load_buf; 877 int ret; 878 879 memset(&images, '\0', sizeof(images)); 880 images.verify = 1; 881 noffset = fit_image_load(&images, (ulong)fit, 882 NULL, &fit_uname_config, 883 IH_ARCH_DEFAULT, req_image_type, -1, 884 FIT_LOAD_IGNORED, &data, &len); 885 if (noffset < 0) 886 return noffset; 887 if (fit_image_get_type(fit, noffset, &image_type)) { 888 puts("Can't get image type!\n"); 889 return -EINVAL; 890 } 891 892 if (fit_image_get_comp(fit, noffset, &imape_comp)) { 893 puts("Can't get image compression!\n"); 894 return -EINVAL; 895 } 896 897 /* Allow the image to expand by a factor of 4, should be safe */ 898 load_buf = malloc((1 << 20) + len * 4); 899 ret = bootm_decomp_image(imape_comp, 0, data, image_type, load_buf, 900 (void *)data, len, CONFIG_SYS_BOOTM_LEN, 901 &load_end); 902 free(load_buf); 903 904 if (ret && ret != BOOTM_ERR_UNIMPLEMENTED) 905 return ret; 906 907 return 0; 908 } 909 910 int bootm_host_load_images(const void *fit, int cfg_noffset) 911 { 912 static uint8_t image_types[] = { 913 IH_TYPE_KERNEL, 914 IH_TYPE_FLATDT, 915 IH_TYPE_RAMDISK, 916 }; 917 int err = 0; 918 int i; 919 920 for (i = 0; i < ARRAY_SIZE(image_types); i++) { 921 int ret; 922 923 ret = bootm_host_load_image(fit, image_types[i]); 924 if (!err && ret && ret != -ENOENT) 925 err = ret; 926 } 927 928 /* Return the first error we found */ 929 return err; 930 } 931 932 #endif /* ndef USE_HOSTCC */ 933