[people/ms/u-boot.git] / common / spl / spl_ubi.c

/*
 * Copyright (C) 2016
 * Ladislav Michl <ladis@linux-mips.org>
 *
 * SPDX-License-Identifier: GPL 2.0+ BSD-3-Clause
 */

#include <common.h>
#include <config.h>
#include <nand.h>
#include <onenand_uboot.h>
#include <ubispl.h>
#include <spl.h>

int spl_ubi_load_image(struct spl_image_info *spl_image,
		       struct spl_boot_device *bootdev)
{
	struct image_header *header;
	struct ubispl_info info;
	struct ubispl_load volumes[2];
	int ret = 1;

	switch (bootdev->boot_device) {
#ifdef CONFIG_SPL_NAND_SUPPORT
	case BOOT_DEVICE_NAND:
		nand_init();
		info.read = nand_spl_read_block;
		info.peb_size = CONFIG_SYS_NAND_BLOCK_SIZE;
		break;
#endif
#ifdef CONFIG_SPL_ONENAND_SUPPORT
	case BOOT_DEVICE_ONENAND:
		info.read = onenand_spl_read_block;
		info.peb_size = CONFIG_SYS_ONENAND_BLOCK_SIZE;
		break;
#endif
	default:
		goto out;
	}
	info.ubi = (struct ubi_scan_info *)CONFIG_SPL_UBI_INFO_ADDR;
	info.fastmap = IS_ENABLED(CONFIG_MTD_UBI_FASTMAP);

	info.peb_offset = CONFIG_SPL_UBI_PEB_OFFSET;
	info.vid_offset = CONFIG_SPL_UBI_VID_OFFSET;
	info.leb_start = CONFIG_SPL_UBI_LEB_START;
	info.peb_count = CONFIG_SPL_UBI_MAX_PEBS - info.peb_offset;

#ifdef CONFIG_SPL_OS_BOOT
	if (!spl_start_uboot()) {
		volumes[0].vol_id = CONFIG_SPL_UBI_LOAD_KERNEL_ID;
		volumes[0].load_addr = (void *)CONFIG_SYS_LOAD_ADDR;
		volumes[1].vol_id = CONFIG_SPL_UBI_LOAD_ARGS_ID;
		volumes[1].load_addr = (void *)CONFIG_SYS_SPL_ARGS_ADDR;

		ret = ubispl_load_volumes(&info, volumes, 2);
		if (!ret) {
			header = (struct image_header *)volumes[0].load_addr;
			spl_parse_image_header(spl_image, header);
			puts("Linux loaded.\n");
			goto out;
		}
		puts("Loading Linux failed, falling back to U-Boot.\n");
	}
#endif
	header = (struct image_header *)
		(CONFIG_SYS_TEXT_BASE - sizeof(struct image_header));
	volumes[0].vol_id = CONFIG_SPL_UBI_LOAD_MONITOR_ID;
	volumes[0].load_addr = (void *)header;

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