Fix the pwd and group functions. The bb_ stuff was a leftover from
[oweals/busybox.git] / sh.c
diff --git a/sh.c b/sh.c
index dd1d3aa5c37b9c2f58cc88f9a618e0ec65fddf62..bd6945605867c4b5ccfda9be50ea64b713dd7bb9 100644 (file)
--- a/sh.c
+++ b/sh.c
@@ -2,7 +2,7 @@
 /*
  * lash -- the BusyBox Lame-Ass SHell
  *
- * Copyright (C) 2000 by Lineo, inc.
+ * Copyright (C) 1999,2000,2001 by Lineo, inc.
  * Written by Erik Andersen <andersen@lineo.com>, <andersee@debian.org>
  *
  * Based in part on ladsh.c by Michael K. Johnson and Erik W. Troan, which is
  *
  */
 
+/* The parsing engine of this program is officially at a dead-end.
+ * Future work in that direction should move to the work posted
+ * at http://doolittle.faludi.com/~larry/parser.html .
+ * A start on the integration of that work with the rest of sh.c
+ * is at http://codepoet.org/sh.c .
+ */
 //
 //This works pretty well now, and is now on by default.
 #define BB_FEATURE_SH_ENVIRONMENT
@@ -58,8 +64,8 @@
 #include <getopt.h>
 #include "cmdedit.h"
 
-#define MAX_LINE       256     /* size of input buffer for cwd data */
-#define MAX_READ       128     /* size of input buffer for `read' builtin */
+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"
 extern size_t NUM_APPLETS;
 
@@ -73,8 +79,6 @@ static const unsigned int IF_TRUE_CONTEXT=0x2;
 static const unsigned int IF_FALSE_CONTEXT=0x4;
 static const unsigned int THEN_EXP_CONTEXT=0x8;
 static const unsigned int ELSE_EXP_CONTEXT=0x10;
-unsigned int shell_context = 0;
-
 
 
 struct jobset {
@@ -119,6 +123,11 @@ struct built_in_command {
        int (*function) (struct child_prog *);  /* function ptr */
 };
 
+struct close_me {
+       int fd;
+       struct close_me *next;
+};
+
 /* function prototypes for builtins */
 static int builtin_cd(struct child_prog *cmd);
 static int builtin_env(struct child_prog *dummy);
@@ -143,6 +152,8 @@ static int run_command_predicate(char *cmd);
 
 
 /* function prototypes for shell stuff */
+static void mark_open(int fd);
+static void mark_closed(int fd);
 static void checkjobs(struct jobset *job_list);
 static int get_command(FILE * source, char *command);
 static int parse_command(char **command_ptr, struct job *job, int *inbg);
@@ -184,11 +195,19 @@ static struct built_in_command bltins_forking[] = {
        {NULL, NULL, NULL}
 };
 
+
+/* Variables we export */
+unsigned int shell_context;  /* Used in cmdedit.c to reset the
+                                                               context when someone hits ^C */
+
+
+/* Globals that are static to this file */
 static char *cwd;
 static char *local_pending_command = NULL;
 static struct jobset job_list = { NULL, NULL };
 static int argc;
 static char **argv;
+static struct close_me *close_me_head = NULL;
 #ifdef BB_FEATURE_SH_ENVIRONMENT
 static int last_bg_pid=-1;
 static int last_return_code=-1;
@@ -271,7 +290,7 @@ static int builtin_env(struct child_prog *dummy)
        char **e;
 
        for (e = environ; *e; e++) {
-               fprintf(stdout, "%s\n", *e);
+               printf( "%s\n", *e);
        }
        return (0);
 }
@@ -387,7 +406,7 @@ static int builtin_jobs(struct child_prog *child)
 static int builtin_pwd(struct child_prog *dummy)
 {
        getcwd(cwd, MAX_LINE);
-       fprintf(stdout, "%s\n", cwd);
+       printf( "%s\n", cwd);
        return EXIT_SUCCESS;
 }
 
@@ -540,19 +559,23 @@ static int builtin_source(struct child_prog *child)
 {
        FILE *input;
        int status;
+       int fd;
 
        if (child->argv[1] == NULL)
                return EXIT_FAILURE;
 
        input = fopen(child->argv[1], "r");
        if (!input) {
-               fprintf(stdout, "Couldn't open file '%s'\n", child->argv[1]);
+               printf( "Couldn't open file '%s'\n", child->argv[1]);
                return EXIT_FAILURE;
        }
 
+       fd=fileno(input);
+       mark_open(fd);
        /* Now run the file */
        status = busy_loop(input);
        fclose(input);
+       mark_closed(fd);
        return (status);
 }
 
@@ -560,7 +583,7 @@ static int builtin_source(struct child_prog *child)
 static int builtin_unset(struct child_prog *child)
 {
        if (child->argv[1] == NULL) {
-               fprintf(stdout, "unset: parameter required.\n");
+               printf( "unset: parameter required.\n");
                return EXIT_FAILURE;
        }
        unsetenv(child->argv[1]);
@@ -587,6 +610,34 @@ static int run_command_predicate(char *cmd)
 }
 #endif
 
+static void mark_open(int fd)
+{
+       struct close_me *new = xmalloc(sizeof(struct close_me));
+       new->fd = fd;
+       new->next = close_me_head;
+       close_me_head = new;
+}
+
+static void mark_closed(int fd)
+{
+       struct close_me *tmp;
+       if (close_me_head == NULL || close_me_head->fd != fd)
+               error_msg_and_die("corrupt close_me");
+       tmp = close_me_head;
+       close_me_head = close_me_head->next;
+       free(tmp);
+}
+
+static void close_all()
+{
+       struct close_me *c;
+       for (c=close_me_head; c; c=c->next) {
+               close(c->fd);
+       }
+       close_me_head = NULL;
+}
+
+
 /* free up all memory from a job */
 static void free_job(struct job *cmd)
 {
@@ -762,7 +813,7 @@ static char* setup_prompt_string(int state)
 static int get_command(FILE * source, char *command)
 {
        char *prompt_str;
-       
+
        if (source == NULL) {
                if (local_pending_command) {
                        /* a command specified (-c option): return it & mark it done */
@@ -774,9 +825,9 @@ static int get_command(FILE * source, char *command)
                return 1;
        }
 
-       prompt_str = setup_prompt_string(shell_context);
-       
        if (source == stdin) {
+               prompt_str = setup_prompt_string(shell_context);
+
 #ifdef BB_FEATURE_SH_COMMAND_EDITING
                /*
                ** enable command line editing only while a command line
@@ -786,11 +837,12 @@ static int get_command(FILE * source, char *command)
                */
                cmdedit_init();
                cmdedit_read_input(prompt_str, command);
-               free( prompt_str);
+               free(prompt_str);
                cmdedit_terminate();
                return 0;
 #else
-               fprintf(stdout, "%s", prompt_str);
+               fputs(prompt_str, stdout);
+               free(prompt_str);
 #endif
        }
 
@@ -1146,6 +1198,9 @@ static int parse_command(char **command_ptr, struct job *job, int *inbg)
 #ifdef BB_FEATURE_SH_BACKTICKS
                        case '`':
                                /* Exec a backtick-ed command */
+                               /* Besides any previous brokenness, I have not
+                                * updated backtick handling for close_me support.
+                                * I don't know if it needs it or not.  -- LRD */
                                {
                                        char* charptr1=NULL, *charptr2;
                                        char* ptr=NULL;
@@ -1354,8 +1409,7 @@ static int pseudo_exec(struct child_prog *child)
 #endif
 
        execvp(child->argv[0], child->argv);
-       error_msg_and_die("%s: %s\n", child->argv[0],
-                       strerror(errno));
+       perror_msg_and_die("%s", child->argv[0]);
 }
 
 static void insert_job(struct job *newjob, int inbg)
@@ -1459,6 +1513,8 @@ static int run_command(struct job *newjob, int inbg, int outpipe[2])
                if (!(child->pid = fork())) {
                        signal(SIGTTOU, SIG_DFL);
 
+                       close_all();
+
                        if (outpipe[1]!=-1) {
                                close(outpipe[0]);
                        }
@@ -1554,7 +1610,8 @@ static int busy_loop(FILE * input)
                        while (!job_list.fg->progs[i].pid ||
                                   job_list.fg->progs[i].is_stopped == 1) i++;
 
-                       waitpid(job_list.fg->progs[i].pid, &status, WUNTRACED);
+                       if (waitpid(job_list.fg->progs[i].pid, &status, WUNTRACED)<0)
+                               perror_msg_and_die("waitpid(%d)",job_list.fg->progs[i].pid);
 
                        if (WIFEXITED(status) || WIFSIGNALED(status)) {
                                /* the child exited */
@@ -1608,8 +1665,6 @@ static int busy_loop(FILE * input)
 #ifdef BB_FEATURE_CLEAN_UP
 void free_memory(void)
 {
-       if (prompt_str)
-               free(prompt_str);
        if (cwd)
                free(cwd);
        if (local_pending_command)
@@ -1629,17 +1684,33 @@ int shell_main(int argc_l, char **argv_l)
        argc = argc_l;
        argv = argv_l;
 
+       shell_context = 0;
+       cwd=NULL;
+#ifdef BB_FEATURE_SH_STANDALONE_SHELL
+       /* These variables need re-initializing when recursing */
+       local_pending_command = NULL;
+       job_list.head = NULL;
+       job_list.fg = NULL;
+#ifdef BB_FEATURE_SH_ENVIRONMENT
+       last_bg_pid=-1;
+       last_return_code=-1;
+       show_x_trace=FALSE;
+#endif
+#endif
 
        if (argv[0] && argv[0][0] == '-') {
-                 FILE *input;
-                 input = fopen("/etc/profile", "r");
-                 if (!input) {
-                         fprintf(stdout, "Couldn't open file '/etc/profile'\n");
-                 } else {
-                         /* Now run the file */
-                         busy_loop(input);
-                         fclose(input);
-                 }
+               FILE *prof_input;
+               prof_input = fopen("/etc/profile", "r");
+               if (!prof_input) {
+                       printf( "Couldn't open file '/etc/profile'\n");
+               } else {
+                       int tmp_fd = fileno(prof_input);
+                       mark_open(tmp_fd);      
+                       /* Now run the file */
+                       busy_loop(prof_input);
+                       fclose(prof_input);
+                       mark_closed(tmp_fd);
+               }
        }
 
        while ((opt = getopt(argc_l, argv_l, "cxi")) > 0) {
@@ -1676,13 +1747,14 @@ int shell_main(int argc_l, char **argv_l)
                interactive=TRUE;
        }
        if (interactive==TRUE) {
-               //fprintf(stdout, "optind=%d  argv[optind]='%s'\n", optind, argv[optind]);
+               //printf( "optind=%d  argv[optind]='%s'\n", optind, argv[optind]);
                /* Looks like they want an interactive shell */
-               fprintf(stdout, "\n\nBusyBox v%s (%s) Built-in shell (lash)\n", BB_VER, BB_BT);
-               fprintf(stdout, "Enter 'help' for a list of built-in commands.\n\n");
+               printf( "\n\nBusyBox v%s (%s) Built-in shell (lash)\n", BB_VER, BB_BT);
+               printf( "Enter 'help' for a list of built-in commands.\n\n");
        } else if (local_pending_command==NULL) {
-               //fprintf(stdout, "optind=%d  argv[optind]='%s'\n", optind, argv[optind]);
+               //printf( "optind=%d  argv[optind]='%s'\n", optind, argv[optind]);
                input = xfopen(argv[optind], "r");
+               mark_open(fileno(input));  /* be lazy, never mark this closed */
        }
 
        /* initialize the cwd -- this is never freed...*/