xref: /rk3399_rockchip-uboot/common/spl/spl_ubi.c (revision ecdfd69a4be55363589e8185ff151b02e6c36cfa)
1bf55cd4fSLadislav Michl /*
2bf55cd4fSLadislav Michl  * Copyright (C) 2016
3bf55cd4fSLadislav Michl  * Ladislav Michl <ladis@linux-mips.org>
4bf55cd4fSLadislav Michl  *
5bf55cd4fSLadislav Michl  * SPDX-License-Identifier: GPL 2.0+ BSD-3-Clause
6bf55cd4fSLadislav Michl  */
7bf55cd4fSLadislav Michl 
8bf55cd4fSLadislav Michl #include <common.h>
9bf55cd4fSLadislav Michl #include <config.h>
10bf55cd4fSLadislav Michl #include <nand.h>
11bf55cd4fSLadislav Michl #include <onenand_uboot.h>
12bf55cd4fSLadislav Michl #include <ubispl.h>
13bf55cd4fSLadislav Michl #include <spl.h>
14bf55cd4fSLadislav Michl 
15*ecdfd69aSSimon Glass int spl_ubi_load_image(struct spl_boot_device *bootdev)
16bf55cd4fSLadislav Michl {
17bf55cd4fSLadislav Michl 	struct image_header *header;
18bf55cd4fSLadislav Michl 	struct ubispl_info info;
19bf55cd4fSLadislav Michl 	struct ubispl_load volumes[2];
20bf55cd4fSLadislav Michl 	int ret = 1;
21bf55cd4fSLadislav Michl 
22*ecdfd69aSSimon Glass 	switch (bootdev->boot_device) {
23bf55cd4fSLadislav Michl #ifdef CONFIG_SPL_NAND_SUPPORT
24bf55cd4fSLadislav Michl 	case BOOT_DEVICE_NAND:
25bf55cd4fSLadislav Michl 		nand_init();
26bf55cd4fSLadislav Michl 		info.read = nand_spl_read_block;
27bf55cd4fSLadislav Michl 		info.peb_size = CONFIG_SYS_NAND_BLOCK_SIZE;
28bf55cd4fSLadislav Michl 		break;
29bf55cd4fSLadislav Michl #endif
30bf55cd4fSLadislav Michl #ifdef CONFIG_SPL_ONENAND_SUPPORT
31bf55cd4fSLadislav Michl 	case BOOT_DEVICE_ONENAND:
32bf55cd4fSLadislav Michl 		info.read = onenand_spl_read_block;
33bf55cd4fSLadislav Michl 		info.peb_size = CONFIG_SYS_ONENAND_BLOCK_SIZE;
34bf55cd4fSLadislav Michl 		break;
35bf55cd4fSLadislav Michl #endif
36bf55cd4fSLadislav Michl 	default:
37bf55cd4fSLadislav Michl 		goto out;
38bf55cd4fSLadislav Michl 	}
39bf55cd4fSLadislav Michl 	info.ubi = (struct ubi_scan_info *)CONFIG_SPL_UBI_INFO_ADDR;
40bf55cd4fSLadislav Michl 	info.fastmap = 1;
41bf55cd4fSLadislav Michl 
42bf55cd4fSLadislav Michl 	info.peb_offset = CONFIG_SPL_UBI_PEB_OFFSET;
43bf55cd4fSLadislav Michl 	info.vid_offset = CONFIG_SPL_UBI_VID_OFFSET;
44bf55cd4fSLadislav Michl 	info.leb_start = CONFIG_SPL_UBI_LEB_START;
45bf55cd4fSLadislav Michl 	info.peb_count = CONFIG_SPL_UBI_MAX_PEBS - info.peb_offset;
46bf55cd4fSLadislav Michl 
47bf55cd4fSLadislav Michl #ifdef CONFIG_SPL_OS_BOOT
48bf55cd4fSLadislav Michl 	if (!spl_start_uboot()) {
49bf55cd4fSLadislav Michl 		volumes[0].vol_id = CONFIG_SPL_UBI_LOAD_KERNEL_ID;
50bf55cd4fSLadislav Michl 		volumes[0].load_addr = (void *)CONFIG_SYS_LOAD_ADDR;
51bf55cd4fSLadislav Michl 		volumes[1].vol_id = CONFIG_SPL_UBI_LOAD_ARGS_ID;
52bf55cd4fSLadislav Michl 		volumes[1].load_addr = (void *)CONFIG_SYS_SPL_ARGS_ADDR;
53bf55cd4fSLadislav Michl 
54bf55cd4fSLadislav Michl 		ret = ubispl_load_volumes(&info, volumes, 2);
55bf55cd4fSLadislav Michl 		if (!ret) {
56bf55cd4fSLadislav Michl 			header = (struct image_header *)volumes[0].load_addr;
5771316c1dSSimon Glass 			spl_parse_image_header(&spl_image, header);
58bf55cd4fSLadislav Michl 			puts("Linux loaded.\n");
59bf55cd4fSLadislav Michl 			goto out;
60bf55cd4fSLadislav Michl 		}
61bf55cd4fSLadislav Michl 		puts("Loading Linux failed, falling back to U-Boot.\n");
62bf55cd4fSLadislav Michl 	}
63bf55cd4fSLadislav Michl #endif
64bf55cd4fSLadislav Michl 	header = (struct image_header *)
65bf55cd4fSLadislav Michl 		(CONFIG_SYS_TEXT_BASE - sizeof(struct image_header));
66bf55cd4fSLadislav Michl 	volumes[0].vol_id = CONFIG_SPL_UBI_LOAD_MONITOR_ID;
67bf55cd4fSLadislav Michl 	volumes[0].load_addr = (void *)header;
68bf55cd4fSLadislav Michl 
69bf55cd4fSLadislav Michl 	ret = ubispl_load_volumes(&info, volumes, 1);
70bf55cd4fSLadislav Michl 	if (!ret)
7171316c1dSSimon Glass 		spl_parse_image_header(&spl_image, header);
72bf55cd4fSLadislav Michl out:
73bf55cd4fSLadislav Michl #ifdef CONFIG_SPL_NAND_SUPPORT
74*ecdfd69aSSimon Glass 	if (bootdev->boot_device == BOOT_DEVICE_NAND)
75bf55cd4fSLadislav Michl 		nand_deselect();
76bf55cd4fSLadislav Michl #endif
77bf55cd4fSLadislav Michl 	return ret;
78bf55cd4fSLadislav Michl }
79