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