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