imximage: fix commands other than write_data
[platform/kernel/u-boot.git] / common / cmd_spl.c
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 */
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
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
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         );