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