1*4882a593Smuzhiyun /*
2*4882a593Smuzhiyun * (C) Copyright 2019 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 <android_ab.h>
9*4882a593Smuzhiyun #include <bootm.h>
10*4882a593Smuzhiyun #include <boot_rkimg.h>
11*4882a593Smuzhiyun #include <image.h>
12*4882a593Smuzhiyun #include <malloc.h>
13*4882a593Smuzhiyun #include <sysmem.h>
14*4882a593Smuzhiyun #include <asm/arch/fit.h>
15*4882a593Smuzhiyun
16*4882a593Smuzhiyun DECLARE_GLOBAL_DATA_PTR;
17*4882a593Smuzhiyun
do_boot_fit_storage(ulong * size)18*4882a593Smuzhiyun static void *do_boot_fit_storage(ulong *size)
19*4882a593Smuzhiyun {
20*4882a593Smuzhiyun return fit_image_load_bootables(size);
21*4882a593Smuzhiyun }
22*4882a593Smuzhiyun
do_boot_fit_ram(char * const argv[],ulong * data_size)23*4882a593Smuzhiyun static void *do_boot_fit_ram(char *const argv[], ulong *data_size)
24*4882a593Smuzhiyun {
25*4882a593Smuzhiyun void *fit;
26*4882a593Smuzhiyun int size;
27*4882a593Smuzhiyun
28*4882a593Smuzhiyun fit = (void *)simple_strtoul(argv[1], NULL, 16);
29*4882a593Smuzhiyun if (!fit || fdt_check_header(fit)) {
30*4882a593Smuzhiyun FIT_I("Invalid header\n");
31*4882a593Smuzhiyun return NULL;
32*4882a593Smuzhiyun }
33*4882a593Smuzhiyun
34*4882a593Smuzhiyun size = fit_image_get_bootables_size(fit);
35*4882a593Smuzhiyun if (!size) {
36*4882a593Smuzhiyun FIT_I("Failed to get bootable image size\n");
37*4882a593Smuzhiyun return NULL;
38*4882a593Smuzhiyun }
39*4882a593Smuzhiyun
40*4882a593Smuzhiyun /* reserve this full FIT image */
41*4882a593Smuzhiyun if (!sysmem_alloc_base(MEM_FIT_USER,
42*4882a593Smuzhiyun (phys_addr_t)fit, ALIGN(size, 512)))
43*4882a593Smuzhiyun return NULL;
44*4882a593Smuzhiyun
45*4882a593Smuzhiyun *data_size = size;
46*4882a593Smuzhiyun
47*4882a593Smuzhiyun return fit;
48*4882a593Smuzhiyun }
49*4882a593Smuzhiyun
50*4882a593Smuzhiyun /*
51*4882a593Smuzhiyun * argc == 1:
52*4882a593Smuzhiyun * FIT image is loaded from storage(eg. CONFIG_BOOTCOMMAND).
53*4882a593Smuzhiyun *
54*4882a593Smuzhiyun * argc == 2:
55*4882a593Smuzhiyun * FIT image is already in ram, the booflow is:
56*4882a593Smuzhiyun * CLI cmd "bootm <fit_addr>" => do_bootm() =>
57*4882a593Smuzhiyun * board_do_bootm() => boot_fit <fit_addr>
58*4882a593Smuzhiyun */
do_boot_fit(cmd_tbl_t * cmdtp,int flag,int argc,char * const argv[])59*4882a593Smuzhiyun static int do_boot_fit(cmd_tbl_t *cmdtp, int flag, int argc, char *const argv[])
60*4882a593Smuzhiyun {
61*4882a593Smuzhiyun char *bootm_args[1];
62*4882a593Smuzhiyun char fit_addr[12];
63*4882a593Smuzhiyun ulong size;
64*4882a593Smuzhiyun void *fit;
65*4882a593Smuzhiyun int ret;
66*4882a593Smuzhiyun
67*4882a593Smuzhiyun if (argc > 2)
68*4882a593Smuzhiyun return CMD_RET_USAGE;
69*4882a593Smuzhiyun
70*4882a593Smuzhiyun printf("## Booting FIT Image ");
71*4882a593Smuzhiyun
72*4882a593Smuzhiyun if (argc == 1)
73*4882a593Smuzhiyun fit = do_boot_fit_storage(&size);
74*4882a593Smuzhiyun else
75*4882a593Smuzhiyun fit = do_boot_fit_ram(argv, &size);
76*4882a593Smuzhiyun
77*4882a593Smuzhiyun if (!fit) {
78*4882a593Smuzhiyun FIT_I("No FIT image\n");
79*4882a593Smuzhiyun goto out;
80*4882a593Smuzhiyun }
81*4882a593Smuzhiyun
82*4882a593Smuzhiyun if (fdt_check_header(fit)) {
83*4882a593Smuzhiyun FIT_I("Invalid FIT format\n");
84*4882a593Smuzhiyun goto out;
85*4882a593Smuzhiyun }
86*4882a593Smuzhiyun
87*4882a593Smuzhiyun /* fixup entry/load and alloc sysmem */
88*4882a593Smuzhiyun if (fit_image_pre_process(fit))
89*4882a593Smuzhiyun goto out;
90*4882a593Smuzhiyun
91*4882a593Smuzhiyun env_set("bootm-no-reloc", "y");
92*4882a593Smuzhiyun snprintf(fit_addr, sizeof(fit_addr), "0x%lx", (ulong)fit);
93*4882a593Smuzhiyun bootm_args[0] = fit_addr;
94*4882a593Smuzhiyun
95*4882a593Smuzhiyun printf("at %s with size 0x%08lx\n", fit_addr, size);
96*4882a593Smuzhiyun
97*4882a593Smuzhiyun #ifdef CONFIG_ANDROID_AB
98*4882a593Smuzhiyun char slot_suffix[3] = {0};
99*4882a593Smuzhiyun char slot_info[21] = "android_slotsufix=";
100*4882a593Smuzhiyun
101*4882a593Smuzhiyun if (ab_get_slot_suffix(slot_suffix))
102*4882a593Smuzhiyun goto out;
103*4882a593Smuzhiyun
104*4882a593Smuzhiyun strcat(slot_info, slot_suffix);
105*4882a593Smuzhiyun env_update("bootargs", slot_info);
106*4882a593Smuzhiyun #endif
107*4882a593Smuzhiyun
108*4882a593Smuzhiyun ret = do_bootm_states(NULL, 0, ARRAY_SIZE(bootm_args), bootm_args,
109*4882a593Smuzhiyun BOOTM_STATE_START |
110*4882a593Smuzhiyun BOOTM_STATE_FINDOS | BOOTM_STATE_FINDOTHER |
111*4882a593Smuzhiyun BOOTM_STATE_LOADOS |
112*4882a593Smuzhiyun #ifdef CONFIG_SYS_BOOT_RAMDISK_HIGH
113*4882a593Smuzhiyun BOOTM_STATE_RAMDISK |
114*4882a593Smuzhiyun #endif
115*4882a593Smuzhiyun BOOTM_STATE_OS_PREP | BOOTM_STATE_OS_FAKE_GO |
116*4882a593Smuzhiyun BOOTM_STATE_OS_GO, &images, 1);
117*4882a593Smuzhiyun
118*4882a593Smuzhiyun if (ret && argc != 1) {
119*4882a593Smuzhiyun fit_image_fail_process(fit);
120*4882a593Smuzhiyun goto out;
121*4882a593Smuzhiyun }
122*4882a593Smuzhiyun
123*4882a593Smuzhiyun return CMD_RET_SUCCESS;
124*4882a593Smuzhiyun out:
125*4882a593Smuzhiyun return CMD_RET_FAILURE;
126*4882a593Smuzhiyun }
127*4882a593Smuzhiyun
128*4882a593Smuzhiyun U_BOOT_CMD_ALWAYS(
129*4882a593Smuzhiyun boot_fit, 2, 1, do_boot_fit,
130*4882a593Smuzhiyun "Boot FIT Image from memory or boot/recovery partition",
131*4882a593Smuzhiyun "boot_fit [addr]"
132*4882a593Smuzhiyun );
133