1*4882a593Smuzhiyun /*
2*4882a593Smuzhiyun * Copyright (c) 2020 Rockchip Electronics Co., Ltd
3*4882a593Smuzhiyun *
4*4882a593Smuzhiyun * SPDX-License-Identifier: GPL-2.0+
5*4882a593Smuzhiyun */
6*4882a593Smuzhiyun
7*4882a593Smuzhiyun #include <common.h>
8*4882a593Smuzhiyun #include <blk.h>
9*4882a593Smuzhiyun #include <boot_rkimg.h>
10*4882a593Smuzhiyun #include <dm.h>
11*4882a593Smuzhiyun #include <errno.h>
12*4882a593Smuzhiyun #include <image.h>
13*4882a593Smuzhiyun #include <malloc.h>
14*4882a593Smuzhiyun #include <mtd_blk.h>
15*4882a593Smuzhiyun #include <part.h>
16*4882a593Smuzhiyun #include <spl.h>
17*4882a593Smuzhiyun #include <spl_ab.h>
18*4882a593Smuzhiyun #include <spl_rkfw.h>
19*4882a593Smuzhiyun #include <asm/u-boot.h>
20*4882a593Smuzhiyun #include <dm/device-internal.h>
21*4882a593Smuzhiyun #include <linux/compiler.h>
22*4882a593Smuzhiyun
find_rknand_device(int dev_num)23*4882a593Smuzhiyun static struct blk_desc *find_rknand_device(int dev_num)
24*4882a593Smuzhiyun {
25*4882a593Smuzhiyun struct udevice *dev;
26*4882a593Smuzhiyun struct blk_desc *desc;
27*4882a593Smuzhiyun int ret;
28*4882a593Smuzhiyun
29*4882a593Smuzhiyun ret = blk_find_device(IF_TYPE_RKNAND, dev_num, &dev);
30*4882a593Smuzhiyun if (ret)
31*4882a593Smuzhiyun return NULL;
32*4882a593Smuzhiyun
33*4882a593Smuzhiyun ret = device_probe(dev);
34*4882a593Smuzhiyun if (ret) {
35*4882a593Smuzhiyun debug("RKNAND Device %d not found\n", dev_num);
36*4882a593Smuzhiyun return NULL;
37*4882a593Smuzhiyun }
38*4882a593Smuzhiyun
39*4882a593Smuzhiyun desc = dev_get_uclass_platdata(dev);
40*4882a593Smuzhiyun if (!desc)
41*4882a593Smuzhiyun return NULL;
42*4882a593Smuzhiyun
43*4882a593Smuzhiyun return desc;
44*4882a593Smuzhiyun }
45*4882a593Smuzhiyun
h_spl_load_read(struct spl_load_info * load,ulong sector,ulong count,void * buf)46*4882a593Smuzhiyun static ulong h_spl_load_read(struct spl_load_info *load, ulong sector,
47*4882a593Smuzhiyun ulong count, void *buf)
48*4882a593Smuzhiyun {
49*4882a593Smuzhiyun return blk_dread(load->dev, sector, count, buf);
50*4882a593Smuzhiyun }
51*4882a593Smuzhiyun
spl_rknand_load_image(struct spl_image_info * spl_image,struct spl_boot_device * bootdev)52*4882a593Smuzhiyun static int spl_rknand_load_image(struct spl_image_info *spl_image,
53*4882a593Smuzhiyun struct spl_boot_device *bootdev)
54*4882a593Smuzhiyun {
55*4882a593Smuzhiyun lbaint_t image_sector = CONFIG_RKNAND_BLK_U_BOOT_OFFS;
56*4882a593Smuzhiyun struct image_header *header;
57*4882a593Smuzhiyun struct spl_load_info load;
58*4882a593Smuzhiyun struct blk_desc *desc;
59*4882a593Smuzhiyun int ret = -1;
60*4882a593Smuzhiyun
61*4882a593Smuzhiyun desc = find_rknand_device(0);
62*4882a593Smuzhiyun if (!desc)
63*4882a593Smuzhiyun return -ENODEV;
64*4882a593Smuzhiyun
65*4882a593Smuzhiyun load.dev = desc;
66*4882a593Smuzhiyun load.priv = NULL;
67*4882a593Smuzhiyun load.filename = NULL;
68*4882a593Smuzhiyun load.bl_len = desc->blksz;
69*4882a593Smuzhiyun load.read = h_spl_load_read;
70*4882a593Smuzhiyun
71*4882a593Smuzhiyun #ifdef CONFIG_SPL_LIBDISK_SUPPORT
72*4882a593Smuzhiyun disk_partition_t info;
73*4882a593Smuzhiyun
74*4882a593Smuzhiyun ret = part_get_info_by_name(desc, PART_UBOOT, &info);
75*4882a593Smuzhiyun if (ret > 0)
76*4882a593Smuzhiyun image_sector = info.start;
77*4882a593Smuzhiyun #endif
78*4882a593Smuzhiyun
79*4882a593Smuzhiyun if (IS_ENABLED(CONFIG_SPL_LOAD_FIT)) {
80*4882a593Smuzhiyun header = (struct image_header *)(CONFIG_SYS_TEXT_BASE -
81*4882a593Smuzhiyun sizeof(struct image_header));
82*4882a593Smuzhiyun ret = blk_dread(desc, image_sector, 1, header);
83*4882a593Smuzhiyun if (ret != 1)
84*4882a593Smuzhiyun return -ENODEV;
85*4882a593Smuzhiyun
86*4882a593Smuzhiyun #ifdef CONFIG_SPL_FIT_IMAGE_MULTIPLE
87*4882a593Smuzhiyun if (image_get_magic(header) == FDT_MAGIC ||
88*4882a593Smuzhiyun CONFIG_SPL_FIT_IMAGE_MULTIPLE > 1) {
89*4882a593Smuzhiyun #else
90*4882a593Smuzhiyun if (image_get_magic(header) == FDT_MAGIC) {
91*4882a593Smuzhiyun #endif
92*4882a593Smuzhiyun ret = spl_load_simple_fit(spl_image, &load,
93*4882a593Smuzhiyun image_sector,
94*4882a593Smuzhiyun header);
95*4882a593Smuzhiyun }
96*4882a593Smuzhiyun }
97*4882a593Smuzhiyun
98*4882a593Smuzhiyun if (!ret)
99*4882a593Smuzhiyun return 0;
100*4882a593Smuzhiyun
101*4882a593Smuzhiyun #ifdef CONFIG_SPL_LOAD_RKFW
102*4882a593Smuzhiyun ret = spl_load_rkfw_image(spl_image, &load);
103*4882a593Smuzhiyun #endif
104*4882a593Smuzhiyun
105*4882a593Smuzhiyun return ret;
106*4882a593Smuzhiyun }
107*4882a593Smuzhiyun
108*4882a593Smuzhiyun SPL_LOAD_IMAGE_METHOD("RKNAND", 0, BOOT_DEVICE_RKNAND, spl_rknand_load_image);
109