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