ash: Expand here-documents in the current shell environment

[oweals/busybox.git] / libbb / bb_askpass.c

diff --git a/libbb/bb_askpass.c b/libbb/bb_askpass.c
index cf384e52b9d1915b2f2275048c2b4443f898bfb5..2dcead35ab2149d47de280f562259f81edf67f4e 100644 (file)
--- a/libbb/bb_askpass.c
+++ b/libbb/bb_askpass.c
@@ -1,75 +1,99 @@
 /* vi: set sw=4 ts=4: */
 /*
  * Ask for a password
- * I use a static buffer in this function.  Plan accordingly.
  *
  * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.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 <unistd.h>
-#include <fcntl.h>
-#include <signal.h>
-#include <termios.h>
-#include <sys/ioctl.h>
-
 #include "libbb.h"
 
 /* do nothing signal handler */
-static void askpass_timeout(int ATTRIBUTE_UNUSED ignore)
+static void askpass_timeout(int UNUSED_PARAM ignore)
 {
 }
 
-char *bb_askpass(int timeout, const char * prompt)
+char* FAST_FUNC bb_ask_noecho(int fd, int timeout, const char *prompt)
 {
-       static char passwd[64];
-
+#define MAX_LINE 0xfff
        char *ret;
        int i;
-       struct sigaction sa;
-       struct termios old, new;
-
-       tcgetattr(STDIN_FILENO, &old);
-       tcflush(STDIN_FILENO, TCIFLUSH);
-
-       memset(passwd, 0, sizeof(passwd));
+       struct sigaction sa, oldsa;
+       struct termios tio, oldtio;
 
+       tcflush(fd, TCIFLUSH);
+       /* Was buggy: was printing prompt *before* flushing input,
+        * which was upsetting "expect" based scripts of some users.
+        */
        fputs(prompt, stdout);
-       fflush(stdout);
+       fflush_all();
 
-       tcgetattr(STDIN_FILENO, &new);
-       new.c_iflag &= ~(IUCLC|IXON|IXOFF|IXANY);
-       new.c_lflag &= ~(ECHO|ECHOE|ECHOK|ECHONL|TOSTOP);
-       tcsetattr(STDIN_FILENO, TCSANOW, &new);
+       tcgetattr(fd, &oldtio);
+       tio = oldtio;
+       /* Switch off echo. ECHOxyz meaning:
+        * ECHO    echo input chars
+        * ECHOE   echo BS-SP-BS on erase character
+        * ECHOK   echo kill char specially, not as ^c (ECHOKE controls how exactly)
+        * ECHOKE  erase all input via BS-SP-BS on kill char (else go to next line)
+        * ECHOCTL Echo ctrl chars as ^c (else echo verbatim:
+        *         e.g. up arrow emits "ESC-something" and thus moves cursor up!)
+        * ECHONL  Echo NL even if ECHO is not set
+        * ECHOPRT On erase, echo erased chars
+        *         [qwe<BS><BS><BS> input looks like "qwe\ewq/" on screen]
+        */
+       tio.c_lflag &= ~(ECHO|ECHOE|ECHOK|ECHONL);
+       tcsetattr(fd, TCSANOW, &tio);
 
+       memset(&sa, 0, sizeof(sa));
+       /* sa.sa_flags = 0; - no SA_RESTART! */
+       /* SIGINT and SIGALRM will interrupt reads below */
+       sa.sa_handler = askpass_timeout;
+       sigaction(SIGINT, &sa, &oldsa);
        if (timeout) {
-               sa.sa_flags = 0;
-               sa.sa_handler = askpass_timeout;
-               sigaction(SIGALRM, &sa, NULL);
+               sigaction_set(SIGALRM, &sa);
                alarm(timeout);
        }
 
        ret = NULL;
-       if (read(STDIN_FILENO, passwd, sizeof(passwd)-1) > 0) {
-               ret = passwd;
-               i = 0;
-               /* Last byte is guaranteed to be 0
-                  (read did not overwrite it) */
-               do {
-                       if (passwd[i] == '\r' || passwd[i] == '\n')
-                               passwd[i] = 0;
-               } while (passwd[i++]);
+       i = 0;
+       while (1) {
+               int r;
+
+               /* User input is uber-slow, no need to optimize reallocs.
+                * Grow it on every char.
+                */
+               ret = xrealloc(ret, i + 2);
+               r = read(fd, &ret[i], 1);
+
+               if ((i == 0 && r == 0) /* EOF (^D) with no password */
+                || r < 0 /* read is interrupted by timeout or ^C */
+               ) {
+                       ret[i] = '\0'; /* paranoia */
+                       nuke_str(ret); /* paranoia */
+                       free(ret);
+                       ret = NULL;
+                       break;
+               }
+
+               if (r == 0 /* EOF */
+                || ret[i] == '\r' || ret[i] == '\n' /* EOL */
+                || ++i == MAX_LINE /* line limit */
+               ) {
+                       ret[i] = '\0';
+                       break;
+               }
        }
 
        if (timeout) {
                alarm(0);
        }
-
-       tcsetattr(STDIN_FILENO, TCSANOW, &old);
-       puts("");
-       fflush(stdout);
+       sigaction_set(SIGINT, &oldsa);
+       tcsetattr(fd, TCSANOW, &oldtio);
+       bb_putchar('\n');
+       fflush_all();
        return ret;
 }
+char* FAST_FUNC bb_ask_noecho_stdin(const char *prompt)
+{
+       return bb_ask_noecho(STDIN_FILENO, 0, prompt);
+}