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 */ 27 28 /* #define DEBUG */ 29 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 36 37 #ifdef CONFIG_SYS_HUSH_PARSER 38 #include <hush.h> 39 #endif 40 41 #include <post.h> 42 43 #if defined(CONFIG_SILENT_CONSOLE) || defined(CONFIG_POST) || defined(CONFIG_CMDLINE_EDITING) 44 DECLARE_GLOBAL_DATA_PTR; 45 #endif 46 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"))); 52 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 56 57 extern int do_bootd (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[]); 58 59 #if defined(CONFIG_UPDATE_TFTP) 60 void update_tftp (void); 61 #endif /* CONFIG_UPDATE_TFTP */ 62 63 #define MAX_DELAY_STOP_STR 32 64 65 #if defined(CONFIG_BOOTDELAY) && (CONFIG_BOOTDELAY >= 0) 66 static int abortboot(int); 67 #endif 68 69 #undef DEBUG_PARSER 70 71 char console_buffer[CONFIG_SYS_CBSIZE + 1]; /* console I/O buffer */ 72 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 */ 76 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 81 82 #define endtick(seconds) (get_ticks() + (uint64_t)(seconds) * get_tbclk()) 83 84 #ifndef CONFIG_BOOT_RETRY_MIN 85 #define CONFIG_BOOT_RETRY_MIN CONFIG_BOOT_RETRY_TIME 86 #endif 87 88 #ifdef CONFIG_MODEM_SUPPORT 89 int do_mdm_init = 0; 90 extern void mdm_init(void); /* defined in board.c */ 91 #endif 92 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 }; 115 116 char presskey [MAX_DELAY_STOP_STR]; 117 u_int presskey_len = 0; 118 u_int presskey_max = 0; 119 u_int i; 120 121 # ifdef CONFIG_AUTOBOOT_PROMPT 122 printf(CONFIG_AUTOBOOT_PROMPT); 123 # endif 124 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 141 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; 147 148 presskey_max = presskey_max > delaykey[i].len ? 149 presskey_max : delaykey[i].len; 150 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 } 157 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]; 169 170 presskey [i] = getc(); 171 } 172 } 173 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 184 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); 194 195 # if DEBUG_BOOTKEYS 196 if (!abort) 197 puts("key timeout\n"); 198 # endif 199 200 #ifdef CONFIG_SILENT_CONSOLE 201 if (abort) 202 gd->flags &= ~GD_FLG_SILENT; 203 #endif 204 205 return abort; 206 } 207 208 # else /* !defined(CONFIG_AUTOBOOT_KEYED) */ 209 210 #ifdef CONFIG_MENUKEY 211 static int menukey = 0; 212 #endif 213 214 static __inline__ int abortboot(int bootdelay) 215 { 216 int abort = 0; 217 218 #ifdef CONFIG_MENUPROMPT 219 printf(CONFIG_MENUPROMPT); 220 #else 221 printf("Hit any key to stop autoboot: %2d ", bootdelay); 222 #endif 223 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 237 238 while ((bootdelay > 0) && (!abort)) { 239 int i; 240 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 } 256 257 printf("\b\b\b%2d ", bootdelay); 258 } 259 260 putc('\n'); 261 262 #ifdef CONFIG_SILENT_CONSOLE 263 if (abort) 264 gd->flags &= ~GD_FLG_SILENT; 265 #endif 266 267 return abort; 268 } 269 # endif /* CONFIG_AUTOBOOT_KEYED */ 270 #endif /* CONFIG_BOOTDELAY >= 0 */ 271 272 /****************************************************************************/ 273 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 282 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 */ 296 297 #if defined(CONFIG_VFD) && defined(VFD_TEST_LOGO) 298 ulong bmp = 0; /* default bitmap */ 299 extern int trab_vfd (ulong bitmap); 300 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 */ 307 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 */ 317 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 */ 328 329 #ifdef CONFIG_VERSION_VARIABLE 330 { 331 extern char version_string[]; 332 333 setenv ("ver", version_string); /* set version variable */ 334 } 335 #endif /* CONFIG_VERSION_VARIABLE */ 336 337 #ifdef CONFIG_SYS_HUSH_PARSER 338 u_boot_hush_start (); 339 #endif 340 341 #if defined(CONFIG_HUSH_INIT_VAR) 342 hush_init_var (); 343 #endif 344 345 #ifdef CONFIG_AUTO_COMPLETE 346 install_auto_complete(); 347 #endif 348 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 354 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 361 362 # ifdef CONFIG_AUTOBOOT_KEYED 363 disable_ctrlc(prev); /* restore Control C checking */ 364 # endif 365 } 366 #endif /* CONFIG_PREBOOT */ 367 368 #if defined(CONFIG_UPDATE_TFTP) 369 update_tftp (); 370 #endif /* CONFIG_UPDATE_TFTP */ 371 372 #if defined(CONFIG_BOOTDELAY) && (CONFIG_BOOTDELAY >= 0) 373 s = getenv ("bootdelay"); 374 bootdelay = s ? (int)simple_strtol(s, NULL, 10) : CONFIG_BOOTDELAY; 375 376 debug ("### main_loop entered: bootdelay=%d\n\n", bootdelay); 377 378 # ifdef CONFIG_BOOT_RETRY_TIME 379 init_cmd_timeout (); 380 # endif /* CONFIG_BOOT_RETRY_TIME */ 381 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"); 397 398 debug ("### main_loop: bootcmd=\"%s\"\n", s ? s : "<UNDEFINED>"); 399 400 if (bootdelay >= 0 && s && !abortboot (bootdelay)) { 401 # ifdef CONFIG_AUTOBOOT_KEYED 402 int prev = disable_ctrlc(1); /* disable Control C checking */ 403 # endif 404 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 411 412 # ifdef CONFIG_AUTOBOOT_KEYED 413 disable_ctrlc(prev); /* restore Control C checking */ 414 # endif 415 } 416 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 */ 431 432 #ifdef CONFIG_AMIGAONEG3SE 433 { 434 extern void video_banner(void); 435 video_banner(); 436 } 437 #endif 438 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); 457 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 476 477 if (len == -1) 478 puts ("<INTERRUPT>\n"); 479 else 480 rc = run_command (lastcommand, flag); 481 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 } 489 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"); 497 498 if (s != NULL) 499 retry_time = (int)simple_strtol(s, NULL, 10); 500 else 501 retry_time = CONFIG_BOOT_RETRY_TIME; 502 503 if (retry_time >= 0 && retry_time < CONFIG_BOOT_RETRY_MIN) 504 retry_time = CONFIG_BOOT_RETRY_MIN; 505 } 506 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 515 516 #ifdef CONFIG_CMDLINE_EDITING 517 518 /* 519 * cmdline-editing related codes from vivi. 520 * Author: Janghoon Lyu <nandy@mizi.com> 521 */ 522 523 #define putnstr(str,n) do { \ 524 printf ("%.*s", (int)n, str); \ 525 } while (0) 526 527 #define CTL_CH(c) ((c) - 'a' + 1) 528 529 #define MAX_CMDBUF_SIZE CONFIG_SYS_CBSIZE 530 531 #define CTL_BACKSPACE ('\b') 532 #define DEL ((char)255) 533 #define DEL7 ((char)127) 534 #define CREAD_HIST_CHAR ('!') 535 536 #define getcmd_putch(ch) putc(ch) 537 #define getcmd_getch() getc() 538 #define getcmd_cbeep() getcmd_putch('\a') 539 540 #define HIST_MAX 20 541 #define HIST_SIZE MAX_CMDBUF_SIZE 542 543 static int hist_max = 0; 544 static int hist_add_idx = 0; 545 static int hist_cur = -1; 546 unsigned hist_num = 0; 547 548 char* hist_list[HIST_MAX]; 549 char hist_lines[HIST_MAX][HIST_SIZE + 1]; /* Save room for NULL */ 550 551 #define add_idx_minus_one() ((hist_add_idx == 0) ? hist_max : hist_add_idx-1) 552 553 static void hist_init(void) 554 { 555 int i; 556 557 hist_max = 0; 558 hist_add_idx = 0; 559 hist_cur = -1; 560 hist_num = 0; 561 562 for (i = 0; i < HIST_MAX; i++) { 563 hist_list[i] = hist_lines[i]; 564 hist_list[i][0] = '\0'; 565 } 566 } 567 568 static void cread_add_to_hist(char *line) 569 { 570 strcpy(hist_list[hist_add_idx], line); 571 572 if (++hist_add_idx >= HIST_MAX) 573 hist_add_idx = 0; 574 575 if (hist_add_idx > hist_max) 576 hist_max = hist_add_idx; 577 578 hist_num++; 579 } 580 581 static char* hist_prev(void) 582 { 583 char *ret; 584 int old_cur; 585 586 if (hist_cur < 0) 587 return NULL; 588 589 old_cur = hist_cur; 590 if (--hist_cur < 0) 591 hist_cur = hist_max; 592 593 if (hist_cur == hist_add_idx) { 594 hist_cur = old_cur; 595 ret = NULL; 596 } else 597 ret = hist_list[hist_cur]; 598 599 return (ret); 600 } 601 602 static char* hist_next(void) 603 { 604 char *ret; 605 606 if (hist_cur < 0) 607 return NULL; 608 609 if (hist_cur == hist_add_idx) 610 return NULL; 611 612 if (++hist_cur > hist_max) 613 hist_cur = 0; 614 615 if (hist_cur == hist_add_idx) { 616 ret = ""; 617 } else 618 ret = hist_list[hist_cur]; 619 620 return (ret); 621 } 622 623 #ifndef CONFIG_CMDLINE_EDITING 624 static void cread_print_hist_list(void) 625 { 626 int i; 627 unsigned long n; 628 629 n = hist_num - hist_max; 630 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 */ 643 644 #define BEGINNING_OF_LINE() { \ 645 while (num) { \ 646 getcmd_putch(CTL_BACKSPACE); \ 647 num--; \ 648 } \ 649 } 650 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 } 661 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 } 669 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; 674 675 /* room ??? */ 676 if (insert || *num == *eol_num) { 677 if (*eol_num > len - 1) { 678 getcmd_cbeep(); 679 return; 680 } 681 (*eol_num)++; 682 } 683 684 if (insert) { 685 wlen = *eol_num - *num; 686 if (wlen > 1) { 687 memmove(&buf[*num+1], &buf[*num], wlen-1); 688 } 689 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 } 704 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 } 713 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); 724 725 if (init_len) 726 cread_add_str(buf, init_len, 1, &num, &eol_num, buf, *len); 727 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 736 737 ichar = getcmd_getch(); 738 739 if ((ichar == '\n') || (ichar == '\r')) { 740 putc('\n'); 741 break; 742 } 743 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 } 759 760 switch (ichar) { 761 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 } 790 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; 801 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 } 827 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; 869 870 esc_len = 0; 871 872 if (ichar == CTL_CH('p')) 873 hline = hist_prev(); 874 else 875 hline = hist_next(); 876 877 if (!hline) { 878 getcmd_cbeep(); 879 continue; 880 } 881 882 /* nuke the current line */ 883 /* first, go home */ 884 BEGINNING_OF_LINE(); 885 886 /* erase to end of line */ 887 ERASE_TO_EOL(); 888 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; 898 899 /* do not autocomplete when in the middle */ 900 if (num < eol_num) { 901 getcmd_cbeep(); 902 break; 903 } 904 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 */ 923 924 if (buf[0] && buf[0] != CREAD_HIST_CHAR) 925 cread_add_to_hist(buf); 926 hist_cur = hist_add_idx; 927 928 return 0; 929 } 930 931 #endif /* CONFIG_CMDLINE_EDITING */ 932 933 /****************************************************************************/ 934 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'; 950 951 return readline_into_buffer(prompt, console_buffer); 952 } 953 954 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; 962 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 } 974 975 if (prompt) 976 puts (prompt); 977 978 rc = cread_line(prompt, p, &len); 979 return rc < 0 ? rc : len; 980 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; 988 989 /* print prompt */ 990 if (prompt) { 991 plen = strlen (prompt); 992 puts (prompt); 993 } 994 col = plen; 995 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 */ 1005 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(); 1014 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); 1024 1025 case '\0': /* nul */ 1026 continue; 1027 1028 case 0x03: /* ^C - break */ 1029 p_buf[0] = '\0'; /* discard input */ 1030 return (-1); 1031 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; 1040 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; 1047 1048 case 0x08: /* ^H - backspace */ 1049 case 0x7F: /* DEL - backspace */ 1050 p=delete_char(p_buf, p, &col, &n, plen); 1051 continue; 1052 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 } 1084 1085 /****************************************************************************/ 1086 1087 static char * delete_char (char *buffer, char *p, int *colp, int *np, int plen) 1088 { 1089 char *s; 1090 1091 if (*np == 0) { 1092 return (p); 1093 } 1094 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 } 1116 1117 /****************************************************************************/ 1118 1119 int parse_line (char *line, char *argv[]) 1120 { 1121 int nargs = 0; 1122 1123 #ifdef DEBUG_PARSER 1124 printf ("parse_line: \"%s\"\n", line); 1125 #endif 1126 while (nargs < CONFIG_SYS_MAXARGS) { 1127 1128 /* skip any white space */ 1129 while ((*line == ' ') || (*line == '\t')) { 1130 ++line; 1131 } 1132 1133 if (*line == '\0') { /* end of line, no more args */ 1134 argv[nargs] = NULL; 1135 #ifdef DEBUG_PARSER 1136 printf ("parse_line: nargs=%d\n", nargs); 1137 #endif 1138 return (nargs); 1139 } 1140 1141 argv[nargs++] = line; /* begin of argument string */ 1142 1143 /* find end of string */ 1144 while (*line && (*line != ' ') && (*line != '\t')) { 1145 ++line; 1146 } 1147 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 } 1155 1156 *line++ = '\0'; /* terminate current arg */ 1157 } 1158 1159 printf ("** Too many args (max. %d) **\n", CONFIG_SYS_MAXARGS); 1160 1161 #ifdef DEBUG_PARSER 1162 printf ("parse_line: nargs=%d\n", nargs); 1163 #endif 1164 return (nargs); 1165 } 1166 1167 /****************************************************************************/ 1168 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 '$' */ 1176 1177 /* 1 = waiting for '(' or '{' */ 1178 /* 2 = waiting for ')' or '}' */ 1179 /* 3 = waiting for ''' */ 1180 #ifdef DEBUG_PARSER 1181 char *output_start = output; 1182 1183 printf ("[PROCESS_MACROS] INPUT len %d: \"%s\"\n", strlen (input), 1184 input); 1185 #endif 1186 1187 prev = '\0'; /* previous character */ 1188 1189 while (inputcnt && outputcnt) { 1190 c = *input++; 1191 inputcnt--; 1192 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 } 1202 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--; 1224 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 */ 1236 1237 /* Get the varname */ 1238 for (i = 0; i < envcnt; i++) { 1239 envname[i] = varname_start[i]; 1240 } 1241 envname[i] = 0; 1242 1243 /* Get its value */ 1244 envval = getenv (envname); 1245 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 } 1267 1268 if (outputcnt) 1269 *output = 0; 1270 else 1271 *(output - 1) = 0; 1272 1273 #ifdef DEBUG_PARSER 1274 printf ("[PROCESS_MACROS] OUTPUT len %d: \"%s\"\n", 1275 strlen (output_start), output_start); 1276 #endif 1277 } 1278 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 */ 1295 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; 1308 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 1314 1315 clear_ctrlc(); /* forget any previous Control C */ 1316 1317 if (!cmd || !*cmd) { 1318 return -1; /* empty command */ 1319 } 1320 1321 if (strlen(cmd) >= CONFIG_SYS_CBSIZE) { 1322 puts ("## Command too long!\n"); 1323 return -1; 1324 } 1325 1326 strcpy (cmdbuf, cmd); 1327 1328 /* Process separators and check for invalid 1329 * repeatable commands 1330 */ 1331 1332 #ifdef DEBUG_PARSER 1333 printf ("[PROCESS_SEPARATORS] %s\n", cmd); 1334 #endif 1335 while (*str) { 1336 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; 1345 1346 if (!inquotes && 1347 (*sep == ';') && /* separator */ 1348 ( sep != str) && /* past string start */ 1349 (*(sep-1) != '\\')) /* and NOT escaped */ 1350 break; 1351 } 1352 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 1366 1367 /* find macros in this token and replace them */ 1368 process_macros (token, finaltoken); 1369 1370 /* Extract arguments */ 1371 if ((argc = parse_line (finaltoken, argv)) == 0) { 1372 rc = -1; /* no command at all */ 1373 continue; 1374 } 1375 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 } 1382 1383 /* found - check max args */ 1384 if (argc > cmdtp->maxargs) { 1385 cmd_usage(cmdtp); 1386 rc = -1; 1387 continue; 1388 } 1389 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 1405 1406 /* OK - call function to do the command */ 1407 if ((cmdtp->cmd) (cmdtp, flag, argc, argv) != 0) { 1408 rc = -1; 1409 } 1410 1411 repeatable &= cmdtp->repeatable; 1412 1413 /* Did the user stop this? */ 1414 if (had_ctrlc ()) 1415 return -1; /* if stopped then not repeatable */ 1416 } 1417 1418 return rc ? rc : repeatable; 1419 } 1420 1421 /****************************************************************************/ 1422 1423 #if defined(CONFIG_CMD_RUN) 1424 int do_run (cmd_tbl_t * cmdtp, int flag, int argc, char *argv[]) 1425 { 1426 int i; 1427 1428 if (argc < 2) { 1429 cmd_usage(cmdtp); 1430 return 1; 1431 } 1432 1433 for (i=1; i<argc; ++i) { 1434 char *arg; 1435 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 1452