X-Git-Url: http://cloudbase.mooo.com/gitweb/z180-stamp.git/blobdiff_plain/bbd45c46a75edd930486980c0d6a94c52d8cf403..8ed660166ce9cdeb63b4cf710c663407b7ec9128:/avr/cli.c

diff --git a/avr/cli.c b/avr/cli.c
index 6b26b6f..aa2c388 100644
--- a/avr/cli.c
+++ b/avr/cli.c
@@ -8,9 +8,10 @@
  * (C) Copyright 2005
  * JinHua Luo, GuangDong Linux Center, <luo.jinhua@gd-linux.com>
  *
- * SPDX-License-Identifier:	GPL-2.0+
+ * SPDX-License-Identifier:	GPL-2.0
  */
 
+#include "cli.h"
 #include "common.h"
 
 #include <string.h>
@@ -25,7 +26,6 @@
 #include "env.h"
 #include "cli_readline.h"
 #include "con-utils.h"
-#include "cli.h"
 
 
 /* FIXME: Quoting problems */
@@ -44,9 +44,7 @@ static int cli_parse_line(char *line, char *argv[])
 
 	debug_parser("%s: \"%s\"\n", __func__, line);
 
-	for (outp = inp = line, quote = '\0' ;
-			nargs < CONFIG_SYS_MAXARGS && (c = *inp) != '\0';
-			inp++) {
+	for (outp = inp = line, quote = '\0'; (c = *inp) != '\0'; inp++) {
 
 		switch (state) {
 		case 0:						/* before arg string, waiting for arg start */
@@ -91,6 +89,11 @@ static int cli_parse_line(char *line, char *argv[])
 			break;
 
 		}
+
+		if (nargs > CONFIG_SYS_MAXARGS) {
+			--nargs;
+			break;
+		}
 		*outp++ = c;
 	}
 
@@ -174,7 +177,7 @@ char *process_macros(char *input, char *output)
 				if (c == '}') {
 					/* Terminate variable name */
 					*(inp-1) = '\0';
-					const char *envval = getenv(varname);
+					const char *envval = getenv_char(varname);
 					*(inp-1) = '}';
 					/* Copy into the line if it exists */
 					if (envval != NULL)
@@ -207,7 +210,7 @@ char *process_macros(char *input, char *output)
  * WARNING:
  *
  * We must create a temporary copy of the command since the command we get
- * may be the result from getenv(), which returns a pointer directly to
+ * may be the result from getenv_char(), which returns a pointer directly to
  * the environment data, which may change magicly when the command we run
  * creates or modifies environment variables (like "bootp" does).
  *
@@ -313,7 +316,7 @@ static int cli_run_command_list(const char *cmd)
 
 
 /*
- * Run a command using the selected parser.
+ * Run a command.
  *
  * @param cmd	Command to run
  * @param flag	Execution flags (CMD_FLAG_...)
@@ -361,7 +364,7 @@ void cli_loop(void)
 	int rc = 1;
 
 	for (;;) {
-		len = cli_readline(PSTR(CONFIG_SYS_PROMPT));
+		len = cli_readline(PSTR(CONFIG_SYS_PROMPT), 1);
 
 		flag = 0;	/* assume no special flags for now */
 		if (len > 0) {
@@ -395,9 +398,9 @@ command_ret_t do_run(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
 	for (i = 1; i < argc; ++i) {
 		char *arg;
 
-		arg = getenv(argv[i]);
+		arg = getenv_char(argv[i]);
 		if (arg == NULL) {
-			printf_P(PSTR("## Error: \"%s\" not defined\n"), argv[i]);
+			printf_P(PSTR("## Error: \"%s\" is not set\n"), argv[i]);
 			return CMD_RET_FAILURE;
 		}