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