Fix env usage information.
[oweals/busybox.git] / lash.c
diff --git a/lash.c b/lash.c
index 49fb6b5365840aed117bdca178410f56dc3d75df..1d128355cafd1cfd63e250d106ffec683ad22057 100644 (file)
--- a/lash.c
+++ b/lash.c
 #include <sys/wait.h>
 #include <unistd.h>
 #include <getopt.h>
+#include "busybox.h"
+#include "cmdedit.h"
 
-#undef BB_FEATURE_SH_WORDEXP
+#ifdef BB_LOCALE_SUPPORT
+#include <locale.h>
+#endif
 
-#if BB_FEATURE_SH_WORDEXP
+//#define BB_FEATURE_SH_WORDEXP
+
+#ifdef BB_FEATURE_SH_WORDEXP
 #include <wordexp.h>
 #define expand_t       wordexp_t
 #undef BB_FEATURE_SH_BACKTICKS
 #include <glob.h>
 #define expand_t       glob_t
 #endif 
-#include "busybox.h"
-#include "cmdedit.h"
 
 
-static const int MAX_LINE = 256;       /* size of input buffer for cwd data */
 static const int MAX_READ = 128;       /* size of input buffer for `read' builtin */
 #define JOB_STATUS_FORMAT "[%d] %-22s %.40s\n"
 
@@ -142,7 +145,6 @@ struct close_me {
 
 /* function prototypes for builtins */
 static int builtin_cd(struct child_prog *cmd);
-static int builtin_env(struct child_prog *dummy);
 static int builtin_exec(struct child_prog *cmd);
 static int builtin_exit(struct child_prog *cmd);
 static int builtin_fg_bg(struct child_prog *cmd);
@@ -202,7 +204,6 @@ static struct built_in_command bltins[] = {
 /* Table of forking built-in functions (things that fork cannot change global
  * variables in the parent process, such as the current working directory) */
 static struct built_in_command bltins_forking[] = {
-       {"env", "Print all environment variables", builtin_env},
        {"pwd", "Print current directory", builtin_pwd},
        {"help", "List shell built-in commands", builtin_help},
        {NULL, NULL, NULL}
@@ -231,7 +232,7 @@ static char syntax_err[]="syntax error near unexpected token";
 #endif
 
 static char *PS1;
-static char *PS2;
+static char *PS2 = "> ";
 
 
 #ifdef DEBUG_SHELL
@@ -255,14 +256,13 @@ static inline void debug_printf(const char *format, ...) { }
 builtin   previous use      notes
 ------ -----------------  ---------
 cd      cmd->progs[0]
-env     0
 exec    cmd->progs[0]  squashed bug: didn't look for applets or forking builtins
 exit    cmd->progs[0]
 fg_bg   cmd->progs[0], job_list->head, job_list->fg
 help    0
 jobs    job_list->head
 pwd     0
-export  cmd->progs[0]  passes cmd, job_list to builtin_env(), which ignores them
+export  cmd->progs[0]
 source  cmd->progs[0]
 unset   cmd->progs[0]
 read    cmd->progs[0]
@@ -296,22 +296,11 @@ static int builtin_cd(struct child_prog *child)
                printf("cd: %s: %m\n", newdir);
                return EXIT_FAILURE;
        }
-       getcwd(cwd, sizeof(char)*MAX_LINE);
+       cwd = xgetcwd(cwd);
 
        return EXIT_SUCCESS;
 }
 
-/* built-in 'env' handler */
-static int builtin_env(struct child_prog *dummy)
-{
-       char **e;
-
-       for (e = environ; *e; e++) {
-               printf( "%s\n", *e);
-       }
-       return (0);
-}
-
 /* built-in 'exec' handler */
 static int builtin_exec(struct child_prog *child)
 {
@@ -423,7 +412,6 @@ static int builtin_jobs(struct child_prog *child)
 /* built-in 'pwd' handler */
 static int builtin_pwd(struct child_prog *dummy)
 {
-       getcwd(cwd, MAX_LINE);
        printf( "%s\n", cwd);
        return EXIT_SUCCESS;
 }
@@ -435,17 +423,27 @@ static int builtin_export(struct child_prog *child)
        char *v = child->argv[1];
 
        if (v == NULL) {
-               return (builtin_env(child));
+               char **e;
+               for (e = environ; *e; e++) {
+                       printf( "%s\n", *e);
+               }
+               return 0;
        }
        res = putenv(v);
        if (res)
                fprintf(stderr, "export: %m\n");
-#ifndef BB_FEATURE_SIMPLE_PROMPT
+#ifndef BB_FEATURE_SH_SIMPLE_PROMPT
        if (strncmp(v, "PS1=", 4)==0)
                PS1 = getenv("PS1");
-       else if (strncmp(v, "PS2=", 4)==0)
-               PS2 = getenv("PS2");
 #endif
+
+#ifdef BB_LOCALE_SUPPORT
+       if(strncmp(v, "LC_ALL=", 7)==0)
+               setlocale(LC_ALL, getenv("LC_ALL"));
+       if(strncmp(v, "LC_CTYPE=", 9)==0)
+               setlocale(LC_CTYPE, getenv("LC_CTYPE"));
+#endif
+
        return (res);
 }
 
@@ -627,10 +625,7 @@ static int builtin_unset(struct child_prog *child)
  */
 static int run_command_predicate(char *cmd)
 {
-       int n=strlen(cmd);
-       local_pending_command = xmalloc(n+1);
-       strncpy(local_pending_command, cmd, n); 
-       local_pending_command[n]='\0';
+       local_pending_command = xstrdup(cmd);
        return( busy_loop(NULL));
 }
 #endif
@@ -669,29 +664,34 @@ static void close_all()
 static void free_job(struct job *cmd)
 {
        int i;
+       struct jobset *keep;
 
        for (i = 0; i < cmd->num_progs; i++) {
                free(cmd->progs[i].argv);
                if (cmd->progs[i].redirects)
                        free(cmd->progs[i].redirects);
        }
-       free(cmd->progs);
+       if (cmd->progs)
+               free(cmd->progs);
        if (cmd->text)
                free(cmd->text);
-       free(cmd->cmdbuf);
+       if (cmd->cmdbuf)
+               free(cmd->cmdbuf);
+       keep = cmd->job_list;
        memset(cmd, 0, sizeof(struct job));
+       cmd->job_list = keep;
 }
 
-/* remove a job from the job_list */
-static void remove_job(struct jobset *job_list, struct job *job)
+/* remove a job from a jobset */
+static void remove_job(struct jobset *j_list, struct job *job)
 {
        struct job *prevjob;
 
        free_job(job);
-       if (job == job_list->head) {
-               job_list->head = job->next;
+       if (job == j_list->head) {
+               j_list->head = job->next;
        } else {
-               prevjob = job_list->head;
+               prevjob = j_list->head;
                while (prevjob->next != job)
                        prevjob = prevjob->next;
                prevjob->next = job->next;
@@ -702,7 +702,7 @@ static void remove_job(struct jobset *job_list, struct job *job)
 
 /* Checks to see if any background processes have exited -- if they 
    have, figure out why and see if a job has completed */
-static void checkjobs(struct jobset *job_list)
+static void checkjobs(struct jobset *j_list)
 {
        struct job *job;
        pid_t childpid;
@@ -710,7 +710,7 @@ static void checkjobs(struct jobset *job_list)
        int prognum = 0;
 
        while ((childpid = waitpid(-1, &status, WNOHANG | WUNTRACED)) > 0) {
-               for (job = job_list->head; job; job = job->next) {
+               for (job = j_list->head; job; job = job->next) {
                        prognum = 0;
                        while (prognum < job->num_progs &&
                                   job->progs[prognum].pid != childpid) prognum++;
@@ -729,7 +729,7 @@ static void checkjobs(struct jobset *job_list)
 
                        if (!job->running_progs) {
                                printf(JOB_STATUS_FORMAT, job->jobid, "Done", job->text);
-                               remove_job(job_list, job);
+                               remove_job(j_list, job);
                        }
                } else {
                        /* child stopped */
@@ -805,23 +805,18 @@ static void restore_redirects(int squirrel[])
 
 static inline void cmdedit_set_initial_prompt(void)
 {
-#ifdef BB_FEATURE_SIMPLE_PROMPT
+#ifdef BB_FEATURE_SH_SIMPLE_PROMPT
        PS1 = NULL;
-       PS2 = "> ";
 #else
        PS1 = getenv("PS1");
-       if(PS1==0) {
+       if(PS1==0)
                PS1 = "\\w \\$ ";
-       }
-       PS2 = getenv("PS2");
-       if(PS2==0) 
-               PS2 = "> ";
 #endif 
 }
 
 static inline void setup_prompt_string(char **prompt_str)
 {
-#ifdef BB_FEATURE_SIMPLE_PROMPT
+#ifdef BB_FEATURE_SH_SIMPLE_PROMPT
        /* Set up the prompt */
        if (shell_context == 0) {
                if (PS1)
@@ -876,9 +871,6 @@ static int get_command(FILE * source, char *command)
                return 1;
        }
 
-       /* remove trailing newline */
-       chomp(command);
-
        return 0;
 }
 
@@ -905,35 +897,35 @@ static char* itoa(register int i)
 #endif 
 
 #if defined BB_FEATURE_SH_ENVIRONMENT && ! defined BB_FEATURE_SH_WORDEXP
-char * strsep_space( char *string, int * index)
+char * strsep_space( char *string, int * ix)
 {
        char *token, *begin;
 
        begin = string;
 
        /* Short circuit the trivial case */
-       if ( !string || ! string[*index])
+       if ( !string || ! string[*ix])
                return NULL;
 
        /* Find the end of the token. */
-       while( string && string[*index] && !isspace(string[*index]) ) {
-               (*index)++;
+       while( string && string[*ix] && !isspace(string[*ix]) ) {
+               (*ix)++;
        }
 
        /* Find the end of any whitespace trailing behind 
         * the token and let that be part of the token */
-       while( string && string[*index] && isspace(string[*index]) ) {
-               (*index)++;
+       while( string && string[*ix] && isspace(string[*ix]) ) {
+               (*ix)++;
        }
 
-       if (! string && *index==0) {
+       if (! string && *ix==0) {
                /* Nothing useful was found */
                return NULL;
        }
 
-       token = xmalloc(*index);
-       token[*index] = '\0';
-       strncpy(token, string,  *index); 
+       token = xmalloc(*ix+1);
+       token[*ix] = '\0';
+       strncpy(token, string,  *ix); 
 
        return token;
 }
@@ -945,7 +937,7 @@ static int expand_arguments(char *command)
 #ifdef BB_FEATURE_SH_ENVIRONMENT
        expand_t expand_result;
        char *src, *dst, *var;
-       int index = 0;
+       int ix = 0;
        int i=0, length, total_length=0, retval;
        const char *out_of_space = "out of space during expansion"; 
 #endif
@@ -954,19 +946,19 @@ static int expand_arguments(char *command)
        chomp(command);
        
        /* Fix up escape sequences to be the Real Thing(tm) */
-       while( command && command[index]) {
-               if (command[index] == '\\') {
-                       char *tmp = command+index+1;
-                       command[index+1] = process_escape_sequence(  &tmp );
-                       memmove(command+index, command+index+1, strlen(command+index));
+       while( command && command[ix]) {
+               if (command[ix] == '\\') {
+                       const char *tmp = command+ix+1;
+                       command[ix] = process_escape_sequence(  &tmp );
+                       memmove(command+ix + 1, tmp, strlen(tmp)+1);
                }
-               index++;
+               ix++;
        }
 
 #ifdef BB_FEATURE_SH_ENVIRONMENT
 
 
-#if BB_FEATURE_SH_WORDEXP
+#ifdef BB_FEATURE_SH_WORDEXP
        /* This first part uses wordexp() which is a wonderful C lib 
         * function which expands nearly everything.  */ 
        retval = wordexp (command, &expand_result, WRDE_SHOWERR);
@@ -1010,16 +1002,25 @@ static int expand_arguments(char *command)
         * (char*) into cmd (char**, one word per string) */
        {
         
-               int flags = GLOB_NOCHECK|GLOB_BRACE|GLOB_TILDE;
+               int flags = GLOB_NOCHECK
+#ifdef GLOB_BRACE
+                               | GLOB_BRACE
+#endif 
+#ifdef GLOB_TILDE
+                               | GLOB_TILDE
+#endif 
+                       ;
                char *tmpcmd, *cmd, *cmd_copy;
                /* We need a clean copy, so strsep can mess up the copy while
                 * we write stuff into the original (in a minute) */
                cmd = cmd_copy = strdup(command);
                *command = '\0';
-               for (index = 0, tmpcmd = cmd; 
-                               (tmpcmd = strsep_space(cmd, &index)) != NULL; cmd += index, index=0) {
+               for (ix = 0, tmpcmd = cmd; 
+                               (tmpcmd = strsep_space(cmd, &ix)) != NULL; cmd += ix, ix=0) {
                        if (*tmpcmd == '\0')
                                break;
+                       /* we need to trim() the result for glob! */
+                       trim(tmpcmd);
                        retval = glob(tmpcmd, flags, NULL, &expand_result);
                        free(tmpcmd); /* Free mem allocated by strsep_space */
                        if (retval == GLOB_NOSPACE) {
@@ -1037,13 +1038,14 @@ static int expand_arguments(char *command)
                        /* Convert from char** (one word per string) to a simple char*,
                         * but don't overflow command which is BUFSIZ in length */
                                for (i=0; i < expand_result.gl_pathc; i++) {
-                                       length=strlen(expand_result.gl_pathv[i])+1;
-                                       if (BUFSIZ-total_length-length <= 0) {
+                                       length=strlen(expand_result.gl_pathv[i]);
+                                       if (total_length+length+1 >= BUFSIZ) {
                                                error_msg(out_of_space);
                                                return FALSE;
                                        }
-                                       strcat(command+total_length, expand_result.gl_pathv[i]);
                                        strcat(command+total_length, " ");
+                                       total_length+=1;
+                                       strcat(command+total_length, expand_result.gl_pathv[i]);
                                        total_length+=length;
                                }
                                globfree (&expand_result);
@@ -1084,11 +1086,11 @@ static int expand_arguments(char *command)
                        case '0':case '1':case '2':case '3':case '4':
                        case '5':case '6':case '7':case '8':case '9':
                                {
-                                       int index=*(dst + 1)-48;
-                                       if (index >= argc) {
+                                       int ixx=*(dst + 1)-48;
+                                       if (ixx >= argc) {
                                                var='\0';
                                        } else {
-                                               var = argv[index];
+                                               var = argv[ixx];
                                        }
                                }
                                break;
@@ -1101,25 +1103,24 @@ static int expand_arguments(char *command)
                } else {
                        /* Looks like an environment variable */
                        char delim_hold;
-                       int num_skip_chars=1;
+                       int num_skip_chars=0;
                        int dstlen = strlen(dst);
                        /* Is this a ${foo} type variable? */
                        if (dstlen >=2 && *(dst+1) == '{') {
                                src=strchr(dst+1, '}');
-                               num_skip_chars=2;
+                               num_skip_chars=1;
                        } else {
-                               src=strpbrk(dst+1, " \t~`!$^&*()=|\\[];\"'<>?./");
+                               src=dst+1;
+                               while(isalnum(*src) || *src=='_') src++;
                        }
                        if (src == NULL) {
                                src = dst+dstlen;
                        }
                        delim_hold=*src;
                        *src='\0';  /* temporary */
-                       var = getenv(dst + num_skip_chars);
+                       var = getenv(dst + 1 + num_skip_chars);
                        *src=delim_hold;
-                       if (num_skip_chars==2) {
-                               src++;
-                       }
+                       src += num_skip_chars;
                }
                if (var == NULL) {
                        /* Seems we got an un-expandable variable.  So delete it. */
@@ -1134,8 +1135,7 @@ static int expand_arguments(char *command)
                        }
                        /* Move stuff to the end of the string to accommodate
                         * filling the created gap with the new stuff */
-                       memmove(dst+subst_len, src, trail_len);
-                       *(dst+subst_len+trail_len)='\0';
+                       memmove(dst+subst_len, src, trail_len+1);
                        /* Now copy in the new stuff */
                        memcpy(dst, var, subst_len);
                        src = dst+subst_len;
@@ -1159,7 +1159,7 @@ static int parse_command(char **command_ptr, struct job *job, int *inbg)
        int argc_l = 0;
        int done = 0;
        int argv_alloced;
-       int i;
+       int i, saw_quote = 0;
        char quote = '\0';
        int count;
        struct child_prog *prog;
@@ -1221,7 +1221,7 @@ static int parse_command(char **command_ptr, struct job *job, int *inbg)
                                           *src == ']') *buf++ = '\\';
                        *buf++ = *src;
                } else if (isspace(*src)) {
-                       if (*prog->argv[argc_l]) {
+                       if (*prog->argv[argc_l] || saw_quote) {
                                buf++, argc_l++;
                                /* +1 here leaves room for the NULL which ends argv */
                                if ((argc_l + 1) == argv_alloced) {
@@ -1231,12 +1231,14 @@ static int parse_command(char **command_ptr, struct job *job, int *inbg)
                                                                                  argv_alloced);
                                }
                                prog->argv[argc_l] = buf;
+                               saw_quote = 0;
                        }
                } else
                        switch (*src) {
                        case '"':
                        case '\'':
                                quote = *src;
+                               saw_quote = 1;
                                break;
 
                        case '#':                       /* comment */
@@ -1289,7 +1291,7 @@ static int parse_command(char **command_ptr, struct job *job, int *inbg)
                                        chptr++;
 
                                if (!*chptr) {
-                                       error_msg("file name expected after %c", *src);
+                                       error_msg("file name expected after %c", *(src-1));
                                        free_job(job);
                                        job->num_progs=0;
                                        return 1;
@@ -1305,7 +1307,7 @@ static int parse_command(char **command_ptr, struct job *job, int *inbg)
 
                        case '|':                       /* pipe */
                                /* finish this command */
-                               if (*prog->argv[argc_l])
+                               if (*prog->argv[argc_l] || saw_quote)
                                        argc_l++;
                                if (!argc_l) {
                                        error_msg("empty command in pipe");
@@ -1460,7 +1462,6 @@ static int parse_command(char **command_ptr, struct job *job, int *inbg)
                                        break;
 #else
                                        error_msg("character expected after \\");
-                                       free(command);
                                        free_job(job);
                                        return 1;
 #endif
@@ -1475,7 +1476,7 @@ static int parse_command(char **command_ptr, struct job *job, int *inbg)
                src++;
        }
 
-       if (*prog->argv[argc_l]) {
+       if (*prog->argv[argc_l] || saw_quote) {
                argc_l++;
        }
        if (!argc_l) {
@@ -1551,9 +1552,9 @@ static int pseudo_exec(struct child_prog *child)
 #endif
 
        {
-           char** argv=child->argv;
+           char** argv_l=child->argv;
            int argc_l;
-           for(argc_l=0;*argv!=NULL; argv++, argc_l++);
+           for(argc_l=0;*argv_l!=NULL; argv_l++, argc_l++);
            optind = 1;
            run_applet_by_name(name, argc_l, child->argv);
        }
@@ -1566,19 +1567,19 @@ static int pseudo_exec(struct child_prog *child)
 static void insert_job(struct job *newjob, int inbg)
 {
        struct job *thejob;
-       struct jobset *job_list=newjob->job_list;
+       struct jobset *j_list=newjob->job_list;
 
        /* find the ID for thejob to use */
        newjob->jobid = 1;
-       for (thejob = job_list->head; thejob; thejob = thejob->next)
+       for (thejob = j_list->head; thejob; thejob = thejob->next)
                if (thejob->jobid >= newjob->jobid)
                        newjob->jobid = thejob->jobid + 1;
 
        /* add thejob to the list of running jobs */
-       if (!job_list->head) {
-               thejob = job_list->head = xmalloc(sizeof(*thejob));
+       if (!j_list->head) {
+               thejob = j_list->head = xmalloc(sizeof(*thejob));
        } else {
-               for (thejob = job_list->head; thejob->next; thejob = thejob->next) /* nothing */;
+               for (thejob = j_list->head; thejob->next; thejob = thejob->next) /* nothing */;
                thejob->next = xmalloc(sizeof(*thejob));
                thejob = thejob->next;
        }
@@ -1802,7 +1803,7 @@ static int busy_loop(FILE * input)
                        if (!job_list.fg) {
                                /* move the shell to the foreground */
                                /* suppress messages when run from /linuxrc mag@sysgo.de */
-                               if (tcsetpgrp(0, getpid()) && errno != ENOTTY)
+                               if (tcsetpgrp(0, getpgrp()) && errno != ENOTTY)
                                        perror_msg("tcsetpgrp"); 
                        }
                }
@@ -1824,8 +1825,10 @@ static int busy_loop(FILE * input)
 #ifdef BB_FEATURE_CLEAN_UP
 void free_memory(void)
 {
-       if (cwd)
+       if (cwd) {
                free(cwd);
+               cwd = NULL;
+       }
        if (local_pending_command)
                free(local_pending_command);
 
@@ -1845,7 +1848,6 @@ int shell_main(int argc_l, char **argv_l)
 
        /* These variables need re-initializing when recursing */
        shell_context = 0;
-       cwd=NULL;
        local_pending_command = NULL;
        close_me_head = NULL;
        job_list.head = NULL;
@@ -1907,7 +1909,7 @@ int shell_main(int argc_l, char **argv_l)
        if (interactive==TRUE) {
                //printf( "optind=%d  argv[optind]='%s'\n", optind, argv[optind]);
                /* Looks like they want an interactive shell */
-               printf( "\n\nBusyBox v%s (%s) Built-in shell (lash)\n", BB_VER, BB_BT);
+               printf( "\n\n" BB_BANNER " Built-in shell (lash)\n");
                printf( "Enter 'help' for a list of built-in commands.\n\n");
        } else if (local_pending_command==NULL) {
                //printf( "optind=%d  argv[optind]='%s'\n", optind, argv[optind]);
@@ -1916,8 +1918,7 @@ int shell_main(int argc_l, char **argv_l)
        }
 
        /* initialize the cwd -- this is never freed...*/
-       cwd=(char*)xmalloc(sizeof(char)*MAX_LINE+1);
-       getcwd(cwd, sizeof(char)*MAX_LINE);
+       cwd = xgetcwd(0);
 
 #ifdef BB_FEATURE_CLEAN_UP
        atexit(free_memory);
@@ -1927,7 +1928,6 @@ int shell_main(int argc_l, char **argv_l)
        cmdedit_set_initial_prompt();
 #else
        PS1 = NULL;
-       PS2 = "> ";
 #endif
        
        return (busy_loop(input));