]>
Commit | Line | Data |
---|---|---|
83d290c5 | 1 | // SPDX-License-Identifier: GPL-2.0+ |
30354978 SG |
2 | /* |
3 | * (C) Copyright 2000 | |
4 | * Wolfgang Denk, DENX Software Engineering, [email protected]. | |
5 | * | |
6 | * Add to readline cmdline-editing by | |
7 | * (C) Copyright 2005 | |
8 | * JinHua Luo, GuangDong Linux Center, <[email protected]> | |
30354978 SG |
9 | */ |
10 | ||
11 | #include <common.h> | |
52f24238 | 12 | #include <bootstage.h> |
30354978 SG |
13 | #include <cli.h> |
14 | #include <cli_hush.h> | |
288b29e4 | 15 | #include <command.h> |
24b852a7 | 16 | #include <console.h> |
7b51b576 | 17 | #include <env.h> |
affb2156 | 18 | #include <fdtdec.h> |
db41d65a | 19 | #include <hang.h> |
30354978 | 20 | #include <malloc.h> |
401d1c4f | 21 | #include <asm/global_data.h> |
30354978 | 22 | |
affb2156 SG |
23 | DECLARE_GLOBAL_DATA_PTR; |
24 | ||
f8bb6964 | 25 | #ifdef CONFIG_CMDLINE |
30354978 SG |
26 | /* |
27 | * Run a command using the selected parser. | |
28 | * | |
29 | * @param cmd Command to run | |
30 | * @param flag Execution flags (CMD_FLAG_...) | |
31 | * @return 0 on success, or != 0 on error. | |
32 | */ | |
33 | int run_command(const char *cmd, int flag) | |
34 | { | |
2dd468db | 35 | #if !CONFIG_IS_ENABLED(HUSH_PARSER) |
30354978 SG |
36 | /* |
37 | * cli_run_command can return 0 or 1 for success, so clean up | |
38 | * its result. | |
39 | */ | |
40 | if (cli_simple_run_command(cmd, flag) == -1) | |
41 | return 1; | |
42 | ||
43 | return 0; | |
44 | #else | |
87b6398b SG |
45 | int hush_flags = FLAG_PARSE_SEMICOLON | FLAG_EXIT_FROM_LOOP; |
46 | ||
47 | if (flag & CMD_FLAG_ENV) | |
48 | hush_flags |= FLAG_CONT_ON_NEWLINE; | |
49 | return parse_string_outer(cmd, hush_flags); | |
30354978 SG |
50 | #endif |
51 | } | |
52 | ||
1d43bfd2 TB |
53 | /* |
54 | * Run a command using the selected parser, and check if it is repeatable. | |
55 | * | |
56 | * @param cmd Command to run | |
57 | * @param flag Execution flags (CMD_FLAG_...) | |
58 | * @return 0 (not repeatable) or 1 (repeatable) on success, -1 on error. | |
59 | */ | |
60 | int run_command_repeatable(const char *cmd, int flag) | |
61 | { | |
f1f9d4fa | 62 | #ifndef CONFIG_HUSH_PARSER |
1d43bfd2 TB |
63 | return cli_simple_run_command(cmd, flag); |
64 | #else | |
65 | /* | |
66 | * parse_string_outer() returns 1 for failure, so clean up | |
67 | * its result. | |
68 | */ | |
69 | if (parse_string_outer(cmd, | |
70 | FLAG_PARSE_SEMICOLON | FLAG_EXIT_FROM_LOOP)) | |
71 | return -1; | |
72 | ||
73 | return 0; | |
74 | #endif | |
75 | } | |
19464f4f SA |
76 | #else |
77 | __weak int board_run_command(const char *cmdline) | |
78 | { | |
79 | printf("## Commands are disabled. Please enable CONFIG_CMDLINE.\n"); | |
80 | ||
81 | return 1; | |
82 | } | |
f8bb6964 | 83 | #endif /* CONFIG_CMDLINE */ |
1d43bfd2 | 84 | |
30354978 SG |
85 | int run_command_list(const char *cmd, int len, int flag) |
86 | { | |
87 | int need_buff = 1; | |
88 | char *buff = (char *)cmd; /* cast away const */ | |
89 | int rcode = 0; | |
90 | ||
91 | if (len == -1) { | |
92 | len = strlen(cmd); | |
f1f9d4fa | 93 | #ifdef CONFIG_HUSH_PARSER |
30354978 SG |
94 | /* hush will never change our string */ |
95 | need_buff = 0; | |
96 | #else | |
97 | /* the built-in parser will change our string if it sees \n */ | |
98 | need_buff = strchr(cmd, '\n') != NULL; | |
99 | #endif | |
100 | } | |
101 | if (need_buff) { | |
102 | buff = malloc(len + 1); | |
103 | if (!buff) | |
104 | return 1; | |
105 | memcpy(buff, cmd, len); | |
106 | buff[len] = '\0'; | |
107 | } | |
f1f9d4fa | 108 | #ifdef CONFIG_HUSH_PARSER |
30354978 SG |
109 | rcode = parse_string_outer(buff, FLAG_PARSE_SEMICOLON); |
110 | #else | |
111 | /* | |
112 | * This function will overwrite any \n it sees with a \0, which | |
113 | * is why it can't work with a const char *. Here we are making | |
114 | * using of internal knowledge of this function, to avoid always | |
115 | * doing a malloc() which is actually required only in a case that | |
116 | * is pretty rare. | |
117 | */ | |
f8bb6964 | 118 | #ifdef CONFIG_CMDLINE |
30354978 | 119 | rcode = cli_simple_run_command_list(buff, flag); |
f8bb6964 SG |
120 | #else |
121 | rcode = board_run_command(buff); | |
122 | #endif | |
09a78862 | 123 | #endif |
30354978 SG |
124 | if (need_buff) |
125 | free(buff); | |
30354978 SG |
126 | |
127 | return rcode; | |
128 | } | |
129 | ||
130 | /****************************************************************************/ | |
131 | ||
132 | #if defined(CONFIG_CMD_RUN) | |
09140113 | 133 | int do_run(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[]) |
30354978 SG |
134 | { |
135 | int i; | |
136 | ||
137 | if (argc < 2) | |
138 | return CMD_RET_USAGE; | |
139 | ||
140 | for (i = 1; i < argc; ++i) { | |
141 | char *arg; | |
142 | ||
00caae6d | 143 | arg = env_get(argv[i]); |
30354978 SG |
144 | if (arg == NULL) { |
145 | printf("## Error: \"%s\" not defined\n", argv[i]); | |
146 | return 1; | |
147 | } | |
148 | ||
87b6398b | 149 | if (run_command(arg, flag | CMD_FLAG_ENV) != 0) |
30354978 SG |
150 | return 1; |
151 | } | |
152 | return 0; | |
153 | } | |
154 | #endif | |
c1bb2cd0 | 155 | |
0f925822 | 156 | #if CONFIG_IS_ENABLED(OF_CONTROL) |
affb2156 SG |
157 | bool cli_process_fdt(const char **cmdp) |
158 | { | |
159 | /* Allow the fdt to override the boot command */ | |
160 | char *env = fdtdec_get_config_string(gd->fdt_blob, "bootcmd"); | |
161 | if (env) | |
162 | *cmdp = env; | |
163 | /* | |
164 | * If the bootsecure option was chosen, use secure_boot_cmd(). | |
165 | * Always use 'env' in this case, since bootsecure requres that the | |
166 | * bootcmd was specified in the FDT too. | |
167 | */ | |
168 | return fdtdec_get_config_int(gd->fdt_blob, "bootsecure", 0) != 0; | |
169 | } | |
170 | ||
171 | /* | |
172 | * Runs the given boot command securely. Specifically: | |
173 | * - Doesn't run the command with the shell (run_command or parse_string_outer), | |
174 | * since that's a lot of code surface that an attacker might exploit. | |
175 | * Because of this, we don't do any argument parsing--the secure boot command | |
176 | * has to be a full-fledged u-boot command. | |
177 | * - Doesn't check for keypresses before booting, since that could be a | |
178 | * security hole; also disables Ctrl-C. | |
179 | * - Doesn't allow the command to return. | |
180 | * | |
181 | * Upon any failures, this function will drop into an infinite loop after | |
182 | * printing the error message to console. | |
183 | */ | |
184 | void cli_secure_boot_cmd(const char *cmd) | |
185 | { | |
f8bb6964 | 186 | #ifdef CONFIG_CMDLINE |
09140113 | 187 | struct cmd_tbl *cmdtp; |
f8bb6964 | 188 | #endif |
affb2156 SG |
189 | int rc; |
190 | ||
191 | if (!cmd) { | |
192 | printf("## Error: Secure boot command not specified\n"); | |
193 | goto err; | |
194 | } | |
195 | ||
196 | /* Disable Ctrl-C just in case some command is used that checks it. */ | |
197 | disable_ctrlc(1); | |
198 | ||
199 | /* Find the command directly. */ | |
f8bb6964 | 200 | #ifdef CONFIG_CMDLINE |
affb2156 SG |
201 | cmdtp = find_cmd(cmd); |
202 | if (!cmdtp) { | |
203 | printf("## Error: \"%s\" not defined\n", cmd); | |
204 | goto err; | |
205 | } | |
206 | ||
207 | /* Run the command, forcing no flags and faking argc and argv. */ | |
208 | rc = (cmdtp->cmd)(cmdtp, 0, 1, (char **)&cmd); | |
209 | ||
f8bb6964 SG |
210 | #else |
211 | rc = board_run_command(cmd); | |
212 | #endif | |
213 | ||
affb2156 SG |
214 | /* Shouldn't ever return from boot command. */ |
215 | printf("## Error: \"%s\" returned (code %d)\n", cmd, rc); | |
216 | ||
217 | err: | |
218 | /* | |
219 | * Not a whole lot to do here. Rebooting won't help much, since we'll | |
220 | * just end up right back here. Just loop. | |
221 | */ | |
222 | hang(); | |
223 | } | |
0f925822 | 224 | #endif /* CONFIG_IS_ENABLED(OF_CONTROL) */ |
affb2156 | 225 | |
c1bb2cd0 SG |
226 | void cli_loop(void) |
227 | { | |
b07cc48c | 228 | bootstage_mark(BOOTSTAGE_ID_ENTER_CLI_LOOP); |
f1f9d4fa | 229 | #ifdef CONFIG_HUSH_PARSER |
c1bb2cd0 SG |
230 | parse_file_outer(); |
231 | /* This point is never reached */ | |
232 | for (;;); | |
30eae26b | 233 | #elif defined(CONFIG_CMDLINE) |
c1bb2cd0 | 234 | cli_simple_loop(); |
f8bb6964 SG |
235 | #else |
236 | printf("## U-Boot command line is disabled. Please enable CONFIG_CMDLINE\n"); | |
f1f9d4fa | 237 | #endif /*CONFIG_HUSH_PARSER*/ |
c1bb2cd0 SG |
238 | } |
239 | ||
240 | void cli_init(void) | |
241 | { | |
f1f9d4fa | 242 | #ifdef CONFIG_HUSH_PARSER |
c1bb2cd0 SG |
243 | u_boot_hush_start(); |
244 | #endif | |
245 | ||
246 | #if defined(CONFIG_HUSH_INIT_VAR) | |
247 | hush_init_var(); | |
248 | #endif | |
249 | } |