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

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

#include <common.h>
#include <config.h>
#include <image.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 legacy_img_hdr *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 = CFG_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;

#if CONFIG_IS_ENABLED(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_SPL_PAYLOAD_ARGS_ADDR;

		ret = ubispl_load_volumes(&info, volumes, 2);
		if (!ret) {
			header = (struct legacy_img_hdr *)volumes[0].load_addr;
			spl_parse_image_header(spl_image, bootdev, header);
			puts("Linux loaded.\n");
			goto out;
		}
		puts("Loading Linux failed, falling back to U-Boot.\n");
	}
#endif
	header = spl_get_load_buffer(-sizeof(*header), sizeof(header));
#ifdef CONFIG_SPL_UBI_LOAD_BY_VOLNAME
	volumes[0].vol_id = -1;
	strncpy(volumes[0].name,
		CONFIG_SPL_UBI_LOAD_MONITOR_VOLNAME,
		UBI_VOL_NAME_MAX + 1);
#else
	volumes[0].vol_id = CONFIG_SPL_UBI_LOAD_MONITOR_ID;
#endif
	volumes[0].load_addr = (void *)header;

	ret = ubispl_load_volumes(&info, volumes, 1);
	if (!ret)
		spl_parse_image_header(spl_image, bootdev, 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
4549e789	1	// SPDX-License-Identifier: GPL-2.0+ OR BSD-3-Clause
bf55cd4f LM	2	/*
	3	* Copyright (C) 2016
	4	* Ladislav Michl <ladis@linux-mips.org>
bf55cd4f LM	5	*/
bf55cd4f LM	6
d678a59d	7	#include <common.h>
bf55cd4f	8	#include <config.h>
4d72caa5	9	#include <image.h>
bf55cd4f LM	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	17	{
f3543e69	18	struct legacy_img_hdr *header;
bf55cd4f LM	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;
65cc0e2a	34	info.peb_size = CFG_SYS_ONENAND_BLOCK_SIZE;
bf55cd4f LM	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
7115007c	48	#if CONFIG_IS_ENABLED(OS_BOOT)
bf55cd4f LM	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;
9cbdc3a0	53	volumes[1].load_addr = (void *)CONFIG_SPL_PAYLOAD_ARGS_ADDR;
bf55cd4f LM	54
	55	ret = ubispl_load_volumes(&info, volumes, 2);
	56	if (!ret) {
f3543e69	57	header = (struct legacy_img_hdr *)volumes[0].load_addr;
2e0429bc	58	spl_parse_image_header(spl_image, bootdev, 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
04ce5427	65	header = spl_get_load_buffer(-sizeof(*header), sizeof(header));
6ea31cce HG	66	#ifdef CONFIG_SPL_UBI_LOAD_BY_VOLNAME
	67	volumes[0].vol_id = -1;
	68	strncpy(volumes[0].name,
	69	CONFIG_SPL_UBI_LOAD_MONITOR_VOLNAME,
	70	UBI_VOL_NAME_MAX + 1);
	71	#else
bf55cd4f	72	volumes[0].vol_id = CONFIG_SPL_UBI_LOAD_MONITOR_ID;
6ea31cce	73	#endif
bf55cd4f LM	74	volumes[0].load_addr = (void *)header;
	75
	76	ret = ubispl_load_volumes(&info, volumes, 1);
	77	if (!ret)
2e0429bc	78	spl_parse_image_header(spl_image, bootdev, header);
bf55cd4f LM	79	out:
bf55cd4f LM	80	#ifdef CONFIG_SPL_NAND_SUPPORT
ecdfd69a	81	if (bootdev->boot_device == BOOT_DEVICE_NAND)
bf55cd4f LM	82	nand_deselect();
	83	#endif
	84	return ret;
	85	}
7d7dd821	86	/* Use priorty 0 so that Ubi will override NAND and ONENAND methods */
ebc4ef61 SG	87	SPL_LOAD_IMAGE_METHOD("NAND", 0, BOOT_DEVICE_NAND, spl_ubi_load_image);
ebc4ef61 SG	88	SPL_LOAD_IMAGE_METHOD("OneNAND", 0, BOOT_DEVICE_ONENAND, spl_ubi_load_image);