1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * (C) Copyright 2000-2009
4  * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
5  */
6 
7 #include <bootm.h>
8 #include <command.h>
9 #include <env.h>
10 #include <image.h>
11 #include <irq_func.h>
12 #include <lmb.h>
13 #include <log.h>
14 #include <mapmem.h>
15 #include <asm/global_data.h>
16 #include <linux/kernel.h>
17 #include <linux/sizes.h>
18 
19 DECLARE_GLOBAL_DATA_PTR;
20 /*
21  * Image booting support
22  */
booti_start(struct bootm_info * bmi)23 static int booti_start(struct bootm_info *bmi)
24 {
25 	struct bootm_headers *images = bmi->images;
26 	int ret;
27 	ulong ld;
28 	ulong relocated_addr;
29 	ulong image_size;
30 	uint8_t *temp;
31 	ulong dest;
32 	ulong dest_end;
33 	phys_addr_t ep_addr;
34 	unsigned long comp_len;
35 	unsigned long decomp_len;
36 	int ctype;
37 
38 	ret = bootm_run_states(bmi, BOOTM_STATE_START);
39 
40 	/* Setup Linux kernel Image entry point */
41 	if (!bmi->addr_img) {
42 		ld = image_load_addr;
43 		debug("*  kernel: default image load address = 0x%08lx\n",
44 				image_load_addr);
45 	} else {
46 		ld = hextoul(bmi->addr_img, NULL);
47 		debug("*  kernel: cmdline image address = 0x%08lx\n", ld);
48 	}
49 
50 	temp = map_sysmem(ld, 0);
51 	ctype = image_decomp_type(temp, 2);
52 	if (ctype > 0) {
53 		dest = env_get_ulong("kernel_comp_addr_r", 16, 0);
54 		comp_len = env_get_ulong("kernel_comp_size", 16, 0);
55 		if (!dest || !comp_len) {
56 			puts("kernel_comp_addr_r or kernel_comp_size is not provided!\n");
57 			return -EINVAL;
58 		}
59 		if (dest < gd->ram_base || dest > gd->ram_top) {
60 			puts("kernel_comp_addr_r is outside of DRAM range!\n");
61 			return -EINVAL;
62 		}
63 
64 		debug("kernel image compression type %d size = 0x%08lx address = 0x%08lx\n",
65 			ctype, comp_len, (ulong)dest);
66 		decomp_len = comp_len * 10;
67 		ret = image_decomp(ctype, 0, ld, IH_TYPE_KERNEL,
68 				 (void *)dest, (void *)ld, comp_len,
69 				 decomp_len, &dest_end);
70 		if (ret)
71 			return ret;
72 		/* dest_end contains the uncompressed Image size */
73 		memmove((void *) ld, (void *)dest, dest_end);
74 	}
75 	unmap_sysmem((void *)ld);
76 
77 	ret = booti_setup(ld, &relocated_addr, &image_size, false);
78 	if (ret)
79 		return 1;
80 
81 	/* Handle BOOTM_STATE_LOADOS */
82 	if (relocated_addr != ld) {
83 		printf("Moving Image from 0x%lx to 0x%lx, end=0x%lx\n", ld,
84 		       relocated_addr, relocated_addr + image_size);
85 		memmove((void *)relocated_addr, (void *)ld, image_size);
86 	}
87 
88 	images->ep = relocated_addr;
89 	images->os.start = relocated_addr;
90 	images->os.end = relocated_addr + image_size;
91 
92 	ep_addr = (phys_addr_t)images->ep;
93 	ret = lmb_alloc_mem(LMB_MEM_ALLOC_ADDR, 0, &ep_addr,
94 			    le32_to_cpu(image_size), LMB_NONE);
95 	if (ret) {
96 		printf("Failed to allocate memory for the image at %#llx\n",
97 		       (unsigned long long)images->ep);
98 		return 1;
99 	}
100 
101 	/*
102 	 * Handle the BOOTM_STATE_FINDOTHER state ourselves as we do not
103 	 * have a header that provide this informaiton.
104 	 */
105 	if (bootm_find_images(image_load_addr, bmi->conf_ramdisk, bmi->conf_fdt,
106 			      relocated_addr, image_size))
107 		return 1;
108 
109 	return 0;
110 }
111 
do_booti(struct cmd_tbl * cmdtp,int flag,int argc,char * const argv[])112 int do_booti(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[])
113 {
114 	struct bootm_info bmi;
115 	int states;
116 	int ret;
117 
118 	/* Consume 'booti' */
119 	argc--; argv++;
120 
121 	bootm_init(&bmi);
122 	if (argc)
123 		bmi.addr_img = argv[0];
124 	if (argc > 1)
125 		bmi.conf_ramdisk = argv[1];
126 	if (argc > 2)
127 		bmi.conf_fdt = argv[2];
128 	bmi.boot_progress = true;
129 	bmi.cmd_name = "booti";
130 	/* do not set up argc and argv[] since nothing uses them */
131 
132 	if (booti_start(&bmi))
133 		return 1;
134 
135 	/*
136 	 * We are doing the BOOTM_STATE_LOADOS state ourselves, so must
137 	 * disable interrupts ourselves
138 	 */
139 	bootm_disable_interrupts();
140 
141 	images.os.os = IH_OS_LINUX;
142 	if (IS_ENABLED(CONFIG_RISCV_SMODE))
143 		images.os.arch = IH_ARCH_RISCV;
144 	else if (IS_ENABLED(CONFIG_ARM64))
145 		images.os.arch = IH_ARCH_ARM64;
146 
147 	states = BOOTM_STATE_MEASURE | BOOTM_STATE_OS_PREP |
148 		BOOTM_STATE_OS_FAKE_GO | BOOTM_STATE_OS_GO;
149 	if (IS_ENABLED(CONFIG_SYS_BOOT_RAMDISK_HIGH))
150 		states |= BOOTM_STATE_RAMDISK;
151 
152 	ret = bootm_run_states(&bmi, states);
153 
154 	return ret;
155 }
156 
157 U_BOOT_LONGHELP(booti,
158 	"[addr [initrd[:size]] [fdt]]\n"
159 	"    - boot Linux flat or compressed 'Image' stored at 'addr'\n"
160 	"\tThe argument 'initrd' is optional and specifies the address\n"
161 	"\tof an initrd in memory. The optional parameter ':size' allows\n"
162 	"\tspecifying the size of a RAW initrd.\n"
163 	"\tCurrently only booting from gz, bz2, lzma and lz4 compression\n"
164 	"\ttypes are supported. In order to boot from any of these compressed\n"
165 	"\timages, user have to set kernel_comp_addr_r and kernel_comp_size environment\n"
166 	"\tvariables beforehand.\n"
167 #if defined(CONFIG_OF_LIBFDT)
168 	"\tSince booting a Linux kernel requires a flat device-tree, a\n"
169 	"\tthird argument providing the address of the device-tree blob\n"
170 	"\tis required. To boot a kernel with a device-tree blob but\n"
171 	"\twithout an initrd image, use a '-' for the initrd argument.\n"
172 #endif
173 	);
174 
175 U_BOOT_CMD(
176 	booti,	CONFIG_SYS_MAXARGS,	1,	do_booti,
177 	"boot Linux kernel 'Image' format from memory", booti_help_text
178 );
179