Eric Andersen | 6f9a778 | 2004-05-01 01:27:30 +0000 | [diff] [blame] | 1 | /* vi: set sw=4 ts=4: */ |
| 2 | /* |
| 3 | * Ask for a password |
Eric Andersen | 6f9a778 | 2004-05-01 01:27:30 +0000 | [diff] [blame] | 4 | * |
| 5 | * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org> |
| 6 | * |
Denys Vlasenko | 0ef64bd | 2010-08-16 20:14:46 +0200 | [diff] [blame] | 7 | * Licensed under GPLv2 or later, see file LICENSE in this source tree. |
Eric Andersen | 6f9a778 | 2004-05-01 01:27:30 +0000 | [diff] [blame] | 8 | */ |
| 9 | |
Bernhard Reutner-Fischer | 20f4000 | 2006-01-30 17:17:14 +0000 | [diff] [blame] | 10 | #include "libbb.h" |
Eric Andersen | 6f9a778 | 2004-05-01 01:27:30 +0000 | [diff] [blame] | 11 | |
| 12 | /* do nothing signal handler */ |
Denis Vlasenko | a60f84e | 2008-07-05 09:18:54 +0000 | [diff] [blame] | 13 | static void askpass_timeout(int UNUSED_PARAM ignore) |
Eric Andersen | 6f9a778 | 2004-05-01 01:27:30 +0000 | [diff] [blame] | 14 | { |
| 15 | } |
| 16 | |
Bernhard Reutner-Fischer | 82b1429 | 2008-12-03 18:48:39 +0000 | [diff] [blame] | 17 | char* FAST_FUNC bb_ask_stdin(const char *prompt) |
| 18 | { |
| 19 | return bb_ask(STDIN_FILENO, 0, prompt); |
| 20 | } |
| 21 | char* FAST_FUNC bb_ask(const int fd, int timeout, const char *prompt) |
Eric Andersen | 6f9a778 | 2004-05-01 01:27:30 +0000 | [diff] [blame] | 22 | { |
Denis Vlasenko | 91e149a | 2007-06-18 10:35:06 +0000 | [diff] [blame] | 23 | /* Was static char[BIGNUM] */ |
| 24 | enum { sizeof_passwd = 128 }; |
Denis Vlasenko | 6429aab | 2006-09-23 12:22:11 +0000 | [diff] [blame] | 25 | |
Denys Vlasenko | 02859aa | 2015-10-09 18:16:40 +0200 | [diff] [blame] | 26 | char *passwd; |
Eric Andersen | 6f9a778 | 2004-05-01 01:27:30 +0000 | [diff] [blame] | 27 | char *ret; |
Denis Vlasenko | 6429aab | 2006-09-23 12:22:11 +0000 | [diff] [blame] | 28 | int i; |
Denis Vlasenko | e5387a0 | 2007-10-20 19:20:22 +0000 | [diff] [blame] | 29 | struct sigaction sa, oldsa; |
| 30 | struct termios tio, oldtio; |
Eric Andersen | 6f9a778 | 2004-05-01 01:27:30 +0000 | [diff] [blame] | 31 | |
Denys Vlasenko | d32fc64 | 2014-07-01 13:20:22 +0200 | [diff] [blame] | 32 | tcflush(fd, TCIFLUSH); |
| 33 | /* Was buggy: was printing prompt *before* flushing input, |
| 34 | * which was upsetting "expect" based scripts of some users. |
| 35 | */ |
Denys Vlasenko | 7449e18 | 2011-10-22 06:27:41 +0200 | [diff] [blame] | 36 | fputs(prompt, stdout); |
| 37 | fflush_all(); |
Denys Vlasenko | 7449e18 | 2011-10-22 06:27:41 +0200 | [diff] [blame] | 38 | |
| 39 | tcgetattr(fd, &oldtio); |
Denis Vlasenko | e5387a0 | 2007-10-20 19:20:22 +0000 | [diff] [blame] | 40 | tio = oldtio; |
Denys Vlasenko | 7449e18 | 2011-10-22 06:27:41 +0200 | [diff] [blame] | 41 | #if 0 |
| 42 | /* Switch off UPPERCASE->lowercase conversion (never used since 198x) |
| 43 | * and XON/XOFF (why we want to mess with this??) |
| 44 | */ |
| 45 | # ifndef IUCLC |
| 46 | # define IUCLC 0 |
| 47 | # endif |
Denis Vlasenko | e5387a0 | 2007-10-20 19:20:22 +0000 | [diff] [blame] | 48 | tio.c_iflag &= ~(IUCLC|IXON|IXOFF|IXANY); |
Denys Vlasenko | 7449e18 | 2011-10-22 06:27:41 +0200 | [diff] [blame] | 49 | #endif |
| 50 | /* Switch off echo */ |
| 51 | tio.c_lflag &= ~(ECHO|ECHOE|ECHOK|ECHONL); |
Denys Vlasenko | 557deb1 | 2010-02-03 12:17:06 +0100 | [diff] [blame] | 52 | tcsetattr(fd, TCSANOW, &tio); |
Eric Andersen | 6f9a778 | 2004-05-01 01:27:30 +0000 | [diff] [blame] | 53 | |
Denis Vlasenko | e5387a0 | 2007-10-20 19:20:22 +0000 | [diff] [blame] | 54 | memset(&sa, 0, sizeof(sa)); |
| 55 | /* sa.sa_flags = 0; - no SA_RESTART! */ |
Denys Vlasenko | e936c6d | 2010-02-01 04:55:30 +0100 | [diff] [blame] | 56 | /* SIGINT and SIGALRM will interrupt reads below */ |
Denis Vlasenko | e5387a0 | 2007-10-20 19:20:22 +0000 | [diff] [blame] | 57 | sa.sa_handler = askpass_timeout; |
| 58 | sigaction(SIGINT, &sa, &oldsa); |
Eric Andersen | 6f9a778 | 2004-05-01 01:27:30 +0000 | [diff] [blame] | 59 | if (timeout) { |
Denis Vlasenko | 8e2cfec | 2008-03-12 23:19:35 +0000 | [diff] [blame] | 60 | sigaction_set(SIGALRM, &sa); |
Eric Andersen | 6f9a778 | 2004-05-01 01:27:30 +0000 | [diff] [blame] | 61 | alarm(timeout); |
| 62 | } |
| 63 | |
Denys Vlasenko | 02859aa | 2015-10-09 18:16:40 +0200 | [diff] [blame] | 64 | passwd = auto_string(xmalloc(sizeof_passwd)); |
Denys Vlasenko | e936c6d | 2010-02-01 04:55:30 +0100 | [diff] [blame] | 65 | ret = passwd; |
| 66 | i = 0; |
| 67 | while (1) { |
| 68 | int r = read(fd, &ret[i], 1); |
Jonathan Liu | b6dc13c | 2013-05-21 17:01:55 +0200 | [diff] [blame] | 69 | if ((i == 0 && r == 0) /* EOF (^D) with no password */ |
| 70 | || r < 0 |
| 71 | ) { |
Denys Vlasenko | e936c6d | 2010-02-01 04:55:30 +0100 | [diff] [blame] | 72 | /* read is interrupted by timeout or ^C */ |
| 73 | ret = NULL; |
| 74 | break; |
| 75 | } |
| 76 | if (r == 0 /* EOF */ |
| 77 | || ret[i] == '\r' || ret[i] == '\n' /* EOL */ |
| 78 | || ++i == sizeof_passwd-1 /* line limit */ |
| 79 | ) { |
| 80 | ret[i] = '\0'; |
| 81 | break; |
| 82 | } |
Eric Andersen | 6f9a778 | 2004-05-01 01:27:30 +0000 | [diff] [blame] | 83 | } |
| 84 | |
| 85 | if (timeout) { |
| 86 | alarm(0); |
| 87 | } |
Denis Vlasenko | 8e2cfec | 2008-03-12 23:19:35 +0000 | [diff] [blame] | 88 | sigaction_set(SIGINT, &oldsa); |
Denys Vlasenko | 557deb1 | 2010-02-03 12:17:06 +0100 | [diff] [blame] | 89 | tcsetattr(fd, TCSANOW, &oldtio); |
Denis Vlasenko | 4daad90 | 2007-09-27 10:20:47 +0000 | [diff] [blame] | 90 | bb_putchar('\n'); |
Denys Vlasenko | 8131eea | 2009-11-02 14:19:51 +0100 | [diff] [blame] | 91 | fflush_all(); |
Eric Andersen | 6f9a778 | 2004-05-01 01:27:30 +0000 | [diff] [blame] | 92 | return ret; |
| 93 | } |