blob: defe2d499d898f57b474a61284b649ff8524f1e9 [file] [log] [blame]
wdenkc6097192002-11-03 00:24:07 +00001/*
2 * (C) Copyright 2000
3 * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
4 *
Wolfgang Denk501090a2006-07-21 11:33:45 +02005 * Add to readline cmdline-editing by
6 * (C) Copyright 2005
7 * JinHua Luo, GuangDong Linux Center, <luo.jinhua@gd-linux.com>
8 *
wdenkc6097192002-11-03 00:24:07 +00009 * See file CREDITS for list of people who contributed to this
10 * project.
11 *
12 * This program is free software; you can redistribute it and/or
13 * modify it under the terms of the GNU General Public License as
14 * published by the Free Software Foundation; either version 2 of
15 * the License, or (at your option) any later version.
16 *
17 * This program is distributed in the hope that it will be useful,
18 * but WITHOUT ANY WARRANTY; without even the implied warranty of
19 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
20 * GNU General Public License for more details.
21 *
22 * You should have received a copy of the GNU General Public License
23 * along with this program; if not, write to the Free Software
24 * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
25 * MA 02111-1307 USA
26 */
27
wdenka6c7ad22002-12-03 21:28:10 +000028/* #define DEBUG */
29
wdenkc6097192002-11-03 00:24:07 +000030#include <common.h>
31#include <watchdog.h>
32#include <command.h>
Andreas Bießmann09c2e902011-07-18 20:24:04 +020033#include <version.h>
Wolfgang Denkd9631ec2005-09-30 15:18:23 +020034#ifdef CONFIG_MODEM_SUPPORT
35#include <malloc.h> /* for free() prototype */
36#endif
wdenkc6097192002-11-03 00:24:07 +000037
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +020038#ifdef CONFIG_SYS_HUSH_PARSER
wdenkc6097192002-11-03 00:24:07 +000039#include <hush.h>
40#endif
41
wdenkbdccc4f2003-08-05 17:43:17 +000042#include <post.h>
Jason Hobbs4d91a6e2011-08-23 11:06:54 +000043#include <linux/ctype.h>
Heiko Schocher317d6c52012-01-16 21:13:35 +000044#include <menu.h>
wdenkbdccc4f2003-08-05 17:43:17 +000045
James Yang597f6c22008-05-05 10:22:53 -050046#if defined(CONFIG_SILENT_CONSOLE) || defined(CONFIG_POST) || defined(CONFIG_CMDLINE_EDITING)
Wolfgang Denkd87080b2006-03-31 18:32:53 +020047DECLARE_GLOBAL_DATA_PTR;
48#endif
49
Heiko Schocherfad63402007-07-13 09:54:17 +020050/*
51 * Board-specific Platform code can reimplement show_boot_progress () if needed
52 */
53void inline __show_boot_progress (int val) {}
Emil Medve5e2c08c2009-05-12 13:48:32 -050054void show_boot_progress (int val) __attribute__((weak, alias("__show_boot_progress")));
Heiko Schocherfad63402007-07-13 09:54:17 +020055
Bartlomiej Sieka4bae9092008-10-01 15:26:31 +020056#if defined(CONFIG_UPDATE_TFTP)
Andreas Pretzsch8d6b7322011-07-16 05:50:59 +000057int update_tftp (ulong addr);
Bartlomiej Sieka4bae9092008-10-01 15:26:31 +020058#endif /* CONFIG_UPDATE_TFTP */
wdenk8bde7f72003-06-27 21:31:46 +000059
wdenkc6097192002-11-03 00:24:07 +000060#define MAX_DELAY_STOP_STR 32
61
wdenkc6097192002-11-03 00:24:07 +000062#undef DEBUG_PARSER
63
John Schmoller6475b9f2010-03-12 09:49:23 -060064char console_buffer[CONFIG_SYS_CBSIZE + 1]; /* console I/O buffer */
wdenkc6097192002-11-03 00:24:07 +000065
Stefan Roese3ca91222006-07-27 16:11:19 +020066static char * delete_char (char *buffer, char *p, int *colp, int *np, int plen);
Mike Frysinger82359ae2010-12-22 09:40:45 -050067static const char erase_seq[] = "\b \b"; /* erase sequence */
68static const char tab_seq[] = " "; /* used to expand TABs */
wdenkc6097192002-11-03 00:24:07 +000069
70#ifdef CONFIG_BOOT_RETRY_TIME
71static uint64_t endtime = 0; /* must be set, default is instant timeout */
72static int retry_time = -1; /* -1 so can call readline before main_loop */
73#endif
74
75#define endtick(seconds) (get_ticks() + (uint64_t)(seconds) * get_tbclk())
76
77#ifndef CONFIG_BOOT_RETRY_MIN
78#define CONFIG_BOOT_RETRY_MIN CONFIG_BOOT_RETRY_TIME
79#endif
80
81#ifdef CONFIG_MODEM_SUPPORT
82int do_mdm_init = 0;
83extern void mdm_init(void); /* defined in board.c */
84#endif
85
86/***************************************************************************
87 * Watch for 'delay' seconds for autoboot stop or autoboot delay string.
Jason Hobbsc8a20792011-08-31 05:37:24 +000088 * returns: 0 - no key string, allow autoboot 1 - got key string, abort
wdenkc6097192002-11-03 00:24:07 +000089 */
90#if defined(CONFIG_BOOTDELAY) && (CONFIG_BOOTDELAY >= 0)
91# if defined(CONFIG_AUTOBOOT_KEYED)
Jason Hobbsb41bc5a2011-08-23 11:06:50 +000092#ifndef CONFIG_MENU
93static inline
94#endif
95int abortboot(int bootdelay)
wdenkc6097192002-11-03 00:24:07 +000096{
97 int abort = 0;
98 uint64_t etime = endtick(bootdelay);
Wolfgang Denk19973b62006-10-28 00:38:39 +020099 struct {
wdenkc6097192002-11-03 00:24:07 +0000100 char* str;
101 u_int len;
102 int retry;
103 }
Wolfgang Denk19973b62006-10-28 00:38:39 +0200104 delaykey [] = {
wdenkc6097192002-11-03 00:24:07 +0000105 { str: getenv ("bootdelaykey"), retry: 1 },
106 { str: getenv ("bootdelaykey2"), retry: 1 },
107 { str: getenv ("bootstopkey"), retry: 0 },
108 { str: getenv ("bootstopkey2"), retry: 0 },
109 };
110
111 char presskey [MAX_DELAY_STOP_STR];
112 u_int presskey_len = 0;
113 u_int presskey_max = 0;
114 u_int i;
115
116# ifdef CONFIG_AUTOBOOT_PROMPT
Stefan Roesef2302d42008-08-06 14:05:38 +0200117 printf(CONFIG_AUTOBOOT_PROMPT);
wdenkc6097192002-11-03 00:24:07 +0000118# endif
119
120# ifdef CONFIG_AUTOBOOT_DELAY_STR
121 if (delaykey[0].str == NULL)
122 delaykey[0].str = CONFIG_AUTOBOOT_DELAY_STR;
123# endif
124# ifdef CONFIG_AUTOBOOT_DELAY_STR2
125 if (delaykey[1].str == NULL)
126 delaykey[1].str = CONFIG_AUTOBOOT_DELAY_STR2;
127# endif
128# ifdef CONFIG_AUTOBOOT_STOP_STR
129 if (delaykey[2].str == NULL)
130 delaykey[2].str = CONFIG_AUTOBOOT_STOP_STR;
131# endif
132# ifdef CONFIG_AUTOBOOT_STOP_STR2
133 if (delaykey[3].str == NULL)
134 delaykey[3].str = CONFIG_AUTOBOOT_STOP_STR2;
135# endif
136
137 for (i = 0; i < sizeof(delaykey) / sizeof(delaykey[0]); i ++) {
138 delaykey[i].len = delaykey[i].str == NULL ?
139 0 : strlen (delaykey[i].str);
140 delaykey[i].len = delaykey[i].len > MAX_DELAY_STOP_STR ?
141 MAX_DELAY_STOP_STR : delaykey[i].len;
142
143 presskey_max = presskey_max > delaykey[i].len ?
144 presskey_max : delaykey[i].len;
145
146# if DEBUG_BOOTKEYS
147 printf("%s key:<%s>\n",
148 delaykey[i].retry ? "delay" : "stop",
149 delaykey[i].str ? delaykey[i].str : "NULL");
150# endif
151 }
152
153 /* In order to keep up with incoming data, check timeout only
154 * when catch up.
155 */
Peter Korsgaardc3284b02008-12-10 16:24:16 +0100156 do {
157 if (tstc()) {
158 if (presskey_len < presskey_max) {
159 presskey [presskey_len ++] = getc();
160 }
161 else {
162 for (i = 0; i < presskey_max - 1; i ++)
163 presskey [i] = presskey [i + 1];
164
165 presskey [i] = getc();
166 }
167 }
168
wdenkc6097192002-11-03 00:24:07 +0000169 for (i = 0; i < sizeof(delaykey) / sizeof(delaykey[0]); i ++) {
170 if (delaykey[i].len > 0 &&
171 presskey_len >= delaykey[i].len &&
172 memcmp (presskey + presskey_len - delaykey[i].len,
wdenk8bde7f72003-06-27 21:31:46 +0000173 delaykey[i].str,
wdenkc6097192002-11-03 00:24:07 +0000174 delaykey[i].len) == 0) {
175# if DEBUG_BOOTKEYS
176 printf("got %skey\n",
177 delaykey[i].retry ? "delay" : "stop");
178# endif
179
180# ifdef CONFIG_BOOT_RETRY_TIME
181 /* don't retry auto boot */
182 if (! delaykey[i].retry)
183 retry_time = -1;
184# endif
185 abort = 1;
186 }
187 }
Peter Korsgaardc3284b02008-12-10 16:24:16 +0100188 } while (!abort && get_ticks() <= etime);
wdenkc6097192002-11-03 00:24:07 +0000189
wdenkc6097192002-11-03 00:24:07 +0000190# if DEBUG_BOOTKEYS
191 if (!abort)
Ladislav Michlada4d402007-04-25 16:01:26 +0200192 puts("key timeout\n");
wdenkc6097192002-11-03 00:24:07 +0000193# endif
194
dzu8cb81432003-10-24 13:14:45 +0000195#ifdef CONFIG_SILENT_CONSOLE
Ladislav Michl4ec5bd52007-04-25 16:01:26 +0200196 if (abort)
197 gd->flags &= ~GD_FLG_SILENT;
dzu8cb81432003-10-24 13:14:45 +0000198#endif
199
wdenkc6097192002-11-03 00:24:07 +0000200 return abort;
201}
202
203# else /* !defined(CONFIG_AUTOBOOT_KEYED) */
204
wdenkc7de8292002-11-19 11:04:11 +0000205#ifdef CONFIG_MENUKEY
206static int menukey = 0;
207#endif
208
Jason Hobbsb41bc5a2011-08-23 11:06:50 +0000209#ifndef CONFIG_MENU
210static inline
211#endif
212int abortboot(int bootdelay)
wdenkc6097192002-11-03 00:24:07 +0000213{
214 int abort = 0;
215
wdenkc7de8292002-11-19 11:04:11 +0000216#ifdef CONFIG_MENUPROMPT
Stefan Roesef2302d42008-08-06 14:05:38 +0200217 printf(CONFIG_MENUPROMPT);
wdenkc7de8292002-11-19 11:04:11 +0000218#else
wdenkc6097192002-11-03 00:24:07 +0000219 printf("Hit any key to stop autoboot: %2d ", bootdelay);
wdenkc7de8292002-11-19 11:04:11 +0000220#endif
wdenkc6097192002-11-03 00:24:07 +0000221
222#if defined CONFIG_ZERO_BOOTDELAY_CHECK
wdenk8bde7f72003-06-27 21:31:46 +0000223 /*
224 * Check if key already pressed
225 * Don't check if bootdelay < 0
226 */
wdenkc6097192002-11-03 00:24:07 +0000227 if (bootdelay >= 0) {
228 if (tstc()) { /* we got a key press */
229 (void) getc(); /* consume input */
wdenk4b9206e2004-03-23 22:14:11 +0000230 puts ("\b\b\b 0");
Ladislav Michl4ec5bd52007-04-25 16:01:26 +0200231 abort = 1; /* don't auto boot */
wdenkc6097192002-11-03 00:24:07 +0000232 }
wdenk8bde7f72003-06-27 21:31:46 +0000233 }
wdenkc6097192002-11-03 00:24:07 +0000234#endif
235
wdenkf72da342003-10-10 10:05:42 +0000236 while ((bootdelay > 0) && (!abort)) {
wdenkc6097192002-11-03 00:24:07 +0000237 int i;
238
239 --bootdelay;
240 /* delay 100 * 10ms */
241 for (i=0; !abort && i<100; ++i) {
242 if (tstc()) { /* we got a key press */
243 abort = 1; /* don't auto boot */
244 bootdelay = 0; /* no more delay */
wdenkc7de8292002-11-19 11:04:11 +0000245# ifdef CONFIG_MENUKEY
246 menukey = getc();
247# else
wdenkc6097192002-11-03 00:24:07 +0000248 (void) getc(); /* consume input */
wdenkc7de8292002-11-19 11:04:11 +0000249# endif
wdenkc6097192002-11-03 00:24:07 +0000250 break;
251 }
Ladislav Michlada4d402007-04-25 16:01:26 +0200252 udelay(10000);
wdenkc6097192002-11-03 00:24:07 +0000253 }
254
Ladislav Michlada4d402007-04-25 16:01:26 +0200255 printf("\b\b\b%2d ", bootdelay);
wdenkc6097192002-11-03 00:24:07 +0000256 }
257
Ladislav Michlada4d402007-04-25 16:01:26 +0200258 putc('\n');
wdenkc6097192002-11-03 00:24:07 +0000259
wdenkf72da342003-10-10 10:05:42 +0000260#ifdef CONFIG_SILENT_CONSOLE
Ladislav Michl4ec5bd52007-04-25 16:01:26 +0200261 if (abort)
262 gd->flags &= ~GD_FLG_SILENT;
wdenkf72da342003-10-10 10:05:42 +0000263#endif
264
wdenkc6097192002-11-03 00:24:07 +0000265 return abort;
266}
267# endif /* CONFIG_AUTOBOOT_KEYED */
268#endif /* CONFIG_BOOTDELAY >= 0 */
269
Jason Hobbsc8a20792011-08-31 05:37:24 +0000270/*
271 * Return 0 on success, or != 0 on error.
272 */
Jason Hobbsc8a20792011-08-31 05:37:24 +0000273int run_command2(const char *cmd, int flag)
274{
275#ifndef CONFIG_SYS_HUSH_PARSER
276 /*
277 * run_command can return 0 or 1 for success, so clean up its result.
278 */
279 if (run_command(cmd, flag) == -1)
280 return 1;
281
282 return 0;
283#else
284 return parse_string_outer(cmd,
285 FLAG_PARSE_SEMICOLON | FLAG_EXIT_FROM_LOOP);
286#endif
287}
288
wdenkc6097192002-11-03 00:24:07 +0000289/****************************************************************************/
290
291void main_loop (void)
292{
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +0200293#ifndef CONFIG_SYS_HUSH_PARSER
294 static char lastcommand[CONFIG_SYS_CBSIZE] = { 0, };
wdenkc6097192002-11-03 00:24:07 +0000295 int len;
296 int rc = 1;
297 int flag;
298#endif
299
300#if defined(CONFIG_BOOTDELAY) && (CONFIG_BOOTDELAY >= 0)
301 char *s;
302 int bootdelay;
303#endif
304#ifdef CONFIG_PREBOOT
305 char *p;
306#endif
wdenkbdccc4f2003-08-05 17:43:17 +0000307#ifdef CONFIG_BOOTCOUNT_LIMIT
308 unsigned long bootcount = 0;
309 unsigned long bootlimit = 0;
310 char *bcs;
311 char bcs_set[16];
312#endif /* CONFIG_BOOTCOUNT_LIMIT */
wdenkc6097192002-11-03 00:24:07 +0000313
wdenkbdccc4f2003-08-05 17:43:17 +0000314#ifdef CONFIG_BOOTCOUNT_LIMIT
315 bootcount = bootcount_load();
316 bootcount++;
317 bootcount_store (bootcount);
318 sprintf (bcs_set, "%lu", bootcount);
319 setenv ("bootcount", bcs_set);
320 bcs = getenv ("bootlimit");
321 bootlimit = bcs ? simple_strtoul (bcs, NULL, 10) : 0;
322#endif /* CONFIG_BOOTCOUNT_LIMIT */
323
wdenkc6097192002-11-03 00:24:07 +0000324#ifdef CONFIG_MODEM_SUPPORT
325 debug ("DEBUG: main_loop: do_mdm_init=%d\n", do_mdm_init);
326 if (do_mdm_init) {
Wolfgang Denk77ddac92005-10-13 16:45:02 +0200327 char *str = strdup(getenv("mdm_cmd"));
wdenkc6097192002-11-03 00:24:07 +0000328 setenv ("preboot", str); /* set or delete definition */
329 if (str != NULL)
330 free (str);
331 mdm_init(); /* wait for modem connection */
332 }
333#endif /* CONFIG_MODEM_SUPPORT */
334
stroese05875972003-04-04 15:44:49 +0000335#ifdef CONFIG_VERSION_VARIABLE
336 {
stroese155cb012003-07-11 08:00:33 +0000337 setenv ("ver", version_string); /* set version variable */
stroese05875972003-04-04 15:44:49 +0000338 }
339#endif /* CONFIG_VERSION_VARIABLE */
340
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +0200341#ifdef CONFIG_SYS_HUSH_PARSER
wdenkc6097192002-11-03 00:24:07 +0000342 u_boot_hush_start ();
343#endif
344
Heiko Schocher81473f62008-10-15 09:40:28 +0200345#if defined(CONFIG_HUSH_INIT_VAR)
346 hush_init_var ();
347#endif
348
wdenkc6097192002-11-03 00:24:07 +0000349#ifdef CONFIG_PREBOOT
350 if ((p = getenv ("preboot")) != NULL) {
351# ifdef CONFIG_AUTOBOOT_KEYED
352 int prev = disable_ctrlc(1); /* disable Control C checking */
353# endif
354
Jason Hobbsc8a20792011-08-31 05:37:24 +0000355 run_command2(p, 0);
wdenkc6097192002-11-03 00:24:07 +0000356
357# ifdef CONFIG_AUTOBOOT_KEYED
358 disable_ctrlc(prev); /* restore Control C checking */
359# endif
360 }
361#endif /* CONFIG_PREBOOT */
362
Wolfgang Denk143cd212010-03-11 23:56:03 +0100363#if defined(CONFIG_UPDATE_TFTP)
Andreas Pretzsch8d6b7322011-07-16 05:50:59 +0000364 update_tftp (0UL);
Wolfgang Denk143cd212010-03-11 23:56:03 +0100365#endif /* CONFIG_UPDATE_TFTP */
366
wdenkc6097192002-11-03 00:24:07 +0000367#if defined(CONFIG_BOOTDELAY) && (CONFIG_BOOTDELAY >= 0)
368 s = getenv ("bootdelay");
369 bootdelay = s ? (int)simple_strtol(s, NULL, 10) : CONFIG_BOOTDELAY;
370
wdenka6c7ad22002-12-03 21:28:10 +0000371 debug ("### main_loop entered: bootdelay=%d\n\n", bootdelay);
wdenkc6097192002-11-03 00:24:07 +0000372
Heiko Schocher317d6c52012-01-16 21:13:35 +0000373#if defined(CONFIG_MENU_SHOW)
374 bootdelay = menu_show(bootdelay);
375#endif
wdenkc6097192002-11-03 00:24:07 +0000376# ifdef CONFIG_BOOT_RETRY_TIME
wdenk6dd652f2003-06-19 23:40:20 +0000377 init_cmd_timeout ();
wdenkc6097192002-11-03 00:24:07 +0000378# endif /* CONFIG_BOOT_RETRY_TIME */
379
Yuri Tikhonovb428f6a2008-02-04 14:11:03 +0100380#ifdef CONFIG_POST
381 if (gd->flags & GD_FLG_POSTFAIL) {
382 s = getenv("failbootcmd");
383 }
384 else
385#endif /* CONFIG_POST */
wdenkbdccc4f2003-08-05 17:43:17 +0000386#ifdef CONFIG_BOOTCOUNT_LIMIT
387 if (bootlimit && (bootcount > bootlimit)) {
388 printf ("Warning: Bootlimit (%u) exceeded. Using altbootcmd.\n",
389 (unsigned)bootlimit);
390 s = getenv ("altbootcmd");
391 }
392 else
393#endif /* CONFIG_BOOTCOUNT_LIMIT */
394 s = getenv ("bootcmd");
wdenka6c7ad22002-12-03 21:28:10 +0000395
396 debug ("### main_loop: bootcmd=\"%s\"\n", s ? s : "<UNDEFINED>");
397
wdenkc6097192002-11-03 00:24:07 +0000398 if (bootdelay >= 0 && s && !abortboot (bootdelay)) {
399# ifdef CONFIG_AUTOBOOT_KEYED
400 int prev = disable_ctrlc(1); /* disable Control C checking */
401# endif
402
Jason Hobbsc8a20792011-08-31 05:37:24 +0000403 run_command2(s, 0);
wdenkc6097192002-11-03 00:24:07 +0000404
405# ifdef CONFIG_AUTOBOOT_KEYED
406 disable_ctrlc(prev); /* restore Control C checking */
407# endif
408 }
wdenkc7de8292002-11-19 11:04:11 +0000409
410# ifdef CONFIG_MENUKEY
wdenka6c7ad22002-12-03 21:28:10 +0000411 if (menukey == CONFIG_MENUKEY) {
Jason Hobbs370d1e32011-06-23 08:27:30 +0000412 s = getenv("menucmd");
Jason Hobbsc8a20792011-08-31 05:37:24 +0000413 if (s)
414 run_command2(s, 0);
wdenkc7de8292002-11-19 11:04:11 +0000415 }
416#endif /* CONFIG_MENUKEY */
Wolfgang Denk953b7e62010-06-13 18:28:54 +0200417#endif /* CONFIG_BOOTDELAY */
wdenkc7de8292002-11-19 11:04:11 +0000418
wdenkc6097192002-11-03 00:24:07 +0000419 /*
420 * Main Loop for Monitor Command Processing
421 */
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +0200422#ifdef CONFIG_SYS_HUSH_PARSER
wdenkc6097192002-11-03 00:24:07 +0000423 parse_file_outer();
424 /* This point is never reached */
425 for (;;);
426#else
427 for (;;) {
428#ifdef CONFIG_BOOT_RETRY_TIME
429 if (rc >= 0) {
430 /* Saw enough of a valid command to
431 * restart the timeout.
432 */
433 reset_cmd_timeout();
434 }
435#endif
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +0200436 len = readline (CONFIG_SYS_PROMPT);
wdenkc6097192002-11-03 00:24:07 +0000437
438 flag = 0; /* assume no special flags for now */
439 if (len > 0)
440 strcpy (lastcommand, console_buffer);
441 else if (len == 0)
442 flag |= CMD_FLAG_REPEAT;
443#ifdef CONFIG_BOOT_RETRY_TIME
444 else if (len == -2) {
445 /* -2 means timed out, retry autoboot
446 */
wdenk4b9206e2004-03-23 22:14:11 +0000447 puts ("\nTimed out waiting for command\n");
wdenkc6097192002-11-03 00:24:07 +0000448# ifdef CONFIG_RESET_TO_RETRY
449 /* Reinit board to run initialization code again */
450 do_reset (NULL, 0, 0, NULL);
451# else
452 return; /* retry autoboot */
453# endif
454 }
455#endif
456
457 if (len == -1)
wdenk4b9206e2004-03-23 22:14:11 +0000458 puts ("<INTERRUPT>\n");
wdenkc6097192002-11-03 00:24:07 +0000459 else
460 rc = run_command (lastcommand, flag);
461
462 if (rc <= 0) {
463 /* invalid command or not repeatable, forget it */
464 lastcommand[0] = 0;
465 }
466 }
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +0200467#endif /*CONFIG_SYS_HUSH_PARSER*/
wdenkc6097192002-11-03 00:24:07 +0000468}
469
wdenk6dd652f2003-06-19 23:40:20 +0000470#ifdef CONFIG_BOOT_RETRY_TIME
471/***************************************************************************
Wolfgang Denk19973b62006-10-28 00:38:39 +0200472 * initialize command line timeout
wdenk6dd652f2003-06-19 23:40:20 +0000473 */
474void init_cmd_timeout(void)
475{
476 char *s = getenv ("bootretry");
477
478 if (s != NULL)
wdenkb028f712003-12-07 21:39:28 +0000479 retry_time = (int)simple_strtol(s, NULL, 10);
wdenk6dd652f2003-06-19 23:40:20 +0000480 else
481 retry_time = CONFIG_BOOT_RETRY_TIME;
482
483 if (retry_time >= 0 && retry_time < CONFIG_BOOT_RETRY_MIN)
484 retry_time = CONFIG_BOOT_RETRY_MIN;
485}
486
wdenkc6097192002-11-03 00:24:07 +0000487/***************************************************************************
488 * reset command line timeout to retry_time seconds
489 */
wdenkc6097192002-11-03 00:24:07 +0000490void reset_cmd_timeout(void)
491{
492 endtime = endtick(retry_time);
493}
494#endif
495
Wolfgang Denk501090a2006-07-21 11:33:45 +0200496#ifdef CONFIG_CMDLINE_EDITING
497
498/*
499 * cmdline-editing related codes from vivi.
500 * Author: Janghoon Lyu <nandy@mizi.com>
501 */
502
Wolfgang Denk501090a2006-07-21 11:33:45 +0200503#define putnstr(str,n) do { \
Andrew Klossnerdc4b0b32008-07-07 06:41:14 -0700504 printf ("%.*s", (int)n, str); \
Wolfgang Denk501090a2006-07-21 11:33:45 +0200505 } while (0)
Wolfgang Denk501090a2006-07-21 11:33:45 +0200506
507#define CTL_CH(c) ((c) - 'a' + 1)
Wolfgang Denk501090a2006-07-21 11:33:45 +0200508#define CTL_BACKSPACE ('\b')
509#define DEL ((char)255)
510#define DEL7 ((char)127)
511#define CREAD_HIST_CHAR ('!')
512
513#define getcmd_putch(ch) putc(ch)
514#define getcmd_getch() getc()
515#define getcmd_cbeep() getcmd_putch('\a')
516
517#define HIST_MAX 20
Peter Tyser8804ae32010-09-29 13:30:56 -0500518#define HIST_SIZE CONFIG_SYS_CBSIZE
Wolfgang Denk501090a2006-07-21 11:33:45 +0200519
520static int hist_max = 0;
521static int hist_add_idx = 0;
522static int hist_cur = -1;
523unsigned hist_num = 0;
524
525char* hist_list[HIST_MAX];
John Schmoller6475b9f2010-03-12 09:49:23 -0600526char hist_lines[HIST_MAX][HIST_SIZE + 1]; /* Save room for NULL */
Wolfgang Denk501090a2006-07-21 11:33:45 +0200527
528#define add_idx_minus_one() ((hist_add_idx == 0) ? hist_max : hist_add_idx-1)
529
530static void hist_init(void)
531{
532 int i;
533
534 hist_max = 0;
535 hist_add_idx = 0;
536 hist_cur = -1;
537 hist_num = 0;
538
539 for (i = 0; i < HIST_MAX; i++) {
540 hist_list[i] = hist_lines[i];
541 hist_list[i][0] = '\0';
542 }
543}
544
545static void cread_add_to_hist(char *line)
546{
547 strcpy(hist_list[hist_add_idx], line);
548
549 if (++hist_add_idx >= HIST_MAX)
550 hist_add_idx = 0;
551
552 if (hist_add_idx > hist_max)
553 hist_max = hist_add_idx;
554
555 hist_num++;
556}
557
558static char* hist_prev(void)
559{
560 char *ret;
561 int old_cur;
562
563 if (hist_cur < 0)
564 return NULL;
565
566 old_cur = hist_cur;
567 if (--hist_cur < 0)
568 hist_cur = hist_max;
569
570 if (hist_cur == hist_add_idx) {
571 hist_cur = old_cur;
572 ret = NULL;
573 } else
574 ret = hist_list[hist_cur];
575
576 return (ret);
577}
578
579static char* hist_next(void)
580{
581 char *ret;
582
583 if (hist_cur < 0)
584 return NULL;
585
586 if (hist_cur == hist_add_idx)
587 return NULL;
588
589 if (++hist_cur > hist_max)
590 hist_cur = 0;
591
592 if (hist_cur == hist_add_idx) {
593 ret = "";
594 } else
595 ret = hist_list[hist_cur];
596
597 return (ret);
598}
599
Stefan Roese3ca91222006-07-27 16:11:19 +0200600#ifndef CONFIG_CMDLINE_EDITING
Wolfgang Denk501090a2006-07-21 11:33:45 +0200601static void cread_print_hist_list(void)
602{
603 int i;
604 unsigned long n;
605
606 n = hist_num - hist_max;
607
608 i = hist_add_idx + 1;
609 while (1) {
610 if (i > hist_max)
611 i = 0;
612 if (i == hist_add_idx)
613 break;
614 printf("%s\n", hist_list[i]);
615 n++;
616 i++;
617 }
618}
Stefan Roese3ca91222006-07-27 16:11:19 +0200619#endif /* CONFIG_CMDLINE_EDITING */
Wolfgang Denk501090a2006-07-21 11:33:45 +0200620
621#define BEGINNING_OF_LINE() { \
622 while (num) { \
623 getcmd_putch(CTL_BACKSPACE); \
624 num--; \
625 } \
626}
627
628#define ERASE_TO_EOL() { \
629 if (num < eol_num) { \
Mike Frysinger8faba482010-07-23 05:28:15 -0400630 printf("%*s", (int)(eol_num - num), ""); \
631 do { \
Wolfgang Denk501090a2006-07-21 11:33:45 +0200632 getcmd_putch(CTL_BACKSPACE); \
Mike Frysinger8faba482010-07-23 05:28:15 -0400633 } while (--eol_num > num); \
Wolfgang Denk501090a2006-07-21 11:33:45 +0200634 } \
635}
636
637#define REFRESH_TO_EOL() { \
638 if (num < eol_num) { \
639 wlen = eol_num - num; \
640 putnstr(buf + num, wlen); \
641 num = eol_num; \
642 } \
643}
644
645static void cread_add_char(char ichar, int insert, unsigned long *num,
646 unsigned long *eol_num, char *buf, unsigned long len)
647{
648 unsigned long wlen;
649
650 /* room ??? */
651 if (insert || *num == *eol_num) {
652 if (*eol_num > len - 1) {
653 getcmd_cbeep();
654 return;
655 }
656 (*eol_num)++;
657 }
658
659 if (insert) {
660 wlen = *eol_num - *num;
661 if (wlen > 1) {
662 memmove(&buf[*num+1], &buf[*num], wlen-1);
663 }
664
665 buf[*num] = ichar;
666 putnstr(buf + *num, wlen);
667 (*num)++;
668 while (--wlen) {
669 getcmd_putch(CTL_BACKSPACE);
670 }
671 } else {
672 /* echo the character */
673 wlen = 1;
674 buf[*num] = ichar;
675 putnstr(buf + *num, wlen);
676 (*num)++;
677 }
678}
679
680static void cread_add_str(char *str, int strsize, int insert, unsigned long *num,
681 unsigned long *eol_num, char *buf, unsigned long len)
682{
683 while (strsize--) {
684 cread_add_char(*str, insert, num, eol_num, buf, len);
685 str++;
686 }
687}
688
Heiko Schocher9c348312012-01-16 21:13:05 +0000689static int cread_line(const char *const prompt, char *buf, unsigned int *len,
690 int timeout)
Wolfgang Denk501090a2006-07-21 11:33:45 +0200691{
692 unsigned long num = 0;
693 unsigned long eol_num = 0;
Wolfgang Denk501090a2006-07-21 11:33:45 +0200694 unsigned long wlen;
695 char ichar;
696 int insert = 1;
697 int esc_len = 0;
Wolfgang Denk501090a2006-07-21 11:33:45 +0200698 char esc_save[8];
Peter Tyserecc55002009-10-25 15:12:54 -0500699 int init_len = strlen(buf);
Heiko Schocher9c348312012-01-16 21:13:05 +0000700 int first = 1;
Peter Tyserecc55002009-10-25 15:12:54 -0500701
702 if (init_len)
703 cread_add_str(buf, init_len, 1, &num, &eol_num, buf, *len);
Wolfgang Denk501090a2006-07-21 11:33:45 +0200704
705 while (1) {
Andreas Engel00ac50e2008-01-09 17:10:56 +0100706#ifdef CONFIG_BOOT_RETRY_TIME
707 while (!tstc()) { /* while no incoming data */
708 if (retry_time >= 0 && get_ticks() > endtime)
709 return (-2); /* timed out */
Jens Scharsig30dc1652010-04-09 19:02:38 +0200710 WATCHDOG_RESET();
Andreas Engel00ac50e2008-01-09 17:10:56 +0100711 }
712#endif
Heiko Schocher9c348312012-01-16 21:13:05 +0000713 if (first && timeout) {
714 uint64_t etime = endtick(timeout);
715
716 while (!tstc()) { /* while no incoming data */
717 if (get_ticks() >= etime)
718 return -2; /* timed out */
719 WATCHDOG_RESET();
720 }
721 first = 0;
722 }
Andreas Engel00ac50e2008-01-09 17:10:56 +0100723
Wolfgang Denk501090a2006-07-21 11:33:45 +0200724 ichar = getcmd_getch();
725
726 if ((ichar == '\n') || (ichar == '\r')) {
Wolfgang Denkdd9f06f2006-07-21 11:34:34 +0200727 putc('\n');
Wolfgang Denk501090a2006-07-21 11:33:45 +0200728 break;
729 }
730
731 /*
732 * handle standard linux xterm esc sequences for arrow key, etc.
733 */
734 if (esc_len != 0) {
735 if (esc_len == 1) {
736 if (ichar == '[') {
737 esc_save[esc_len] = ichar;
738 esc_len = 2;
739 } else {
740 cread_add_str(esc_save, esc_len, insert,
741 &num, &eol_num, buf, *len);
742 esc_len = 0;
743 }
744 continue;
745 }
746
747 switch (ichar) {
748
749 case 'D': /* <- key */
750 ichar = CTL_CH('b');
751 esc_len = 0;
752 break;
753 case 'C': /* -> key */
754 ichar = CTL_CH('f');
755 esc_len = 0;
756 break; /* pass off to ^F handler */
757 case 'H': /* Home key */
758 ichar = CTL_CH('a');
759 esc_len = 0;
760 break; /* pass off to ^A handler */
761 case 'A': /* up arrow */
762 ichar = CTL_CH('p');
763 esc_len = 0;
764 break; /* pass off to ^P handler */
765 case 'B': /* down arrow */
766 ichar = CTL_CH('n');
767 esc_len = 0;
768 break; /* pass off to ^N handler */
769 default:
770 esc_save[esc_len++] = ichar;
771 cread_add_str(esc_save, esc_len, insert,
772 &num, &eol_num, buf, *len);
773 esc_len = 0;
774 continue;
775 }
776 }
777
778 switch (ichar) {
779 case 0x1b:
780 if (esc_len == 0) {
781 esc_save[esc_len] = ichar;
782 esc_len = 1;
783 } else {
Wolfgang Denkdd9f06f2006-07-21 11:34:34 +0200784 puts("impossible condition #876\n");
Wolfgang Denk501090a2006-07-21 11:33:45 +0200785 esc_len = 0;
786 }
787 break;
788
789 case CTL_CH('a'):
790 BEGINNING_OF_LINE();
791 break;
792 case CTL_CH('c'): /* ^C - break */
793 *buf = '\0'; /* discard input */
794 return (-1);
795 case CTL_CH('f'):
796 if (num < eol_num) {
797 getcmd_putch(buf[num]);
798 num++;
799 }
800 break;
801 case CTL_CH('b'):
802 if (num) {
803 getcmd_putch(CTL_BACKSPACE);
804 num--;
805 }
806 break;
807 case CTL_CH('d'):
808 if (num < eol_num) {
809 wlen = eol_num - num - 1;
810 if (wlen) {
811 memmove(&buf[num], &buf[num+1], wlen);
812 putnstr(buf + num, wlen);
813 }
814
815 getcmd_putch(' ');
816 do {
817 getcmd_putch(CTL_BACKSPACE);
818 } while (wlen--);
819 eol_num--;
820 }
821 break;
822 case CTL_CH('k'):
823 ERASE_TO_EOL();
824 break;
825 case CTL_CH('e'):
826 REFRESH_TO_EOL();
827 break;
828 case CTL_CH('o'):
829 insert = !insert;
830 break;
831 case CTL_CH('x'):
Jean-Christophe PLAGNIOL-VILLARD23d0baf2007-12-22 15:52:58 +0100832 case CTL_CH('u'):
Wolfgang Denk501090a2006-07-21 11:33:45 +0200833 BEGINNING_OF_LINE();
834 ERASE_TO_EOL();
835 break;
836 case DEL:
837 case DEL7:
838 case 8:
839 if (num) {
840 wlen = eol_num - num;
841 num--;
842 memmove(&buf[num], &buf[num+1], wlen);
843 getcmd_putch(CTL_BACKSPACE);
844 putnstr(buf + num, wlen);
845 getcmd_putch(' ');
846 do {
847 getcmd_putch(CTL_BACKSPACE);
848 } while (wlen--);
849 eol_num--;
850 }
851 break;
852 case CTL_CH('p'):
853 case CTL_CH('n'):
854 {
855 char * hline;
856
857 esc_len = 0;
858
859 if (ichar == CTL_CH('p'))
860 hline = hist_prev();
861 else
862 hline = hist_next();
863
864 if (!hline) {
865 getcmd_cbeep();
866 continue;
867 }
868
869 /* nuke the current line */
870 /* first, go home */
871 BEGINNING_OF_LINE();
872
873 /* erase to end of line */
874 ERASE_TO_EOL();
875
876 /* copy new line into place and display */
877 strcpy(buf, hline);
878 eol_num = strlen(buf);
879 REFRESH_TO_EOL();
880 continue;
881 }
Jean-Christophe PLAGNIOL-VILLARD23d0baf2007-12-22 15:52:58 +0100882#ifdef CONFIG_AUTO_COMPLETE
883 case '\t': {
884 int num2, col;
885
886 /* do not autocomplete when in the middle */
887 if (num < eol_num) {
888 getcmd_cbeep();
889 break;
890 }
891
892 buf[num] = '\0';
893 col = strlen(prompt) + eol_num;
894 num2 = num;
895 if (cmd_auto_complete(prompt, buf, &num2, &col)) {
896 col = num2 - num;
897 num += col;
898 eol_num += col;
899 }
900 break;
901 }
902#endif
Wolfgang Denk501090a2006-07-21 11:33:45 +0200903 default:
904 cread_add_char(ichar, insert, &num, &eol_num, buf, *len);
905 break;
906 }
907 }
908 *len = eol_num;
909 buf[eol_num] = '\0'; /* lose the newline */
910
911 if (buf[0] && buf[0] != CREAD_HIST_CHAR)
912 cread_add_to_hist(buf);
913 hist_cur = hist_add_idx;
914
Peter Tyserf9239432009-10-25 15:12:53 -0500915 return 0;
Wolfgang Denk501090a2006-07-21 11:33:45 +0200916}
917
918#endif /* CONFIG_CMDLINE_EDITING */
919
wdenkc6097192002-11-03 00:24:07 +0000920/****************************************************************************/
921
922/*
923 * Prompt for input and read a line.
924 * If CONFIG_BOOT_RETRY_TIME is defined and retry_time >= 0,
925 * time out when time goes past endtime (timebase time in ticks).
926 * Return: number of read characters
927 * -1 if break
928 * -2 if timed out
929 */
930int readline (const char *const prompt)
931{
Peter Tyserecc55002009-10-25 15:12:54 -0500932 /*
933 * If console_buffer isn't 0-length the user will be prompted to modify
934 * it instead of entering it from scratch as desired.
935 */
936 console_buffer[0] = '\0';
937
Heiko Schocher9c348312012-01-16 21:13:05 +0000938 return readline_into_buffer(prompt, console_buffer, 0);
James Yang6636b622008-01-09 11:17:49 -0600939}
940
941
Heiko Schocher9c348312012-01-16 21:13:05 +0000942int readline_into_buffer(const char *const prompt, char *buffer, int timeout)
James Yang6636b622008-01-09 11:17:49 -0600943{
944 char *p = buffer;
Wolfgang Denk501090a2006-07-21 11:33:45 +0200945#ifdef CONFIG_CMDLINE_EDITING
Peter Tyser8804ae32010-09-29 13:30:56 -0500946 unsigned int len = CONFIG_SYS_CBSIZE;
Stefan Roesed8f961b2006-08-07 15:08:44 +0200947 int rc;
Wolfgang Denk501090a2006-07-21 11:33:45 +0200948 static int initted = 0;
949
James Yang597f6c22008-05-05 10:22:53 -0500950 /*
951 * History uses a global array which is not
952 * writable until after relocation to RAM.
953 * Revert to non-history version if still
954 * running from flash.
955 */
956 if (gd->flags & GD_FLG_RELOC) {
957 if (!initted) {
958 hist_init();
959 initted = 1;
960 }
961
Peter Tysere491a712009-10-25 15:12:52 -0500962 if (prompt)
963 puts (prompt);
James Yang597f6c22008-05-05 10:22:53 -0500964
Heiko Schocher9c348312012-01-16 21:13:05 +0000965 rc = cread_line(prompt, p, &len, timeout);
James Yang597f6c22008-05-05 10:22:53 -0500966 return rc < 0 ? rc : len;
967
968 } else {
969#endif /* CONFIG_CMDLINE_EDITING */
Kumar Gala0ec59522008-01-10 02:22:05 -0600970 char * p_buf = p;
wdenkc6097192002-11-03 00:24:07 +0000971 int n = 0; /* buffer index */
972 int plen = 0; /* prompt length */
973 int col; /* output column cnt */
974 char c;
975
976 /* print prompt */
977 if (prompt) {
978 plen = strlen (prompt);
979 puts (prompt);
980 }
981 col = plen;
982
983 for (;;) {
984#ifdef CONFIG_BOOT_RETRY_TIME
985 while (!tstc()) { /* while no incoming data */
986 if (retry_time >= 0 && get_ticks() > endtime)
987 return (-2); /* timed out */
Jens Scharsig30dc1652010-04-09 19:02:38 +0200988 WATCHDOG_RESET();
wdenkc6097192002-11-03 00:24:07 +0000989 }
990#endif
991 WATCHDOG_RESET(); /* Trigger watchdog, if needed */
992
993#ifdef CONFIG_SHOW_ACTIVITY
994 while (!tstc()) {
995 extern void show_activity(int arg);
996 show_activity(0);
Jens Scharsig30dc1652010-04-09 19:02:38 +0200997 WATCHDOG_RESET();
wdenkc6097192002-11-03 00:24:07 +0000998 }
999#endif
1000 c = getc();
1001
1002 /*
1003 * Special character handling
1004 */
1005 switch (c) {
1006 case '\r': /* Enter */
1007 case '\n':
1008 *p = '\0';
1009 puts ("\r\n");
James Yang6636b622008-01-09 11:17:49 -06001010 return (p - p_buf);
wdenkc6097192002-11-03 00:24:07 +00001011
wdenk27aa8182004-03-23 22:37:33 +00001012 case '\0': /* nul */
1013 continue;
1014
wdenkc6097192002-11-03 00:24:07 +00001015 case 0x03: /* ^C - break */
James Yang6636b622008-01-09 11:17:49 -06001016 p_buf[0] = '\0'; /* discard input */
wdenkc6097192002-11-03 00:24:07 +00001017 return (-1);
1018
1019 case 0x15: /* ^U - erase line */
1020 while (col > plen) {
1021 puts (erase_seq);
1022 --col;
1023 }
James Yang6636b622008-01-09 11:17:49 -06001024 p = p_buf;
wdenkc6097192002-11-03 00:24:07 +00001025 n = 0;
1026 continue;
1027
Wolfgang Denk1636d1c2007-06-22 23:59:00 +02001028 case 0x17: /* ^W - erase word */
James Yang6636b622008-01-09 11:17:49 -06001029 p=delete_char(p_buf, p, &col, &n, plen);
wdenkc6097192002-11-03 00:24:07 +00001030 while ((n > 0) && (*p != ' ')) {
James Yang6636b622008-01-09 11:17:49 -06001031 p=delete_char(p_buf, p, &col, &n, plen);
wdenkc6097192002-11-03 00:24:07 +00001032 }
1033 continue;
1034
1035 case 0x08: /* ^H - backspace */
1036 case 0x7F: /* DEL - backspace */
James Yang6636b622008-01-09 11:17:49 -06001037 p=delete_char(p_buf, p, &col, &n, plen);
wdenkc6097192002-11-03 00:24:07 +00001038 continue;
1039
1040 default:
1041 /*
1042 * Must be a normal character then
1043 */
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +02001044 if (n < CONFIG_SYS_CBSIZE-2) {
wdenkc6097192002-11-03 00:24:07 +00001045 if (c == '\t') { /* expand TABs */
wdenk04a85b32004-04-15 18:22:41 +00001046#ifdef CONFIG_AUTO_COMPLETE
1047 /* if auto completion triggered just continue */
1048 *p = '\0';
1049 if (cmd_auto_complete(prompt, console_buffer, &n, &col)) {
James Yang6636b622008-01-09 11:17:49 -06001050 p = p_buf + n; /* reset */
wdenk04a85b32004-04-15 18:22:41 +00001051 continue;
1052 }
1053#endif
wdenkc6097192002-11-03 00:24:07 +00001054 puts (tab_seq+(col&07));
1055 col += 8 - (col&07);
1056 } else {
1057 ++col; /* echo input */
1058 putc (c);
1059 }
1060 *p++ = c;
1061 ++n;
1062 } else { /* Buffer full */
1063 putc ('\a');
1064 }
1065 }
1066 }
James Yang597f6c22008-05-05 10:22:53 -05001067#ifdef CONFIG_CMDLINE_EDITING
1068 }
1069#endif
wdenkc6097192002-11-03 00:24:07 +00001070}
1071
1072/****************************************************************************/
1073
1074static char * delete_char (char *buffer, char *p, int *colp, int *np, int plen)
1075{
1076 char *s;
1077
1078 if (*np == 0) {
1079 return (p);
1080 }
1081
1082 if (*(--p) == '\t') { /* will retype the whole line */
1083 while (*colp > plen) {
1084 puts (erase_seq);
1085 (*colp)--;
1086 }
1087 for (s=buffer; s<p; ++s) {
1088 if (*s == '\t') {
1089 puts (tab_seq+((*colp) & 07));
1090 *colp += 8 - ((*colp) & 07);
1091 } else {
1092 ++(*colp);
1093 putc (*s);
1094 }
1095 }
1096 } else {
1097 puts (erase_seq);
1098 (*colp)--;
1099 }
1100 (*np)--;
1101 return (p);
1102}
1103
1104/****************************************************************************/
1105
1106int parse_line (char *line, char *argv[])
1107{
1108 int nargs = 0;
1109
1110#ifdef DEBUG_PARSER
1111 printf ("parse_line: \"%s\"\n", line);
1112#endif
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +02001113 while (nargs < CONFIG_SYS_MAXARGS) {
wdenkc6097192002-11-03 00:24:07 +00001114
1115 /* skip any white space */
Jason Hobbs4d91a6e2011-08-23 11:06:54 +00001116 while (isblank(*line))
wdenkc6097192002-11-03 00:24:07 +00001117 ++line;
wdenkc6097192002-11-03 00:24:07 +00001118
1119 if (*line == '\0') { /* end of line, no more args */
1120 argv[nargs] = NULL;
1121#ifdef DEBUG_PARSER
1122 printf ("parse_line: nargs=%d\n", nargs);
1123#endif
1124 return (nargs);
1125 }
1126
1127 argv[nargs++] = line; /* begin of argument string */
1128
1129 /* find end of string */
Jason Hobbs4d91a6e2011-08-23 11:06:54 +00001130 while (*line && !isblank(*line))
wdenkc6097192002-11-03 00:24:07 +00001131 ++line;
wdenkc6097192002-11-03 00:24:07 +00001132
1133 if (*line == '\0') { /* end of line, no more args */
1134 argv[nargs] = NULL;
1135#ifdef DEBUG_PARSER
1136 printf ("parse_line: nargs=%d\n", nargs);
1137#endif
1138 return (nargs);
1139 }
1140
1141 *line++ = '\0'; /* terminate current arg */
1142 }
1143
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +02001144 printf ("** Too many args (max. %d) **\n", CONFIG_SYS_MAXARGS);
wdenkc6097192002-11-03 00:24:07 +00001145
1146#ifdef DEBUG_PARSER
1147 printf ("parse_line: nargs=%d\n", nargs);
1148#endif
1149 return (nargs);
1150}
1151
1152/****************************************************************************/
1153
1154static void process_macros (const char *input, char *output)
1155{
1156 char c, prev;
1157 const char *varname_start = NULL;
Wolfgang Denk19973b62006-10-28 00:38:39 +02001158 int inputcnt = strlen (input);
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +02001159 int outputcnt = CONFIG_SYS_CBSIZE;
Wolfgang Denk19973b62006-10-28 00:38:39 +02001160 int state = 0; /* 0 = waiting for '$' */
1161
1162 /* 1 = waiting for '(' or '{' */
1163 /* 2 = waiting for ')' or '}' */
1164 /* 3 = waiting for ''' */
wdenkc6097192002-11-03 00:24:07 +00001165#ifdef DEBUG_PARSER
1166 char *output_start = output;
1167
Wolfgang Denk19973b62006-10-28 00:38:39 +02001168 printf ("[PROCESS_MACROS] INPUT len %d: \"%s\"\n", strlen (input),
1169 input);
wdenkc6097192002-11-03 00:24:07 +00001170#endif
1171
Wolfgang Denk19973b62006-10-28 00:38:39 +02001172 prev = '\0'; /* previous character */
wdenkc6097192002-11-03 00:24:07 +00001173
1174 while (inputcnt && outputcnt) {
wdenk8bde7f72003-06-27 21:31:46 +00001175 c = *input++;
Wolfgang Denk19973b62006-10-28 00:38:39 +02001176 inputcnt--;
wdenkc6097192002-11-03 00:24:07 +00001177
Wolfgang Denk19973b62006-10-28 00:38:39 +02001178 if (state != 3) {
1179 /* remove one level of escape characters */
1180 if ((c == '\\') && (prev != '\\')) {
1181 if (inputcnt-- == 0)
1182 break;
1183 prev = c;
1184 c = *input++;
1185 }
wdenka25f8622003-01-02 23:57:29 +00001186 }
wdenkc6097192002-11-03 00:24:07 +00001187
Wolfgang Denk19973b62006-10-28 00:38:39 +02001188 switch (state) {
1189 case 0: /* Waiting for (unescaped) $ */
1190 if ((c == '\'') && (prev != '\\')) {
1191 state = 3;
1192 break;
1193 }
1194 if ((c == '$') && (prev != '\\')) {
1195 state++;
1196 } else {
wdenkc6097192002-11-03 00:24:07 +00001197 *(output++) = c;
1198 outputcnt--;
1199 }
Wolfgang Denk19973b62006-10-28 00:38:39 +02001200 break;
1201 case 1: /* Waiting for ( */
1202 if (c == '(' || c == '{') {
1203 state++;
1204 varname_start = input;
1205 } else {
1206 state = 0;
1207 *(output++) = '$';
1208 outputcnt--;
wdenkc6097192002-11-03 00:24:07 +00001209
Wolfgang Denk19973b62006-10-28 00:38:39 +02001210 if (outputcnt) {
1211 *(output++) = c;
wdenkc6097192002-11-03 00:24:07 +00001212 outputcnt--;
1213 }
Wolfgang Denk19973b62006-10-28 00:38:39 +02001214 }
1215 break;
1216 case 2: /* Waiting for ) */
1217 if (c == ')' || c == '}') {
1218 int i;
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +02001219 char envname[CONFIG_SYS_CBSIZE], *envval;
Wolfgang Denk19973b62006-10-28 00:38:39 +02001220 int envcnt = input - varname_start - 1; /* Varname # of chars */
1221
1222 /* Get the varname */
1223 for (i = 0; i < envcnt; i++) {
1224 envname[i] = varname_start[i];
1225 }
1226 envname[i] = 0;
1227
1228 /* Get its value */
1229 envval = getenv (envname);
1230
1231 /* Copy into the line if it exists */
1232 if (envval != NULL)
1233 while ((*envval) && outputcnt) {
1234 *(output++) = *(envval++);
1235 outputcnt--;
1236 }
1237 /* Look for another '$' */
1238 state = 0;
1239 }
1240 break;
1241 case 3: /* Waiting for ' */
1242 if ((c == '\'') && (prev != '\\')) {
1243 state = 0;
1244 } else {
1245 *(output++) = c;
1246 outputcnt--;
1247 }
1248 break;
wdenkc6097192002-11-03 00:24:07 +00001249 }
Wolfgang Denk19973b62006-10-28 00:38:39 +02001250 prev = c;
wdenkc6097192002-11-03 00:24:07 +00001251 }
1252
1253 if (outputcnt)
1254 *output = 0;
Bartlomiej Sieka9160b962007-05-27 17:04:18 +02001255 else
1256 *(output - 1) = 0;
wdenkc6097192002-11-03 00:24:07 +00001257
1258#ifdef DEBUG_PARSER
1259 printf ("[PROCESS_MACROS] OUTPUT len %d: \"%s\"\n",
Wolfgang Denk19973b62006-10-28 00:38:39 +02001260 strlen (output_start), output_start);
wdenkc6097192002-11-03 00:24:07 +00001261#endif
1262}
1263
1264/****************************************************************************
1265 * returns:
1266 * 1 - command executed, repeatable
1267 * 0 - command executed but not repeatable, interrupted commands are
1268 * always considered not repeatable
1269 * -1 - not executed (unrecognized, bootd recursion or too many args)
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +02001270 * (If cmd is NULL or "" or longer than CONFIG_SYS_CBSIZE-1 it is
wdenkc6097192002-11-03 00:24:07 +00001271 * considered unrecognized)
1272 *
1273 * WARNING:
1274 *
1275 * We must create a temporary copy of the command since the command we get
1276 * may be the result from getenv(), which returns a pointer directly to
1277 * the environment data, which may change magicly when the command we run
1278 * creates or modifies environment variables (like "bootp" does).
1279 */
1280
1281int run_command (const char *cmd, int flag)
1282{
1283 cmd_tbl_t *cmdtp;
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +02001284 char cmdbuf[CONFIG_SYS_CBSIZE]; /* working copy of cmd */
wdenkc6097192002-11-03 00:24:07 +00001285 char *token; /* start of token in cmdbuf */
1286 char *sep; /* end of token (separator) in cmdbuf */
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +02001287 char finaltoken[CONFIG_SYS_CBSIZE];
wdenkc6097192002-11-03 00:24:07 +00001288 char *str = cmdbuf;
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +02001289 char *argv[CONFIG_SYS_MAXARGS + 1]; /* NULL terminated */
wdenkf07771c2003-05-28 08:06:31 +00001290 int argc, inquotes;
wdenkc6097192002-11-03 00:24:07 +00001291 int repeatable = 1;
wdenkf07771c2003-05-28 08:06:31 +00001292 int rc = 0;
wdenkc6097192002-11-03 00:24:07 +00001293
1294#ifdef DEBUG_PARSER
1295 printf ("[RUN_COMMAND] cmd[%p]=\"", cmd);
1296 puts (cmd ? cmd : "NULL"); /* use puts - string may be loooong */
1297 puts ("\"\n");
1298#endif
1299
1300 clear_ctrlc(); /* forget any previous Control C */
1301
1302 if (!cmd || !*cmd) {
1303 return -1; /* empty command */
1304 }
1305
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +02001306 if (strlen(cmd) >= CONFIG_SYS_CBSIZE) {
wdenkc6097192002-11-03 00:24:07 +00001307 puts ("## Command too long!\n");
1308 return -1;
1309 }
1310
1311 strcpy (cmdbuf, cmd);
1312
1313 /* Process separators and check for invalid
1314 * repeatable commands
1315 */
1316
1317#ifdef DEBUG_PARSER
1318 printf ("[PROCESS_SEPARATORS] %s\n", cmd);
1319#endif
1320 while (*str) {
1321
1322 /*
1323 * Find separator, or string end
1324 * Allow simple escape of ';' by writing "\;"
1325 */
wdenka25f8622003-01-02 23:57:29 +00001326 for (inquotes = 0, sep = str; *sep; sep++) {
1327 if ((*sep=='\'') &&
1328 (*(sep-1) != '\\'))
1329 inquotes=!inquotes;
1330
1331 if (!inquotes &&
1332 (*sep == ';') && /* separator */
wdenkc6097192002-11-03 00:24:07 +00001333 ( sep != str) && /* past string start */
1334 (*(sep-1) != '\\')) /* and NOT escaped */
1335 break;
1336 }
1337
1338 /*
1339 * Limit the token to data between separators
1340 */
1341 token = str;
1342 if (*sep) {
1343 str = sep + 1; /* start of command for next pass */
1344 *sep = '\0';
1345 }
1346 else
1347 str = sep; /* no more commands for next pass */
1348#ifdef DEBUG_PARSER
1349 printf ("token: \"%s\"\n", token);
1350#endif
1351
1352 /* find macros in this token and replace them */
1353 process_macros (token, finaltoken);
1354
1355 /* Extract arguments */
Wolfgang Denk1264b402006-03-12 02:20:55 +01001356 if ((argc = parse_line (finaltoken, argv)) == 0) {
1357 rc = -1; /* no command at all */
1358 continue;
1359 }
wdenkc6097192002-11-03 00:24:07 +00001360
1361 /* Look up command in command table */
1362 if ((cmdtp = find_cmd(argv[0])) == NULL) {
1363 printf ("Unknown command '%s' - try 'help'\n", argv[0]);
wdenkf07771c2003-05-28 08:06:31 +00001364 rc = -1; /* give up after bad command */
1365 continue;
wdenkc6097192002-11-03 00:24:07 +00001366 }
1367
1368 /* found - check max args */
1369 if (argc > cmdtp->maxargs) {
Peter Tyser62c3ae72009-01-27 18:03:10 -06001370 cmd_usage(cmdtp);
wdenkf07771c2003-05-28 08:06:31 +00001371 rc = -1;
1372 continue;
wdenkc6097192002-11-03 00:24:07 +00001373 }
1374
Jon Loeligerc3517f92007-07-08 18:10:08 -05001375#if defined(CONFIG_CMD_BOOTD)
wdenkc6097192002-11-03 00:24:07 +00001376 /* avoid "bootd" recursion */
1377 if (cmdtp->cmd == do_bootd) {
1378#ifdef DEBUG_PARSER
1379 printf ("[%s]\n", finaltoken);
1380#endif
1381 if (flag & CMD_FLAG_BOOTD) {
wdenk4b9206e2004-03-23 22:14:11 +00001382 puts ("'bootd' recursion detected\n");
wdenkf07771c2003-05-28 08:06:31 +00001383 rc = -1;
1384 continue;
Wolfgang Denk1264b402006-03-12 02:20:55 +01001385 } else {
wdenkc6097192002-11-03 00:24:07 +00001386 flag |= CMD_FLAG_BOOTD;
Wolfgang Denk1264b402006-03-12 02:20:55 +01001387 }
wdenkc6097192002-11-03 00:24:07 +00001388 }
Jon Loeliger90253172007-07-10 11:02:44 -05001389#endif
wdenkc6097192002-11-03 00:24:07 +00001390
1391 /* OK - call function to do the command */
1392 if ((cmdtp->cmd) (cmdtp, flag, argc, argv) != 0) {
wdenkf07771c2003-05-28 08:06:31 +00001393 rc = -1;
wdenkc6097192002-11-03 00:24:07 +00001394 }
1395
1396 repeatable &= cmdtp->repeatable;
1397
1398 /* Did the user stop this? */
1399 if (had_ctrlc ())
Detlev Zundel5afb2022007-05-23 18:47:48 +02001400 return -1; /* if stopped then not repeatable */
wdenkc6097192002-11-03 00:24:07 +00001401 }
1402
wdenkf07771c2003-05-28 08:06:31 +00001403 return rc ? rc : repeatable;
wdenkc6097192002-11-03 00:24:07 +00001404}
1405
1406/****************************************************************************/
1407
Jon Loeligerc3517f92007-07-08 18:10:08 -05001408#if defined(CONFIG_CMD_RUN)
Wolfgang Denk54841ab2010-06-28 22:00:46 +02001409int do_run (cmd_tbl_t * cmdtp, int flag, int argc, char * const argv[])
wdenkc6097192002-11-03 00:24:07 +00001410{
1411 int i;
wdenkc6097192002-11-03 00:24:07 +00001412
Wolfgang Denk47e26b12010-07-17 01:06:04 +02001413 if (argc < 2)
1414 return cmd_usage(cmdtp);
wdenkc6097192002-11-03 00:24:07 +00001415
1416 for (i=1; i<argc; ++i) {
wdenk3e386912003-04-05 00:53:31 +00001417 char *arg;
1418
1419 if ((arg = getenv (argv[i])) == NULL) {
1420 printf ("## Error: \"%s\" not defined\n", argv[i]);
1421 return 1;
1422 }
Jason Hobbsc8a20792011-08-31 05:37:24 +00001423
1424 if (run_command2(arg, flag) != 0)
wdenk3e386912003-04-05 00:53:31 +00001425 return 1;
wdenkc6097192002-11-03 00:24:07 +00001426 }
wdenk3e386912003-04-05 00:53:31 +00001427 return 0;
wdenkc6097192002-11-03 00:24:07 +00001428}
Jon Loeliger90253172007-07-10 11:02:44 -05001429#endif