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