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