Merge branch 'busybox' into merge

1 files changed, 123 insertions, 118 deletions

diff --git a/shell/hush.c b/shell/hush.c
index b3ae73b9b..4b08232a4 100644
--- a/shell/hush.c
+++ b/shell/hush.c
@@ -466,9 +466,9 @@
 
 #define JOB_STATUS_FORMAT    "[%u] %-22s %.40s\n"
 
-#define _SPECIAL_VARS_STR     "_*@$!?#"
-#define SPECIAL_VARS_STR     ("_*@$!?#" + 1)
-#define NUMERIC_SPECVARS_STR ("_*@$!?#" + 3)
+#define _SPECIAL_VARS_STR     "_*@$!?#-"
+#define SPECIAL_VARS_STR     ("_*@$!?#-" + 1)
+#define NUMERIC_SPECVARS_STR ("_*@$!?#-" + 3)
 #if BASH_PATTERN_SUBST
 /* Support / and // replace ops */
 /* Note that // is stored as \ in "encoded" string representation */
@@ -854,8 +854,7 @@ struct globals {
         /* 'interactive_fd' is a fd# open to ctty, if we have one
          * _AND_ if we decided to act interactively */
         int interactive_fd;
-        const char *PS1;
-        IF_FEATURE_EDITING_FANCY_PROMPT(const char *PS2;)
+        IF_NOT_FEATURE_EDITING_FANCY_PROMPT(char *PS1;)
 # define G_interactive_fd (G.interactive_fd)
 #else
 # define G_interactive_fd 0
@@ -903,6 +902,7 @@ struct globals {
 #else
 # define G_x_mode 0
 #endif
+        char opt_s;
 #if ENABLE_HUSH_INTERACTIVE
         smallint promptmode; /* 0: PS1, 1: PS2 */
 #endif
@@ -968,8 +968,8 @@ struct globals {
         smallint we_have_children;
 #endif
 #if ENABLE_HUSH_LINENO_VAR
-        unsigned lineno;
-        char *lineno_var;
+        unsigned parse_lineno;
+        unsigned execute_lineno;
 #endif
         HFILE *HFILE_list;
         /* Which signals have non-DFL handler (even with no traps set)?
@@ -1009,6 +1009,7 @@ struct globals {
         int debug_indent;
 #endif
         struct sigaction sa;
+        char optstring_buf[sizeof("eixs")];
 #if BASH_EPOCH_VARS
         char epoch_buf[sizeof("%lu.nnnnnn") + sizeof(long)*3];
 #endif
@@ -1448,13 +1449,6 @@ static void syntax_error_unexpected_ch(unsigned lineno UNUSED_PARAM, int ch)
 #endif
 
 
-#if ENABLE_HUSH_INTERACTIVE && ENABLE_FEATURE_EDITING_FANCY_PROMPT
-static void cmdedit_update_prompt(void);
-#else
-# define cmdedit_update_prompt() ((void)0)
-#endif
-
-
 /* Utility functions
  */
 /* Replace each \x with x in place, return ptr past NUL. */
@@ -2039,7 +2033,8 @@ static sighandler_t pick_sighandler(unsigned sig)
 static void hush_exit(int exitcode)
 {
 #if ENABLE_FEATURE_EDITING_SAVE_ON_EXIT
-        save_history(G.line_input_state);
+        if (G.line_input_state)
+                save_history(G.line_input_state);
 #endif
 
         fflush_all();
@@ -2229,6 +2224,10 @@ static const char* FAST_FUNC get_local_var_value(const char *name)
         if (strcmp(name, "RANDOM") == 0)
                 return utoa(next_random(&G.random_gen));
 #endif
+#if ENABLE_HUSH_LINENO_VAR
+        if (strcmp(name, "LINENO") == 0)
+                return utoa(G.execute_lineno);
+#endif
 #if BASH_EPOCH_VARS
         {
                 const char *fmt = NULL;
@@ -2248,32 +2247,20 @@ static const char* FAST_FUNC get_local_var_value(const char *name)
         return NULL;
 }
 
+#if ENABLE_HUSH_GETOPTS
 static void handle_changed_special_names(const char *name, unsigned name_len)
 {
-        if (ENABLE_HUSH_INTERACTIVE && ENABLE_FEATURE_EDITING_FANCY_PROMPT
-         && name_len == 3 && name[0] == 'P' && name[1] == 'S'
-        ) {
-                cmdedit_update_prompt();
-                return;
-        }
-
-        if ((ENABLE_HUSH_LINENO_VAR || ENABLE_HUSH_GETOPTS)
-         && name_len == 6
-        ) {
-#if ENABLE_HUSH_LINENO_VAR
-                if (strncmp(name, "LINENO", 6) == 0) {
-                        G.lineno_var = NULL;
-                        return;
-                }
-#endif
-#if ENABLE_HUSH_GETOPTS
+        if (name_len == 6) {
                 if (strncmp(name, "OPTIND", 6) == 0) {
                         G.getopt_count = 0;
                         return;
                 }
-#endif
         }
 }
+#else
+/* Do not even bother evaluating arguments */
+# define handle_changed_special_names(...) ((void)0)
+#endif
 
 /* str holds "NAME=VAL" and is expected to be malloced.
  * We take ownership of it.
@@ -2289,6 +2276,7 @@ static int set_local_var(char *str, unsigned flags)
         char *free_me = NULL;
         char *eq_sign;
         int name_len;
+        int retval;
         unsigned local_lvl = (flags >> SETFLAG_VARLVL_SHIFT);
 
         eq_sign = strchr(str, '=');
@@ -2402,24 +2390,24 @@ static int set_local_var(char *str, unsigned flags)
 #endif
         if (flags & SETFLAG_EXPORT)
                 cur->flg_export = 1;
+        retval = 0;
         if (cur->flg_export) {
                 if (flags & SETFLAG_UNEXPORT) {
                         cur->flg_export = 0;
                         /* unsetenv was already done */
                 } else {
-                        int i;
                         debug_printf_env("%s: putenv '%s'/%u\n", __func__, cur->varstr, cur->var_nest_level);
-                        i = putenv(cur->varstr);
-                        /* only now we can free old exported malloced string */
-                        free(free_me);
-                        return i;
+                        retval = putenv(cur->varstr);
+                        /* fall through to "free(free_me)" -
+                         * only now we can free old exported malloced string
+                         */
                 }
         }
         free(free_me);
 
         handle_changed_special_names(cur->varstr, name_len - 1);
 
-        return 0;
+        return retval;
 }
 
 static void FAST_FUNC set_local_var_from_halves(const char *name, const char *val)
@@ -2462,7 +2450,7 @@ static int unset_local_var_len(const char *name, int name_len)
                 cur_pp = &cur->next;
         }
 
-        /* Handle "unset PS1" et al even if did not find the variable to unset */
+        /* Handle "unset LINENO" et al even if did not find the variable to unset */
         handle_changed_special_names(name, name_len);
 
         return EXIT_SUCCESS;
@@ -2581,36 +2569,27 @@ static void reinit_unicode_for_hush(void)
  *      \
  * It exercises a lot of corner cases.
  */
-# if ENABLE_FEATURE_EDITING_FANCY_PROMPT
-static void cmdedit_update_prompt(void)
-{
-        G.PS1 = get_local_var_value("PS1");
-        if (G.PS1 == NULL)
-                G.PS1 = "";
-        G.PS2 = get_local_var_value("PS2");
-        if (G.PS2 == NULL)
-                G.PS2 = "";
-}
-# endif
 static const char *setup_prompt_string(void)
 {
         const char *prompt_str;
 
         debug_printf_prompt("%s promptmode:%d\n", __func__, G.promptmode);
 
-        IF_FEATURE_EDITING_FANCY_PROMPT(    prompt_str = G.PS2;)
-        IF_NOT_FEATURE_EDITING_FANCY_PROMPT(prompt_str = "> ";)
+# if ENABLE_FEATURE_EDITING_FANCY_PROMPT
+        prompt_str = get_local_var_value(G.promptmode == 0 ? "PS1" : "PS2");
+        if (!prompt_str)
+                prompt_str = "";
+# else
+        prompt_str = "> "; /* if PS2, else... */
         if (G.promptmode == 0) { /* PS1 */
-                if (!ENABLE_FEATURE_EDITING_FANCY_PROMPT) {
-                        /* No fancy prompts supported, (re)generate "CURDIR $ " by hand */
-                        free((char*)G.PS1);
-                        /* bash uses $PWD value, even if it is set by user.
-                         * It uses current dir only if PWD is unset.
-                         * We always use current dir. */
-                        G.PS1 = xasprintf("%s %c ", get_cwd(0), (geteuid() != 0) ? '$' : '#');
-                }
-                prompt_str = G.PS1;
+                /* No fancy prompts supported, (re)generate "CURDIR $ " by hand */
+                free(G.PS1);
+                /* bash uses $PWD value, even if it is set by user.
+                 * It uses current dir only if PWD is unset.
+                 * We always use current dir. */
+                G.PS1 = xasprintf("%s %c ", get_cwd(0), (geteuid() != 0) ? '$' : '#');
         }
+# endif
         debug_printf("prompt_str '%s'\n", prompt_str);
         return prompt_str;
 }
@@ -2747,8 +2726,8 @@ static int i_getch(struct in_str *i)
         i->last_char = ch;
 #if ENABLE_HUSH_LINENO_VAR
         if (ch == '\n') {
-                G.lineno++;
-                debug_printf_parse("G.lineno++ = %u\n", G.lineno);
+                G.parse_lineno++;
+                debug_printf_parse("G.parse_lineno++ = %u\n", G.parse_lineno);
         }
 #endif
         return ch;
@@ -3750,8 +3729,8 @@ static int done_command(struct parse_context *ctx)
  clear_and_ret:
         memset(command, 0, sizeof(*command));
 #if ENABLE_HUSH_LINENO_VAR
-        command->lineno = G.lineno;
-        debug_printf_parse("command->lineno = G.lineno (%u)\n", G.lineno);
+        command->lineno = G.parse_lineno;
+        debug_printf_parse("command->lineno = G.parse_lineno (%u)\n", G.parse_lineno);
 #endif
         return pi->num_cmds; /* used only for 0/nonzero check */
 }
@@ -4198,7 +4177,7 @@ static int done_word(struct parse_context *ctx)
 #if ENABLE_HUSH_LOOPS
         if (ctx->ctx_res_w == RES_FOR) {
                 if (ctx->word.has_quoted_part
-                 || !is_well_formed_var_name(command->argv[0], '\0')
+                 || endofname(command->argv[0])[0] != '\0'
                 ) {
                         /* bash says just "not a valid identifier" */
                         syntax_error("not a valid identifier in for");
@@ -4912,6 +4891,7 @@ static int parse_dollar(o_string *as_string,
         case '#': /* number of args */
         case '*': /* args */
         case '@': /* args */
+        case '-': /* $- option flags set by set builtin or shell options (-i etc) */
                 goto make_one_char_var;
         case '{': {
                 char len_single_ch;
@@ -5086,11 +5066,10 @@ static int parse_dollar(o_string *as_string,
         case '_':
                 goto make_var;
 #if 0
-        /* TODO: $_ and $-: */
+        /* TODO: $_: */
         /* $_ Shell or shell script name; or last argument of last command
          * (if last command wasn't a pipe; if it was, bash sets $_ to "");
          * but in command's env, set to full pathname used to invoke it */
-        /* $- Option flags set by set builtin or shell options (-i etc) */
                 ch = i_getch(input);
                 nommu_addchr(as_string, ch);
                 ch = i_peek_and_eat_bkslash_nl(input);
@@ -5372,7 +5351,7 @@ static struct pipe *parse_stream(char **pstring,
                         if ((ctx.is_assignment == MAYBE_ASSIGNMENT
                             || ctx.is_assignment == WORD_IS_KEYWORD)
                          && ch == '='
-                         && is_well_formed_var_name(ctx.word.data, '=')
+                         && endofname(ctx.word.data)[0] == '='
                         ) {
                                 ctx.is_assignment = DEFINITELY_ASSIGNMENT;
                                 debug_printf_parse("ctx.is_assignment='%s'\n", assignment_flag[ctx.is_assignment]);
@@ -6119,6 +6098,12 @@ static int encode_then_append_var_plusminus(o_string *output, int n,
                 /* string has no special chars
                  * && string has no $IFS chars
                  */
+                if (dquoted) {
+                        /* Prints 1 (quoted expansion is a "" word, not nothing):
+                         * set -- "${notexist-}"; echo $#
+                         */
+                        output->has_quoted_part = 1;
+                }
                 return expand_vars_to_list(output, n, str);
         }
 
@@ -6415,6 +6400,26 @@ static NOINLINE int expand_one_var(o_string *output, int n,
                 case '#': /* argc */
                         val = utoa(G.global_argc ? G.global_argc-1 : 0);
                         break;
+                case '-': { /* active options */
+                        /* Check set_mode() to see what option chars we support */
+                        char *cp;
+                        val = cp = G.optstring_buf;
+                        if (G.o_opt[OPT_O_ERREXIT])
+                                *cp++ = 'e';
+                        if (G_interactive_fd)
+                                *cp++ = 'i';
+                        if (G_x_mode)
+                                *cp++ = 'x';
+                        /* If G.o_opt[OPT_O_NOEXEC] is true,
+                         * commands read but are not executed,
+                         * so $- can not execute too, 'n' is never seen in $-.
+                         */
+                        if (G.opt_s)
+                                *cp++ = 's';
+//TODO: show 'c' if executed via "hush -c 'CMDS'" (bash only, not ash)
+                        *cp = '\0';
+                        break;
+                }
                 default:
                         val = get_local_var_value(var);
                 }
@@ -7275,22 +7280,22 @@ static void parse_and_run_stream(struct in_str *inp, int end_trigger)
 static void parse_and_run_string(const char *s)
 {
         struct in_str input;
-        //IF_HUSH_LINENO_VAR(unsigned sv = G.lineno;)
+        //IF_HUSH_LINENO_VAR(unsigned sv = G.parse_lineno;)
 
         setup_string_in_str(&input, s);
         parse_and_run_stream(&input, '\0');
-        //IF_HUSH_LINENO_VAR(G.lineno = sv;)
+        //IF_HUSH_LINENO_VAR(G.parse_lineno = sv;)
 }
 
 static void parse_and_run_file(HFILE *fp)
 {
         struct in_str input;
-        IF_HUSH_LINENO_VAR(unsigned sv = G.lineno;)
+        IF_HUSH_LINENO_VAR(unsigned sv = G.parse_lineno;)
 
-        IF_HUSH_LINENO_VAR(G.lineno = 1;)
+        IF_HUSH_LINENO_VAR(G.parse_lineno = 1;)
         setup_file_in_str(&input, fp);
         parse_and_run_stream(&input, ';');
-        IF_HUSH_LINENO_VAR(G.lineno = sv;)
+        IF_HUSH_LINENO_VAR(G.parse_lineno = sv;)
 }
 
 #if ENABLE_HUSH_TICK
@@ -7598,10 +7603,10 @@ static int save_fd_on_redirect(int fd, int avoid_fd, struct squirrel **sqp)
                 avoid_fd = 9;
 
 #if ENABLE_HUSH_INTERACTIVE
-        if (fd == G.interactive_fd) {
+        if (fd == G_interactive_fd) {
                 /* Testcase: "ls -l /proc/$$/fd 255>&-" should work */
-                G.interactive_fd = xdup_CLOEXEC_and_close(G.interactive_fd, avoid_fd);
-                debug_printf_redir("redirect_fd %d: matches interactive_fd, moving it to %d\n", fd, G.interactive_fd);
+                G_interactive_fd = xdup_CLOEXEC_and_close(G_interactive_fd, avoid_fd);
+                debug_printf_redir("redirect_fd %d: matches interactive_fd, moving it to %d\n", fd, G_interactive_fd);
                 return 1; /* "we closed fd" */
         }
 #endif
@@ -7677,7 +7682,7 @@ static void restore_redirects(struct squirrel *sq)
                 free(sq);
         }
 
-        /* If moved, G.interactive_fd stays on new fd, not restoring it */
+        /* If moved, G_interactive_fd stays on new fd, not restoring it */
 }
 
 #if ENABLE_FEATURE_SH_STANDALONE && BB_MMU
@@ -7694,7 +7699,7 @@ static int internally_opened_fd(int fd, struct squirrel *sq)
         int i;
 
 #if ENABLE_HUSH_INTERACTIVE
-        if (fd == G.interactive_fd)
+        if (fd == G_interactive_fd)
                 return 1;
 #endif
         /* If this one of script's fds? */
@@ -8989,8 +8994,7 @@ static NOINLINE int run_pipe(struct pipe *pi)
                 struct variable *old_vars;
 
 #if ENABLE_HUSH_LINENO_VAR
-                if (G.lineno_var)
-                        strcpy(G.lineno_var + sizeof("LINENO=")-1, utoa(command->lineno));
+                G.execute_lineno = command->lineno;
 #endif
 
                 if (argv[command->assignment_cnt] == NULL) {
@@ -9223,8 +9227,7 @@ static NOINLINE int run_pipe(struct pipe *pi)
                         xpiped_pair(pipefds);
 
 #if ENABLE_HUSH_LINENO_VAR
-                if (G.lineno_var)
-                        strcpy(G.lineno_var + sizeof("LINENO=")-1, utoa(command->lineno));
+                G.execute_lineno = command->lineno;
 #endif
 
                 command->pid = BB_MMU ? fork() : vfork();
@@ -9908,14 +9911,6 @@ int hush_main(int argc, char **argv)
         /* Export PWD */
         set_pwd_var(SETFLAG_EXPORT);
 
-#if ENABLE_HUSH_INTERACTIVE && ENABLE_FEATURE_EDITING_FANCY_PROMPT
-        /* Set (but not export) PS1/2 unless already set */
-        if (!get_local_var_value("PS1"))
-                set_local_var_from_halves("PS1", "\\w \\$ ");
-        if (!get_local_var_value("PS2"))
-                set_local_var_from_halves("PS2", "> ");
-#endif
-
 #if BASH_HOSTNAME_VAR
         /* Set (but not export) HOSTNAME unless already set */
         if (!get_local_var_value("HOSTNAME")) {
@@ -9927,6 +9922,11 @@ int hush_main(int argc, char **argv)
         /* IFS is not inherited from the parent environment */
         set_local_var_from_halves("IFS", defifs);
 
+        if (!get_local_var_value("PATH"))
+                set_local_var_from_halves("PATH", bb_default_root_path);
+
+        /* PS1/PS2 are set later, if we determine that we are interactive */
+
         /* bash also exports SHLVL and _,
          * and sets (but doesn't export) the following variables:
          * BASH=/bin/bash
@@ -9960,21 +9960,7 @@ int hush_main(int argc, char **argv)
          * PS4='+ '
          */
 
-#if ENABLE_HUSH_LINENO_VAR
-        if (ENABLE_HUSH_LINENO_VAR) {
-                char *p = xasprintf("LINENO=%*s", (int)(sizeof(int)*3), "");
-                set_local_var(p, /*flags*/ 0);
-                G.lineno_var = p; /* can't assign before set_local_var("LINENO=...") */
-        }
-#endif
-
-#if ENABLE_FEATURE_EDITING
-        G.line_input_state = new_line_input_t(FOR_SHELL);
-#endif
-
         /* Initialize some more globals to non-zero values */
-        cmdedit_update_prompt();
-
         die_func = restore_ttypgrp_and__exit;
 
         /* Shell is non-interactive at first. We need to call
@@ -10192,6 +10178,7 @@ int hush_main(int argc, char **argv)
 #endif
                 goto final_return;
         }
+        G.opt_s = 1;
 
         /* Up to here, shell was non-interactive. Now it may become one.
          * NB: don't forget to (re)run install_special_sighandlers() as needed.
@@ -10260,6 +10247,9 @@ int hush_main(int argc, char **argv)
                 }
                 enable_restore_tty_pgrp_on_exit();
 
+# if ENABLE_FEATURE_EDITING
+                G.line_input_state = new_line_input_t(FOR_SHELL);
+# endif
 # if ENABLE_HUSH_SAVEHISTORY && MAX_HISTORY > 0
                 {
                         const char *hp = get_local_var_value("HISTFILE");
@@ -10308,14 +10298,23 @@ int hush_main(int argc, char **argv)
          * (--norc turns this off, --rcfile <file> overrides)
          */
 
-        if (!ENABLE_FEATURE_SH_EXTRA_QUIET && G_interactive_fd) {
-                /* note: ash and hush share this string */
-                printf("\n\n%s %s\n"
-                        IF_HUSH_HELP("Enter 'help' for a list of built-in commands.\n")
-                        "\n",
-                        bb_banner,
-                        "hush - the humble shell"
-                );
+        if (G_interactive_fd) {
+#if ENABLE_HUSH_INTERACTIVE && ENABLE_FEATURE_EDITING_FANCY_PROMPT
+                /* Set (but not export) PS1/2 unless already set */
+                if (!get_local_var_value("PS1"))
+                        set_local_var_from_halves("PS1", "\\w \\$ ");
+                if (!get_local_var_value("PS2"))
+                        set_local_var_from_halves("PS2", "> ");
+#endif
+                if (!ENABLE_FEATURE_SH_EXTRA_QUIET) {
+                        /* note: ash and hush share this string */
+                        printf("\n\n%s %s\n"
+                                IF_HUSH_HELP("Enter 'help' for a list of built-in commands.\n")
+                                "\n",
+                                bb_banner,
+                                "hush - the humble shell"
+                        );
+                }
         }
 
         parse_and_run_file(hfopen(NULL)); /* stdin */
@@ -10378,7 +10377,8 @@ static int FAST_FUNC builtin_help(char **argv UNUSED_PARAM)
 #if MAX_HISTORY && ENABLE_FEATURE_EDITING
 static int FAST_FUNC builtin_history(char **argv UNUSED_PARAM)
 {
-        show_history(G.line_input_state);
+        if (G.line_input_state)
+                show_history(G.line_input_state);
         return EXIT_SUCCESS;
 }
 #endif
@@ -10687,9 +10687,7 @@ static int helper_export_local(char **argv, unsigned flags)
 {
         do {
                 char *name = *argv;
-                char *name_end = strchrnul(name, '=');
-
-                /* So far we do not check that name is valid (TODO?) */
+                const char *name_end = endofname(name);
 
                 if (*name_end == '\0') {
                         struct variable *var, **vpp;
@@ -10747,8 +10745,15 @@ static int helper_export_local(char **argv, unsigned flags)
                          */
                         name = xasprintf("%s=", name);
                 } else {
+                        if (*name_end != '=') {
+                                bb_error_msg("'%s': bad variable name", name);
+                                /* do not parse following argv[]s: */
+                                return 1;
+                        }
                         /* (Un)exporting/making local NAME=VALUE */
                         name = xstrdup(name);
+                        /* Testcase: export PS1='\w \$ ' */
+                        unbackslash(name);
                 }
                 debug_printf_env("%s: set_local_var('%s')\n", __func__, name);
                 if (set_local_var(name, flags))