1 /*
2 * (C) Copyright 2000
3 * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
4 *
5 * Add to readline cmdline-editing by
6 * (C) Copyright 2005
7 * JinHua Luo, GuangDong Linux Center, <luo.jinhua@gd-linux.com>
8 *
9 * 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 */
28 /* #define DEBUG */
30 #include <common.h>
31 #include <watchdog.h>
32 #include <command.h>
33 #ifdef CONFIG_MODEM_SUPPORT
34 #include <malloc.h> /* for free() prototype */
35 #endif
37 #ifdef CONFIG_SYS_HUSH_PARSER
38 #include <hush.h>
39 #endif
41 #include <post.h>
43 #if defined(CONFIG_SILENT_CONSOLE) || defined(CONFIG_POST) || defined(CONFIG_CMDLINE_EDITING)
44 DECLARE_GLOBAL_DATA_PTR;
45 #endif
47 /*
48 * Board-specific Platform code can reimplement show_boot_progress () if needed
49 */
50 void inline __show_boot_progress (int val) {}
51 void show_boot_progress (int val) __attribute__((weak, alias("__show_boot_progress")));
53 #if defined(CONFIG_BOOT_RETRY_TIME) && defined(CONFIG_RESET_TO_RETRY)
54 extern int do_reset (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[]); /* for do_reset() prototype */
55 #endif
57 extern int do_bootd (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[]);
59 #if defined(CONFIG_UPDATE_TFTP)
60 void update_tftp (void);
61 #endif /* CONFIG_UPDATE_TFTP */
63 #define MAX_DELAY_STOP_STR 32
65 #if defined(CONFIG_BOOTDELAY) && (CONFIG_BOOTDELAY >= 0)
66 static int abortboot(int);
67 #endif
69 #undef DEBUG_PARSER
71 char console_buffer[CONFIG_SYS_CBSIZE + 1]; /* console I/O buffer */
73 static char * delete_char (char *buffer, char *p, int *colp, int *np, int plen);
74 static char erase_seq[] = "\b \b"; /* erase sequence */
75 static char tab_seq[] = " "; /* used to expand TABs */
77 #ifdef CONFIG_BOOT_RETRY_TIME
78 static uint64_t endtime = 0; /* must be set, default is instant timeout */
79 static int retry_time = -1; /* -1 so can call readline before main_loop */
80 #endif
82 #define endtick(seconds) (get_ticks() + (uint64_t)(seconds) * get_tbclk())
84 #ifndef CONFIG_BOOT_RETRY_MIN
85 #define CONFIG_BOOT_RETRY_MIN CONFIG_BOOT_RETRY_TIME
86 #endif
88 #ifdef CONFIG_MODEM_SUPPORT
89 int do_mdm_init = 0;
90 extern void mdm_init(void); /* defined in board.c */
91 #endif
93 /***************************************************************************
94 * Watch for 'delay' seconds for autoboot stop or autoboot delay string.
95 * returns: 0 - no key string, allow autoboot
96 * 1 - got key string, abort
97 */
98 #if defined(CONFIG_BOOTDELAY) && (CONFIG_BOOTDELAY >= 0)
99 # if defined(CONFIG_AUTOBOOT_KEYED)
100 static __inline__ int abortboot(int bootdelay)
101 {
102 int abort = 0;
103 uint64_t etime = endtick(bootdelay);
104 struct {
105 char* str;
106 u_int len;
107 int retry;
108 }
109 delaykey [] = {
110 { str: getenv ("bootdelaykey"), retry: 1 },
111 { str: getenv ("bootdelaykey2"), retry: 1 },
112 { str: getenv ("bootstopkey"), retry: 0 },
113 { str: getenv ("bootstopkey2"), retry: 0 },
114 };
116 char presskey [MAX_DELAY_STOP_STR];
117 u_int presskey_len = 0;
118 u_int presskey_max = 0;
119 u_int i;
121 # ifdef CONFIG_AUTOBOOT_PROMPT
122 printf(CONFIG_AUTOBOOT_PROMPT);
123 # endif
125 # ifdef CONFIG_AUTOBOOT_DELAY_STR
126 if (delaykey[0].str == NULL)
127 delaykey[0].str = CONFIG_AUTOBOOT_DELAY_STR;
128 # endif
129 # ifdef CONFIG_AUTOBOOT_DELAY_STR2
130 if (delaykey[1].str == NULL)
131 delaykey[1].str = CONFIG_AUTOBOOT_DELAY_STR2;
132 # endif
133 # ifdef CONFIG_AUTOBOOT_STOP_STR
134 if (delaykey[2].str == NULL)
135 delaykey[2].str = CONFIG_AUTOBOOT_STOP_STR;
136 # endif
137 # ifdef CONFIG_AUTOBOOT_STOP_STR2
138 if (delaykey[3].str == NULL)
139 delaykey[3].str = CONFIG_AUTOBOOT_STOP_STR2;
140 # endif
142 for (i = 0; i < sizeof(delaykey) / sizeof(delaykey[0]); i ++) {
143 delaykey[i].len = delaykey[i].str == NULL ?
144 0 : strlen (delaykey[i].str);
145 delaykey[i].len = delaykey[i].len > MAX_DELAY_STOP_STR ?
146 MAX_DELAY_STOP_STR : delaykey[i].len;
148 presskey_max = presskey_max > delaykey[i].len ?
149 presskey_max : delaykey[i].len;
151 # if DEBUG_BOOTKEYS
152 printf("%s key:<%s>\n",
153 delaykey[i].retry ? "delay" : "stop",
154 delaykey[i].str ? delaykey[i].str : "NULL");
155 # endif
156 }
158 /* In order to keep up with incoming data, check timeout only
159 * when catch up.
160 */
161 do {
162 if (tstc()) {
163 if (presskey_len < presskey_max) {
164 presskey [presskey_len ++] = getc();
165 }
166 else {
167 for (i = 0; i < presskey_max - 1; i ++)
168 presskey [i] = presskey [i + 1];
170 presskey [i] = getc();
171 }
172 }
174 for (i = 0; i < sizeof(delaykey) / sizeof(delaykey[0]); i ++) {
175 if (delaykey[i].len > 0 &&
176 presskey_len >= delaykey[i].len &&
177 memcmp (presskey + presskey_len - delaykey[i].len,
178 delaykey[i].str,
179 delaykey[i].len) == 0) {
180 # if DEBUG_BOOTKEYS
181 printf("got %skey\n",
182 delaykey[i].retry ? "delay" : "stop");
183 # endif
185 # ifdef CONFIG_BOOT_RETRY_TIME
186 /* don't retry auto boot */
187 if (! delaykey[i].retry)
188 retry_time = -1;
189 # endif
190 abort = 1;
191 }
192 }
193 } while (!abort && get_ticks() <= etime);
195 # if DEBUG_BOOTKEYS
196 if (!abort)
197 puts("key timeout\n");
198 # endif
200 #ifdef CONFIG_SILENT_CONSOLE
201 if (abort)
202 gd->flags &= ~GD_FLG_SILENT;
203 #endif
205 return abort;
206 }
208 # else /* !defined(CONFIG_AUTOBOOT_KEYED) */
210 #ifdef CONFIG_MENUKEY
211 static int menukey = 0;
212 #endif
214 static __inline__ int abortboot(int bootdelay)
215 {
216 int abort = 0;
218 #ifdef CONFIG_MENUPROMPT
219 printf(CONFIG_MENUPROMPT);
220 #else
221 printf("Hit any key to stop autoboot: %2d ", bootdelay);
222 #endif
224 #if defined CONFIG_ZERO_BOOTDELAY_CHECK
225 /*
226 * Check if key already pressed
227 * Don't check if bootdelay < 0
228 */
229 if (bootdelay >= 0) {
230 if (tstc()) { /* we got a key press */
231 (void) getc(); /* consume input */
232 puts ("\b\b\b 0");
233 abort = 1; /* don't auto boot */
234 }
235 }
236 #endif
238 while ((bootdelay > 0) && (!abort)) {
239 int i;
241 --bootdelay;
242 /* delay 100 * 10ms */
243 for (i=0; !abort && i<100; ++i) {
244 if (tstc()) { /* we got a key press */
245 abort = 1; /* don't auto boot */
246 bootdelay = 0; /* no more delay */
247 # ifdef CONFIG_MENUKEY
248 menukey = getc();
249 # else
250 (void) getc(); /* consume input */
251 # endif
252 break;
253 }
254 udelay(10000);
255 }
257 printf("\b\b\b%2d ", bootdelay);
258 }
260 putc('\n');
262 #ifdef CONFIG_SILENT_CONSOLE
263 if (abort)
264 gd->flags &= ~GD_FLG_SILENT;
265 #endif
267 return abort;
268 }
269 # endif /* CONFIG_AUTOBOOT_KEYED */
270 #endif /* CONFIG_BOOTDELAY >= 0 */
272 /****************************************************************************/
274 void main_loop (void)
275 {
276 #ifndef CONFIG_SYS_HUSH_PARSER
277 static char lastcommand[CONFIG_SYS_CBSIZE] = { 0, };
278 int len;
279 int rc = 1;
280 int flag;
281 #endif
283 #if defined(CONFIG_BOOTDELAY) && (CONFIG_BOOTDELAY >= 0)
284 char *s;
285 int bootdelay;
286 #endif
287 #ifdef CONFIG_PREBOOT
288 char *p;
289 #endif
290 #ifdef CONFIG_BOOTCOUNT_LIMIT
291 unsigned long bootcount = 0;
292 unsigned long bootlimit = 0;
293 char *bcs;
294 char bcs_set[16];
295 #endif /* CONFIG_BOOTCOUNT_LIMIT */
297 #if defined(CONFIG_VFD) && defined(VFD_TEST_LOGO)
298 ulong bmp = 0; /* default bitmap */
299 extern int trab_vfd (ulong bitmap);
301 #ifdef CONFIG_MODEM_SUPPORT
302 if (do_mdm_init)
303 bmp = 1; /* alternate bitmap */
304 #endif
305 trab_vfd (bmp);
306 #endif /* CONFIG_VFD && VFD_TEST_LOGO */
308 #ifdef CONFIG_BOOTCOUNT_LIMIT
309 bootcount = bootcount_load();
310 bootcount++;
311 bootcount_store (bootcount);
312 sprintf (bcs_set, "%lu", bootcount);
313 setenv ("bootcount", bcs_set);
314 bcs = getenv ("bootlimit");
315 bootlimit = bcs ? simple_strtoul (bcs, NULL, 10) : 0;
316 #endif /* CONFIG_BOOTCOUNT_LIMIT */
318 #ifdef CONFIG_MODEM_SUPPORT
319 debug ("DEBUG: main_loop: do_mdm_init=%d\n", do_mdm_init);
320 if (do_mdm_init) {
321 char *str = strdup(getenv("mdm_cmd"));
322 setenv ("preboot", str); /* set or delete definition */
323 if (str != NULL)
324 free (str);
325 mdm_init(); /* wait for modem connection */
326 }
327 #endif /* CONFIG_MODEM_SUPPORT */
329 #ifdef CONFIG_VERSION_VARIABLE
330 {
331 extern char version_string[];
333 setenv ("ver", version_string); /* set version variable */
334 }
335 #endif /* CONFIG_VERSION_VARIABLE */
337 #ifdef CONFIG_SYS_HUSH_PARSER
338 u_boot_hush_start ();
339 #endif
341 #if defined(CONFIG_HUSH_INIT_VAR)
342 hush_init_var ();
343 #endif
345 #ifdef CONFIG_AUTO_COMPLETE
346 install_auto_complete();
347 #endif
349 #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
355 # ifndef CONFIG_SYS_HUSH_PARSER
356 run_command (p, 0);
357 # else
358 parse_string_outer(p, FLAG_PARSE_SEMICOLON |
359 FLAG_EXIT_FROM_LOOP);
360 # endif
362 # ifdef CONFIG_AUTOBOOT_KEYED
363 disable_ctrlc(prev); /* restore Control C checking */
364 # endif
365 }
366 #endif /* CONFIG_PREBOOT */
368 #if defined(CONFIG_UPDATE_TFTP)
369 update_tftp ();
370 #endif /* CONFIG_UPDATE_TFTP */
372 #if defined(CONFIG_BOOTDELAY) && (CONFIG_BOOTDELAY >= 0)
373 s = getenv ("bootdelay");
374 bootdelay = s ? (int)simple_strtol(s, NULL, 10) : CONFIG_BOOTDELAY;
376 debug ("### main_loop entered: bootdelay=%d\n\n", bootdelay);
378 # ifdef CONFIG_BOOT_RETRY_TIME
379 init_cmd_timeout ();
380 # endif /* CONFIG_BOOT_RETRY_TIME */
382 #ifdef CONFIG_POST
383 if (gd->flags & GD_FLG_POSTFAIL) {
384 s = getenv("failbootcmd");
385 }
386 else
387 #endif /* CONFIG_POST */
388 #ifdef CONFIG_BOOTCOUNT_LIMIT
389 if (bootlimit && (bootcount > bootlimit)) {
390 printf ("Warning: Bootlimit (%u) exceeded. Using altbootcmd.\n",
391 (unsigned)bootlimit);
392 s = getenv ("altbootcmd");
393 }
394 else
395 #endif /* CONFIG_BOOTCOUNT_LIMIT */
396 s = getenv ("bootcmd");
398 debug ("### main_loop: bootcmd=\"%s\"\n", s ? s : "<UNDEFINED>");
400 if (bootdelay >= 0 && s && !abortboot (bootdelay)) {
401 # ifdef CONFIG_AUTOBOOT_KEYED
402 int prev = disable_ctrlc(1); /* disable Control C checking */
403 # endif
405 # ifndef CONFIG_SYS_HUSH_PARSER
406 run_command (s, 0);
407 # else
408 parse_string_outer(s, FLAG_PARSE_SEMICOLON |
409 FLAG_EXIT_FROM_LOOP);
410 # endif
412 # ifdef CONFIG_AUTOBOOT_KEYED
413 disable_ctrlc(prev); /* restore Control C checking */
414 # endif
415 }
417 # ifdef CONFIG_MENUKEY
418 if (menukey == CONFIG_MENUKEY) {
419 s = getenv("menucmd");
420 if (s) {
421 # ifndef CONFIG_SYS_HUSH_PARSER
422 run_command (s, 0);
423 # else
424 parse_string_outer(s, FLAG_PARSE_SEMICOLON |
425 FLAG_EXIT_FROM_LOOP);
426 # endif
427 }
428 }
429 #endif /* CONFIG_MENUKEY */
430 #endif /* CONFIG_BOOTDELAY */
432 #ifdef CONFIG_AMIGAONEG3SE
433 {
434 extern void video_banner(void);
435 video_banner();
436 }
437 #endif
439 /*
440 * Main Loop for Monitor Command Processing
441 */
442 #ifdef CONFIG_SYS_HUSH_PARSER
443 parse_file_outer();
444 /* This point is never reached */
445 for (;;);
446 #else
447 for (;;) {
448 #ifdef CONFIG_BOOT_RETRY_TIME
449 if (rc >= 0) {
450 /* Saw enough of a valid command to
451 * restart the timeout.
452 */
453 reset_cmd_timeout();
454 }
455 #endif
456 len = readline (CONFIG_SYS_PROMPT);
458 flag = 0; /* assume no special flags for now */
459 if (len > 0)
460 strcpy (lastcommand, console_buffer);
461 else if (len == 0)
462 flag |= CMD_FLAG_REPEAT;
463 #ifdef CONFIG_BOOT_RETRY_TIME
464 else if (len == -2) {
465 /* -2 means timed out, retry autoboot
466 */
467 puts ("\nTimed out waiting for command\n");
468 # ifdef CONFIG_RESET_TO_RETRY
469 /* Reinit board to run initialization code again */
470 do_reset (NULL, 0, 0, NULL);
471 # else
472 return; /* retry autoboot */
473 # endif
474 }
475 #endif
477 if (len == -1)
478 puts ("<INTERRUPT>\n");
479 else
480 rc = run_command (lastcommand, flag);
482 if (rc <= 0) {
483 /* invalid command or not repeatable, forget it */
484 lastcommand[0] = 0;
485 }
486 }
487 #endif /*CONFIG_SYS_HUSH_PARSER*/
488 }
490 #ifdef CONFIG_BOOT_RETRY_TIME
491 /***************************************************************************
492 * initialize command line timeout
493 */
494 void init_cmd_timeout(void)
495 {
496 char *s = getenv ("bootretry");
498 if (s != NULL)
499 retry_time = (int)simple_strtol(s, NULL, 10);
500 else
501 retry_time = CONFIG_BOOT_RETRY_TIME;
503 if (retry_time >= 0 && retry_time < CONFIG_BOOT_RETRY_MIN)
504 retry_time = CONFIG_BOOT_RETRY_MIN;
505 }
507 /***************************************************************************
508 * reset command line timeout to retry_time seconds
509 */
510 void reset_cmd_timeout(void)
511 {
512 endtime = endtick(retry_time);
513 }
514 #endif
516 #ifdef CONFIG_CMDLINE_EDITING
518 /*
519 * cmdline-editing related codes from vivi.
520 * Author: Janghoon Lyu <nandy@mizi.com>
521 */
523 #define putnstr(str,n) do { \
524 printf ("%.*s", (int)n, str); \
525 } while (0)
527 #define CTL_CH(c) ((c) - 'a' + 1)
529 #define MAX_CMDBUF_SIZE CONFIG_SYS_CBSIZE
531 #define CTL_BACKSPACE ('\b')
532 #define DEL ((char)255)
533 #define DEL7 ((char)127)
534 #define CREAD_HIST_CHAR ('!')
536 #define getcmd_putch(ch) putc(ch)
537 #define getcmd_getch() getc()
538 #define getcmd_cbeep() getcmd_putch('\a')
540 #define HIST_MAX 20
541 #define HIST_SIZE MAX_CMDBUF_SIZE
543 static int hist_max = 0;
544 static int hist_add_idx = 0;
545 static int hist_cur = -1;
546 unsigned hist_num = 0;
548 char* hist_list[HIST_MAX];
549 char hist_lines[HIST_MAX][HIST_SIZE + 1]; /* Save room for NULL */
551 #define add_idx_minus_one() ((hist_add_idx == 0) ? hist_max : hist_add_idx-1)
553 static void hist_init(void)
554 {
555 int i;
557 hist_max = 0;
558 hist_add_idx = 0;
559 hist_cur = -1;
560 hist_num = 0;
562 for (i = 0; i < HIST_MAX; i++) {
563 hist_list[i] = hist_lines[i];
564 hist_list[i][0] = '\0';
565 }
566 }
568 static void cread_add_to_hist(char *line)
569 {
570 strcpy(hist_list[hist_add_idx], line);
572 if (++hist_add_idx >= HIST_MAX)
573 hist_add_idx = 0;
575 if (hist_add_idx > hist_max)
576 hist_max = hist_add_idx;
578 hist_num++;
579 }
581 static char* hist_prev(void)
582 {
583 char *ret;
584 int old_cur;
586 if (hist_cur < 0)
587 return NULL;
589 old_cur = hist_cur;
590 if (--hist_cur < 0)
591 hist_cur = hist_max;
593 if (hist_cur == hist_add_idx) {
594 hist_cur = old_cur;
595 ret = NULL;
596 } else
597 ret = hist_list[hist_cur];
599 return (ret);
600 }
602 static char* hist_next(void)
603 {
604 char *ret;
606 if (hist_cur < 0)
607 return NULL;
609 if (hist_cur == hist_add_idx)
610 return NULL;
612 if (++hist_cur > hist_max)
613 hist_cur = 0;
615 if (hist_cur == hist_add_idx) {
616 ret = "";
617 } else
618 ret = hist_list[hist_cur];
620 return (ret);
621 }
623 #ifndef CONFIG_CMDLINE_EDITING
624 static void cread_print_hist_list(void)
625 {
626 int i;
627 unsigned long n;
629 n = hist_num - hist_max;
631 i = hist_add_idx + 1;
632 while (1) {
633 if (i > hist_max)
634 i = 0;
635 if (i == hist_add_idx)
636 break;
637 printf("%s\n", hist_list[i]);
638 n++;
639 i++;
640 }
641 }
642 #endif /* CONFIG_CMDLINE_EDITING */
644 #define BEGINNING_OF_LINE() { \
645 while (num) { \
646 getcmd_putch(CTL_BACKSPACE); \
647 num--; \
648 } \
649 }
651 #define ERASE_TO_EOL() { \
652 if (num < eol_num) { \
653 int tmp; \
654 for (tmp = num; tmp < eol_num; tmp++) \
655 getcmd_putch(' '); \
656 while (tmp-- > num) \
657 getcmd_putch(CTL_BACKSPACE); \
658 eol_num = num; \
659 } \
660 }
662 #define REFRESH_TO_EOL() { \
663 if (num < eol_num) { \
664 wlen = eol_num - num; \
665 putnstr(buf + num, wlen); \
666 num = eol_num; \
667 } \
668 }
670 static void cread_add_char(char ichar, int insert, unsigned long *num,
671 unsigned long *eol_num, char *buf, unsigned long len)
672 {
673 unsigned long wlen;
675 /* room ??? */
676 if (insert || *num == *eol_num) {
677 if (*eol_num > len - 1) {
678 getcmd_cbeep();
679 return;
680 }
681 (*eol_num)++;
682 }
684 if (insert) {
685 wlen = *eol_num - *num;
686 if (wlen > 1) {
687 memmove(&buf[*num+1], &buf[*num], wlen-1);
688 }
690 buf[*num] = ichar;
691 putnstr(buf + *num, wlen);
692 (*num)++;
693 while (--wlen) {
694 getcmd_putch(CTL_BACKSPACE);
695 }
696 } else {
697 /* echo the character */
698 wlen = 1;
699 buf[*num] = ichar;
700 putnstr(buf + *num, wlen);
701 (*num)++;
702 }
703 }
705 static void cread_add_str(char *str, int strsize, int insert, unsigned long *num,
706 unsigned long *eol_num, char *buf, unsigned long len)
707 {
708 while (strsize--) {
709 cread_add_char(*str, insert, num, eol_num, buf, len);
710 str++;
711 }
712 }
714 static int cread_line(const char *const prompt, char *buf, unsigned int *len)
715 {
716 unsigned long num = 0;
717 unsigned long eol_num = 0;
718 unsigned long wlen;
719 char ichar;
720 int insert = 1;
721 int esc_len = 0;
722 char esc_save[8];
723 int init_len = strlen(buf);
725 if (init_len)
726 cread_add_str(buf, init_len, 1, &num, &eol_num, buf, *len);
728 while (1) {
729 #ifdef CONFIG_BOOT_RETRY_TIME
730 while (!tstc()) { /* while no incoming data */
731 if (retry_time >= 0 && get_ticks() > endtime)
732 return (-2); /* timed out */
733 WATCHDOG_RESET();
734 }
735 #endif
737 ichar = getcmd_getch();
739 if ((ichar == '\n') || (ichar == '\r')) {
740 putc('\n');
741 break;
742 }
744 /*
745 * handle standard linux xterm esc sequences for arrow key, etc.
746 */
747 if (esc_len != 0) {
748 if (esc_len == 1) {
749 if (ichar == '[') {
750 esc_save[esc_len] = ichar;
751 esc_len = 2;
752 } else {
753 cread_add_str(esc_save, esc_len, insert,
754 &num, &eol_num, buf, *len);
755 esc_len = 0;
756 }
757 continue;
758 }
760 switch (ichar) {
762 case 'D': /* <- key */
763 ichar = CTL_CH('b');
764 esc_len = 0;
765 break;
766 case 'C': /* -> key */
767 ichar = CTL_CH('f');
768 esc_len = 0;
769 break; /* pass off to ^F handler */
770 case 'H': /* Home key */
771 ichar = CTL_CH('a');
772 esc_len = 0;
773 break; /* pass off to ^A handler */
774 case 'A': /* up arrow */
775 ichar = CTL_CH('p');
776 esc_len = 0;
777 break; /* pass off to ^P handler */
778 case 'B': /* down arrow */
779 ichar = CTL_CH('n');
780 esc_len = 0;
781 break; /* pass off to ^N handler */
782 default:
783 esc_save[esc_len++] = ichar;
784 cread_add_str(esc_save, esc_len, insert,
785 &num, &eol_num, buf, *len);
786 esc_len = 0;
787 continue;
788 }
789 }
791 switch (ichar) {
792 case 0x1b:
793 if (esc_len == 0) {
794 esc_save[esc_len] = ichar;
795 esc_len = 1;
796 } else {
797 puts("impossible condition #876\n");
798 esc_len = 0;
799 }
800 break;
802 case CTL_CH('a'):
803 BEGINNING_OF_LINE();
804 break;
805 case CTL_CH('c'): /* ^C - break */
806 *buf = '\0'; /* discard input */
807 return (-1);
808 case CTL_CH('f'):
809 if (num < eol_num) {
810 getcmd_putch(buf[num]);
811 num++;
812 }
813 break;
814 case CTL_CH('b'):
815 if (num) {
816 getcmd_putch(CTL_BACKSPACE);
817 num--;
818 }
819 break;
820 case CTL_CH('d'):
821 if (num < eol_num) {
822 wlen = eol_num - num - 1;
823 if (wlen) {
824 memmove(&buf[num], &buf[num+1], wlen);
825 putnstr(buf + num, wlen);
826 }
828 getcmd_putch(' ');
829 do {
830 getcmd_putch(CTL_BACKSPACE);
831 } while (wlen--);
832 eol_num--;
833 }
834 break;
835 case CTL_CH('k'):
836 ERASE_TO_EOL();
837 break;
838 case CTL_CH('e'):
839 REFRESH_TO_EOL();
840 break;
841 case CTL_CH('o'):
842 insert = !insert;
843 break;
844 case CTL_CH('x'):
845 case CTL_CH('u'):
846 BEGINNING_OF_LINE();
847 ERASE_TO_EOL();
848 break;
849 case DEL:
850 case DEL7:
851 case 8:
852 if (num) {
853 wlen = eol_num - num;
854 num--;
855 memmove(&buf[num], &buf[num+1], wlen);
856 getcmd_putch(CTL_BACKSPACE);
857 putnstr(buf + num, wlen);
858 getcmd_putch(' ');
859 do {
860 getcmd_putch(CTL_BACKSPACE);
861 } while (wlen--);
862 eol_num--;
863 }
864 break;
865 case CTL_CH('p'):
866 case CTL_CH('n'):
867 {
868 char * hline;
870 esc_len = 0;
872 if (ichar == CTL_CH('p'))
873 hline = hist_prev();
874 else
875 hline = hist_next();
877 if (!hline) {
878 getcmd_cbeep();
879 continue;
880 }
882 /* nuke the current line */
883 /* first, go home */
884 BEGINNING_OF_LINE();
886 /* erase to end of line */
887 ERASE_TO_EOL();
889 /* copy new line into place and display */
890 strcpy(buf, hline);
891 eol_num = strlen(buf);
892 REFRESH_TO_EOL();
893 continue;
894 }
895 #ifdef CONFIG_AUTO_COMPLETE
896 case '\t': {
897 int num2, col;
899 /* do not autocomplete when in the middle */
900 if (num < eol_num) {
901 getcmd_cbeep();
902 break;
903 }
905 buf[num] = '\0';
906 col = strlen(prompt) + eol_num;
907 num2 = num;
908 if (cmd_auto_complete(prompt, buf, &num2, &col)) {
909 col = num2 - num;
910 num += col;
911 eol_num += col;
912 }
913 break;
914 }
915 #endif
916 default:
917 cread_add_char(ichar, insert, &num, &eol_num, buf, *len);
918 break;
919 }
920 }
921 *len = eol_num;
922 buf[eol_num] = '\0'; /* lose the newline */
924 if (buf[0] && buf[0] != CREAD_HIST_CHAR)
925 cread_add_to_hist(buf);
926 hist_cur = hist_add_idx;
928 return 0;
929 }
931 #endif /* CONFIG_CMDLINE_EDITING */
933 /****************************************************************************/
935 /*
936 * Prompt for input and read a line.
937 * If CONFIG_BOOT_RETRY_TIME is defined and retry_time >= 0,
938 * time out when time goes past endtime (timebase time in ticks).
939 * Return: number of read characters
940 * -1 if break
941 * -2 if timed out
942 */
943 int readline (const char *const prompt)
944 {
945 /*
946 * If console_buffer isn't 0-length the user will be prompted to modify
947 * it instead of entering it from scratch as desired.
948 */
949 console_buffer[0] = '\0';
951 return readline_into_buffer(prompt, console_buffer);
952 }
955 int readline_into_buffer (const char *const prompt, char * buffer)
956 {
957 char *p = buffer;
958 #ifdef CONFIG_CMDLINE_EDITING
959 unsigned int len=MAX_CMDBUF_SIZE;
960 int rc;
961 static int initted = 0;
963 /*
964 * History uses a global array which is not
965 * writable until after relocation to RAM.
966 * Revert to non-history version if still
967 * running from flash.
968 */
969 if (gd->flags & GD_FLG_RELOC) {
970 if (!initted) {
971 hist_init();
972 initted = 1;
973 }
975 if (prompt)
976 puts (prompt);
978 rc = cread_line(prompt, p, &len);
979 return rc < 0 ? rc : len;
981 } else {
982 #endif /* CONFIG_CMDLINE_EDITING */
983 char * p_buf = p;
984 int n = 0; /* buffer index */
985 int plen = 0; /* prompt length */
986 int col; /* output column cnt */
987 char c;
989 /* print prompt */
990 if (prompt) {
991 plen = strlen (prompt);
992 puts (prompt);
993 }
994 col = plen;
996 for (;;) {
997 #ifdef CONFIG_BOOT_RETRY_TIME
998 while (!tstc()) { /* while no incoming data */
999 if (retry_time >= 0 && get_ticks() > endtime)
1000 return (-2); /* timed out */
1001 WATCHDOG_RESET();
1002 }
1003 #endif
1004 WATCHDOG_RESET(); /* Trigger watchdog, if needed */
1006 #ifdef CONFIG_SHOW_ACTIVITY
1007 while (!tstc()) {
1008 extern void show_activity(int arg);
1009 show_activity(0);
1010 WATCHDOG_RESET();
1011 }
1012 #endif
1013 c = getc();
1015 /*
1016 * Special character handling
1017 */
1018 switch (c) {
1019 case '\r': /* Enter */
1020 case '\n':
1021 *p = '\0';
1022 puts ("\r\n");
1023 return (p - p_buf);
1025 case '\0': /* nul */
1026 continue;
1028 case 0x03: /* ^C - break */
1029 p_buf[0] = '\0'; /* discard input */
1030 return (-1);
1032 case 0x15: /* ^U - erase line */
1033 while (col > plen) {
1034 puts (erase_seq);
1035 --col;
1036 }
1037 p = p_buf;
1038 n = 0;
1039 continue;
1041 case 0x17: /* ^W - erase word */
1042 p=delete_char(p_buf, p, &col, &n, plen);
1043 while ((n > 0) && (*p != ' ')) {
1044 p=delete_char(p_buf, p, &col, &n, plen);
1045 }
1046 continue;
1048 case 0x08: /* ^H - backspace */
1049 case 0x7F: /* DEL - backspace */
1050 p=delete_char(p_buf, p, &col, &n, plen);
1051 continue;
1053 default:
1054 /*
1055 * Must be a normal character then
1056 */
1057 if (n < CONFIG_SYS_CBSIZE-2) {
1058 if (c == '\t') { /* expand TABs */
1059 #ifdef CONFIG_AUTO_COMPLETE
1060 /* if auto completion triggered just continue */
1061 *p = '\0';
1062 if (cmd_auto_complete(prompt, console_buffer, &n, &col)) {
1063 p = p_buf + n; /* reset */
1064 continue;
1065 }
1066 #endif
1067 puts (tab_seq+(col&07));
1068 col += 8 - (col&07);
1069 } else {
1070 ++col; /* echo input */
1071 putc (c);
1072 }
1073 *p++ = c;
1074 ++n;
1075 } else { /* Buffer full */
1076 putc ('\a');
1077 }
1078 }
1079 }
1080 #ifdef CONFIG_CMDLINE_EDITING
1081 }
1082 #endif
1083 }
1085 /****************************************************************************/
1087 static char * delete_char (char *buffer, char *p, int *colp, int *np, int plen)
1088 {
1089 char *s;
1091 if (*np == 0) {
1092 return (p);
1093 }
1095 if (*(--p) == '\t') { /* will retype the whole line */
1096 while (*colp > plen) {
1097 puts (erase_seq);
1098 (*colp)--;
1099 }
1100 for (s=buffer; s<p; ++s) {
1101 if (*s == '\t') {
1102 puts (tab_seq+((*colp) & 07));
1103 *colp += 8 - ((*colp) & 07);
1104 } else {
1105 ++(*colp);
1106 putc (*s);
1107 }
1108 }
1109 } else {
1110 puts (erase_seq);
1111 (*colp)--;
1112 }
1113 (*np)--;
1114 return (p);
1115 }
1117 /****************************************************************************/
1119 int parse_line (char *line, char *argv[])
1120 {
1121 int nargs = 0;
1123 #ifdef DEBUG_PARSER
1124 printf ("parse_line: \"%s\"\n", line);
1125 #endif
1126 while (nargs < CONFIG_SYS_MAXARGS) {
1128 /* skip any white space */
1129 while ((*line == ' ') || (*line == '\t')) {
1130 ++line;
1131 }
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 }
1141 argv[nargs++] = line; /* begin of argument string */
1143 /* find end of string */
1144 while (*line && (*line != ' ') && (*line != '\t')) {
1145 ++line;
1146 }
1148 if (*line == '\0') { /* end of line, no more args */
1149 argv[nargs] = NULL;
1150 #ifdef DEBUG_PARSER
1151 printf ("parse_line: nargs=%d\n", nargs);
1152 #endif
1153 return (nargs);
1154 }
1156 *line++ = '\0'; /* terminate current arg */
1157 }
1159 printf ("** Too many args (max. %d) **\n", CONFIG_SYS_MAXARGS);
1161 #ifdef DEBUG_PARSER
1162 printf ("parse_line: nargs=%d\n", nargs);
1163 #endif
1164 return (nargs);
1165 }
1167 /****************************************************************************/
1169 static void process_macros (const char *input, char *output)
1170 {
1171 char c, prev;
1172 const char *varname_start = NULL;
1173 int inputcnt = strlen (input);
1174 int outputcnt = CONFIG_SYS_CBSIZE;
1175 int state = 0; /* 0 = waiting for '$' */
1177 /* 1 = waiting for '(' or '{' */
1178 /* 2 = waiting for ')' or '}' */
1179 /* 3 = waiting for ''' */
1180 #ifdef DEBUG_PARSER
1181 char *output_start = output;
1183 printf ("[PROCESS_MACROS] INPUT len %d: \"%s\"\n", strlen (input),
1184 input);
1185 #endif
1187 prev = '\0'; /* previous character */
1189 while (inputcnt && outputcnt) {
1190 c = *input++;
1191 inputcnt--;
1193 if (state != 3) {
1194 /* remove one level of escape characters */
1195 if ((c == '\\') && (prev != '\\')) {
1196 if (inputcnt-- == 0)
1197 break;
1198 prev = c;
1199 c = *input++;
1200 }
1201 }
1203 switch (state) {
1204 case 0: /* Waiting for (unescaped) $ */
1205 if ((c == '\'') && (prev != '\\')) {
1206 state = 3;
1207 break;
1208 }
1209 if ((c == '$') && (prev != '\\')) {
1210 state++;
1211 } else {
1212 *(output++) = c;
1213 outputcnt--;
1214 }
1215 break;
1216 case 1: /* Waiting for ( */
1217 if (c == '(' || c == '{') {
1218 state++;
1219 varname_start = input;
1220 } else {
1221 state = 0;
1222 *(output++) = '$';
1223 outputcnt--;
1225 if (outputcnt) {
1226 *(output++) = c;
1227 outputcnt--;
1228 }
1229 }
1230 break;
1231 case 2: /* Waiting for ) */
1232 if (c == ')' || c == '}') {
1233 int i;
1234 char envname[CONFIG_SYS_CBSIZE], *envval;
1235 int envcnt = input - varname_start - 1; /* Varname # of chars */
1237 /* Get the varname */
1238 for (i = 0; i < envcnt; i++) {
1239 envname[i] = varname_start[i];
1240 }
1241 envname[i] = 0;
1243 /* Get its value */
1244 envval = getenv (envname);
1246 /* Copy into the line if it exists */
1247 if (envval != NULL)
1248 while ((*envval) && outputcnt) {
1249 *(output++) = *(envval++);
1250 outputcnt--;
1251 }
1252 /* Look for another '$' */
1253 state = 0;
1254 }
1255 break;
1256 case 3: /* Waiting for ' */
1257 if ((c == '\'') && (prev != '\\')) {
1258 state = 0;
1259 } else {
1260 *(output++) = c;
1261 outputcnt--;
1262 }
1263 break;
1264 }
1265 prev = c;
1266 }
1268 if (outputcnt)
1269 *output = 0;
1270 else
1271 *(output - 1) = 0;
1273 #ifdef DEBUG_PARSER
1274 printf ("[PROCESS_MACROS] OUTPUT len %d: \"%s\"\n",
1275 strlen (output_start), output_start);
1276 #endif
1277 }
1279 /****************************************************************************
1280 * returns:
1281 * 1 - command executed, repeatable
1282 * 0 - command executed but not repeatable, interrupted commands are
1283 * always considered not repeatable
1284 * -1 - not executed (unrecognized, bootd recursion or too many args)
1285 * (If cmd is NULL or "" or longer than CONFIG_SYS_CBSIZE-1 it is
1286 * considered unrecognized)
1287 *
1288 * WARNING:
1289 *
1290 * We must create a temporary copy of the command since the command we get
1291 * may be the result from getenv(), which returns a pointer directly to
1292 * the environment data, which may change magicly when the command we run
1293 * creates or modifies environment variables (like "bootp" does).
1294 */
1296 int run_command (const char *cmd, int flag)
1297 {
1298 cmd_tbl_t *cmdtp;
1299 char cmdbuf[CONFIG_SYS_CBSIZE]; /* working copy of cmd */
1300 char *token; /* start of token in cmdbuf */
1301 char *sep; /* end of token (separator) in cmdbuf */
1302 char finaltoken[CONFIG_SYS_CBSIZE];
1303 char *str = cmdbuf;
1304 char *argv[CONFIG_SYS_MAXARGS + 1]; /* NULL terminated */
1305 int argc, inquotes;
1306 int repeatable = 1;
1307 int rc = 0;
1309 #ifdef DEBUG_PARSER
1310 printf ("[RUN_COMMAND] cmd[%p]=\"", cmd);
1311 puts (cmd ? cmd : "NULL"); /* use puts - string may be loooong */
1312 puts ("\"\n");
1313 #endif
1315 clear_ctrlc(); /* forget any previous Control C */
1317 if (!cmd || !*cmd) {
1318 return -1; /* empty command */
1319 }
1321 if (strlen(cmd) >= CONFIG_SYS_CBSIZE) {
1322 puts ("## Command too long!\n");
1323 return -1;
1324 }
1326 strcpy (cmdbuf, cmd);
1328 /* Process separators and check for invalid
1329 * repeatable commands
1330 */
1332 #ifdef DEBUG_PARSER
1333 printf ("[PROCESS_SEPARATORS] %s\n", cmd);
1334 #endif
1335 while (*str) {
1337 /*
1338 * Find separator, or string end
1339 * Allow simple escape of ';' by writing "\;"
1340 */
1341 for (inquotes = 0, sep = str; *sep; sep++) {
1342 if ((*sep=='\'') &&
1343 (*(sep-1) != '\\'))
1344 inquotes=!inquotes;
1346 if (!inquotes &&
1347 (*sep == ';') && /* separator */
1348 ( sep != str) && /* past string start */
1349 (*(sep-1) != '\\')) /* and NOT escaped */
1350 break;
1351 }
1353 /*
1354 * Limit the token to data between separators
1355 */
1356 token = str;
1357 if (*sep) {
1358 str = sep + 1; /* start of command for next pass */
1359 *sep = '\0';
1360 }
1361 else
1362 str = sep; /* no more commands for next pass */
1363 #ifdef DEBUG_PARSER
1364 printf ("token: \"%s\"\n", token);
1365 #endif
1367 /* find macros in this token and replace them */
1368 process_macros (token, finaltoken);
1370 /* Extract arguments */
1371 if ((argc = parse_line (finaltoken, argv)) == 0) {
1372 rc = -1; /* no command at all */
1373 continue;
1374 }
1376 /* Look up command in command table */
1377 if ((cmdtp = find_cmd(argv[0])) == NULL) {
1378 printf ("Unknown command '%s' - try 'help'\n", argv[0]);
1379 rc = -1; /* give up after bad command */
1380 continue;
1381 }
1383 /* found - check max args */
1384 if (argc > cmdtp->maxargs) {
1385 cmd_usage(cmdtp);
1386 rc = -1;
1387 continue;
1388 }
1390 #if defined(CONFIG_CMD_BOOTD)
1391 /* avoid "bootd" recursion */
1392 if (cmdtp->cmd == do_bootd) {
1393 #ifdef DEBUG_PARSER
1394 printf ("[%s]\n", finaltoken);
1395 #endif
1396 if (flag & CMD_FLAG_BOOTD) {
1397 puts ("'bootd' recursion detected\n");
1398 rc = -1;
1399 continue;
1400 } else {
1401 flag |= CMD_FLAG_BOOTD;
1402 }
1403 }
1404 #endif
1406 /* OK - call function to do the command */
1407 if ((cmdtp->cmd) (cmdtp, flag, argc, argv) != 0) {
1408 rc = -1;
1409 }
1411 repeatable &= cmdtp->repeatable;
1413 /* Did the user stop this? */
1414 if (had_ctrlc ())
1415 return -1; /* if stopped then not repeatable */
1416 }
1418 return rc ? rc : repeatable;
1419 }
1421 /****************************************************************************/
1423 #if defined(CONFIG_CMD_RUN)
1424 int do_run (cmd_tbl_t * cmdtp, int flag, int argc, char *argv[])
1425 {
1426 int i;
1428 if (argc < 2) {
1429 cmd_usage(cmdtp);
1430 return 1;
1431 }
1433 for (i=1; i<argc; ++i) {
1434 char *arg;
1436 if ((arg = getenv (argv[i])) == NULL) {
1437 printf ("## Error: \"%s\" not defined\n", argv[i]);
1438 return 1;
1439 }
1440 #ifndef CONFIG_SYS_HUSH_PARSER
1441 if (run_command (arg, flag) == -1)
1442 return 1;
1443 #else
1444 if (parse_string_outer(arg,
1445 FLAG_PARSE_SEMICOLON | FLAG_EXIT_FROM_LOOP) != 0)
1446 return 1;
1447 #endif
1448 }
1449 return 0;
1450 }
1451 #endif