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