blob: 248744ba74d3449518e8ecf2a4197a00a8a56f51 [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>
wdenkbdccc4f2003-08-05 17:43:17 +000044
James Yang597f6c22008-05-05 10:22:53 -050045#if defined(CONFIG_SILENT_CONSOLE) || defined(CONFIG_POST) || defined(CONFIG_CMDLINE_EDITING)
Wolfgang Denkd87080b2006-03-31 18:32:53 +020046DECLARE_GLOBAL_DATA_PTR;
47#endif
48
Heiko Schocherfad63402007-07-13 09:54:17 +020049/*
50 * Board-specific Platform code can reimplement show_boot_progress () if needed
51 */
52void inline __show_boot_progress (int val) {}
Emil Medve5e2c08c2009-05-12 13:48:32 -050053void show_boot_progress (int val) __attribute__((weak, alias("__show_boot_progress")));
Heiko Schocherfad63402007-07-13 09:54:17 +020054
Bartlomiej Sieka4bae9092008-10-01 15:26:31 +020055#if defined(CONFIG_UPDATE_TFTP)
Andreas Pretzsch8d6b7322011-07-16 05:50:59 +000056int update_tftp (ulong addr);
Bartlomiej Sieka4bae9092008-10-01 15:26:31 +020057#endif /* CONFIG_UPDATE_TFTP */
wdenk8bde7f72003-06-27 21:31:46 +000058
wdenkc6097192002-11-03 00:24:07 +000059#define MAX_DELAY_STOP_STR 32
60
wdenkc6097192002-11-03 00:24:07 +000061#undef DEBUG_PARSER
62
John Schmoller6475b9f2010-03-12 09:49:23 -060063char console_buffer[CONFIG_SYS_CBSIZE + 1]; /* console I/O buffer */
wdenkc6097192002-11-03 00:24:07 +000064
Stefan Roese3ca91222006-07-27 16:11:19 +020065static char * delete_char (char *buffer, char *p, int *colp, int *np, int plen);
Mike Frysinger82359ae2010-12-22 09:40:45 -050066static const char erase_seq[] = "\b \b"; /* erase sequence */
67static const char tab_seq[] = " "; /* used to expand TABs */
wdenkc6097192002-11-03 00:24:07 +000068
69#ifdef CONFIG_BOOT_RETRY_TIME
70static uint64_t endtime = 0; /* must be set, default is instant timeout */
71static int retry_time = -1; /* -1 so can call readline before main_loop */
72#endif
73
74#define endtick(seconds) (get_ticks() + (uint64_t)(seconds) * get_tbclk())
75
76#ifndef CONFIG_BOOT_RETRY_MIN
77#define CONFIG_BOOT_RETRY_MIN CONFIG_BOOT_RETRY_TIME
78#endif
79
80#ifdef CONFIG_MODEM_SUPPORT
81int do_mdm_init = 0;
82extern void mdm_init(void); /* defined in board.c */
83#endif
84
85/***************************************************************************
86 * Watch for 'delay' seconds for autoboot stop or autoboot delay string.
Jason Hobbsc8a20792011-08-31 05:37:24 +000087 * returns: 0 - no key string, allow autoboot 1 - got key string, abort
wdenkc6097192002-11-03 00:24:07 +000088 */
89#if defined(CONFIG_BOOTDELAY) && (CONFIG_BOOTDELAY >= 0)
90# if defined(CONFIG_AUTOBOOT_KEYED)
Jason Hobbsb41bc5a2011-08-23 11:06:50 +000091#ifndef CONFIG_MENU
92static inline
93#endif
94int abortboot(int bootdelay)
wdenkc6097192002-11-03 00:24:07 +000095{
96 int abort = 0;
97 uint64_t etime = endtick(bootdelay);
Wolfgang Denk19973b62006-10-28 00:38:39 +020098 struct {
wdenkc6097192002-11-03 00:24:07 +000099 char* str;
100 u_int len;
101 int retry;
102 }
Wolfgang Denk19973b62006-10-28 00:38:39 +0200103 delaykey [] = {
wdenkc6097192002-11-03 00:24:07 +0000104 { str: getenv ("bootdelaykey"), retry: 1 },
105 { str: getenv ("bootdelaykey2"), retry: 1 },
106 { str: getenv ("bootstopkey"), retry: 0 },
107 { str: getenv ("bootstopkey2"), retry: 0 },
108 };
109
110 char presskey [MAX_DELAY_STOP_STR];
111 u_int presskey_len = 0;
112 u_int presskey_max = 0;
113 u_int i;
114
115# ifdef CONFIG_AUTOBOOT_PROMPT
Stefan Roesef2302d42008-08-06 14:05:38 +0200116 printf(CONFIG_AUTOBOOT_PROMPT);
wdenkc6097192002-11-03 00:24:07 +0000117# endif
118
119# ifdef CONFIG_AUTOBOOT_DELAY_STR
120 if (delaykey[0].str == NULL)
121 delaykey[0].str = CONFIG_AUTOBOOT_DELAY_STR;
122# endif
123# ifdef CONFIG_AUTOBOOT_DELAY_STR2
124 if (delaykey[1].str == NULL)
125 delaykey[1].str = CONFIG_AUTOBOOT_DELAY_STR2;
126# endif
127# ifdef CONFIG_AUTOBOOT_STOP_STR
128 if (delaykey[2].str == NULL)
129 delaykey[2].str = CONFIG_AUTOBOOT_STOP_STR;
130# endif
131# ifdef CONFIG_AUTOBOOT_STOP_STR2
132 if (delaykey[3].str == NULL)
133 delaykey[3].str = CONFIG_AUTOBOOT_STOP_STR2;
134# endif
135
136 for (i = 0; i < sizeof(delaykey) / sizeof(delaykey[0]); i ++) {
137 delaykey[i].len = delaykey[i].str == NULL ?
138 0 : strlen (delaykey[i].str);
139 delaykey[i].len = delaykey[i].len > MAX_DELAY_STOP_STR ?
140 MAX_DELAY_STOP_STR : delaykey[i].len;
141
142 presskey_max = presskey_max > delaykey[i].len ?
143 presskey_max : delaykey[i].len;
144
145# if DEBUG_BOOTKEYS
146 printf("%s key:<%s>\n",
147 delaykey[i].retry ? "delay" : "stop",
148 delaykey[i].str ? delaykey[i].str : "NULL");
149# endif
150 }
151
152 /* In order to keep up with incoming data, check timeout only
153 * when catch up.
154 */
Peter Korsgaardc3284b02008-12-10 16:24:16 +0100155 do {
156 if (tstc()) {
157 if (presskey_len < presskey_max) {
158 presskey [presskey_len ++] = getc();
159 }
160 else {
161 for (i = 0; i < presskey_max - 1; i ++)
162 presskey [i] = presskey [i + 1];
163
164 presskey [i] = getc();
165 }
166 }
167
wdenkc6097192002-11-03 00:24:07 +0000168 for (i = 0; i < sizeof(delaykey) / sizeof(delaykey[0]); i ++) {
169 if (delaykey[i].len > 0 &&
170 presskey_len >= delaykey[i].len &&
171 memcmp (presskey + presskey_len - delaykey[i].len,
wdenk8bde7f72003-06-27 21:31:46 +0000172 delaykey[i].str,
wdenkc6097192002-11-03 00:24:07 +0000173 delaykey[i].len) == 0) {
174# if DEBUG_BOOTKEYS
175 printf("got %skey\n",
176 delaykey[i].retry ? "delay" : "stop");
177# endif
178
179# ifdef CONFIG_BOOT_RETRY_TIME
180 /* don't retry auto boot */
181 if (! delaykey[i].retry)
182 retry_time = -1;
183# endif
184 abort = 1;
185 }
186 }
Peter Korsgaardc3284b02008-12-10 16:24:16 +0100187 } while (!abort && get_ticks() <= etime);
wdenkc6097192002-11-03 00:24:07 +0000188
wdenkc6097192002-11-03 00:24:07 +0000189# if DEBUG_BOOTKEYS
190 if (!abort)
Ladislav Michlada4d402007-04-25 16:01:26 +0200191 puts("key timeout\n");
wdenkc6097192002-11-03 00:24:07 +0000192# endif
193
dzu8cb81432003-10-24 13:14:45 +0000194#ifdef CONFIG_SILENT_CONSOLE
Ladislav Michl4ec5bd52007-04-25 16:01:26 +0200195 if (abort)
196 gd->flags &= ~GD_FLG_SILENT;
dzu8cb81432003-10-24 13:14:45 +0000197#endif
198
wdenkc6097192002-11-03 00:24:07 +0000199 return abort;
200}
201
202# else /* !defined(CONFIG_AUTOBOOT_KEYED) */
203
wdenkc7de8292002-11-19 11:04:11 +0000204#ifdef CONFIG_MENUKEY
205static int menukey = 0;
206#endif
207
Jason Hobbsb41bc5a2011-08-23 11:06:50 +0000208#ifndef CONFIG_MENU
209static inline
210#endif
211int abortboot(int bootdelay)
wdenkc6097192002-11-03 00:24:07 +0000212{
213 int abort = 0;
214
wdenkc7de8292002-11-19 11:04:11 +0000215#ifdef CONFIG_MENUPROMPT
Stefan Roesef2302d42008-08-06 14:05:38 +0200216 printf(CONFIG_MENUPROMPT);
wdenkc7de8292002-11-19 11:04:11 +0000217#else
wdenkc6097192002-11-03 00:24:07 +0000218 printf("Hit any key to stop autoboot: %2d ", bootdelay);
wdenkc7de8292002-11-19 11:04:11 +0000219#endif
wdenkc6097192002-11-03 00:24:07 +0000220
221#if defined CONFIG_ZERO_BOOTDELAY_CHECK
wdenk8bde7f72003-06-27 21:31:46 +0000222 /*
223 * Check if key already pressed
224 * Don't check if bootdelay < 0
225 */
wdenkc6097192002-11-03 00:24:07 +0000226 if (bootdelay >= 0) {
227 if (tstc()) { /* we got a key press */
228 (void) getc(); /* consume input */
wdenk4b9206e2004-03-23 22:14:11 +0000229 puts ("\b\b\b 0");
Ladislav Michl4ec5bd52007-04-25 16:01:26 +0200230 abort = 1; /* don't auto boot */
wdenkc6097192002-11-03 00:24:07 +0000231 }
wdenk8bde7f72003-06-27 21:31:46 +0000232 }
wdenkc6097192002-11-03 00:24:07 +0000233#endif
234
wdenkf72da342003-10-10 10:05:42 +0000235 while ((bootdelay > 0) && (!abort)) {
wdenkc6097192002-11-03 00:24:07 +0000236 int i;
237
238 --bootdelay;
239 /* delay 100 * 10ms */
240 for (i=0; !abort && i<100; ++i) {
241 if (tstc()) { /* we got a key press */
242 abort = 1; /* don't auto boot */
243 bootdelay = 0; /* no more delay */
wdenkc7de8292002-11-19 11:04:11 +0000244# ifdef CONFIG_MENUKEY
245 menukey = getc();
246# else
wdenkc6097192002-11-03 00:24:07 +0000247 (void) getc(); /* consume input */
wdenkc7de8292002-11-19 11:04:11 +0000248# endif
wdenkc6097192002-11-03 00:24:07 +0000249 break;
250 }
Ladislav Michlada4d402007-04-25 16:01:26 +0200251 udelay(10000);
wdenkc6097192002-11-03 00:24:07 +0000252 }
253
Ladislav Michlada4d402007-04-25 16:01:26 +0200254 printf("\b\b\b%2d ", bootdelay);
wdenkc6097192002-11-03 00:24:07 +0000255 }
256
Ladislav Michlada4d402007-04-25 16:01:26 +0200257 putc('\n');
wdenkc6097192002-11-03 00:24:07 +0000258
wdenkf72da342003-10-10 10:05:42 +0000259#ifdef CONFIG_SILENT_CONSOLE
Ladislav Michl4ec5bd52007-04-25 16:01:26 +0200260 if (abort)
261 gd->flags &= ~GD_FLG_SILENT;
wdenkf72da342003-10-10 10:05:42 +0000262#endif
263
wdenkc6097192002-11-03 00:24:07 +0000264 return abort;
265}
266# endif /* CONFIG_AUTOBOOT_KEYED */
267#endif /* CONFIG_BOOTDELAY >= 0 */
268
Jason Hobbsc8a20792011-08-31 05:37:24 +0000269/*
270 * Return 0 on success, or != 0 on error.
271 */
Jason Hobbs06283a62011-08-31 10:37:30 -0500272#ifndef CONFIG_CMD_PXE
Jason Hobbsc8a20792011-08-31 05:37:24 +0000273static inline
Jason Hobbs06283a62011-08-31 10:37:30 -0500274#endif
Jason Hobbsc8a20792011-08-31 05:37:24 +0000275int run_command2(const char *cmd, int flag)
276{
277#ifndef CONFIG_SYS_HUSH_PARSER
278 /*
279 * run_command can return 0 or 1 for success, so clean up its result.
280 */
281 if (run_command(cmd, flag) == -1)
282 return 1;
283
284 return 0;
285#else
286 return parse_string_outer(cmd,
287 FLAG_PARSE_SEMICOLON | FLAG_EXIT_FROM_LOOP);
288#endif
289}
290
wdenkc6097192002-11-03 00:24:07 +0000291/****************************************************************************/
292
293void main_loop (void)
294{
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +0200295#ifndef CONFIG_SYS_HUSH_PARSER
296 static char lastcommand[CONFIG_SYS_CBSIZE] = { 0, };
wdenkc6097192002-11-03 00:24:07 +0000297 int len;
298 int rc = 1;
299 int flag;
300#endif
301
302#if defined(CONFIG_BOOTDELAY) && (CONFIG_BOOTDELAY >= 0)
303 char *s;
304 int bootdelay;
305#endif
306#ifdef CONFIG_PREBOOT
307 char *p;
308#endif
wdenkbdccc4f2003-08-05 17:43:17 +0000309#ifdef CONFIG_BOOTCOUNT_LIMIT
310 unsigned long bootcount = 0;
311 unsigned long bootlimit = 0;
312 char *bcs;
313 char bcs_set[16];
314#endif /* CONFIG_BOOTCOUNT_LIMIT */
wdenkc6097192002-11-03 00:24:07 +0000315
wdenkbdccc4f2003-08-05 17:43:17 +0000316#ifdef CONFIG_BOOTCOUNT_LIMIT
317 bootcount = bootcount_load();
318 bootcount++;
319 bootcount_store (bootcount);
320 sprintf (bcs_set, "%lu", bootcount);
321 setenv ("bootcount", bcs_set);
322 bcs = getenv ("bootlimit");
323 bootlimit = bcs ? simple_strtoul (bcs, NULL, 10) : 0;
324#endif /* CONFIG_BOOTCOUNT_LIMIT */
325
wdenkc6097192002-11-03 00:24:07 +0000326#ifdef CONFIG_MODEM_SUPPORT
327 debug ("DEBUG: main_loop: do_mdm_init=%d\n", do_mdm_init);
328 if (do_mdm_init) {
Wolfgang Denk77ddac92005-10-13 16:45:02 +0200329 char *str = strdup(getenv("mdm_cmd"));
wdenkc6097192002-11-03 00:24:07 +0000330 setenv ("preboot", str); /* set or delete definition */
331 if (str != NULL)
332 free (str);
333 mdm_init(); /* wait for modem connection */
334 }
335#endif /* CONFIG_MODEM_SUPPORT */
336
stroese05875972003-04-04 15:44:49 +0000337#ifdef CONFIG_VERSION_VARIABLE
338 {
stroese155cb012003-07-11 08:00:33 +0000339 setenv ("ver", version_string); /* set version variable */
stroese05875972003-04-04 15:44:49 +0000340 }
341#endif /* CONFIG_VERSION_VARIABLE */
342
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +0200343#ifdef CONFIG_SYS_HUSH_PARSER
wdenkc6097192002-11-03 00:24:07 +0000344 u_boot_hush_start ();
345#endif
346
Heiko Schocher81473f62008-10-15 09:40:28 +0200347#if defined(CONFIG_HUSH_INIT_VAR)
348 hush_init_var ();
349#endif
350
wdenkc6097192002-11-03 00:24:07 +0000351#ifdef CONFIG_PREBOOT
352 if ((p = getenv ("preboot")) != NULL) {
353# ifdef CONFIG_AUTOBOOT_KEYED
354 int prev = disable_ctrlc(1); /* disable Control C checking */
355# endif
356
Jason Hobbsc8a20792011-08-31 05:37:24 +0000357 run_command2(p, 0);
wdenkc6097192002-11-03 00:24:07 +0000358
359# ifdef CONFIG_AUTOBOOT_KEYED
360 disable_ctrlc(prev); /* restore Control C checking */
361# endif
362 }
363#endif /* CONFIG_PREBOOT */
364
Wolfgang Denk143cd212010-03-11 23:56:03 +0100365#if defined(CONFIG_UPDATE_TFTP)
Andreas Pretzsch8d6b7322011-07-16 05:50:59 +0000366 update_tftp (0UL);
Wolfgang Denk143cd212010-03-11 23:56:03 +0100367#endif /* CONFIG_UPDATE_TFTP */
368
wdenkc6097192002-11-03 00:24:07 +0000369#if defined(CONFIG_BOOTDELAY) && (CONFIG_BOOTDELAY >= 0)
370 s = getenv ("bootdelay");
371 bootdelay = s ? (int)simple_strtol(s, NULL, 10) : CONFIG_BOOTDELAY;
372
wdenka6c7ad22002-12-03 21:28:10 +0000373 debug ("### main_loop entered: bootdelay=%d\n\n", bootdelay);
wdenkc6097192002-11-03 00:24:07 +0000374
375# ifdef CONFIG_BOOT_RETRY_TIME
wdenk6dd652f2003-06-19 23:40:20 +0000376 init_cmd_timeout ();
wdenkc6097192002-11-03 00:24:07 +0000377# endif /* CONFIG_BOOT_RETRY_TIME */
378
Yuri Tikhonovb428f6a2008-02-04 14:11:03 +0100379#ifdef CONFIG_POST
380 if (gd->flags & GD_FLG_POSTFAIL) {
381 s = getenv("failbootcmd");
382 }
383 else
384#endif /* CONFIG_POST */
wdenkbdccc4f2003-08-05 17:43:17 +0000385#ifdef CONFIG_BOOTCOUNT_LIMIT
386 if (bootlimit && (bootcount > bootlimit)) {
387 printf ("Warning: Bootlimit (%u) exceeded. Using altbootcmd.\n",
388 (unsigned)bootlimit);
389 s = getenv ("altbootcmd");
390 }
391 else
392#endif /* CONFIG_BOOTCOUNT_LIMIT */
393 s = getenv ("bootcmd");
wdenka6c7ad22002-12-03 21:28:10 +0000394
395 debug ("### main_loop: bootcmd=\"%s\"\n", s ? s : "<UNDEFINED>");
396
wdenkc6097192002-11-03 00:24:07 +0000397 if (bootdelay >= 0 && s && !abortboot (bootdelay)) {
398# ifdef CONFIG_AUTOBOOT_KEYED
399 int prev = disable_ctrlc(1); /* disable Control C checking */
400# endif
401
Jason Hobbsc8a20792011-08-31 05:37:24 +0000402 run_command2(s, 0);
wdenkc6097192002-11-03 00:24:07 +0000403
404# ifdef CONFIG_AUTOBOOT_KEYED
405 disable_ctrlc(prev); /* restore Control C checking */
406# endif
407 }
wdenkc7de8292002-11-19 11:04:11 +0000408
409# ifdef CONFIG_MENUKEY
wdenka6c7ad22002-12-03 21:28:10 +0000410 if (menukey == CONFIG_MENUKEY) {
Jason Hobbs370d1e32011-06-23 08:27:30 +0000411 s = getenv("menucmd");
Jason Hobbsc8a20792011-08-31 05:37:24 +0000412 if (s)
413 run_command2(s, 0);
wdenkc7de8292002-11-19 11:04:11 +0000414 }
415#endif /* CONFIG_MENUKEY */
Wolfgang Denk953b7e62010-06-13 18:28:54 +0200416#endif /* CONFIG_BOOTDELAY */
wdenkc7de8292002-11-19 11:04:11 +0000417
wdenkc6097192002-11-03 00:24:07 +0000418 /*
419 * Main Loop for Monitor Command Processing
420 */
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +0200421#ifdef CONFIG_SYS_HUSH_PARSER
wdenkc6097192002-11-03 00:24:07 +0000422 parse_file_outer();
423 /* This point is never reached */
424 for (;;);
425#else
426 for (;;) {
427#ifdef CONFIG_BOOT_RETRY_TIME
428 if (rc >= 0) {
429 /* Saw enough of a valid command to
430 * restart the timeout.
431 */
432 reset_cmd_timeout();
433 }
434#endif
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +0200435 len = readline (CONFIG_SYS_PROMPT);
wdenkc6097192002-11-03 00:24:07 +0000436
437 flag = 0; /* assume no special flags for now */
438 if (len > 0)
439 strcpy (lastcommand, console_buffer);
440 else if (len == 0)
441 flag |= CMD_FLAG_REPEAT;
442#ifdef CONFIG_BOOT_RETRY_TIME
443 else if (len == -2) {
444 /* -2 means timed out, retry autoboot
445 */
wdenk4b9206e2004-03-23 22:14:11 +0000446 puts ("\nTimed out waiting for command\n");
wdenkc6097192002-11-03 00:24:07 +0000447# ifdef CONFIG_RESET_TO_RETRY
448 /* Reinit board to run initialization code again */
449 do_reset (NULL, 0, 0, NULL);
450# else
451 return; /* retry autoboot */
452# endif
453 }
454#endif
455
456 if (len == -1)
wdenk4b9206e2004-03-23 22:14:11 +0000457 puts ("<INTERRUPT>\n");
wdenkc6097192002-11-03 00:24:07 +0000458 else
459 rc = run_command (lastcommand, flag);
460
461 if (rc <= 0) {
462 /* invalid command or not repeatable, forget it */
463 lastcommand[0] = 0;
464 }
465 }
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +0200466#endif /*CONFIG_SYS_HUSH_PARSER*/
wdenkc6097192002-11-03 00:24:07 +0000467}
468
wdenk6dd652f2003-06-19 23:40:20 +0000469#ifdef CONFIG_BOOT_RETRY_TIME
470/***************************************************************************
Wolfgang Denk19973b62006-10-28 00:38:39 +0200471 * initialize command line timeout
wdenk6dd652f2003-06-19 23:40:20 +0000472 */
473void init_cmd_timeout(void)
474{
475 char *s = getenv ("bootretry");
476
477 if (s != NULL)
wdenkb028f712003-12-07 21:39:28 +0000478 retry_time = (int)simple_strtol(s, NULL, 10);
wdenk6dd652f2003-06-19 23:40:20 +0000479 else
480 retry_time = CONFIG_BOOT_RETRY_TIME;
481
482 if (retry_time >= 0 && retry_time < CONFIG_BOOT_RETRY_MIN)
483 retry_time = CONFIG_BOOT_RETRY_MIN;
484}
485
wdenkc6097192002-11-03 00:24:07 +0000486/***************************************************************************
487 * reset command line timeout to retry_time seconds
488 */
wdenkc6097192002-11-03 00:24:07 +0000489void reset_cmd_timeout(void)
490{
491 endtime = endtick(retry_time);
492}
493#endif
494
Wolfgang Denk501090a2006-07-21 11:33:45 +0200495#ifdef CONFIG_CMDLINE_EDITING
496
497/*
498 * cmdline-editing related codes from vivi.
499 * Author: Janghoon Lyu <nandy@mizi.com>
500 */
501
Wolfgang Denk501090a2006-07-21 11:33:45 +0200502#define putnstr(str,n) do { \
Andrew Klossnerdc4b0b32008-07-07 06:41:14 -0700503 printf ("%.*s", (int)n, str); \
Wolfgang Denk501090a2006-07-21 11:33:45 +0200504 } while (0)
Wolfgang Denk501090a2006-07-21 11:33:45 +0200505
506#define CTL_CH(c) ((c) - 'a' + 1)
Wolfgang Denk501090a2006-07-21 11:33:45 +0200507#define CTL_BACKSPACE ('\b')
508#define DEL ((char)255)
509#define DEL7 ((char)127)
510#define CREAD_HIST_CHAR ('!')
511
512#define getcmd_putch(ch) putc(ch)
513#define getcmd_getch() getc()
514#define getcmd_cbeep() getcmd_putch('\a')
515
516#define HIST_MAX 20
Peter Tyser8804ae32010-09-29 13:30:56 -0500517#define HIST_SIZE CONFIG_SYS_CBSIZE
Wolfgang Denk501090a2006-07-21 11:33:45 +0200518
519static int hist_max = 0;
520static int hist_add_idx = 0;
521static int hist_cur = -1;
522unsigned hist_num = 0;
523
524char* hist_list[HIST_MAX];
John Schmoller6475b9f2010-03-12 09:49:23 -0600525char hist_lines[HIST_MAX][HIST_SIZE + 1]; /* Save room for NULL */
Wolfgang Denk501090a2006-07-21 11:33:45 +0200526
527#define add_idx_minus_one() ((hist_add_idx == 0) ? hist_max : hist_add_idx-1)
528
529static void hist_init(void)
530{
531 int i;
532
533 hist_max = 0;
534 hist_add_idx = 0;
535 hist_cur = -1;
536 hist_num = 0;
537
538 for (i = 0; i < HIST_MAX; i++) {
539 hist_list[i] = hist_lines[i];
540 hist_list[i][0] = '\0';
541 }
542}
543
544static void cread_add_to_hist(char *line)
545{
546 strcpy(hist_list[hist_add_idx], line);
547
548 if (++hist_add_idx >= HIST_MAX)
549 hist_add_idx = 0;
550
551 if (hist_add_idx > hist_max)
552 hist_max = hist_add_idx;
553
554 hist_num++;
555}
556
557static char* hist_prev(void)
558{
559 char *ret;
560 int old_cur;
561
562 if (hist_cur < 0)
563 return NULL;
564
565 old_cur = hist_cur;
566 if (--hist_cur < 0)
567 hist_cur = hist_max;
568
569 if (hist_cur == hist_add_idx) {
570 hist_cur = old_cur;
571 ret = NULL;
572 } else
573 ret = hist_list[hist_cur];
574
575 return (ret);
576}
577
578static char* hist_next(void)
579{
580 char *ret;
581
582 if (hist_cur < 0)
583 return NULL;
584
585 if (hist_cur == hist_add_idx)
586 return NULL;
587
588 if (++hist_cur > hist_max)
589 hist_cur = 0;
590
591 if (hist_cur == hist_add_idx) {
592 ret = "";
593 } else
594 ret = hist_list[hist_cur];
595
596 return (ret);
597}
598
Stefan Roese3ca91222006-07-27 16:11:19 +0200599#ifndef CONFIG_CMDLINE_EDITING
Wolfgang Denk501090a2006-07-21 11:33:45 +0200600static void cread_print_hist_list(void)
601{
602 int i;
603 unsigned long n;
604
605 n = hist_num - hist_max;
606
607 i = hist_add_idx + 1;
608 while (1) {
609 if (i > hist_max)
610 i = 0;
611 if (i == hist_add_idx)
612 break;
613 printf("%s\n", hist_list[i]);
614 n++;
615 i++;
616 }
617}
Stefan Roese3ca91222006-07-27 16:11:19 +0200618#endif /* CONFIG_CMDLINE_EDITING */
Wolfgang Denk501090a2006-07-21 11:33:45 +0200619
620#define BEGINNING_OF_LINE() { \
621 while (num) { \
622 getcmd_putch(CTL_BACKSPACE); \
623 num--; \
624 } \
625}
626
627#define ERASE_TO_EOL() { \
628 if (num < eol_num) { \
Mike Frysinger8faba482010-07-23 05:28:15 -0400629 printf("%*s", (int)(eol_num - num), ""); \
630 do { \
Wolfgang Denk501090a2006-07-21 11:33:45 +0200631 getcmd_putch(CTL_BACKSPACE); \
Mike Frysinger8faba482010-07-23 05:28:15 -0400632 } while (--eol_num > num); \
Wolfgang Denk501090a2006-07-21 11:33:45 +0200633 } \
634}
635
636#define REFRESH_TO_EOL() { \
637 if (num < eol_num) { \
638 wlen = eol_num - num; \
639 putnstr(buf + num, wlen); \
640 num = eol_num; \
641 } \
642}
643
644static void cread_add_char(char ichar, int insert, unsigned long *num,
645 unsigned long *eol_num, char *buf, unsigned long len)
646{
647 unsigned long wlen;
648
649 /* room ??? */
650 if (insert || *num == *eol_num) {
651 if (*eol_num > len - 1) {
652 getcmd_cbeep();
653 return;
654 }
655 (*eol_num)++;
656 }
657
658 if (insert) {
659 wlen = *eol_num - *num;
660 if (wlen > 1) {
661 memmove(&buf[*num+1], &buf[*num], wlen-1);
662 }
663
664 buf[*num] = ichar;
665 putnstr(buf + *num, wlen);
666 (*num)++;
667 while (--wlen) {
668 getcmd_putch(CTL_BACKSPACE);
669 }
670 } else {
671 /* echo the character */
672 wlen = 1;
673 buf[*num] = ichar;
674 putnstr(buf + *num, wlen);
675 (*num)++;
676 }
677}
678
679static void cread_add_str(char *str, int strsize, int insert, unsigned long *num,
680 unsigned long *eol_num, char *buf, unsigned long len)
681{
682 while (strsize--) {
683 cread_add_char(*str, insert, num, eol_num, buf, len);
684 str++;
685 }
686}
687
Heiko Schocher9c348312012-01-16 21:13:05 +0000688static int cread_line(const char *const prompt, char *buf, unsigned int *len,
689 int timeout)
Wolfgang Denk501090a2006-07-21 11:33:45 +0200690{
691 unsigned long num = 0;
692 unsigned long eol_num = 0;
Wolfgang Denk501090a2006-07-21 11:33:45 +0200693 unsigned long wlen;
694 char ichar;
695 int insert = 1;
696 int esc_len = 0;
Wolfgang Denk501090a2006-07-21 11:33:45 +0200697 char esc_save[8];
Peter Tyserecc55002009-10-25 15:12:54 -0500698 int init_len = strlen(buf);
Heiko Schocher9c348312012-01-16 21:13:05 +0000699 int first = 1;
Peter Tyserecc55002009-10-25 15:12:54 -0500700
701 if (init_len)
702 cread_add_str(buf, init_len, 1, &num, &eol_num, buf, *len);
Wolfgang Denk501090a2006-07-21 11:33:45 +0200703
704 while (1) {
Andreas Engel00ac50e2008-01-09 17:10:56 +0100705#ifdef CONFIG_BOOT_RETRY_TIME
706 while (!tstc()) { /* while no incoming data */
707 if (retry_time >= 0 && get_ticks() > endtime)
708 return (-2); /* timed out */
Jens Scharsig30dc1652010-04-09 19:02:38 +0200709 WATCHDOG_RESET();
Andreas Engel00ac50e2008-01-09 17:10:56 +0100710 }
711#endif
Heiko Schocher9c348312012-01-16 21:13:05 +0000712 if (first && timeout) {
713 uint64_t etime = endtick(timeout);
714
715 while (!tstc()) { /* while no incoming data */
716 if (get_ticks() >= etime)
717 return -2; /* timed out */
718 WATCHDOG_RESET();
719 }
720 first = 0;
721 }
Andreas Engel00ac50e2008-01-09 17:10:56 +0100722
Wolfgang Denk501090a2006-07-21 11:33:45 +0200723 ichar = getcmd_getch();
724
725 if ((ichar == '\n') || (ichar == '\r')) {
Wolfgang Denkdd9f06f2006-07-21 11:34:34 +0200726 putc('\n');
Wolfgang Denk501090a2006-07-21 11:33:45 +0200727 break;
728 }
729
730 /*
731 * handle standard linux xterm esc sequences for arrow key, etc.
732 */
733 if (esc_len != 0) {
734 if (esc_len == 1) {
735 if (ichar == '[') {
736 esc_save[esc_len] = ichar;
737 esc_len = 2;
738 } else {
739 cread_add_str(esc_save, esc_len, insert,
740 &num, &eol_num, buf, *len);
741 esc_len = 0;
742 }
743 continue;
744 }
745
746 switch (ichar) {
747
748 case 'D': /* <- key */
749 ichar = CTL_CH('b');
750 esc_len = 0;
751 break;
752 case 'C': /* -> key */
753 ichar = CTL_CH('f');
754 esc_len = 0;
755 break; /* pass off to ^F handler */
756 case 'H': /* Home key */
757 ichar = CTL_CH('a');
758 esc_len = 0;
759 break; /* pass off to ^A handler */
760 case 'A': /* up arrow */
761 ichar = CTL_CH('p');
762 esc_len = 0;
763 break; /* pass off to ^P handler */
764 case 'B': /* down arrow */
765 ichar = CTL_CH('n');
766 esc_len = 0;
767 break; /* pass off to ^N handler */
768 default:
769 esc_save[esc_len++] = ichar;
770 cread_add_str(esc_save, esc_len, insert,
771 &num, &eol_num, buf, *len);
772 esc_len = 0;
773 continue;
774 }
775 }
776
777 switch (ichar) {
778 case 0x1b:
779 if (esc_len == 0) {
780 esc_save[esc_len] = ichar;
781 esc_len = 1;
782 } else {
Wolfgang Denkdd9f06f2006-07-21 11:34:34 +0200783 puts("impossible condition #876\n");
Wolfgang Denk501090a2006-07-21 11:33:45 +0200784 esc_len = 0;
785 }
786 break;
787
788 case CTL_CH('a'):
789 BEGINNING_OF_LINE();
790 break;
791 case CTL_CH('c'): /* ^C - break */
792 *buf = '\0'; /* discard input */
793 return (-1);
794 case CTL_CH('f'):
795 if (num < eol_num) {
796 getcmd_putch(buf[num]);
797 num++;
798 }
799 break;
800 case CTL_CH('b'):
801 if (num) {
802 getcmd_putch(CTL_BACKSPACE);
803 num--;
804 }
805 break;
806 case CTL_CH('d'):
807 if (num < eol_num) {
808 wlen = eol_num - num - 1;
809 if (wlen) {
810 memmove(&buf[num], &buf[num+1], wlen);
811 putnstr(buf + num, wlen);
812 }
813
814 getcmd_putch(' ');
815 do {
816 getcmd_putch(CTL_BACKSPACE);
817 } while (wlen--);
818 eol_num--;
819 }
820 break;
821 case CTL_CH('k'):
822 ERASE_TO_EOL();
823 break;
824 case CTL_CH('e'):
825 REFRESH_TO_EOL();
826 break;
827 case CTL_CH('o'):
828 insert = !insert;
829 break;
830 case CTL_CH('x'):
Jean-Christophe PLAGNIOL-VILLARD23d0baf2007-12-22 15:52:58 +0100831 case CTL_CH('u'):
Wolfgang Denk501090a2006-07-21 11:33:45 +0200832 BEGINNING_OF_LINE();
833 ERASE_TO_EOL();
834 break;
835 case DEL:
836 case DEL7:
837 case 8:
838 if (num) {
839 wlen = eol_num - num;
840 num--;
841 memmove(&buf[num], &buf[num+1], wlen);
842 getcmd_putch(CTL_BACKSPACE);
843 putnstr(buf + num, wlen);
844 getcmd_putch(' ');
845 do {
846 getcmd_putch(CTL_BACKSPACE);
847 } while (wlen--);
848 eol_num--;
849 }
850 break;
851 case CTL_CH('p'):
852 case CTL_CH('n'):
853 {
854 char * hline;
855
856 esc_len = 0;
857
858 if (ichar == CTL_CH('p'))
859 hline = hist_prev();
860 else
861 hline = hist_next();
862
863 if (!hline) {
864 getcmd_cbeep();
865 continue;
866 }
867
868 /* nuke the current line */
869 /* first, go home */
870 BEGINNING_OF_LINE();
871
872 /* erase to end of line */
873 ERASE_TO_EOL();
874
875 /* copy new line into place and display */
876 strcpy(buf, hline);
877 eol_num = strlen(buf);
878 REFRESH_TO_EOL();
879 continue;
880 }
Jean-Christophe PLAGNIOL-VILLARD23d0baf2007-12-22 15:52:58 +0100881#ifdef CONFIG_AUTO_COMPLETE
882 case '\t': {
883 int num2, col;
884
885 /* do not autocomplete when in the middle */
886 if (num < eol_num) {
887 getcmd_cbeep();
888 break;
889 }
890
891 buf[num] = '\0';
892 col = strlen(prompt) + eol_num;
893 num2 = num;
894 if (cmd_auto_complete(prompt, buf, &num2, &col)) {
895 col = num2 - num;
896 num += col;
897 eol_num += col;
898 }
899 break;
900 }
901#endif
Wolfgang Denk501090a2006-07-21 11:33:45 +0200902 default:
903 cread_add_char(ichar, insert, &num, &eol_num, buf, *len);
904 break;
905 }
906 }
907 *len = eol_num;
908 buf[eol_num] = '\0'; /* lose the newline */
909
910 if (buf[0] && buf[0] != CREAD_HIST_CHAR)
911 cread_add_to_hist(buf);
912 hist_cur = hist_add_idx;
913
Peter Tyserf9239432009-10-25 15:12:53 -0500914 return 0;
Wolfgang Denk501090a2006-07-21 11:33:45 +0200915}
916
917#endif /* CONFIG_CMDLINE_EDITING */
918
wdenkc6097192002-11-03 00:24:07 +0000919/****************************************************************************/
920
921/*
922 * Prompt for input and read a line.
923 * If CONFIG_BOOT_RETRY_TIME is defined and retry_time >= 0,
924 * time out when time goes past endtime (timebase time in ticks).
925 * Return: number of read characters
926 * -1 if break
927 * -2 if timed out
928 */
929int readline (const char *const prompt)
930{
Peter Tyserecc55002009-10-25 15:12:54 -0500931 /*
932 * If console_buffer isn't 0-length the user will be prompted to modify
933 * it instead of entering it from scratch as desired.
934 */
935 console_buffer[0] = '\0';
936
Heiko Schocher9c348312012-01-16 21:13:05 +0000937 return readline_into_buffer(prompt, console_buffer, 0);
James Yang6636b622008-01-09 11:17:49 -0600938}
939
940
Heiko Schocher9c348312012-01-16 21:13:05 +0000941int readline_into_buffer(const char *const prompt, char *buffer, int timeout)
James Yang6636b622008-01-09 11:17:49 -0600942{
943 char *p = buffer;
Wolfgang Denk501090a2006-07-21 11:33:45 +0200944#ifdef CONFIG_CMDLINE_EDITING
Peter Tyser8804ae32010-09-29 13:30:56 -0500945 unsigned int len = CONFIG_SYS_CBSIZE;
Stefan Roesed8f961b2006-08-07 15:08:44 +0200946 int rc;
Wolfgang Denk501090a2006-07-21 11:33:45 +0200947 static int initted = 0;
948
James Yang597f6c22008-05-05 10:22:53 -0500949 /*
950 * History uses a global array which is not
951 * writable until after relocation to RAM.
952 * Revert to non-history version if still
953 * running from flash.
954 */
955 if (gd->flags & GD_FLG_RELOC) {
956 if (!initted) {
957 hist_init();
958 initted = 1;
959 }
960
Peter Tysere491a712009-10-25 15:12:52 -0500961 if (prompt)
962 puts (prompt);
James Yang597f6c22008-05-05 10:22:53 -0500963
Heiko Schocher9c348312012-01-16 21:13:05 +0000964 rc = cread_line(prompt, p, &len, timeout);
James Yang597f6c22008-05-05 10:22:53 -0500965 return rc < 0 ? rc : len;
966
967 } else {
968#endif /* CONFIG_CMDLINE_EDITING */
Kumar Gala0ec59522008-01-10 02:22:05 -0600969 char * p_buf = p;
wdenkc6097192002-11-03 00:24:07 +0000970 int n = 0; /* buffer index */
971 int plen = 0; /* prompt length */
972 int col; /* output column cnt */
973 char c;
974
975 /* print prompt */
976 if (prompt) {
977 plen = strlen (prompt);
978 puts (prompt);
979 }
980 col = plen;
981
982 for (;;) {
983#ifdef CONFIG_BOOT_RETRY_TIME
984 while (!tstc()) { /* while no incoming data */
985 if (retry_time >= 0 && get_ticks() > endtime)
986 return (-2); /* timed out */
Jens Scharsig30dc1652010-04-09 19:02:38 +0200987 WATCHDOG_RESET();
wdenkc6097192002-11-03 00:24:07 +0000988 }
989#endif
990 WATCHDOG_RESET(); /* Trigger watchdog, if needed */
991
992#ifdef CONFIG_SHOW_ACTIVITY
993 while (!tstc()) {
994 extern void show_activity(int arg);
995 show_activity(0);
Jens Scharsig30dc1652010-04-09 19:02:38 +0200996 WATCHDOG_RESET();
wdenkc6097192002-11-03 00:24:07 +0000997 }
998#endif
999 c = getc();
1000
1001 /*
1002 * Special character handling
1003 */
1004 switch (c) {
1005 case '\r': /* Enter */
1006 case '\n':
1007 *p = '\0';
1008 puts ("\r\n");
James Yang6636b622008-01-09 11:17:49 -06001009 return (p - p_buf);
wdenkc6097192002-11-03 00:24:07 +00001010
wdenk27aa8182004-03-23 22:37:33 +00001011 case '\0': /* nul */
1012 continue;
1013
wdenkc6097192002-11-03 00:24:07 +00001014 case 0x03: /* ^C - break */
James Yang6636b622008-01-09 11:17:49 -06001015 p_buf[0] = '\0'; /* discard input */
wdenkc6097192002-11-03 00:24:07 +00001016 return (-1);
1017
1018 case 0x15: /* ^U - erase line */
1019 while (col > plen) {
1020 puts (erase_seq);
1021 --col;
1022 }
James Yang6636b622008-01-09 11:17:49 -06001023 p = p_buf;
wdenkc6097192002-11-03 00:24:07 +00001024 n = 0;
1025 continue;
1026
Wolfgang Denk1636d1c2007-06-22 23:59:00 +02001027 case 0x17: /* ^W - erase word */
James Yang6636b622008-01-09 11:17:49 -06001028 p=delete_char(p_buf, p, &col, &n, plen);
wdenkc6097192002-11-03 00:24:07 +00001029 while ((n > 0) && (*p != ' ')) {
James Yang6636b622008-01-09 11:17:49 -06001030 p=delete_char(p_buf, p, &col, &n, plen);
wdenkc6097192002-11-03 00:24:07 +00001031 }
1032 continue;
1033
1034 case 0x08: /* ^H - backspace */
1035 case 0x7F: /* DEL - backspace */
James Yang6636b622008-01-09 11:17:49 -06001036 p=delete_char(p_buf, p, &col, &n, plen);
wdenkc6097192002-11-03 00:24:07 +00001037 continue;
1038
1039 default:
1040 /*
1041 * Must be a normal character then
1042 */
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +02001043 if (n < CONFIG_SYS_CBSIZE-2) {
wdenkc6097192002-11-03 00:24:07 +00001044 if (c == '\t') { /* expand TABs */
wdenk04a85b32004-04-15 18:22:41 +00001045#ifdef CONFIG_AUTO_COMPLETE
1046 /* if auto completion triggered just continue */
1047 *p = '\0';
1048 if (cmd_auto_complete(prompt, console_buffer, &n, &col)) {
James Yang6636b622008-01-09 11:17:49 -06001049 p = p_buf + n; /* reset */
wdenk04a85b32004-04-15 18:22:41 +00001050 continue;
1051 }
1052#endif
wdenkc6097192002-11-03 00:24:07 +00001053 puts (tab_seq+(col&07));
1054 col += 8 - (col&07);
1055 } else {
1056 ++col; /* echo input */
1057 putc (c);
1058 }
1059 *p++ = c;
1060 ++n;
1061 } else { /* Buffer full */
1062 putc ('\a');
1063 }
1064 }
1065 }
James Yang597f6c22008-05-05 10:22:53 -05001066#ifdef CONFIG_CMDLINE_EDITING
1067 }
1068#endif
wdenkc6097192002-11-03 00:24:07 +00001069}
1070
1071/****************************************************************************/
1072
1073static char * delete_char (char *buffer, char *p, int *colp, int *np, int plen)
1074{
1075 char *s;
1076
1077 if (*np == 0) {
1078 return (p);
1079 }
1080
1081 if (*(--p) == '\t') { /* will retype the whole line */
1082 while (*colp > plen) {
1083 puts (erase_seq);
1084 (*colp)--;
1085 }
1086 for (s=buffer; s<p; ++s) {
1087 if (*s == '\t') {
1088 puts (tab_seq+((*colp) & 07));
1089 *colp += 8 - ((*colp) & 07);
1090 } else {
1091 ++(*colp);
1092 putc (*s);
1093 }
1094 }
1095 } else {
1096 puts (erase_seq);
1097 (*colp)--;
1098 }
1099 (*np)--;
1100 return (p);
1101}
1102
1103/****************************************************************************/
1104
1105int parse_line (char *line, char *argv[])
1106{
1107 int nargs = 0;
1108
1109#ifdef DEBUG_PARSER
1110 printf ("parse_line: \"%s\"\n", line);
1111#endif
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +02001112 while (nargs < CONFIG_SYS_MAXARGS) {
wdenkc6097192002-11-03 00:24:07 +00001113
1114 /* skip any white space */
Jason Hobbs4d91a6e2011-08-23 11:06:54 +00001115 while (isblank(*line))
wdenkc6097192002-11-03 00:24:07 +00001116 ++line;
wdenkc6097192002-11-03 00:24:07 +00001117
1118 if (*line == '\0') { /* end of line, no more args */
1119 argv[nargs] = NULL;
1120#ifdef DEBUG_PARSER
1121 printf ("parse_line: nargs=%d\n", nargs);
1122#endif
1123 return (nargs);
1124 }
1125
1126 argv[nargs++] = line; /* begin of argument string */
1127
1128 /* find end of string */
Jason Hobbs4d91a6e2011-08-23 11:06:54 +00001129 while (*line && !isblank(*line))
wdenkc6097192002-11-03 00:24:07 +00001130 ++line;
wdenkc6097192002-11-03 00:24:07 +00001131
1132 if (*line == '\0') { /* end of line, no more args */
1133 argv[nargs] = NULL;
1134#ifdef DEBUG_PARSER
1135 printf ("parse_line: nargs=%d\n", nargs);
1136#endif
1137 return (nargs);
1138 }
1139
1140 *line++ = '\0'; /* terminate current arg */
1141 }
1142
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +02001143 printf ("** Too many args (max. %d) **\n", CONFIG_SYS_MAXARGS);
wdenkc6097192002-11-03 00:24:07 +00001144
1145#ifdef DEBUG_PARSER
1146 printf ("parse_line: nargs=%d\n", nargs);
1147#endif
1148 return (nargs);
1149}
1150
1151/****************************************************************************/
1152
1153static void process_macros (const char *input, char *output)
1154{
1155 char c, prev;
1156 const char *varname_start = NULL;
Wolfgang Denk19973b62006-10-28 00:38:39 +02001157 int inputcnt = strlen (input);
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +02001158 int outputcnt = CONFIG_SYS_CBSIZE;
Wolfgang Denk19973b62006-10-28 00:38:39 +02001159 int state = 0; /* 0 = waiting for '$' */
1160
1161 /* 1 = waiting for '(' or '{' */
1162 /* 2 = waiting for ')' or '}' */
1163 /* 3 = waiting for ''' */
wdenkc6097192002-11-03 00:24:07 +00001164#ifdef DEBUG_PARSER
1165 char *output_start = output;
1166
Wolfgang Denk19973b62006-10-28 00:38:39 +02001167 printf ("[PROCESS_MACROS] INPUT len %d: \"%s\"\n", strlen (input),
1168 input);
wdenkc6097192002-11-03 00:24:07 +00001169#endif
1170
Wolfgang Denk19973b62006-10-28 00:38:39 +02001171 prev = '\0'; /* previous character */
wdenkc6097192002-11-03 00:24:07 +00001172
1173 while (inputcnt && outputcnt) {
wdenk8bde7f72003-06-27 21:31:46 +00001174 c = *input++;
Wolfgang Denk19973b62006-10-28 00:38:39 +02001175 inputcnt--;
wdenkc6097192002-11-03 00:24:07 +00001176
Wolfgang Denk19973b62006-10-28 00:38:39 +02001177 if (state != 3) {
1178 /* remove one level of escape characters */
1179 if ((c == '\\') && (prev != '\\')) {
1180 if (inputcnt-- == 0)
1181 break;
1182 prev = c;
1183 c = *input++;
1184 }
wdenka25f8622003-01-02 23:57:29 +00001185 }
wdenkc6097192002-11-03 00:24:07 +00001186
Wolfgang Denk19973b62006-10-28 00:38:39 +02001187 switch (state) {
1188 case 0: /* Waiting for (unescaped) $ */
1189 if ((c == '\'') && (prev != '\\')) {
1190 state = 3;
1191 break;
1192 }
1193 if ((c == '$') && (prev != '\\')) {
1194 state++;
1195 } else {
wdenkc6097192002-11-03 00:24:07 +00001196 *(output++) = c;
1197 outputcnt--;
1198 }
Wolfgang Denk19973b62006-10-28 00:38:39 +02001199 break;
1200 case 1: /* Waiting for ( */
1201 if (c == '(' || c == '{') {
1202 state++;
1203 varname_start = input;
1204 } else {
1205 state = 0;
1206 *(output++) = '$';
1207 outputcnt--;
wdenkc6097192002-11-03 00:24:07 +00001208
Wolfgang Denk19973b62006-10-28 00:38:39 +02001209 if (outputcnt) {
1210 *(output++) = c;
wdenkc6097192002-11-03 00:24:07 +00001211 outputcnt--;
1212 }
Wolfgang Denk19973b62006-10-28 00:38:39 +02001213 }
1214 break;
1215 case 2: /* Waiting for ) */
1216 if (c == ')' || c == '}') {
1217 int i;
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +02001218 char envname[CONFIG_SYS_CBSIZE], *envval;
Wolfgang Denk19973b62006-10-28 00:38:39 +02001219 int envcnt = input - varname_start - 1; /* Varname # of chars */
1220
1221 /* Get the varname */
1222 for (i = 0; i < envcnt; i++) {
1223 envname[i] = varname_start[i];
1224 }
1225 envname[i] = 0;
1226
1227 /* Get its value */
1228 envval = getenv (envname);
1229
1230 /* Copy into the line if it exists */
1231 if (envval != NULL)
1232 while ((*envval) && outputcnt) {
1233 *(output++) = *(envval++);
1234 outputcnt--;
1235 }
1236 /* Look for another '$' */
1237 state = 0;
1238 }
1239 break;
1240 case 3: /* Waiting for ' */
1241 if ((c == '\'') && (prev != '\\')) {
1242 state = 0;
1243 } else {
1244 *(output++) = c;
1245 outputcnt--;
1246 }
1247 break;
wdenkc6097192002-11-03 00:24:07 +00001248 }
Wolfgang Denk19973b62006-10-28 00:38:39 +02001249 prev = c;
wdenkc6097192002-11-03 00:24:07 +00001250 }
1251
1252 if (outputcnt)
1253 *output = 0;
Bartlomiej Sieka9160b962007-05-27 17:04:18 +02001254 else
1255 *(output - 1) = 0;
wdenkc6097192002-11-03 00:24:07 +00001256
1257#ifdef DEBUG_PARSER
1258 printf ("[PROCESS_MACROS] OUTPUT len %d: \"%s\"\n",
Wolfgang Denk19973b62006-10-28 00:38:39 +02001259 strlen (output_start), output_start);
wdenkc6097192002-11-03 00:24:07 +00001260#endif
1261}
1262
1263/****************************************************************************
1264 * returns:
1265 * 1 - command executed, repeatable
1266 * 0 - command executed but not repeatable, interrupted commands are
1267 * always considered not repeatable
1268 * -1 - not executed (unrecognized, bootd recursion or too many args)
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +02001269 * (If cmd is NULL or "" or longer than CONFIG_SYS_CBSIZE-1 it is
wdenkc6097192002-11-03 00:24:07 +00001270 * considered unrecognized)
1271 *
1272 * WARNING:
1273 *
1274 * We must create a temporary copy of the command since the command we get
1275 * may be the result from getenv(), which returns a pointer directly to
1276 * the environment data, which may change magicly when the command we run
1277 * creates or modifies environment variables (like "bootp" does).
1278 */
1279
1280int run_command (const char *cmd, int flag)
1281{
1282 cmd_tbl_t *cmdtp;
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +02001283 char cmdbuf[CONFIG_SYS_CBSIZE]; /* working copy of cmd */
wdenkc6097192002-11-03 00:24:07 +00001284 char *token; /* start of token in cmdbuf */
1285 char *sep; /* end of token (separator) in cmdbuf */
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +02001286 char finaltoken[CONFIG_SYS_CBSIZE];
wdenkc6097192002-11-03 00:24:07 +00001287 char *str = cmdbuf;
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +02001288 char *argv[CONFIG_SYS_MAXARGS + 1]; /* NULL terminated */
wdenkf07771c2003-05-28 08:06:31 +00001289 int argc, inquotes;
wdenkc6097192002-11-03 00:24:07 +00001290 int repeatable = 1;
wdenkf07771c2003-05-28 08:06:31 +00001291 int rc = 0;
wdenkc6097192002-11-03 00:24:07 +00001292
1293#ifdef DEBUG_PARSER
1294 printf ("[RUN_COMMAND] cmd[%p]=\"", cmd);
1295 puts (cmd ? cmd : "NULL"); /* use puts - string may be loooong */
1296 puts ("\"\n");
1297#endif
1298
1299 clear_ctrlc(); /* forget any previous Control C */
1300
1301 if (!cmd || !*cmd) {
1302 return -1; /* empty command */
1303 }
1304
Jean-Christophe PLAGNIOL-VILLARD6d0f6bc2008-10-16 15:01:15 +02001305 if (strlen(cmd) >= CONFIG_SYS_CBSIZE) {
wdenkc6097192002-11-03 00:24:07 +00001306 puts ("## Command too long!\n");
1307 return -1;
1308 }
1309
1310 strcpy (cmdbuf, cmd);
1311
1312 /* Process separators and check for invalid
1313 * repeatable commands
1314 */
1315
1316#ifdef DEBUG_PARSER
1317 printf ("[PROCESS_SEPARATORS] %s\n", cmd);
1318#endif
1319 while (*str) {
1320
1321 /*
1322 * Find separator, or string end
1323 * Allow simple escape of ';' by writing "\;"
1324 */
wdenka25f8622003-01-02 23:57:29 +00001325 for (inquotes = 0, sep = str; *sep; sep++) {
1326 if ((*sep=='\'') &&
1327 (*(sep-1) != '\\'))
1328 inquotes=!inquotes;
1329
1330 if (!inquotes &&
1331 (*sep == ';') && /* separator */
wdenkc6097192002-11-03 00:24:07 +00001332 ( sep != str) && /* past string start */
1333 (*(sep-1) != '\\')) /* and NOT escaped */
1334 break;
1335 }
1336
1337 /*
1338 * Limit the token to data between separators
1339 */
1340 token = str;
1341 if (*sep) {
1342 str = sep + 1; /* start of command for next pass */
1343 *sep = '\0';
1344 }
1345 else
1346 str = sep; /* no more commands for next pass */
1347#ifdef DEBUG_PARSER
1348 printf ("token: \"%s\"\n", token);
1349#endif
1350
1351 /* find macros in this token and replace them */
1352 process_macros (token, finaltoken);
1353
1354 /* Extract arguments */
Wolfgang Denk1264b402006-03-12 02:20:55 +01001355 if ((argc = parse_line (finaltoken, argv)) == 0) {
1356 rc = -1; /* no command at all */
1357 continue;
1358 }
wdenkc6097192002-11-03 00:24:07 +00001359
1360 /* Look up command in command table */
1361 if ((cmdtp = find_cmd(argv[0])) == NULL) {
1362 printf ("Unknown command '%s' - try 'help'\n", argv[0]);
wdenkf07771c2003-05-28 08:06:31 +00001363 rc = -1; /* give up after bad command */
1364 continue;
wdenkc6097192002-11-03 00:24:07 +00001365 }
1366
1367 /* found - check max args */
1368 if (argc > cmdtp->maxargs) {
Peter Tyser62c3ae72009-01-27 18:03:10 -06001369 cmd_usage(cmdtp);
wdenkf07771c2003-05-28 08:06:31 +00001370 rc = -1;
1371 continue;
wdenkc6097192002-11-03 00:24:07 +00001372 }
1373
Jon Loeligerc3517f92007-07-08 18:10:08 -05001374#if defined(CONFIG_CMD_BOOTD)
wdenkc6097192002-11-03 00:24:07 +00001375 /* avoid "bootd" recursion */
1376 if (cmdtp->cmd == do_bootd) {
1377#ifdef DEBUG_PARSER
1378 printf ("[%s]\n", finaltoken);
1379#endif
1380 if (flag & CMD_FLAG_BOOTD) {
wdenk4b9206e2004-03-23 22:14:11 +00001381 puts ("'bootd' recursion detected\n");
wdenkf07771c2003-05-28 08:06:31 +00001382 rc = -1;
1383 continue;
Wolfgang Denk1264b402006-03-12 02:20:55 +01001384 } else {
wdenkc6097192002-11-03 00:24:07 +00001385 flag |= CMD_FLAG_BOOTD;
Wolfgang Denk1264b402006-03-12 02:20:55 +01001386 }
wdenkc6097192002-11-03 00:24:07 +00001387 }
Jon Loeliger90253172007-07-10 11:02:44 -05001388#endif
wdenkc6097192002-11-03 00:24:07 +00001389
1390 /* OK - call function to do the command */
1391 if ((cmdtp->cmd) (cmdtp, flag, argc, argv) != 0) {
wdenkf07771c2003-05-28 08:06:31 +00001392 rc = -1;
wdenkc6097192002-11-03 00:24:07 +00001393 }
1394
1395 repeatable &= cmdtp->repeatable;
1396
1397 /* Did the user stop this? */
1398 if (had_ctrlc ())
Detlev Zundel5afb2022007-05-23 18:47:48 +02001399 return -1; /* if stopped then not repeatable */
wdenkc6097192002-11-03 00:24:07 +00001400 }
1401
wdenkf07771c2003-05-28 08:06:31 +00001402 return rc ? rc : repeatable;
wdenkc6097192002-11-03 00:24:07 +00001403}
1404
1405/****************************************************************************/
1406
Jon Loeligerc3517f92007-07-08 18:10:08 -05001407#if defined(CONFIG_CMD_RUN)
Wolfgang Denk54841ab2010-06-28 22:00:46 +02001408int do_run (cmd_tbl_t * cmdtp, int flag, int argc, char * const argv[])
wdenkc6097192002-11-03 00:24:07 +00001409{
1410 int i;
wdenkc6097192002-11-03 00:24:07 +00001411
Wolfgang Denk47e26b12010-07-17 01:06:04 +02001412 if (argc < 2)
1413 return cmd_usage(cmdtp);
wdenkc6097192002-11-03 00:24:07 +00001414
1415 for (i=1; i<argc; ++i) {
wdenk3e386912003-04-05 00:53:31 +00001416 char *arg;
1417
1418 if ((arg = getenv (argv[i])) == NULL) {
1419 printf ("## Error: \"%s\" not defined\n", argv[i]);
1420 return 1;
1421 }
Jason Hobbsc8a20792011-08-31 05:37:24 +00001422
1423 if (run_command2(arg, flag) != 0)
wdenk3e386912003-04-05 00:53:31 +00001424 return 1;
wdenkc6097192002-11-03 00:24:07 +00001425 }
wdenk3e386912003-04-05 00:53:31 +00001426 return 0;
wdenkc6097192002-11-03 00:24:07 +00001427}
Jon Loeliger90253172007-07-10 11:02:44 -05001428#endif