spl: Rework Kconfig to be more menu driven
[platform/kernel/u-boot.git] / common / spl / spl_spi.c
1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * Copyright (C) 2011 OMICRON electronics GmbH
4  *
5  * based on drivers/mtd/nand/raw/nand_spl_load.c
6  *
7  * Copyright (C) 2011
8  * Heiko Schocher, DENX Software Engineering, hs@denx.de.
9  */
10
11 #include <common.h>
12 #include <image.h>
13 #include <log.h>
14 #include <spi.h>
15 #include <spi_flash.h>
16 #include <errno.h>
17 #include <spl.h>
18 #include <asm/global_data.h>
19 #include <dm/ofnode.h>
20
21 #if CONFIG_IS_ENABLED(OS_BOOT)
22 /*
23  * Load the kernel, check for a valid header we can parse, and if found load
24  * the kernel and then device tree.
25  */
26 static int spi_load_image_os(struct spl_image_info *spl_image,
27                              struct spl_boot_device *bootdev,
28                              struct spi_flash *flash,
29                              struct image_header *header)
30 {
31         int err;
32
33         /* Read for a header, parse or error out. */
34         spi_flash_read(flash, CONFIG_SYS_SPI_KERNEL_OFFS, sizeof(*header),
35                        (void *)header);
36
37         if (image_get_magic(header) != IH_MAGIC)
38                 return -1;
39
40         err = spl_parse_image_header(spl_image, bootdev, header);
41         if (err)
42                 return err;
43
44         spi_flash_read(flash, CONFIG_SYS_SPI_KERNEL_OFFS,
45                        spl_image->size, (void *)spl_image->load_addr);
46
47         /* Read device tree. */
48         spi_flash_read(flash, CONFIG_SYS_SPI_ARGS_OFFS,
49                        CONFIG_SYS_SPI_ARGS_SIZE,
50                        (void *)CONFIG_SYS_SPL_ARGS_ADDR);
51
52         return 0;
53 }
54 #endif
55
56 static ulong spl_spi_fit_read(struct spl_load_info *load, ulong sector,
57                               ulong count, void *buf)
58 {
59         struct spi_flash *flash = load->dev;
60         ulong ret;
61
62         ret = spi_flash_read(flash, sector, count, buf);
63         if (!ret)
64                 return count;
65         else
66                 return 0;
67 }
68
69 unsigned int __weak spl_spi_get_uboot_offs(struct spi_flash *flash)
70 {
71         return CONFIG_SYS_SPI_U_BOOT_OFFS;
72 }
73
74 /*
75  * The main entry for SPI booting. It's necessary that SDRAM is already
76  * configured and available since this code loads the main U-Boot image
77  * from SPI into SDRAM and starts it from there.
78  */
79 static int spl_spi_load_image(struct spl_image_info *spl_image,
80                               struct spl_boot_device *bootdev)
81 {
82         int err = 0;
83         unsigned int payload_offs;
84         struct spi_flash *flash;
85         struct image_header *header;
86
87         /*
88          * Load U-Boot image from SPI flash into RAM
89          * In DM mode: defaults speed and mode will be
90          * taken from DT when available
91          */
92
93         flash = spi_flash_probe(CONFIG_SF_DEFAULT_BUS,
94                                 CONFIG_SF_DEFAULT_CS,
95                                 CONFIG_SF_DEFAULT_SPEED,
96                                 CONFIG_SF_DEFAULT_MODE);
97         if (!flash) {
98                 puts("SPI probe failed.\n");
99                 return -ENODEV;
100         }
101
102         payload_offs = spl_spi_get_uboot_offs(flash);
103
104         header = spl_get_load_buffer(-sizeof(*header), sizeof(*header));
105
106         if (CONFIG_IS_ENABLED(OF_REAL)) {
107                 payload_offs = ofnode_conf_read_int("u-boot,spl-payload-offset",
108                                                     payload_offs);
109         }
110
111 #if CONFIG_IS_ENABLED(OS_BOOT)
112         if (spl_start_uboot() || spi_load_image_os(spl_image, bootdev, flash, header))
113 #endif
114         {
115                 /* Load u-boot, mkimage header is 64 bytes. */
116                 err = spi_flash_read(flash, payload_offs, sizeof(*header),
117                                      (void *)header);
118                 if (err) {
119                         debug("%s: Failed to read from SPI flash (err=%d)\n",
120                               __func__, err);
121                         return err;
122                 }
123
124                 if (IS_ENABLED(CONFIG_SPL_LOAD_FIT_FULL) &&
125                     image_get_magic(header) == FDT_MAGIC) {
126                         err = spi_flash_read(flash, payload_offs,
127                                              roundup(fdt_totalsize(header), 4),
128                                              (void *)CONFIG_SYS_LOAD_ADDR);
129                         if (err)
130                                 return err;
131                         err = spl_parse_image_header(spl_image, bootdev,
132                                         (struct image_header *)CONFIG_SYS_LOAD_ADDR);
133                 } else if (IS_ENABLED(CONFIG_SPL_LOAD_FIT) &&
134                            image_get_magic(header) == FDT_MAGIC) {
135                         struct spl_load_info load;
136
137                         debug("Found FIT\n");
138                         load.dev = flash;
139                         load.priv = NULL;
140                         load.filename = NULL;
141                         load.bl_len = 1;
142                         load.read = spl_spi_fit_read;
143                         err = spl_load_simple_fit(spl_image, &load,
144                                                   payload_offs,
145                                                   header);
146                 } else if (IS_ENABLED(CONFIG_SPL_LOAD_IMX_CONTAINER)) {
147                         struct spl_load_info load;
148
149                         load.dev = flash;
150                         load.priv = NULL;
151                         load.filename = NULL;
152                         load.bl_len = 1;
153                         load.read = spl_spi_fit_read;
154
155                         err = spl_load_imx_container(spl_image, &load,
156                                                      payload_offs);
157                 } else {
158                         err = spl_parse_image_header(spl_image, bootdev, header);
159                         if (err)
160                                 return err;
161                         err = spi_flash_read(flash, payload_offs + spl_image->offset,
162                                              spl_image->size,
163                                              (void *)spl_image->load_addr);
164                 }
165         }
166
167         return err;
168 }
169 /* Use priorty 1 so that boards can override this */
170 SPL_LOAD_IMAGE_METHOD("SPI", 1, BOOT_DEVICE_SPI, spl_spi_load_image);