1 /*
2  * Copyright (C) 2011
3  * Corscience GmbH & Co. KG - Simon Schwarz <schwarz@corscience.de>
4  *
5  * SPDX-License-Identifier:	GPL-2.0+
6  */
7 
8 #include <common.h>
9 #include <command.h>
10 #include <cmd_spl.h>
11 
12 DECLARE_GLOBAL_DATA_PTR;
13 
14 static const char **subcmd_list[] = {
15 
16 	[SPL_EXPORT_FDT] = (const char * []) {
17 #ifdef CONFIG_OF_LIBFDT
18 		"start",
19 		"loados",
20 	#ifdef CONFIG_SYS_BOOT_RAMDISK_HIGH
21 		"ramdisk",
22 	#endif
23 		"fdt",
24 		"cmdline",
25 		"bdt",
26 		"prep",
27 #endif
28 		NULL,
29 	},
30 	[SPL_EXPORT_ATAGS] = (const char * []) {
31 #if defined(CONFIG_SETUP_MEMORY_TAGS) || \
32 	defined(CONFIG_CMDLINE_TAG) || \
33 	defined(CONFIG_INITRD_TAG) || \
34 	defined(CONFIG_SERIAL_TAG) || \
35 	defined(CONFIG_REVISION_TAG)
36 		"start",
37 		"loados",
38 #ifdef CONFIG_SYS_BOOT_RAMDISK_HIGH
39 		"ramdisk",
40 #endif
41 		"cmdline",
42 		"bdt",
43 		"prep",
44 #endif
45 		NULL,
46 	},
47 	NULL
48 };
49 
50 /* Calls bootm with the parameters given */
call_bootm(int argc,char * const argv[],const char * subcommand[])51 static int call_bootm(int argc, char * const argv[], const char *subcommand[])
52 {
53 	char *bootm_argv[5];
54 
55 	int i = 0;
56 	int ret = 0;
57 	int j;
58 
59 	/* create paramter array */
60 	bootm_argv[0] = "do_bootm";
61 	switch (argc) {
62 	case 3:
63 		bootm_argv[4] = argv[2]; /* fdt addr */
64 	case 2:
65 		bootm_argv[3] = argv[1]; /* initrd addr */
66 	case 1:
67 		bootm_argv[2] = argv[0]; /* kernel addr */
68 	}
69 
70 
71 	/*
72 	 * - do the work -
73 	 * exec subcommands of do_bootm to init the images
74 	 * data structure
75 	 */
76 	while (subcommand[i] != NULL) {
77 		bootm_argv[1] = (char *)subcommand[i];
78 		debug("args %d: %s %s ", argc, bootm_argv[0], bootm_argv[1]);
79 		for (j = 0; j < argc; j++)
80 			debug("%s ", bootm_argv[j + 2]);
81 		debug("\n");
82 
83 		ret = do_bootm(find_cmd("do_bootm"), 0, argc+2,
84 			bootm_argv);
85 		debug("Subcommand retcode: %d\n", ret);
86 		i++;
87 	}
88 
89 	if (ret) {
90 		printf("ERROR prep subcommand failed!\n");
91 		return -1;
92 	}
93 
94 	return 0;
95 }
96 
97 static cmd_tbl_t cmd_spl_export_sub[] = {
98 	U_BOOT_CMD_MKENT(fdt, 0, 1, (void *)SPL_EXPORT_FDT, "", ""),
99 	U_BOOT_CMD_MKENT(atags, 0, 1, (void *)SPL_EXPORT_ATAGS, "", ""),
100 };
101 
spl_export(cmd_tbl_t * cmdtp,int flag,int argc,char * const argv[])102 static int spl_export(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
103 {
104 	const cmd_tbl_t *c;
105 
106 	if (argc < 2) /* no subcommand */
107 		return cmd_usage(cmdtp);
108 
109 	c = find_cmd_tbl(argv[1], &cmd_spl_export_sub[0],
110 		ARRAY_SIZE(cmd_spl_export_sub));
111 	if ((c) && ((int)c->cmd <= SPL_EXPORT_LAST)) {
112 		argc -= 2;
113 		argv += 2;
114 		if (call_bootm(argc, argv, subcmd_list[(int)c->cmd]))
115 			return -1;
116 		switch ((int)c->cmd) {
117 #ifdef CONFIG_OF_LIBFDT
118 		case SPL_EXPORT_FDT:
119 			printf("Argument image is now in RAM: 0x%p\n",
120 				(void *)images.ft_addr);
121 			break;
122 #endif
123 		case SPL_EXPORT_ATAGS:
124 			printf("Argument image is now in RAM at: 0x%p\n",
125 				(void *)gd->bd->bi_boot_params);
126 			break;
127 		}
128 	} else {
129 		/* Unrecognized command */
130 		return cmd_usage(cmdtp);
131 	}
132 
133 	return 0;
134 }
135 
136 static cmd_tbl_t cmd_spl_sub[] = {
137 	U_BOOT_CMD_MKENT(export, 0, 1, (void *)SPL_EXPORT, "", ""),
138 };
139 
do_spl(cmd_tbl_t * cmdtp,int flag,int argc,char * const argv[])140 static int do_spl(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
141 {
142 	const cmd_tbl_t *c;
143 	int cmd;
144 
145 	if (argc < 2) /* no subcommand */
146 		return cmd_usage(cmdtp);
147 
148 	c = find_cmd_tbl(argv[1], &cmd_spl_sub[0], ARRAY_SIZE(cmd_spl_sub));
149 	if (c) {
150 		cmd = (int)c->cmd;
151 		switch (cmd) {
152 		case SPL_EXPORT:
153 			argc--;
154 			argv++;
155 			if (spl_export(cmdtp, flag, argc, argv))
156 				printf("Subcommand failed\n");
157 			break;
158 		default:
159 			/* unrecognized command */
160 			return cmd_usage(cmdtp);
161 		}
162 	} else {
163 		/* Unrecognized command */
164 		return cmd_usage(cmdtp);
165 	}
166 	return 0;
167 }
168 
169 U_BOOT_CMD(
170 	spl, 6 , 1, do_spl, "SPL configuration",
171 	"export <img=atags|fdt> [kernel_addr] [initrd_addr] [fdt_addr]\n"
172 	"\timg\t\t\"atags\" or \"fdt\"\n"
173 	"\tkernel_addr\taddress where a kernel image is stored.\n"
174 	"\t\t\tkernel is loaded as part of the boot process, but it is not started.\n"
175 	"\tinitrd_addr\taddress of initial ramdisk\n"
176 	"\t\t\tcan be set to \"-\" if fdt_addr without initrd_addr is used.\n"
177 	"\tfdt_addr\tin case of fdt, the address of the device tree.\n"
178 	);
179