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