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