Another whack at scripts/individual. Now builds 212 applets.

[oweals/busybox.git] / coreutils / env.c

diff --git a/coreutils/env.c b/coreutils/env.c
index aca7c3ae2f4f0ddb28153fcf8c9714da8bd5c9d0..eb1f0b2683858b2945e5454688c703323674b1f4 100644 (file)
--- a/coreutils/env.c
+++ b/coreutils/env.c
@@ -5,73 +5,102 @@
  * Copyright (c) 1988, 1993, 1994
  *     The Regents of the University of California.  All rights reserved.
  *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
  *
  * Original copyright notice is retained at the end of this file.
  *
- * Modified for BusyBox by Erik Andersen <andersen@lineo.com>, <andersee@debian.org>
+ * Modified for BusyBox by Erik Andersen <andersen@codepoet.org>
+ */
+
+/* BB_AUDIT SUSv3 compliant */
+/* http://www.opengroup.org/onlinepubs/007904975/utilities/env.html */
+
+/* Mar 16, 2003      Manuel Novoa III   (mjn3@codepoet.org)
+ *
+ * Fixed bug involving exit return codes if execvp fails.  Also added
+ * output error checking.
+ */
+
+/*
+ * Modified by Vladimir Oleynik <dzo@simtreas.ru> (C) 2003
+ * - correct "-" option usage
+ * - multiple "-u unsetenv" support
+ * - GNU long option support
+ * - use bb_default_error_retval
  */
 
+
 #include <stdio.h>
 #include <string.h>
-#include <getopt.h>
 #include <stdlib.h>
+#include <errno.h>
 #include <unistd.h>
+#include <getopt.h> /* struct option */
 #include "busybox.h"
 
-extern int env_main(int argc, char** argv)
+#if ENABLE_FEATURE_ENV_LONG_OPTIONS
+static const struct option env_long_options[] = {
+       { "ignore-environment", 0, NULL, 'i' },
+       { "unset", 1, NULL, 'u' },
+       { 0, 0, 0, 0 }
+};
+#endif
+
+int env_main(int argc, char** argv)
 {
+       static char *cleanenv[1] = { NULL };
+
        char **ep, *p;
-       char *cleanenv[1];
-       int ignore_environment = 0;
-       int ch;
-
-       while ((ch = getopt(argc, argv, "+iu:")) != -1) {
-               switch(ch) {
-               case 'i':
-                       ignore_environment = 1;
-                       break;
-               case 'u':
-                       unsetenv(optarg);
-                       break;
-               default:
-                       show_usage();
-               }
-       }
-       if (optind != argc && !strcmp(argv[optind], "-")) {
-               ignore_environment = 1;
-               argv++;
+       unsigned long opt;
+       llist_t *unset_env = NULL;
+       extern char **environ;
+
+       bb_opt_complementally = "u::";
+#if ENABLE_FEATURE_ENV_LONG_OPTIONS
+       bb_applet_long_options = env_long_options;
+#endif
+
+       opt = bb_getopt_ulflags(argc, argv, "+iu:", &unset_env);
+
+       argv += optind;
+       if (*argv && (argv[0][0] == '-') && !argv[0][1]) {
+               opt |= 1;
+               ++argv;
        }
-       if (ignore_environment) {
+
+       if(opt & 1)
                environ = cleanenv;
-               cleanenv[0] = NULL;
+       else if(opt & 2) {
+               while(unset_env) {
+                       unsetenv(unset_env->data);
+                       unset_env = unset_env->link;
+               }
        }
-       for (argv += optind; *argv && (p = strchr(*argv, '=')); ++argv)
-               putenv(*argv);
+
+       while (*argv && ((p = strchr(*argv, '=')) != NULL)) {
+               if (putenv(*argv) < 0) {
+                       bb_perror_msg_and_die("putenv");
+               }
+               ++argv;
+       }
+
        if (*argv) {
                execvp(*argv, argv);
-               perror_msg_and_die("%s", *argv);
+               /* SUSv3-mandated exit codes. */
+               bb_default_error_retval = (errno == ENOENT) ? 127 : 126;
+               bb_perror_msg_and_die("%s", *argv);
+       }
+
+       for (ep = environ; *ep; ep++) {
+               puts(*ep);
        }
-       for (ep = environ; *ep; ep++)
-               printf("%s\n", *ep);
-       return 0;
+
+       bb_fflush_stdout_and_exit(0);
 }
 
 /*
  * Copyright (c) 1988, 1993, 1994
- *     The Regents of the University of California.  All rights reserved.
+ *      The Regents of the University of California.  All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
@@ -82,8 +111,8 @@ extern int env_main(int argc, char** argv)
  *    notice, this list of conditions and the following disclaimer in the
  *    documentation and/or other materials provided with the distribution.
  *
- * 3. <BSD Advertising Clause omitted per the July 22, 1999 licensing change 
- *             ftp://ftp.cs.berkeley.edu/pub/4bsd/README.Impt.License.Change> 
+ * 3. <BSD Advertising Clause omitted per the July 22, 1999 licensing change
+ *             ftp://ftp.cs.berkeley.edu/pub/4bsd/README.Impt.License.Change>
  *
  * 4. Neither the name of the University nor the names of its contributors
  *    may be used to endorse or promote products derived from this software