3 * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
5 * See file CREDITS for list of people who contributed to this
8 * This program is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU General Public License as
10 * published by the Free Software Foundation; either version 2 of
11 * the License, or (at your option) any later version.
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
18 * You should have received a copy of the GNU General Public License
19 * along with this program; if not, write to the Free Software
20 * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
29 #include <cmd_nvedit.h>
30 #include <cmd_bootm.h>
32 #if defined(CONFIG_BOOT_RETRY_TIME) && defined(CONFIG_RESET_TO_RETRY)
33 #include <cmd_boot.h> /* for do_reset() prototype */
36 #ifdef CFG_HUSH_PARSER
40 #define MAX_DELAY_STOP_STR 32
42 static char * delete_char (char *buffer, char *p, int *colp, int *np, int plen);
43 static int parse_line (char *, char *[]);
44 #if defined(CONFIG_BOOTDELAY) && (CONFIG_BOOTDELAY >= 0)
45 static int abortboot(int);
50 char console_buffer[CFG_CBSIZE]; /* console I/O buffer */
52 static char erase_seq[] = "\b \b"; /* erase sequence */
53 static char tab_seq[] = " "; /* used to expand TABs */
55 #ifdef CONFIG_BOOT_RETRY_TIME
56 static uint64_t endtime = 0; /* must be set, default is instant timeout */
57 static int retry_time = -1; /* -1 so can call readline before main_loop */
60 #define endtick(seconds) (get_ticks() + (uint64_t)(seconds) * get_tbclk())
62 #ifndef CONFIG_BOOT_RETRY_MIN
63 #define CONFIG_BOOT_RETRY_MIN CONFIG_BOOT_RETRY_TIME
66 #ifdef CONFIG_MODEM_SUPPORT
68 extern void mdm_init(void); /* defined in board.c */
71 /***************************************************************************
72 * Watch for 'delay' seconds for autoboot stop or autoboot delay string.
73 * returns: 0 - no key string, allow autoboot
74 * 1 - got key string, abort
76 #if defined(CONFIG_BOOTDELAY) && (CONFIG_BOOTDELAY >= 0)
77 # if defined(CONFIG_AUTOBOOT_KEYED)
78 static __inline__ int abortboot(int bootdelay)
81 uint64_t etime = endtick(bootdelay);
90 { str: getenv ("bootdelaykey"), retry: 1 },
91 { str: getenv ("bootdelaykey2"), retry: 1 },
92 { str: getenv ("bootstopkey"), retry: 0 },
93 { str: getenv ("bootstopkey2"), retry: 0 },
96 char presskey [MAX_DELAY_STOP_STR];
97 u_int presskey_len = 0;
98 u_int presskey_max = 0;
101 # ifdef CONFIG_AUTOBOOT_PROMPT
102 printf (CONFIG_AUTOBOOT_PROMPT, bootdelay);
105 # ifdef CONFIG_AUTOBOOT_DELAY_STR
106 if (delaykey[0].str == NULL)
107 delaykey[0].str = CONFIG_AUTOBOOT_DELAY_STR;
109 # ifdef CONFIG_AUTOBOOT_DELAY_STR2
110 if (delaykey[1].str == NULL)
111 delaykey[1].str = CONFIG_AUTOBOOT_DELAY_STR2;
113 # ifdef CONFIG_AUTOBOOT_STOP_STR
114 if (delaykey[2].str == NULL)
115 delaykey[2].str = CONFIG_AUTOBOOT_STOP_STR;
117 # ifdef CONFIG_AUTOBOOT_STOP_STR2
118 if (delaykey[3].str == NULL)
119 delaykey[3].str = CONFIG_AUTOBOOT_STOP_STR2;
122 for (i = 0; i < sizeof(delaykey) / sizeof(delaykey[0]); i ++) {
123 delaykey[i].len = delaykey[i].str == NULL ?
124 0 : strlen (delaykey[i].str);
125 delaykey[i].len = delaykey[i].len > MAX_DELAY_STOP_STR ?
126 MAX_DELAY_STOP_STR : delaykey[i].len;
128 presskey_max = presskey_max > delaykey[i].len ?
129 presskey_max : delaykey[i].len;
132 printf("%s key:<%s>\n",
133 delaykey[i].retry ? "delay" : "stop",
134 delaykey[i].str ? delaykey[i].str : "NULL");
138 /* In order to keep up with incoming data, check timeout only
141 while (!abort && get_ticks() <= etime) {
142 for (i = 0; i < sizeof(delaykey) / sizeof(delaykey[0]); i ++) {
143 if (delaykey[i].len > 0 &&
144 presskey_len >= delaykey[i].len &&
145 memcmp (presskey + presskey_len - delaykey[i].len,
147 delaykey[i].len) == 0) {
149 printf("got %skey\n",
150 delaykey[i].retry ? "delay" : "stop");
153 # ifdef CONFIG_BOOT_RETRY_TIME
154 /* don't retry auto boot */
155 if (! delaykey[i].retry)
163 if (presskey_len < presskey_max) {
164 presskey [presskey_len ++] = getc();
167 for (i = 0; i < presskey_max - 1; i ++)
168 presskey [i] = presskey [i + 1];
170 presskey [i] = getc();
176 printf("key timeout\n");
182 # else /* !defined(CONFIG_AUTOBOOT_KEYED) */
184 #ifdef CONFIG_MENUKEY
185 static int menukey = 0;
188 static __inline__ int abortboot(int bootdelay)
192 #ifdef CONFIG_MENUPROMPT
193 printf(CONFIG_MENUPROMPT, bootdelay);
195 printf("Hit any key to stop autoboot: %2d ", bootdelay);
198 #if defined CONFIG_ZERO_BOOTDELAY_CHECK
200 * Check if key already pressed
201 * Don't check if bootdelay < 0
203 if (bootdelay >= 0) {
204 if (tstc()) { /* we got a key press */
205 (void) getc(); /* consume input */
206 printf ("\b\b\b 0\n");
207 return 1; /* don't auto boot */
212 while (bootdelay > 0) {
216 /* delay 100 * 10ms */
217 for (i=0; !abort && i<100; ++i) {
218 if (tstc()) { /* we got a key press */
219 abort = 1; /* don't auto boot */
220 bootdelay = 0; /* no more delay */
221 # ifdef CONFIG_MENUKEY
224 (void) getc(); /* consume input */
231 printf ("\b\b\b%2d ", bootdelay);
238 # endif /* CONFIG_AUTOBOOT_KEYED */
239 #endif /* CONFIG_BOOTDELAY >= 0 */
241 /****************************************************************************/
243 void main_loop (void)
245 #ifndef CFG_HUSH_PARSER
246 static char lastcommand[CFG_CBSIZE] = { 0, };
252 #if defined(CONFIG_BOOTDELAY) && (CONFIG_BOOTDELAY >= 0)
256 #ifdef CONFIG_PREBOOT
260 #if defined(CONFIG_VFD) && defined(VFD_TEST_LOGO)
261 ulong bmp = 0; /* default bitmap */
262 extern int trab_vfd (ulong bitmap);
264 #ifdef CONFIG_MODEM_SUPPORT
266 bmp = 1; /* alternate bitmap */
269 #endif /* CONFIG_VFD && VFD_TEST_LOGO */
271 #ifdef CONFIG_MODEM_SUPPORT
272 debug ("DEBUG: main_loop: do_mdm_init=%d\n", do_mdm_init);
274 uchar *str = strdup(getenv("mdm_cmd"));
275 setenv ("preboot", str); /* set or delete definition */
278 mdm_init(); /* wait for modem connection */
280 #endif /* CONFIG_MODEM_SUPPORT */
282 #ifdef CFG_HUSH_PARSER
283 u_boot_hush_start ();
286 #ifdef CONFIG_PREBOOT
287 if ((p = getenv ("preboot")) != NULL) {
288 # ifdef CONFIG_AUTOBOOT_KEYED
289 int prev = disable_ctrlc(1); /* disable Control C checking */
292 # ifndef CFG_HUSH_PARSER
295 parse_string_outer(p, FLAG_PARSE_SEMICOLON |
296 FLAG_EXIT_FROM_LOOP);
299 # ifdef CONFIG_AUTOBOOT_KEYED
300 disable_ctrlc(prev); /* restore Control C checking */
303 #endif /* CONFIG_PREBOOT */
305 #if defined(CONFIG_BOOTDELAY) && (CONFIG_BOOTDELAY >= 0)
306 s = getenv ("bootdelay");
307 bootdelay = s ? (int)simple_strtol(s, NULL, 10) : CONFIG_BOOTDELAY;
309 debug ("### main_loop entered: bootdelay=%d\n\n", bootdelay);
311 # ifdef CONFIG_BOOT_RETRY_TIME
312 s = getenv ("bootretry");
314 retry_time = (int)simple_strtoul(s, NULL, 10);
316 retry_time = CONFIG_BOOT_RETRY_TIME;
317 if (retry_time >= 0 && retry_time < CONFIG_BOOT_RETRY_MIN)
318 retry_time = CONFIG_BOOT_RETRY_MIN;
319 # endif /* CONFIG_BOOT_RETRY_TIME */
321 s = getenv ("bootcmd");
323 debug ("### main_loop: bootcmd=\"%s\"\n", s ? s : "<UNDEFINED>");
325 if (bootdelay >= 0 && s && !abortboot (bootdelay)) {
326 # ifdef CONFIG_AUTOBOOT_KEYED
327 int prev = disable_ctrlc(1); /* disable Control C checking */
330 # ifndef CFG_HUSH_PARSER
333 parse_string_outer(s, FLAG_PARSE_SEMICOLON |
334 FLAG_EXIT_FROM_LOOP);
337 # ifdef CONFIG_AUTOBOOT_KEYED
338 disable_ctrlc(prev); /* restore Control C checking */
342 # ifdef CONFIG_MENUKEY
343 if (menukey == CONFIG_MENUKEY) {
344 s = getenv("menucmd");
346 # ifndef CFG_HUSH_PARSER
347 run_command (s, bd, 0);
349 parse_string_outer(s, FLAG_PARSE_SEMICOLON |
350 FLAG_EXIT_FROM_LOOP);
354 #endif /* CONFIG_MENUKEY */
355 #endif /* CONFIG_BOOTDELAY */
357 #ifdef CONFIG_AMIGAONEG3SE
359 extern void video_banner(void);
365 * Main Loop for Monitor Command Processing
367 #ifdef CFG_HUSH_PARSER
369 /* This point is never reached */
373 #ifdef CONFIG_BOOT_RETRY_TIME
375 /* Saw enough of a valid command to
376 * restart the timeout.
381 len = readline (CFG_PROMPT);
383 flag = 0; /* assume no special flags for now */
385 strcpy (lastcommand, console_buffer);
387 flag |= CMD_FLAG_REPEAT;
388 #ifdef CONFIG_BOOT_RETRY_TIME
389 else if (len == -2) {
390 /* -2 means timed out, retry autoboot
392 printf("\nTimed out waiting for command\n");
393 # ifdef CONFIG_RESET_TO_RETRY
394 /* Reinit board to run initialization code again */
395 do_reset (NULL, 0, 0, NULL);
397 return; /* retry autoboot */
403 printf ("<INTERRUPT>\n");
405 rc = run_command (lastcommand, flag);
408 /* invalid command or not repeatable, forget it */
412 #endif /*CFG_HUSH_PARSER*/
415 /***************************************************************************
416 * reset command line timeout to retry_time seconds
418 #ifdef CONFIG_BOOT_RETRY_TIME
419 void reset_cmd_timeout(void)
421 endtime = endtick(retry_time);
425 /****************************************************************************/
428 * Prompt for input and read a line.
429 * If CONFIG_BOOT_RETRY_TIME is defined and retry_time >= 0,
430 * time out when time goes past endtime (timebase time in ticks).
431 * Return: number of read characters
435 int readline (const char *const prompt)
437 char *p = console_buffer;
438 int n = 0; /* buffer index */
439 int plen = 0; /* prompt length */
440 int col; /* output column cnt */
445 plen = strlen (prompt);
451 #ifdef CONFIG_BOOT_RETRY_TIME
452 while (!tstc()) { /* while no incoming data */
453 if (retry_time >= 0 && get_ticks() > endtime)
454 return (-2); /* timed out */
457 WATCHDOG_RESET(); /* Trigger watchdog, if needed */
459 #ifdef CONFIG_SHOW_ACTIVITY
461 extern void show_activity(int arg);
468 * Special character handling
471 case '\r': /* Enter */
475 return (p - console_buffer);
477 case 0x03: /* ^C - break */
478 console_buffer[0] = '\0'; /* discard input */
481 case 0x15: /* ^U - erase line */
490 case 0x17: /* ^W - erase word */
491 p=delete_char(console_buffer, p, &col, &n, plen);
492 while ((n > 0) && (*p != ' ')) {
493 p=delete_char(console_buffer, p, &col, &n, plen);
497 case 0x08: /* ^H - backspace */
498 case 0x7F: /* DEL - backspace */
499 p=delete_char(console_buffer, p, &col, &n, plen);
504 * Must be a normal character then
506 if (n < CFG_CBSIZE-2) {
507 if (c == '\t') { /* expand TABs */
508 puts (tab_seq+(col&07));
511 ++col; /* echo input */
516 } else { /* Buffer full */
523 /****************************************************************************/
525 static char * delete_char (char *buffer, char *p, int *colp, int *np, int plen)
533 if (*(--p) == '\t') { /* will retype the whole line */
534 while (*colp > plen) {
538 for (s=buffer; s<p; ++s) {
540 puts (tab_seq+((*colp) & 07));
541 *colp += 8 - ((*colp) & 07);
555 /****************************************************************************/
557 int parse_line (char *line, char *argv[])
562 printf ("parse_line: \"%s\"\n", line);
564 while (nargs < CFG_MAXARGS) {
566 /* skip any white space */
567 while ((*line == ' ') || (*line == '\t')) {
571 if (*line == '\0') { /* end of line, no more args */
574 printf ("parse_line: nargs=%d\n", nargs);
579 argv[nargs++] = line; /* begin of argument string */
581 /* find end of string */
582 while (*line && (*line != ' ') && (*line != '\t')) {
586 if (*line == '\0') { /* end of line, no more args */
589 printf ("parse_line: nargs=%d\n", nargs);
594 *line++ = '\0'; /* terminate current arg */
597 printf ("** Too many args (max. %d) **\n", CFG_MAXARGS);
600 printf ("parse_line: nargs=%d\n", nargs);
605 /****************************************************************************/
607 static void process_macros (const char *input, char *output)
610 const char *varname_start = NULL;
611 int inputcnt = strlen (input);
612 int outputcnt = CFG_CBSIZE;
613 int state = 0; /* 0 = waiting for '$' */
614 /* 1 = waiting for '(' */
615 /* 2 = waiting for ')' */
616 /* 3 = waiting for ''' */
618 char *output_start = output;
620 printf ("[PROCESS_MACROS] INPUT len %d: \"%s\"\n", strlen(input), input);
623 prev = '\0'; /* previous character */
625 while (inputcnt && outputcnt) {
630 /* remove one level of escape characters */
631 if ((c == '\\') && (prev != '\\')) {
640 case 0: /* Waiting for (unescaped) $ */
641 if ((c == '\'') && (prev != '\\')) {
647 if ((c == '$') && (prev != '\\')) {
654 case 1: /* Waiting for ( */
657 varname_start = input;
669 case 2: /* Waiting for ) */
672 char envname[CFG_CBSIZE], *envval;
673 int envcnt = input-varname_start-1; /* Varname # of chars */
675 /* Get the varname */
676 for (i = 0; i < envcnt; i++) {
677 envname[i] = varname_start[i];
682 envval = getenv (envname);
684 /* Copy into the line if it exists */
686 while ((*envval) && outputcnt) {
687 *(output++) = *(envval++);
690 /* Look for another '$' */
694 case 3: /* Waiting for ' */
695 if ((c == '\'') && (prev != '\\')) {
712 printf ("[PROCESS_MACROS] OUTPUT len %d: \"%s\"\n",
713 strlen(output_start), output_start);
717 /****************************************************************************
719 * 1 - command executed, repeatable
720 * 0 - command executed but not repeatable, interrupted commands are
721 * always considered not repeatable
722 * -1 - not executed (unrecognized, bootd recursion or too many args)
723 * (If cmd is NULL or "" or longer than CFG_CBSIZE-1 it is
724 * considered unrecognized)
728 * We must create a temporary copy of the command since the command we get
729 * may be the result from getenv(), which returns a pointer directly to
730 * the environment data, which may change magicly when the command we run
731 * creates or modifies environment variables (like "bootp" does).
734 int run_command (const char *cmd, int flag)
737 char cmdbuf[CFG_CBSIZE]; /* working copy of cmd */
738 char *token; /* start of token in cmdbuf */
739 char *sep; /* end of token (separator) in cmdbuf */
740 char finaltoken[CFG_CBSIZE];
742 char *argv[CFG_MAXARGS + 1]; /* NULL terminated */
748 printf ("[RUN_COMMAND] cmd[%p]=\"", cmd);
749 puts (cmd ? cmd : "NULL"); /* use puts - string may be loooong */
753 clear_ctrlc(); /* forget any previous Control C */
756 return -1; /* empty command */
759 if (strlen(cmd) >= CFG_CBSIZE) {
760 puts ("## Command too long!\n");
764 strcpy (cmdbuf, cmd);
766 /* Process separators and check for invalid
767 * repeatable commands
771 printf ("[PROCESS_SEPARATORS] %s\n", cmd);
776 * Find separator, or string end
777 * Allow simple escape of ';' by writing "\;"
779 for (inquotes = 0, sep = str; *sep; sep++) {
785 (*sep == ';') && /* separator */
786 ( sep != str) && /* past string start */
787 (*(sep-1) != '\\')) /* and NOT escaped */
792 * Limit the token to data between separators
796 str = sep + 1; /* start of command for next pass */
800 str = sep; /* no more commands for next pass */
802 printf ("token: \"%s\"\n", token);
805 /* find macros in this token and replace them */
806 process_macros (token, finaltoken);
808 /* Extract arguments */
809 argc = parse_line (finaltoken, argv);
811 /* Look up command in command table */
812 if ((cmdtp = find_cmd(argv[0])) == NULL) {
813 printf ("Unknown command '%s' - try 'help'\n", argv[0]);
814 return -1; /* give up after bad command */
817 /* found - check max args */
818 if (argc > cmdtp->maxargs) {
819 printf ("Usage:\n%s\n", cmdtp->usage);
823 #if (CONFIG_COMMANDS & CFG_CMD_BOOTD)
824 /* avoid "bootd" recursion */
825 if (cmdtp->cmd == do_bootd) {
827 printf ("[%s]\n", finaltoken);
829 if (flag & CMD_FLAG_BOOTD) {
830 printf ("'bootd' recursion detected\n");
834 flag |= CMD_FLAG_BOOTD;
836 #endif /* CFG_CMD_BOOTD */
838 /* OK - call function to do the command */
839 if ((cmdtp->cmd) (cmdtp, flag, argc, argv) != 0) {
843 repeatable &= cmdtp->repeatable;
845 /* Did the user stop this? */
847 return 0; /* if stopped then not repeatable */
853 /****************************************************************************/
855 #if (CONFIG_COMMANDS & CFG_CMD_RUN)
856 int do_run (cmd_tbl_t * cmdtp, int flag, int argc, char *argv[])
862 printf ("Usage:\n%s\n", cmdtp->usage);
866 for (i=1; i<argc; ++i) {
867 #ifndef CFG_HUSH_PARSER
868 if (run_command (getenv (argv[i]), flag) != -1) ++rcode;
870 if (parse_string_outer(getenv (argv[i]),
871 FLAG_PARSE_SEMICOLON | FLAG_EXIT_FROM_LOOP) == 0) ++rcode;
874 return ((rcode == i) ? 0 : 1);