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 <bootstage.h>
7 #include <command.h>
8 #include <cpu_func.h>
9 #include <image.h>
10 #include <log.h>
11 #include <part.h>
12
common_diskboot(struct cmd_tbl * cmdtp,const char * intf,int argc,char * const argv[])13 int common_diskboot(struct cmd_tbl *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 struct disk_partition info;
21 #if defined(CONFIG_LEGACY_IMAGE_FORMAT)
22 struct legacy_img_hdr *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 = hextoul(argv[1], NULL);
39
40 bootstage_mark(BOOTSTAGE_ID_IDE_BOOT_DEVICE);
41
42 part = blk_get_device_part_str(intf, cmd_arg2(argc, argv),
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_LEGACY_IMAGE_FORMAT)
69 case IMAGE_FORMAT_LEGACY:
70 hdr = (struct legacy_img_hdr *)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, IMAGE_SIZE_INVAL)) {
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 }
123 #endif
124
125 flush_cache(addr, (cnt+1)*info.blksz);
126
127 /* Loading ok, update default load address */
128 image_load_addr = addr;
129
130 return bootm_maybe_autostart(cmdtp, argv[0]);
131 }
132