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