1 /* 2 * (C) Copyright 2000-2011 3 * Wolfgang Denk, DENX Software Engineering, wd@denx.de. 4 * 5 * SPDX-License-Identifier: GPL-2.0+ 6 */ 7 #include <common.h> 8 #include <command.h> 9 #include <part.h> 10 11 int common_diskboot(cmd_tbl_t *cmdtp, const char *intf, int argc, 12 char *const argv[]) 13 { 14 __maybe_unused int dev; 15 int part; 16 ulong addr = CONFIG_SYS_LOAD_ADDR; 17 ulong cnt; 18 disk_partition_t info; 19 #if defined(CONFIG_IMAGE_FORMAT_LEGACY) 20 image_header_t *hdr; 21 #endif 22 struct blk_desc *dev_desc; 23 24 #if CONFIG_IS_ENABLED(FIT) 25 const void *fit_hdr = NULL; 26 #endif 27 28 bootstage_mark(BOOTSTAGE_ID_IDE_START); 29 if (argc > 3) { 30 bootstage_error(BOOTSTAGE_ID_IDE_ADDR); 31 return CMD_RET_USAGE; 32 } 33 bootstage_mark(BOOTSTAGE_ID_IDE_ADDR); 34 35 if (argc > 1) 36 addr = simple_strtoul(argv[1], NULL, 16); 37 38 bootstage_mark(BOOTSTAGE_ID_IDE_BOOT_DEVICE); 39 40 part = blk_get_device_part_str(intf, (argc == 3) ? argv[2] : NULL, 41 &dev_desc, &info, 1); 42 if (part < 0) { 43 bootstage_error(BOOTSTAGE_ID_IDE_TYPE); 44 return 1; 45 } 46 47 dev = dev_desc->devnum; 48 bootstage_mark(BOOTSTAGE_ID_IDE_TYPE); 49 50 printf("\nLoading from %s device %d, partition %d: " 51 "Name: %.32s Type: %.32s\n", intf, dev, part, info.name, 52 info.type); 53 54 debug("First Block: " LBAFU ", # of blocks: " LBAFU 55 ", Block Size: %ld\n", 56 info.start, info.size, info.blksz); 57 58 if (blk_dread(dev_desc, info.start, 1, (ulong *)addr) != 1) { 59 printf("** Read error on %d:%d\n", dev, part); 60 bootstage_error(BOOTSTAGE_ID_IDE_PART_READ); 61 return 1; 62 } 63 bootstage_mark(BOOTSTAGE_ID_IDE_PART_READ); 64 65 switch (genimg_get_format((void *) addr)) { 66 #if defined(CONFIG_IMAGE_FORMAT_LEGACY) 67 case IMAGE_FORMAT_LEGACY: 68 hdr = (image_header_t *) addr; 69 70 bootstage_mark(BOOTSTAGE_ID_IDE_FORMAT); 71 72 if (!image_check_hcrc(hdr)) { 73 puts("\n** Bad Header Checksum **\n"); 74 bootstage_error(BOOTSTAGE_ID_IDE_CHECKSUM); 75 return 1; 76 } 77 bootstage_mark(BOOTSTAGE_ID_IDE_CHECKSUM); 78 79 image_print_contents(hdr); 80 81 cnt = image_get_image_size(hdr); 82 break; 83 #endif 84 #if CONFIG_IS_ENABLED(FIT) 85 case IMAGE_FORMAT_FIT: 86 fit_hdr = (const void *) addr; 87 puts("Fit image detected...\n"); 88 89 cnt = fit_get_size(fit_hdr); 90 break; 91 #endif 92 default: 93 bootstage_error(BOOTSTAGE_ID_IDE_FORMAT); 94 puts("** Unknown image type\n"); 95 return 1; 96 } 97 98 cnt += info.blksz - 1; 99 cnt /= info.blksz; 100 cnt -= 1; 101 102 if (blk_dread(dev_desc, info.start + 1, cnt, 103 (ulong *)(addr + info.blksz)) != cnt) { 104 printf("** Read error on %d:%d\n", dev, part); 105 bootstage_error(BOOTSTAGE_ID_IDE_READ); 106 return 1; 107 } 108 bootstage_mark(BOOTSTAGE_ID_IDE_READ); 109 110 #if CONFIG_IS_ENABLED(FIT) 111 /* This cannot be done earlier, 112 * we need complete FIT image in RAM first */ 113 if (genimg_get_format((void *) addr) == IMAGE_FORMAT_FIT) { 114 if (!fit_check_format(fit_hdr)) { 115 bootstage_error(BOOTSTAGE_ID_IDE_FIT_READ); 116 puts("** Bad FIT image format\n"); 117 return 1; 118 } 119 bootstage_mark(BOOTSTAGE_ID_IDE_FIT_READ_OK); 120 fit_print_contents(fit_hdr); 121 } 122 #endif 123 124 flush_cache(addr, (cnt+1)*info.blksz); 125 126 /* Loading ok, update default load address */ 127 load_addr = addr; 128 129 return bootm_maybe_autostart(cmdtp, argv[0]); 130 } 131