lineedit: do not hang on error, but return error indicator.

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

diff --git a/coreutils/printenv.c b/coreutils/printenv.c
index aea88d701063a62a2af6a0ed5a9414493d350965..d38f8fb5f5986cdd134a78aa4b185d7fd4734711 100644 (file)
--- a/coreutils/printenv.c
+++ b/coreutils/printenv.c
@@ -5,37 +5,33 @@
  * Copyright (C) 2005 by Erik Andersen <andersen@codepoet.org>
  * Copyright (C) 2005 by Mike Frysinger <vapier@gentoo.org>
  *
- * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
+ * Licensed under GPLv2 or later, see file LICENSE in this source tree.
  */
 
-#include <stdio.h>
-#include <string.h>
-#include <stdlib.h>
-#include "busybox.h"
+#include "libbb.h"
 
-int printenv_main(int argc, char **argv)
+int printenv_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
+int printenv_main(int argc UNUSED_PARAM, char **argv)
 {
-       extern char **environ;
-       int e = 0;
+       int exit_code = EXIT_SUCCESS;
 
        /* no variables specified, show whole env */
-       if (argc == 1)
+       if (!argv[1]) {
+               int e = 0;
                while (environ[e])
                        puts(environ[e++]);
-
-       /* search for specified variables and print them out if found */
-       else {
-               int i;
-               size_t l;
+       } else {
+               /* search for specified variables and print them out if found */
                char *arg, *env;
 
-               for (i=1; (arg = argv[i]); ++i)
-                       for (; (env = environ[e]); ++e) {
-                               l = strlen(arg);
-                               if (!strncmp(env, arg, l) && env[l] == '=')
-                                       puts(env + l + 1);
-                       }
+               while ((arg = *++argv) != NULL) {
+                       env = getenv(arg);
+                       if (env)
+                               puts(env);
+                       else
+                               exit_code = EXIT_FAILURE;
+               }
        }
 
-       bb_fflush_stdout_and_exit(0);
+       fflush_stdout_and_exit(exit_code);
 }