spl: Pass the loader into spl_load_image()
[platform/kernel/u-boot.git] / common / spl / spl.c
1 /*
2  * (C) Copyright 2010
3  * Texas Instruments, <www.ti.com>
4  *
5  * Aneesh V <aneesh@ti.com>
6  *
7  * SPDX-License-Identifier:     GPL-2.0+
8  */
9 #include <common.h>
10 #include <dm.h>
11 #include <spl.h>
12 #include <asm/u-boot.h>
13 #include <nand.h>
14 #include <fat.h>
15 #include <version.h>
16 #include <image.h>
17 #include <malloc.h>
18 #include <dm/root.h>
19 #include <linux/compiler.h>
20
21 DECLARE_GLOBAL_DATA_PTR;
22
23 #ifndef CONFIG_SYS_UBOOT_START
24 #define CONFIG_SYS_UBOOT_START  CONFIG_SYS_TEXT_BASE
25 #endif
26 #ifndef CONFIG_SYS_MONITOR_LEN
27 /* Unknown U-Boot size, let's assume it will not be more than 200 KB */
28 #define CONFIG_SYS_MONITOR_LEN  (200 * 1024)
29 #endif
30
31 u32 *boot_params_ptr = NULL;
32
33 /* Define board data structure */
34 static bd_t bdata __attribute__ ((section(".data")));
35
36 /*
37  * Board-specific Platform code can reimplement show_boot_progress () if needed
38  */
39 __weak void show_boot_progress(int val) {}
40
41 /*
42  * Default function to determine if u-boot or the OS should
43  * be started. This implementation always returns 1.
44  *
45  * Please implement your own board specific funcion to do this.
46  *
47  * RETURN
48  * 0 to not start u-boot
49  * positive if u-boot should start
50  */
51 #ifdef CONFIG_SPL_OS_BOOT
52 __weak int spl_start_uboot(void)
53 {
54         puts("SPL: Please implement spl_start_uboot() for your board\n");
55         puts("SPL: Direct Linux boot not active!\n");
56         return 1;
57 }
58
59 /*
60  * Weak default function for arch specific zImage check. Return zero
61  * and fill start and end address if image is recognized.
62  */
63 int __weak bootz_setup(ulong image, ulong *start, ulong *end)
64 {
65          return 1;
66 }
67 #endif
68
69 /*
70  * Weak default function for board specific cleanup/preparation before
71  * Linux boot. Some boards/platforms might not need it, so just provide
72  * an empty stub here.
73  */
74 __weak void spl_board_prepare_for_linux(void)
75 {
76         /* Nothing to do! */
77 }
78
79 __weak void spl_board_prepare_for_boot(void)
80 {
81         /* Nothing to do! */
82 }
83
84 void spl_set_header_raw_uboot(struct spl_image_info *spl_image)
85 {
86         spl_image->size = CONFIG_SYS_MONITOR_LEN;
87         spl_image->entry_point = CONFIG_SYS_UBOOT_START;
88         spl_image->load_addr = CONFIG_SYS_TEXT_BASE;
89         spl_image->os = IH_OS_U_BOOT;
90         spl_image->name = "U-Boot";
91 }
92
93 int spl_parse_image_header(struct spl_image_info *spl_image,
94                            const struct image_header *header)
95 {
96         u32 header_size = sizeof(struct image_header);
97
98         if (image_get_magic(header) == IH_MAGIC) {
99                 if (spl_image->flags & SPL_COPY_PAYLOAD_ONLY) {
100                         /*
101                          * On some system (e.g. powerpc), the load-address and
102                          * entry-point is located at address 0. We can't load
103                          * to 0-0x40. So skip header in this case.
104                          */
105                         spl_image->load_addr = image_get_load(header);
106                         spl_image->entry_point = image_get_ep(header);
107                         spl_image->size = image_get_data_size(header);
108                 } else {
109                         spl_image->entry_point = image_get_load(header);
110                         /* Load including the header */
111                         spl_image->load_addr = spl_image->entry_point -
112                                 header_size;
113                         spl_image->size = image_get_data_size(header) +
114                                 header_size;
115                 }
116                 spl_image->os = image_get_os(header);
117                 spl_image->name = image_get_name(header);
118                 debug("spl: payload image: %.*s load addr: 0x%x size: %d\n",
119                         (int)sizeof(spl_image->name), spl_image->name,
120                         spl_image->load_addr, spl_image->size);
121         } else {
122 #ifdef CONFIG_SPL_PANIC_ON_RAW_IMAGE
123                 /*
124                  * CONFIG_SPL_PANIC_ON_RAW_IMAGE is defined when the
125                  * code which loads images in SPL cannot guarantee that
126                  * absolutely all read errors will be reported.
127                  * An example is the LPC32XX MLC NAND driver, which
128                  * will consider that a completely unreadable NAND block
129                  * is bad, and thus should be skipped silently.
130                  */
131                 panic("** no mkimage signature but raw image not supported");
132 #endif
133
134 #ifdef CONFIG_SPL_OS_BOOT
135                 ulong start, end;
136
137                 if (!bootz_setup((ulong)header, &start, &end)) {
138                         spl_image->name = "Linux";
139                         spl_image->os = IH_OS_LINUX;
140                         spl_image->load_addr = CONFIG_SYS_LOAD_ADDR;
141                         spl_image->entry_point = CONFIG_SYS_LOAD_ADDR;
142                         spl_image->size = end - start;
143                         debug("spl: payload zImage, load addr: 0x%x size: %d\n",
144                               spl_image->load_addr, spl_image->size);
145                         return 0;
146                 }
147 #endif
148
149 #ifdef CONFIG_SPL_ABORT_ON_RAW_IMAGE
150                 /* Signature not found, proceed to other boot methods. */
151                 return -EINVAL;
152 #else
153                 /* Signature not found - assume u-boot.bin */
154                 debug("mkimage signature not found - ih_magic = %x\n",
155                         header->ih_magic);
156                 spl_set_header_raw_uboot(spl_image);
157 #endif
158         }
159         return 0;
160 }
161
162 __weak void __noreturn jump_to_image_no_args(struct spl_image_info *spl_image)
163 {
164         typedef void __noreturn (*image_entry_noargs_t)(void);
165
166         image_entry_noargs_t image_entry =
167                 (image_entry_noargs_t)(unsigned long)spl_image->entry_point;
168
169         debug("image entry point: 0x%X\n", spl_image->entry_point);
170         image_entry();
171 }
172
173 #ifndef CONFIG_SPL_LOAD_FIT_ADDRESS
174 # define CONFIG_SPL_LOAD_FIT_ADDRESS    0
175 #endif
176
177 #if defined(CONFIG_SPL_RAM_DEVICE) || defined(CONFIG_SPL_DFU_SUPPORT)
178 static ulong spl_ram_load_read(struct spl_load_info *load, ulong sector,
179                                ulong count, void *buf)
180 {
181         debug("%s: sector %lx, count %lx, buf %lx\n",
182               __func__, sector, count, (ulong)buf);
183         memcpy(buf, (void *)(CONFIG_SPL_LOAD_FIT_ADDRESS + sector), count);
184         return count;
185 }
186
187 static int spl_ram_load_image(struct spl_image_info *spl_image,
188                               struct spl_boot_device *bootdev)
189 {
190         struct image_header *header;
191
192         header = (struct image_header *)CONFIG_SPL_LOAD_FIT_ADDRESS;
193
194 #if defined(CONFIG_SPL_DFU_SUPPORT)
195         if (bootdev->boot_device == BOOT_DEVICE_DFU)
196                 spl_dfu_cmd(0, "dfu_alt_info_ram", "ram", "0");
197 #endif
198
199         if (IS_ENABLED(CONFIG_SPL_LOAD_FIT) &&
200             image_get_magic(header) == FDT_MAGIC) {
201                 struct spl_load_info load;
202
203                 debug("Found FIT\n");
204                 load.bl_len = 1;
205                 load.read = spl_ram_load_read;
206                 spl_load_simple_fit(spl_image, &load, 0, header);
207         } else {
208                 debug("Legacy image\n");
209                 /*
210                  * Get the header.  It will point to an address defined by
211                  * handoff which will tell where the image located inside
212                  * the flash. For now, it will temporary fixed to address
213                  * pointed by U-Boot.
214                  */
215                 header = (struct image_header *)
216                         (CONFIG_SYS_TEXT_BASE - sizeof(struct image_header));
217
218                 spl_parse_image_header(spl_image, header);
219         }
220
221         return 0;
222 }
223 #if defined(CONFIG_SPL_RAM_DEVICE)
224 SPL_LOAD_IMAGE_METHOD("RAM", 0, BOOT_DEVICE_RAM, spl_ram_load_image);
225 #endif
226 #if defined(CONFIG_SPL_DFU_SUPPORT)
227 SPL_LOAD_IMAGE_METHOD("USB DFU", 0, BOOT_DEVICE_DFU, spl_ram_load_image);
228 #endif
229 #endif
230
231 int spl_init(void)
232 {
233         int ret;
234
235         debug("spl_init()\n");
236 #if defined(CONFIG_SYS_MALLOC_F_LEN)
237 #ifdef CONFIG_MALLOC_F_ADDR
238         gd->malloc_base = CONFIG_MALLOC_F_ADDR;
239 #endif
240         gd->malloc_limit = CONFIG_SYS_MALLOC_F_LEN;
241         gd->malloc_ptr = 0;
242 #endif
243         if (CONFIG_IS_ENABLED(OF_CONTROL) && !CONFIG_IS_ENABLED(OF_PLATDATA)) {
244                 ret = fdtdec_setup();
245                 if (ret) {
246                         debug("fdtdec_setup() returned error %d\n", ret);
247                         return ret;
248                 }
249         }
250         if (IS_ENABLED(CONFIG_SPL_DM)) {
251                 /* With CONFIG_OF_PLATDATA, bring in all devices */
252                 ret = dm_init_and_scan(!CONFIG_IS_ENABLED(OF_PLATDATA));
253                 if (ret) {
254                         debug("dm_init_and_scan() returned error %d\n", ret);
255                         return ret;
256                 }
257         }
258         gd->flags |= GD_FLG_SPL_INIT;
259
260         return 0;
261 }
262
263 #ifndef BOOT_DEVICE_NONE
264 #define BOOT_DEVICE_NONE 0xdeadbeef
265 #endif
266
267 __weak void board_boot_order(u32 *spl_boot_list)
268 {
269         spl_boot_list[0] = spl_boot_device();
270 }
271
272 #ifdef CONFIG_SPL_BOARD_LOAD_IMAGE
273 __weak void spl_board_announce_boot_device(void) { }
274 #endif
275
276 #ifdef CONFIG_SPL_LIBCOMMON_SUPPORT
277 struct boot_device_name {
278         u32 boot_dev;
279         const char *name;
280 };
281
282 struct boot_device_name boot_name_table[] = {
283 #ifdef CONFIG_SPL_RAM_DEVICE
284         { BOOT_DEVICE_RAM, "RAM" },
285 #endif
286 #ifdef CONFIG_SPL_MMC_SUPPORT
287         { BOOT_DEVICE_MMC1, "MMC1" },
288         { BOOT_DEVICE_MMC2, "MMC2" },
289         { BOOT_DEVICE_MMC2_2, "MMC2_2" },
290 #endif
291 #ifdef CONFIG_SPL_NAND_SUPPORT
292         { BOOT_DEVICE_NAND, "NAND" },
293 #endif
294 #ifdef CONFIG_SPL_ONENAND_SUPPORT
295         { BOOT_DEVICE_ONENAND, "OneNAND" },
296 #endif
297 #ifdef CONFIG_SPL_NOR_SUPPORT
298         { BOOT_DEVICE_NOR, "NOR" },
299 #endif
300 #ifdef CONFIG_SPL_YMODEM_SUPPORT
301         { BOOT_DEVICE_UART, "UART" },
302 #endif
303 #if defined(CONFIG_SPL_SPI_SUPPORT) || defined(CONFIG_SPL_SPI_FLASH_SUPPORT)
304         { BOOT_DEVICE_SPI, "SPI" },
305 #endif
306 #ifdef CONFIG_SPL_ETH_SUPPORT
307 #ifdef CONFIG_SPL_ETH_DEVICE
308         { BOOT_DEVICE_CPGMAC, "eth device" },
309 #else
310         { BOOT_DEVICE_CPGMAC, "net" },
311 #endif
312 #endif
313 #ifdef CONFIG_SPL_USBETH_SUPPORT
314         { BOOT_DEVICE_USBETH, "USB eth" },
315 #endif
316 #ifdef CONFIG_SPL_USB_SUPPORT
317         { BOOT_DEVICE_USB, "USB" },
318 #endif
319 #ifdef CONFIG_SPL_DFU_SUPPORT
320         { BOOT_DEVICE_DFU, "USB DFU" },
321 #endif
322 #ifdef CONFIG_SPL_SATA_SUPPORT
323         { BOOT_DEVICE_SATA, "SATA" },
324 #endif
325         /* Keep this entry last */
326         { BOOT_DEVICE_NONE, "unknown boot device" },
327 };
328
329 static void announce_boot_device(u32 boot_device)
330 {
331         int i;
332
333         puts("Trying to boot from ");
334
335 #ifdef CONFIG_SPL_BOARD_LOAD_IMAGE
336         if (boot_device == BOOT_DEVICE_BOARD) {
337                 spl_board_announce_boot_device();
338                 puts("\n");
339                 return;
340         }
341 #endif
342         for (i = 0; i < ARRAY_SIZE(boot_name_table) - 1; i++) {
343                 if (boot_name_table[i].boot_dev == boot_device)
344                         break;
345         }
346
347         printf("%s\n", boot_name_table[i].name);
348 }
349 #else
350 static inline void announce_boot_device(u32 boot_device) { }
351 #endif
352
353 static struct spl_image_loader *spl_ll_find_loader(uint boot_device)
354 {
355         struct spl_image_loader *drv =
356                 ll_entry_start(struct spl_image_loader, spl_image_loader);
357         const int n_ents =
358                 ll_entry_count(struct spl_image_loader, spl_image_loader);
359         struct spl_image_loader *entry;
360
361         for (entry = drv; entry != drv + n_ents; entry++) {
362                 if (boot_device == entry->boot_device)
363                         return entry;
364         }
365
366         /* Not found */
367         return NULL;
368 }
369
370 static int spl_load_image(struct spl_image_info *spl_image,
371                           struct spl_image_loader *loader)
372 {
373         struct spl_boot_device bootdev;
374
375         bootdev.boot_device = loader->boot_device;
376         bootdev.boot_device_name = NULL;
377
378         return loader->load_image(spl_image, &bootdev);
379 }
380
381 /**
382  * boot_from_devices() - Try loading an booting U-Boot from a list of devices
383  *
384  * @spl_image: Place to put the image details if successful
385  * @spl_boot_list: List of boot devices to try
386  * @count: Number of elements in spl_boot_list
387  * @return 0 if OK, -ve on error
388  */
389 static int boot_from_devices(struct spl_image_info *spl_image,
390                              u32 spl_boot_list[], int count)
391 {
392         int i;
393
394         for (i = 0; i < count && spl_boot_list[i] != BOOT_DEVICE_NONE; i++) {
395                 struct spl_image_loader *loader;
396
397                 announce_boot_device(spl_boot_list[i]);
398                 loader = spl_ll_find_loader(spl_boot_list[i]);
399 #if defined(CONFIG_SPL_SERIAL_SUPPORT) && defined(CONFIG_SPL_LIBCOMMON_SUPPORT)
400                 if (!loader)
401                         puts("SPL: Unsupported Boot Device!\n");
402 #endif
403                 if (loader && !spl_load_image(spl_image, loader))
404                         return 0;
405         }
406
407         return -ENODEV;
408 }
409
410 void board_init_r(gd_t *dummy1, ulong dummy2)
411 {
412         u32 spl_boot_list[] = {
413                 BOOT_DEVICE_NONE,
414                 BOOT_DEVICE_NONE,
415                 BOOT_DEVICE_NONE,
416                 BOOT_DEVICE_NONE,
417                 BOOT_DEVICE_NONE,
418         };
419         struct spl_image_info spl_image;
420
421         debug(">>spl:board_init_r()\n");
422
423 #if defined(CONFIG_SYS_SPL_MALLOC_START)
424         mem_malloc_init(CONFIG_SYS_SPL_MALLOC_START,
425                         CONFIG_SYS_SPL_MALLOC_SIZE);
426         gd->flags |= GD_FLG_FULL_MALLOC_INIT;
427 #endif
428         if (!(gd->flags & GD_FLG_SPL_INIT)) {
429                 if (spl_init())
430                         hang();
431         }
432 #ifndef CONFIG_PPC
433         /*
434          * timer_init() does not exist on PPC systems. The timer is initialized
435          * and enabled (decrementer) in interrupt_init() here.
436          */
437         timer_init();
438 #endif
439
440 #ifdef CONFIG_SPL_BOARD_INIT
441         spl_board_init();
442 #endif
443
444         memset(&spl_image, '\0', sizeof(spl_image));
445         board_boot_order(spl_boot_list);
446
447         if (boot_from_devices(&spl_image, spl_boot_list,
448                               ARRAY_SIZE(spl_boot_list))) {
449                 puts("SPL: failed to boot from all boot devices\n");
450                 hang();
451         }
452
453         switch (spl_image.os) {
454         case IH_OS_U_BOOT:
455                 debug("Jumping to U-Boot\n");
456                 break;
457 #ifdef CONFIG_SPL_OS_BOOT
458         case IH_OS_LINUX:
459                 debug("Jumping to Linux\n");
460                 spl_board_prepare_for_linux();
461                 jump_to_image_linux(&spl_image,
462                                     (void *)CONFIG_SYS_SPL_ARGS_ADDR);
463 #endif
464         default:
465                 debug("Unsupported OS image.. Jumping nevertheless..\n");
466         }
467 #if defined(CONFIG_SYS_MALLOC_F_LEN) && !defined(CONFIG_SYS_SPL_MALLOC_SIZE)
468         debug("SPL malloc() used %#lx bytes (%ld KB)\n", gd->malloc_ptr,
469               gd->malloc_ptr / 1024);
470 #endif
471
472         debug("loaded - jumping to U-Boot...");
473         spl_board_prepare_for_boot();
474         jump_to_image_no_args(&spl_image);
475 }
476
477 /*
478  * This requires UART clocks to be enabled.  In order for this to work the
479  * caller must ensure that the gd pointer is valid.
480  */
481 void preloader_console_init(void)
482 {
483         gd->bd = &bdata;
484         gd->baudrate = CONFIG_BAUDRATE;
485
486         serial_init();          /* serial communications setup */
487
488         gd->have_console = 1;
489
490         puts("\nU-Boot SPL " PLAIN_VERSION " (" U_BOOT_DATE " - " \
491                         U_BOOT_TIME ")\n");
492 #ifdef CONFIG_SPL_DISPLAY_PRINT
493         spl_display_print();
494 #endif
495 }
496
497 /**
498  * spl_relocate_stack_gd() - Relocate stack ready for board_init_r() execution
499  *
500  * Sometimes board_init_f() runs with a stack in SRAM but we want to use SDRAM
501  * for the main board_init_r() execution. This is typically because we need
502  * more stack space for things like the MMC sub-system.
503  *
504  * This function calculates the stack position, copies the global_data into
505  * place, sets the new gd (except for ARM, for which setting GD within a C
506  * function may not always work) and returns the new stack position. The
507  * caller is responsible for setting up the sp register and, in the case
508  * of ARM, setting up gd.
509  *
510  * All of this is done using the same layout and alignments as done in
511  * board_init_f_init_reserve() / board_init_f_alloc_reserve().
512  *
513  * @return new stack location, or 0 to use the same stack
514  */
515 ulong spl_relocate_stack_gd(void)
516 {
517 #ifdef CONFIG_SPL_STACK_R
518         gd_t *new_gd;
519         ulong ptr = CONFIG_SPL_STACK_R_ADDR;
520
521 #ifdef CONFIG_SPL_SYS_MALLOC_SIMPLE
522         if (CONFIG_SPL_STACK_R_MALLOC_SIMPLE_LEN) {
523                 ptr -= CONFIG_SPL_STACK_R_MALLOC_SIMPLE_LEN;
524                 gd->malloc_base = ptr;
525                 gd->malloc_limit = CONFIG_SPL_STACK_R_MALLOC_SIMPLE_LEN;
526                 gd->malloc_ptr = 0;
527         }
528 #endif
529         /* Get stack position: use 8-byte alignment for ABI compliance */
530         ptr = CONFIG_SPL_STACK_R_ADDR - roundup(sizeof(gd_t),16);
531         new_gd = (gd_t *)ptr;
532         memcpy(new_gd, (void *)gd, sizeof(gd_t));
533 #if CONFIG_IS_ENABLED(DM)
534         dm_fixup_for_gd_move(new_gd);
535 #endif
536 #if !defined(CONFIG_ARM)
537         gd = new_gd;
538 #endif
539         return ptr;
540 #else
541         return 0;
542 #endif
543 }