1 /* 2 * (C) Copyright 2000 3 * Wolfgang Denk, DENX Software Engineering, wd@denx.de. 4 * 5 * See file CREDITS for list of people who contributed to this 6 * project. 7 * 8 * This program is free software; you can redistribute it and/or 9 * modify it under the terms of the GNU General Public License as 10 * published by the Free Software Foundation; either version 2 of 11 * the License, or (at your option) any later version. 12 * 13 * This program is distributed in the hope that it will be useful, 14 * but WITHOUT ANY WARRANTY; without even the implied warranty of 15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 16 * GNU General Public License for more details. 17 * 18 * You should have received a copy of the GNU General Public License 19 * along with this program; if not, write to the Free Software 20 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, 21 * MA 02111-1307 USA 22 */ 23 24 /* #define DEBUG */ 25 26 #include <common.h> 27 #include <watchdog.h> 28 #include <command.h> 29 #include <malloc.h> 30 31 #ifdef CFG_HUSH_PARSER 32 #include <hush.h> 33 #endif 34 35 #include <post.h> 36 37 #if defined(CONFIG_BOOT_RETRY_TIME) && defined(CONFIG_RESET_TO_RETRY) 38 extern int do_reset (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[]); /* for do_reset() prototype */ 39 #endif 40 41 extern int do_bootd (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[]); 42 43 44 #define MAX_DELAY_STOP_STR 32 45 46 static char * delete_char (char *buffer, char *p, int *colp, int *np, int plen); 47 static int parse_line (char *, char *[]); 48 #if defined(CONFIG_BOOTDELAY) && (CONFIG_BOOTDELAY >= 0) 49 static int abortboot(int); 50 #endif 51 52 #undef DEBUG_PARSER 53 54 char console_buffer[CFG_CBSIZE]; /* console I/O buffer */ 55 56 static char erase_seq[] = "\b \b"; /* erase sequence */ 57 static char tab_seq[] = " "; /* used to expand TABs */ 58 59 #ifdef CONFIG_BOOT_RETRY_TIME 60 static uint64_t endtime = 0; /* must be set, default is instant timeout */ 61 static int retry_time = -1; /* -1 so can call readline before main_loop */ 62 #endif 63 64 #define endtick(seconds) (get_ticks() + (uint64_t)(seconds) * get_tbclk()) 65 66 #ifndef CONFIG_BOOT_RETRY_MIN 67 #define CONFIG_BOOT_RETRY_MIN CONFIG_BOOT_RETRY_TIME 68 #endif 69 70 #ifdef CONFIG_MODEM_SUPPORT 71 int do_mdm_init = 0; 72 extern void mdm_init(void); /* defined in board.c */ 73 #endif 74 75 /*************************************************************************** 76 * Watch for 'delay' seconds for autoboot stop or autoboot delay string. 77 * returns: 0 - no key string, allow autoboot 78 * 1 - got key string, abort 79 */ 80 #if defined(CONFIG_BOOTDELAY) && (CONFIG_BOOTDELAY >= 0) 81 # if defined(CONFIG_AUTOBOOT_KEYED) 82 static __inline__ int abortboot(int bootdelay) 83 { 84 int abort = 0; 85 uint64_t etime = endtick(bootdelay); 86 struct 87 { 88 char* str; 89 u_int len; 90 int retry; 91 } 92 delaykey [] = 93 { 94 { str: getenv ("bootdelaykey"), retry: 1 }, 95 { str: getenv ("bootdelaykey2"), retry: 1 }, 96 { str: getenv ("bootstopkey"), retry: 0 }, 97 { str: getenv ("bootstopkey2"), retry: 0 }, 98 }; 99 100 char presskey [MAX_DELAY_STOP_STR]; 101 u_int presskey_len = 0; 102 u_int presskey_max = 0; 103 u_int i; 104 105 #ifdef CONFIG_SILENT_CONSOLE 106 { 107 DECLARE_GLOBAL_DATA_PTR; 108 109 if (gd->flags & GD_FLG_SILENT) { 110 /* Restore serial console */ 111 console_assign (stdout, "serial"); 112 console_assign (stderr, "serial"); 113 } 114 } 115 #endif 116 117 # ifdef CONFIG_AUTOBOOT_PROMPT 118 printf (CONFIG_AUTOBOOT_PROMPT, bootdelay); 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 while (!abort && get_ticks() <= etime) { 158 for (i = 0; i < sizeof(delaykey) / sizeof(delaykey[0]); i ++) { 159 if (delaykey[i].len > 0 && 160 presskey_len >= delaykey[i].len && 161 memcmp (presskey + presskey_len - delaykey[i].len, 162 delaykey[i].str, 163 delaykey[i].len) == 0) { 164 # if DEBUG_BOOTKEYS 165 printf("got %skey\n", 166 delaykey[i].retry ? "delay" : "stop"); 167 # endif 168 169 # ifdef CONFIG_BOOT_RETRY_TIME 170 /* don't retry auto boot */ 171 if (! delaykey[i].retry) 172 retry_time = -1; 173 # endif 174 abort = 1; 175 } 176 } 177 178 if (tstc()) { 179 if (presskey_len < presskey_max) { 180 presskey [presskey_len ++] = getc(); 181 } 182 else { 183 for (i = 0; i < presskey_max - 1; i ++) 184 presskey [i] = presskey [i + 1]; 185 186 presskey [i] = getc(); 187 } 188 } 189 } 190 # if DEBUG_BOOTKEYS 191 if (!abort) 192 puts ("key timeout\n"); 193 # endif 194 195 #ifdef CONFIG_SILENT_CONSOLE 196 { 197 DECLARE_GLOBAL_DATA_PTR; 198 199 if (abort) { 200 /* permanently enable normal console output */ 201 gd->flags &= ~(GD_FLG_SILENT); 202 } else if (gd->flags & GD_FLG_SILENT) { 203 /* Restore silent console */ 204 console_assign (stdout, "nulldev"); 205 console_assign (stderr, "nulldev"); 206 } 207 } 208 #endif 209 210 return abort; 211 } 212 213 # else /* !defined(CONFIG_AUTOBOOT_KEYED) */ 214 215 #ifdef CONFIG_MENUKEY 216 static int menukey = 0; 217 #endif 218 219 static __inline__ int abortboot(int bootdelay) 220 { 221 int abort = 0; 222 223 #ifdef CONFIG_SILENT_CONSOLE 224 { 225 DECLARE_GLOBAL_DATA_PTR; 226 227 if (gd->flags & GD_FLG_SILENT) { 228 /* Restore serial console */ 229 console_assign (stdout, "serial"); 230 console_assign (stderr, "serial"); 231 } 232 } 233 #endif 234 235 #ifdef CONFIG_MENUPROMPT 236 printf(CONFIG_MENUPROMPT, bootdelay); 237 #else 238 printf("Hit any key to stop autoboot: %2d ", bootdelay); 239 #endif 240 241 #if defined CONFIG_ZERO_BOOTDELAY_CHECK 242 /* 243 * Check if key already pressed 244 * Don't check if bootdelay < 0 245 */ 246 if (bootdelay >= 0) { 247 if (tstc()) { /* we got a key press */ 248 (void) getc(); /* consume input */ 249 puts ("\b\b\b 0"); 250 abort = 1; /* don't auto boot */ 251 } 252 } 253 #endif 254 255 while ((bootdelay > 0) && (!abort)) { 256 int i; 257 258 --bootdelay; 259 /* delay 100 * 10ms */ 260 for (i=0; !abort && i<100; ++i) { 261 if (tstc()) { /* we got a key press */ 262 abort = 1; /* don't auto boot */ 263 bootdelay = 0; /* no more delay */ 264 # ifdef CONFIG_MENUKEY 265 menukey = getc(); 266 # else 267 (void) getc(); /* consume input */ 268 # endif 269 break; 270 } 271 udelay (10000); 272 } 273 274 printf ("\b\b\b%2d ", bootdelay); 275 } 276 277 putc ('\n'); 278 279 #ifdef CONFIG_SILENT_CONSOLE 280 { 281 DECLARE_GLOBAL_DATA_PTR; 282 283 if (abort) { 284 /* permanently enable normal console output */ 285 gd->flags &= ~(GD_FLG_SILENT); 286 } else if (gd->flags & GD_FLG_SILENT) { 287 /* Restore silent console */ 288 console_assign (stdout, "nulldev"); 289 console_assign (stderr, "nulldev"); 290 } 291 } 292 #endif 293 294 return abort; 295 } 296 # endif /* CONFIG_AUTOBOOT_KEYED */ 297 #endif /* CONFIG_BOOTDELAY >= 0 */ 298 299 /****************************************************************************/ 300 301 void main_loop (void) 302 { 303 #ifndef CFG_HUSH_PARSER 304 static char lastcommand[CFG_CBSIZE] = { 0, }; 305 int len; 306 int rc = 1; 307 int flag; 308 #endif 309 310 #if defined(CONFIG_BOOTDELAY) && (CONFIG_BOOTDELAY >= 0) 311 char *s; 312 int bootdelay; 313 #endif 314 #ifdef CONFIG_PREBOOT 315 char *p; 316 #endif 317 #ifdef CONFIG_BOOTCOUNT_LIMIT 318 unsigned long bootcount = 0; 319 unsigned long bootlimit = 0; 320 char *bcs; 321 char bcs_set[16]; 322 #endif /* CONFIG_BOOTCOUNT_LIMIT */ 323 324 #if defined(CONFIG_VFD) && defined(VFD_TEST_LOGO) 325 ulong bmp = 0; /* default bitmap */ 326 extern int trab_vfd (ulong bitmap); 327 328 #ifdef CONFIG_MODEM_SUPPORT 329 if (do_mdm_init) 330 bmp = 1; /* alternate bitmap */ 331 #endif 332 trab_vfd (bmp); 333 #endif /* CONFIG_VFD && VFD_TEST_LOGO */ 334 335 #ifdef CONFIG_BOOTCOUNT_LIMIT 336 bootcount = bootcount_load(); 337 bootcount++; 338 bootcount_store (bootcount); 339 sprintf (bcs_set, "%lu", bootcount); 340 setenv ("bootcount", bcs_set); 341 bcs = getenv ("bootlimit"); 342 bootlimit = bcs ? simple_strtoul (bcs, NULL, 10) : 0; 343 #endif /* CONFIG_BOOTCOUNT_LIMIT */ 344 345 #ifdef CONFIG_MODEM_SUPPORT 346 debug ("DEBUG: main_loop: do_mdm_init=%d\n", do_mdm_init); 347 if (do_mdm_init) { 348 uchar *str = strdup(getenv("mdm_cmd")); 349 setenv ("preboot", str); /* set or delete definition */ 350 if (str != NULL) 351 free (str); 352 mdm_init(); /* wait for modem connection */ 353 } 354 #endif /* CONFIG_MODEM_SUPPORT */ 355 356 #ifdef CONFIG_VERSION_VARIABLE 357 { 358 extern char version_string[]; 359 360 setenv ("ver", version_string); /* set version variable */ 361 } 362 #endif /* CONFIG_VERSION_VARIABLE */ 363 364 #ifdef CFG_HUSH_PARSER 365 u_boot_hush_start (); 366 #endif 367 368 #ifdef CONFIG_AUTO_COMPLETE 369 install_auto_complete(); 370 #endif 371 372 #ifdef CONFIG_PREBOOT 373 if ((p = getenv ("preboot")) != NULL) { 374 # ifdef CONFIG_AUTOBOOT_KEYED 375 int prev = disable_ctrlc(1); /* disable Control C checking */ 376 # endif 377 378 # ifndef CFG_HUSH_PARSER 379 run_command (p, 0); 380 # else 381 parse_string_outer(p, FLAG_PARSE_SEMICOLON | 382 FLAG_EXIT_FROM_LOOP); 383 # endif 384 385 # ifdef CONFIG_AUTOBOOT_KEYED 386 disable_ctrlc(prev); /* restore Control C checking */ 387 # endif 388 } 389 #endif /* CONFIG_PREBOOT */ 390 391 #if defined(CONFIG_BOOTDELAY) && (CONFIG_BOOTDELAY >= 0) 392 s = getenv ("bootdelay"); 393 bootdelay = s ? (int)simple_strtol(s, NULL, 10) : CONFIG_BOOTDELAY; 394 395 debug ("### main_loop entered: bootdelay=%d\n\n", bootdelay); 396 397 # ifdef CONFIG_BOOT_RETRY_TIME 398 init_cmd_timeout (); 399 # endif /* CONFIG_BOOT_RETRY_TIME */ 400 401 #ifdef CONFIG_BOOTCOUNT_LIMIT 402 if (bootlimit && (bootcount > bootlimit)) { 403 printf ("Warning: Bootlimit (%u) exceeded. Using altbootcmd.\n", 404 (unsigned)bootlimit); 405 s = getenv ("altbootcmd"); 406 } 407 else 408 #endif /* CONFIG_BOOTCOUNT_LIMIT */ 409 s = getenv ("bootcmd"); 410 411 debug ("### main_loop: bootcmd=\"%s\"\n", s ? s : "<UNDEFINED>"); 412 413 if (bootdelay >= 0 && s && !abortboot (bootdelay)) { 414 # ifdef CONFIG_AUTOBOOT_KEYED 415 int prev = disable_ctrlc(1); /* disable Control C checking */ 416 # endif 417 418 # ifndef CFG_HUSH_PARSER 419 run_command (s, 0); 420 # else 421 parse_string_outer(s, FLAG_PARSE_SEMICOLON | 422 FLAG_EXIT_FROM_LOOP); 423 # endif 424 425 # ifdef CONFIG_AUTOBOOT_KEYED 426 disable_ctrlc(prev); /* restore Control C checking */ 427 # endif 428 } 429 430 # ifdef CONFIG_MENUKEY 431 if (menukey == CONFIG_MENUKEY) { 432 s = getenv("menucmd"); 433 if (s) { 434 # ifndef CFG_HUSH_PARSER 435 run_command (s, bd, 0); 436 # else 437 parse_string_outer(s, FLAG_PARSE_SEMICOLON | 438 FLAG_EXIT_FROM_LOOP); 439 # endif 440 } 441 } 442 #endif /* CONFIG_MENUKEY */ 443 #endif /* CONFIG_BOOTDELAY */ 444 445 #ifdef CONFIG_AMIGAONEG3SE 446 { 447 extern void video_banner(void); 448 video_banner(); 449 } 450 #endif 451 452 /* 453 * Main Loop for Monitor Command Processing 454 */ 455 #ifdef CFG_HUSH_PARSER 456 parse_file_outer(); 457 /* This point is never reached */ 458 for (;;); 459 #else 460 for (;;) { 461 #ifdef CONFIG_BOOT_RETRY_TIME 462 if (rc >= 0) { 463 /* Saw enough of a valid command to 464 * restart the timeout. 465 */ 466 reset_cmd_timeout(); 467 } 468 #endif 469 len = readline (CFG_PROMPT); 470 471 flag = 0; /* assume no special flags for now */ 472 if (len > 0) 473 strcpy (lastcommand, console_buffer); 474 else if (len == 0) 475 flag |= CMD_FLAG_REPEAT; 476 #ifdef CONFIG_BOOT_RETRY_TIME 477 else if (len == -2) { 478 /* -2 means timed out, retry autoboot 479 */ 480 puts ("\nTimed out waiting for command\n"); 481 # ifdef CONFIG_RESET_TO_RETRY 482 /* Reinit board to run initialization code again */ 483 do_reset (NULL, 0, 0, NULL); 484 # else 485 return; /* retry autoboot */ 486 # endif 487 } 488 #endif 489 490 if (len == -1) 491 puts ("<INTERRUPT>\n"); 492 else 493 rc = run_command (lastcommand, flag); 494 495 if (rc <= 0) { 496 /* invalid command or not repeatable, forget it */ 497 lastcommand[0] = 0; 498 } 499 } 500 #endif /*CFG_HUSH_PARSER*/ 501 } 502 503 #ifdef CONFIG_BOOT_RETRY_TIME 504 /*************************************************************************** 505 * initialise command line timeout 506 */ 507 void init_cmd_timeout(void) 508 { 509 char *s = getenv ("bootretry"); 510 511 if (s != NULL) 512 retry_time = (int)simple_strtol(s, NULL, 10); 513 else 514 retry_time = CONFIG_BOOT_RETRY_TIME; 515 516 if (retry_time >= 0 && retry_time < CONFIG_BOOT_RETRY_MIN) 517 retry_time = CONFIG_BOOT_RETRY_MIN; 518 } 519 520 /*************************************************************************** 521 * reset command line timeout to retry_time seconds 522 */ 523 void reset_cmd_timeout(void) 524 { 525 endtime = endtick(retry_time); 526 } 527 #endif 528 529 /****************************************************************************/ 530 531 /* 532 * Prompt for input and read a line. 533 * If CONFIG_BOOT_RETRY_TIME is defined and retry_time >= 0, 534 * time out when time goes past endtime (timebase time in ticks). 535 * Return: number of read characters 536 * -1 if break 537 * -2 if timed out 538 */ 539 int readline (const char *const prompt) 540 { 541 char *p = console_buffer; 542 int n = 0; /* buffer index */ 543 int plen = 0; /* prompt length */ 544 int col; /* output column cnt */ 545 char c; 546 547 /* print prompt */ 548 if (prompt) { 549 plen = strlen (prompt); 550 puts (prompt); 551 } 552 col = plen; 553 554 for (;;) { 555 #ifdef CONFIG_BOOT_RETRY_TIME 556 while (!tstc()) { /* while no incoming data */ 557 if (retry_time >= 0 && get_ticks() > endtime) 558 return (-2); /* timed out */ 559 } 560 #endif 561 WATCHDOG_RESET(); /* Trigger watchdog, if needed */ 562 563 #ifdef CONFIG_SHOW_ACTIVITY 564 while (!tstc()) { 565 extern void show_activity(int arg); 566 show_activity(0); 567 } 568 #endif 569 c = getc(); 570 571 /* 572 * Special character handling 573 */ 574 switch (c) { 575 case '\r': /* Enter */ 576 case '\n': 577 *p = '\0'; 578 puts ("\r\n"); 579 return (p - console_buffer); 580 581 case '\0': /* nul */ 582 continue; 583 584 case 0x03: /* ^C - break */ 585 console_buffer[0] = '\0'; /* discard input */ 586 return (-1); 587 588 case 0x15: /* ^U - erase line */ 589 while (col > plen) { 590 puts (erase_seq); 591 --col; 592 } 593 p = console_buffer; 594 n = 0; 595 continue; 596 597 case 0x17: /* ^W - erase word */ 598 p=delete_char(console_buffer, p, &col, &n, plen); 599 while ((n > 0) && (*p != ' ')) { 600 p=delete_char(console_buffer, p, &col, &n, plen); 601 } 602 continue; 603 604 case 0x08: /* ^H - backspace */ 605 case 0x7F: /* DEL - backspace */ 606 p=delete_char(console_buffer, p, &col, &n, plen); 607 continue; 608 609 default: 610 /* 611 * Must be a normal character then 612 */ 613 if (n < CFG_CBSIZE-2) { 614 if (c == '\t') { /* expand TABs */ 615 #ifdef CONFIG_AUTO_COMPLETE 616 /* if auto completion triggered just continue */ 617 *p = '\0'; 618 if (cmd_auto_complete(prompt, console_buffer, &n, &col)) { 619 p = console_buffer + n; /* reset */ 620 continue; 621 } 622 #endif 623 puts (tab_seq+(col&07)); 624 col += 8 - (col&07); 625 } else { 626 ++col; /* echo input */ 627 putc (c); 628 } 629 *p++ = c; 630 ++n; 631 } else { /* Buffer full */ 632 putc ('\a'); 633 } 634 } 635 } 636 } 637 638 /****************************************************************************/ 639 640 static char * delete_char (char *buffer, char *p, int *colp, int *np, int plen) 641 { 642 char *s; 643 644 if (*np == 0) { 645 return (p); 646 } 647 648 if (*(--p) == '\t') { /* will retype the whole line */ 649 while (*colp > plen) { 650 puts (erase_seq); 651 (*colp)--; 652 } 653 for (s=buffer; s<p; ++s) { 654 if (*s == '\t') { 655 puts (tab_seq+((*colp) & 07)); 656 *colp += 8 - ((*colp) & 07); 657 } else { 658 ++(*colp); 659 putc (*s); 660 } 661 } 662 } else { 663 puts (erase_seq); 664 (*colp)--; 665 } 666 (*np)--; 667 return (p); 668 } 669 670 /****************************************************************************/ 671 672 int parse_line (char *line, char *argv[]) 673 { 674 int nargs = 0; 675 676 #ifdef DEBUG_PARSER 677 printf ("parse_line: \"%s\"\n", line); 678 #endif 679 while (nargs < CFG_MAXARGS) { 680 681 /* skip any white space */ 682 while ((*line == ' ') || (*line == '\t')) { 683 ++line; 684 } 685 686 if (*line == '\0') { /* end of line, no more args */ 687 argv[nargs] = NULL; 688 #ifdef DEBUG_PARSER 689 printf ("parse_line: nargs=%d\n", nargs); 690 #endif 691 return (nargs); 692 } 693 694 argv[nargs++] = line; /* begin of argument string */ 695 696 /* find end of string */ 697 while (*line && (*line != ' ') && (*line != '\t')) { 698 ++line; 699 } 700 701 if (*line == '\0') { /* end of line, no more args */ 702 argv[nargs] = NULL; 703 #ifdef DEBUG_PARSER 704 printf ("parse_line: nargs=%d\n", nargs); 705 #endif 706 return (nargs); 707 } 708 709 *line++ = '\0'; /* terminate current arg */ 710 } 711 712 printf ("** Too many args (max. %d) **\n", CFG_MAXARGS); 713 714 #ifdef DEBUG_PARSER 715 printf ("parse_line: nargs=%d\n", nargs); 716 #endif 717 return (nargs); 718 } 719 720 /****************************************************************************/ 721 722 static void process_macros (const char *input, char *output) 723 { 724 char c, prev; 725 const char *varname_start = NULL; 726 int inputcnt = strlen (input); 727 int outputcnt = CFG_CBSIZE; 728 int state = 0; /* 0 = waiting for '$' */ 729 /* 1 = waiting for '(' or '{' */ 730 /* 2 = waiting for ')' or '}' */ 731 /* 3 = waiting for ''' */ 732 #ifdef DEBUG_PARSER 733 char *output_start = output; 734 735 printf ("[PROCESS_MACROS] INPUT len %d: \"%s\"\n", strlen(input), input); 736 #endif 737 738 prev = '\0'; /* previous character */ 739 740 while (inputcnt && outputcnt) { 741 c = *input++; 742 inputcnt--; 743 744 if (state!=3) { 745 /* remove one level of escape characters */ 746 if ((c == '\\') && (prev != '\\')) { 747 if (inputcnt-- == 0) 748 break; 749 prev = c; 750 c = *input++; 751 } 752 } 753 754 switch (state) { 755 case 0: /* Waiting for (unescaped) $ */ 756 if ((c == '\'') && (prev != '\\')) { 757 state = 3; 758 break; 759 } 760 if ((c == '$') && (prev != '\\')) { 761 state++; 762 } else { 763 *(output++) = c; 764 outputcnt--; 765 } 766 break; 767 case 1: /* Waiting for ( */ 768 if (c == '(' || c == '{') { 769 state++; 770 varname_start = input; 771 } else { 772 state = 0; 773 *(output++) = '$'; 774 outputcnt--; 775 776 if (outputcnt) { 777 *(output++) = c; 778 outputcnt--; 779 } 780 } 781 break; 782 case 2: /* Waiting for ) */ 783 if (c == ')' || c == '}') { 784 int i; 785 char envname[CFG_CBSIZE], *envval; 786 int envcnt = input-varname_start-1; /* Varname # of chars */ 787 788 /* Get the varname */ 789 for (i = 0; i < envcnt; i++) { 790 envname[i] = varname_start[i]; 791 } 792 envname[i] = 0; 793 794 /* Get its value */ 795 envval = getenv (envname); 796 797 /* Copy into the line if it exists */ 798 if (envval != NULL) 799 while ((*envval) && outputcnt) { 800 *(output++) = *(envval++); 801 outputcnt--; 802 } 803 /* Look for another '$' */ 804 state = 0; 805 } 806 break; 807 case 3: /* Waiting for ' */ 808 if ((c == '\'') && (prev != '\\')) { 809 state = 0; 810 } else { 811 *(output++) = c; 812 outputcnt--; 813 } 814 break; 815 } 816 prev = c; 817 } 818 819 if (outputcnt) 820 *output = 0; 821 822 #ifdef DEBUG_PARSER 823 printf ("[PROCESS_MACROS] OUTPUT len %d: \"%s\"\n", 824 strlen(output_start), output_start); 825 #endif 826 } 827 828 /**************************************************************************** 829 * returns: 830 * 1 - command executed, repeatable 831 * 0 - command executed but not repeatable, interrupted commands are 832 * always considered not repeatable 833 * -1 - not executed (unrecognized, bootd recursion or too many args) 834 * (If cmd is NULL or "" or longer than CFG_CBSIZE-1 it is 835 * considered unrecognized) 836 * 837 * WARNING: 838 * 839 * We must create a temporary copy of the command since the command we get 840 * may be the result from getenv(), which returns a pointer directly to 841 * the environment data, which may change magicly when the command we run 842 * creates or modifies environment variables (like "bootp" does). 843 */ 844 845 int run_command (const char *cmd, int flag) 846 { 847 cmd_tbl_t *cmdtp; 848 char cmdbuf[CFG_CBSIZE]; /* working copy of cmd */ 849 char *token; /* start of token in cmdbuf */ 850 char *sep; /* end of token (separator) in cmdbuf */ 851 char finaltoken[CFG_CBSIZE]; 852 char *str = cmdbuf; 853 char *argv[CFG_MAXARGS + 1]; /* NULL terminated */ 854 int argc, inquotes; 855 int repeatable = 1; 856 int rc = 0; 857 858 #ifdef DEBUG_PARSER 859 printf ("[RUN_COMMAND] cmd[%p]=\"", cmd); 860 puts (cmd ? cmd : "NULL"); /* use puts - string may be loooong */ 861 puts ("\"\n"); 862 #endif 863 864 clear_ctrlc(); /* forget any previous Control C */ 865 866 if (!cmd || !*cmd) { 867 return -1; /* empty command */ 868 } 869 870 if (strlen(cmd) >= CFG_CBSIZE) { 871 puts ("## Command too long!\n"); 872 return -1; 873 } 874 875 strcpy (cmdbuf, cmd); 876 877 /* Process separators and check for invalid 878 * repeatable commands 879 */ 880 881 #ifdef DEBUG_PARSER 882 printf ("[PROCESS_SEPARATORS] %s\n", cmd); 883 #endif 884 while (*str) { 885 886 /* 887 * Find separator, or string end 888 * Allow simple escape of ';' by writing "\;" 889 */ 890 for (inquotes = 0, sep = str; *sep; sep++) { 891 if ((*sep=='\'') && 892 (*(sep-1) != '\\')) 893 inquotes=!inquotes; 894 895 if (!inquotes && 896 (*sep == ';') && /* separator */ 897 ( sep != str) && /* past string start */ 898 (*(sep-1) != '\\')) /* and NOT escaped */ 899 break; 900 } 901 902 /* 903 * Limit the token to data between separators 904 */ 905 token = str; 906 if (*sep) { 907 str = sep + 1; /* start of command for next pass */ 908 *sep = '\0'; 909 } 910 else 911 str = sep; /* no more commands for next pass */ 912 #ifdef DEBUG_PARSER 913 printf ("token: \"%s\"\n", token); 914 #endif 915 916 /* find macros in this token and replace them */ 917 process_macros (token, finaltoken); 918 919 /* Extract arguments */ 920 argc = parse_line (finaltoken, argv); 921 922 /* Look up command in command table */ 923 if ((cmdtp = find_cmd(argv[0])) == NULL) { 924 printf ("Unknown command '%s' - try 'help'\n", argv[0]); 925 rc = -1; /* give up after bad command */ 926 continue; 927 } 928 929 /* found - check max args */ 930 if (argc > cmdtp->maxargs) { 931 printf ("Usage:\n%s\n", cmdtp->usage); 932 rc = -1; 933 continue; 934 } 935 936 #if (CONFIG_COMMANDS & CFG_CMD_BOOTD) 937 /* avoid "bootd" recursion */ 938 if (cmdtp->cmd == do_bootd) { 939 #ifdef DEBUG_PARSER 940 printf ("[%s]\n", finaltoken); 941 #endif 942 if (flag & CMD_FLAG_BOOTD) { 943 puts ("'bootd' recursion detected\n"); 944 rc = -1; 945 continue; 946 } 947 else 948 flag |= CMD_FLAG_BOOTD; 949 } 950 #endif /* CFG_CMD_BOOTD */ 951 952 /* OK - call function to do the command */ 953 if ((cmdtp->cmd) (cmdtp, flag, argc, argv) != 0) { 954 rc = -1; 955 } 956 957 repeatable &= cmdtp->repeatable; 958 959 /* Did the user stop this? */ 960 if (had_ctrlc ()) 961 return 0; /* if stopped then not repeatable */ 962 } 963 964 return rc ? rc : repeatable; 965 } 966 967 /****************************************************************************/ 968 969 #if (CONFIG_COMMANDS & CFG_CMD_RUN) 970 int do_run (cmd_tbl_t * cmdtp, int flag, int argc, char *argv[]) 971 { 972 int i; 973 974 if (argc < 2) { 975 printf ("Usage:\n%s\n", cmdtp->usage); 976 return 1; 977 } 978 979 for (i=1; i<argc; ++i) { 980 char *arg; 981 982 if ((arg = getenv (argv[i])) == NULL) { 983 printf ("## Error: \"%s\" not defined\n", argv[i]); 984 return 1; 985 } 986 #ifndef CFG_HUSH_PARSER 987 if (run_command (arg, flag) == -1) 988 return 1; 989 #else 990 if (parse_string_outer(arg, 991 FLAG_PARSE_SEMICOLON | FLAG_EXIT_FROM_LOOP) != 0) 992 return 1; 993 #endif 994 } 995 return 0; 996 } 997 #endif /* CFG_CMD_RUN */ 998