xref: /rk3399_rockchip-uboot/common/spl/spl_mmc.c (revision fbf99d9116409fd59686dfd3f891547b111d65cb)
1 /*
2  * (C) Copyright 2010
3  * Texas Instruments, <www.ti.com>
4  *
5  * Aneesh V <aneesh@ti.com>
6  *
7  * SPDX-License-Identifier:	GPL-2.0+
8  */
9 #include <common.h>
10 #include <dm.h>
11 #include <spl.h>
12 #include <spl_rkfw.h>
13 #include <linux/compiler.h>
14 #include <errno.h>
15 #include <asm/u-boot.h>
16 #include <errno.h>
17 #include <mmc.h>
18 #include <image.h>
19 
20 DECLARE_GLOBAL_DATA_PTR;
21 
22 static int mmc_load_legacy(struct spl_image_info *spl_image, struct mmc *mmc,
23 			   ulong sector, struct image_header *header)
24 {
25 	u32 image_size_sectors;
26 	unsigned long count;
27 	int ret;
28 
29 	ret = spl_parse_image_header(spl_image, header);
30 	if (ret)
31 		return ret;
32 
33 	/* convert size to sectors - round up */
34 	image_size_sectors = (spl_image->size + mmc->read_bl_len - 1) /
35 			     mmc->read_bl_len;
36 
37 	/* Read the header too to avoid extra memcpy */
38 	count = blk_dread(mmc_get_blk_desc(mmc), sector, image_size_sectors,
39 			  (void *)(ulong)spl_image->load_addr);
40 	debug("read %x sectors to %lx\n", image_size_sectors,
41 	      spl_image->load_addr);
42 	if (count != image_size_sectors)
43 		return -EIO;
44 
45 	return 0;
46 }
47 
48 static ulong h_spl_load_read(struct spl_load_info *load, ulong sector,
49 			     ulong count, void *buf)
50 {
51 	struct mmc *mmc = load->dev;
52 
53 	return blk_dread(mmc_get_blk_desc(mmc), sector, count, buf);
54 }
55 
56 static __maybe_unused
57 int mmc_load_image_raw_sector(struct spl_image_info *spl_image,
58 			      struct mmc *mmc, unsigned long sector)
59 {
60 	unsigned long count;
61 	struct image_header *header;
62 	int ret = 0;
63 
64 #ifdef CONFIG_SPL_LOAD_RKFW
65 	struct spl_load_info load;
66 
67 	load.dev = mmc;
68 	load.priv = NULL;
69 	load.filename = NULL;
70 	load.bl_len = mmc->read_bl_len;
71 	load.read = h_spl_load_read;
72 
73 	ret = spl_load_rkfw_image(spl_image, &load,
74 				  CONFIG_RKFW_TRUST_SECTOR,
75 				  CONFIG_RKFW_U_BOOT_SECTOR);
76 	/* If boot successfully or can't try others, just go end */
77 	if (!ret || ret != -EAGAIN)
78 		goto end;
79 #endif
80 
81 	header = (struct image_header *)(CONFIG_SYS_TEXT_BASE -
82 					 sizeof(struct image_header));
83 
84 	/* read image header to find the image size & load address */
85 	count = blk_dread(mmc_get_blk_desc(mmc), sector, 1, header);
86 	debug("hdr read sector %lx, count=%lu\n", sector, count);
87 	if (count == 0) {
88 		ret = -EIO;
89 		goto end;
90 	}
91 
92 	if (IS_ENABLED(CONFIG_SPL_LOAD_FIT) &&
93 	    image_get_magic(header) == FDT_MAGIC) {
94 		struct spl_load_info load;
95 
96 		debug("Found FIT\n");
97 		load.dev = mmc;
98 		load.priv = NULL;
99 		load.filename = NULL;
100 		load.bl_len = mmc->read_bl_len;
101 		load.read = h_spl_load_read;
102 		ret = spl_load_simple_fit(spl_image, &load, sector, header);
103 	} else {
104 		ret = mmc_load_legacy(spl_image, mmc, sector, header);
105 	}
106 
107 end:
108 	if (ret) {
109 #ifdef CONFIG_SPL_LIBCOMMON_SUPPORT
110 		puts("mmc_load_image_raw_sector: mmc block read error\n");
111 #endif
112 		return -1;
113 	}
114 
115 	return 0;
116 }
117 
118 static int spl_mmc_get_device_index(u32 boot_device)
119 {
120 	switch (boot_device) {
121 	case BOOT_DEVICE_MMC1:
122 		return 0;
123 	case BOOT_DEVICE_MMC2:
124 	case BOOT_DEVICE_MMC2_2:
125 		return 1;
126 	}
127 
128 #ifdef CONFIG_SPL_LIBCOMMON_SUPPORT
129 	printf("spl: unsupported mmc boot device.\n");
130 #endif
131 
132 	return -ENODEV;
133 }
134 
135 static int spl_mmc_find_device(struct mmc **mmcp, u32 boot_device)
136 {
137 	int err, mmc_dev;
138 
139 	mmc_dev = spl_mmc_get_device_index(boot_device);
140 	if (mmc_dev < 0)
141 		return mmc_dev;
142 
143 	err = mmc_initialize(NULL);
144 	if (err) {
145 #ifdef CONFIG_SPL_LIBCOMMON_SUPPORT
146 		printf("spl: could not initialize mmc. error: %d\n", err);
147 #endif
148 		return err;
149 	}
150 
151 	*mmcp = find_mmc_device(mmc_dev);
152 	err = *mmcp ? 0 : -ENODEV;
153 	if (err) {
154 #ifdef CONFIG_SPL_LIBCOMMON_SUPPORT
155 		printf("spl: could not find mmc device. error: %d\n", err);
156 #endif
157 		return err;
158 	}
159 
160 	return 0;
161 }
162 
163 #ifdef CONFIG_SYS_MMCSD_RAW_MODE_U_BOOT_USE_PARTITION
164 static int mmc_load_image_raw_partition(struct spl_image_info *spl_image,
165 					struct mmc *mmc, int partition)
166 {
167 	disk_partition_t info;
168 	int err;
169 
170 #ifdef CONFIG_SYS_MMCSD_RAW_MODE_U_BOOT_USE_PARTITION_TYPE
171 	int type_part;
172 	/* Only support MBR so DOS_ENTRY_NUMBERS */
173 	for (type_part = 1; type_part <= DOS_ENTRY_NUMBERS; type_part++) {
174 		err = part_get_info(mmc_get_blk_desc(mmc), type_part, &info);
175 		if (err)
176 			continue;
177 		if (info.sys_ind ==
178 			CONFIG_SYS_MMCSD_RAW_MODE_U_BOOT_PARTITION_TYPE) {
179 			partition = type_part;
180 			break;
181 		}
182 	}
183 #endif
184 
185 	err = part_get_info(mmc_get_blk_desc(mmc), partition, &info);
186 	if (err) {
187 #ifdef CONFIG_SPL_LIBCOMMON_SUPPORT
188 		puts("spl: partition error\n");
189 #endif
190 		return -1;
191 	}
192 
193 #ifdef CONFIG_SYS_MMCSD_RAW_MODE_U_BOOT_USE_SECTOR
194 	return mmc_load_image_raw_sector(spl_image, mmc,
195 			info.start + CONFIG_SYS_MMCSD_RAW_MODE_U_BOOT_SECTOR);
196 #else
197 	return mmc_load_image_raw_sector(spl_image, mmc, info.start);
198 #endif
199 }
200 #endif
201 
202 #ifdef CONFIG_SPL_OS_BOOT
203 static int mmc_load_image_raw_os(struct spl_image_info *spl_image,
204 				 struct mmc *mmc)
205 {
206 	unsigned long count;
207 	int ret;
208 
209 	count = blk_dread(mmc_get_blk_desc(mmc),
210 		CONFIG_SYS_MMCSD_RAW_MODE_ARGS_SECTOR,
211 		CONFIG_SYS_MMCSD_RAW_MODE_ARGS_SECTORS,
212 		(void *) CONFIG_SYS_SPL_ARGS_ADDR);
213 	if (count != CONFIG_SYS_MMCSD_RAW_MODE_ARGS_SECTORS) {
214 #ifdef CONFIG_SPL_LIBCOMMON_SUPPORT
215 		puts("mmc_load_image_raw_os: mmc block read error\n");
216 #endif
217 		return -1;
218 	}
219 
220 	ret = mmc_load_image_raw_sector(spl_image, mmc,
221 		CONFIG_SYS_MMCSD_RAW_MODE_KERNEL_SECTOR);
222 	if (ret)
223 		return ret;
224 
225 	if (spl_image->os != IH_OS_LINUX) {
226 		puts("Expected Linux image is not found. Trying to start U-boot\n");
227 		return -ENOENT;
228 	}
229 
230 	return 0;
231 }
232 #else
233 int spl_start_uboot(void)
234 {
235 	return 1;
236 }
237 static int mmc_load_image_raw_os(struct spl_image_info *spl_image,
238 				 struct mmc *mmc)
239 {
240 	return -ENOSYS;
241 }
242 #endif
243 
244 #ifdef CONFIG_SYS_MMCSD_FS_BOOT_PARTITION
245 static int spl_mmc_do_fs_boot(struct spl_image_info *spl_image, struct mmc *mmc)
246 {
247 	int err = -ENOSYS;
248 
249 #ifdef CONFIG_SPL_FAT_SUPPORT
250 	if (!spl_start_uboot()) {
251 		err = spl_load_image_fat_os(spl_image, mmc_get_blk_desc(mmc),
252 			CONFIG_SYS_MMCSD_FS_BOOT_PARTITION);
253 		if (!err)
254 			return err;
255 	}
256 #ifdef CONFIG_SPL_FS_LOAD_PAYLOAD_NAME
257 	err = spl_load_image_fat(spl_image, mmc_get_blk_desc(mmc),
258 				 CONFIG_SYS_MMCSD_FS_BOOT_PARTITION,
259 				 CONFIG_SPL_FS_LOAD_PAYLOAD_NAME);
260 	if (!err)
261 		return err;
262 #endif
263 #endif
264 #ifdef CONFIG_SPL_EXT_SUPPORT
265 	if (!spl_start_uboot()) {
266 		err = spl_load_image_ext_os(spl_image, mmc_get_blk_desc(mmc),
267 			CONFIG_SYS_MMCSD_FS_BOOT_PARTITION);
268 		if (!err)
269 			return err;
270 	}
271 #ifdef CONFIG_SPL_FS_LOAD_PAYLOAD_NAME
272 	err = spl_load_image_ext(spl_image, mmc_get_blk_desc(mmc),
273 				 CONFIG_SYS_MMCSD_FS_BOOT_PARTITION,
274 				 CONFIG_SPL_FS_LOAD_PAYLOAD_NAME);
275 	if (!err)
276 		return err;
277 #endif
278 #endif
279 
280 #if defined(CONFIG_SPL_FAT_SUPPORT) || defined(CONFIG_SPL_EXT_SUPPORT)
281 	err = -ENOENT;
282 #endif
283 
284 	return err;
285 }
286 #else
287 static int spl_mmc_do_fs_boot(struct spl_image_info *spl_image, struct mmc *mmc)
288 {
289 	return -ENOSYS;
290 }
291 #endif
292 
293 int spl_mmc_load_image(struct spl_image_info *spl_image,
294 		       struct spl_boot_device *bootdev)
295 {
296 	struct mmc *mmc = NULL;
297 	u32 boot_mode;
298 	int err = 0;
299 	__maybe_unused int part;
300 
301 	err = spl_mmc_find_device(&mmc, bootdev->boot_device);
302 	if (err)
303 		return err;
304 
305 	err = mmc_init(mmc);
306 	if (err) {
307 #ifdef CONFIG_SPL_LIBCOMMON_SUPPORT
308 		printf("spl: mmc init failed with error: %d\n", err);
309 #endif
310 		return err;
311 	}
312 
313 	boot_mode = spl_boot_mode(bootdev->boot_device);
314 	err = -EINVAL;
315 	switch (boot_mode) {
316 	case MMCSD_MODE_EMMCBOOT:
317 			/*
318 			 * We need to check what the partition is configured to.
319 			 * 1 and 2 match up to boot0 / boot1 and 7 is user data
320 			 * which is the first physical partition (0).
321 			 */
322 			part = (mmc->part_config >> 3) & PART_ACCESS_MASK;
323 
324 			if (part == 7)
325 				part = 0;
326 
327 			if (CONFIG_IS_ENABLED(MMC_TINY))
328 				err = mmc_switch_part(mmc, part);
329 			else
330 				err = blk_dselect_hwpart(mmc_get_blk_desc(mmc), part);
331 
332 			if (err) {
333 #ifdef CONFIG_SPL_LIBCOMMON_SUPPORT
334 				puts("spl: mmc partition switch failed\n");
335 #endif
336 				return err;
337 			}
338 			/* Fall through */
339 	case MMCSD_MODE_RAW:
340 		debug("spl: mmc boot mode: raw\n");
341 
342 		if (!spl_start_uboot()) {
343 			err = mmc_load_image_raw_os(spl_image, mmc);
344 			if (!err)
345 				return err;
346 		}
347 #ifdef CONFIG_SYS_MMCSD_RAW_MODE_U_BOOT_USE_PARTITION
348 		err = mmc_load_image_raw_partition(spl_image, mmc,
349 			CONFIG_SYS_MMCSD_RAW_MODE_U_BOOT_PARTITION);
350 		if (!err)
351 			return err;
352 #endif
353 #ifdef CONFIG_SYS_MMCSD_RAW_MODE_U_BOOT_USE_SECTOR
354 		err = mmc_load_image_raw_sector(spl_image, mmc,
355 			CONFIG_SYS_MMCSD_RAW_MODE_U_BOOT_SECTOR);
356 		if (!err)
357 			return err;
358 #endif
359 		/* If RAW mode fails, try FS mode. */
360 	case MMCSD_MODE_FS:
361 		debug("spl: mmc boot mode: fs\n");
362 
363 		err = spl_mmc_do_fs_boot(spl_image, mmc);
364 		if (!err)
365 			return err;
366 
367 		break;
368 #ifdef CONFIG_SPL_LIBCOMMON_SUPPORT
369 	default:
370 		puts("spl: mmc: wrong boot mode\n");
371 #endif
372 	}
373 
374 	return err;
375 }
376 
377 SPL_LOAD_IMAGE_METHOD("MMC1", 0, BOOT_DEVICE_MMC1, spl_mmc_load_image);
378 SPL_LOAD_IMAGE_METHOD("MMC2", 0, BOOT_DEVICE_MMC2, spl_mmc_load_image);
379 SPL_LOAD_IMAGE_METHOD("MMC2_2", 0, BOOT_DEVICE_MMC2_2, spl_mmc_load_image);
380