0be618fc916cf21d6d7e6122d0ac847759ae6002
[ira/wip.git] / lib / replace / getpass.c
1 /* Copyright (C) 1992-1998 Free Software Foundation, Inc.
2 This file is part of the GNU C Library.
3
4 The GNU C Library is free software; you can redistribute it and/or
5 modify it under the terms of the GNU Lesser General Public License as
6 published by the Free Software Foundation; either version 3 of the
7 License, or (at your option) any later version.
8
9 The GNU C Library is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12 Library General Public License for more details.
13
14 You should have received a copy of the GNU Lesser General Public
15 License along with the GNU C Library; see the file COPYING.LIB.  If
16 not, see <http://www.gnu.org/licenses/>.  */
17
18 /* Modified to use with samba by Jeremy Allison, 8th July 1995. */
19
20 #include "replace.h"
21 #include "system/filesys.h"
22 #include "system/wait.h"
23 #include "system/terminal.h"
24 #include "system/passwd.h"
25
26 /*
27  * Define additional missing types
28  */
29 #ifndef HAVE_SIG_ATOMIC_T_TYPE
30 typedef int sig_atomic_t;
31 #endif
32
33 #ifndef SIGCLD
34 #define SIGCLD SIGCHLD
35 #endif
36
37 #ifndef SIGNAL_CAST
38 #define SIGNAL_CAST (RETSIGTYPE (*)(int))
39 #endif
40
41 #ifdef SYSV_TERMIO 
42
43 /* SYSTEM V TERMIO HANDLING */
44
45 static struct termio t;
46
47 #define ECHO_IS_ON(t) ((t).c_lflag & ECHO)
48 #define TURN_ECHO_OFF(t) ((t).c_lflag &= ~ECHO)
49 #define TURN_ECHO_ON(t) ((t).c_lflag |= ECHO)
50
51 #ifndef TCSAFLUSH
52 #define TCSAFLUSH 1
53 #endif
54
55 #ifndef TCSANOW
56 #define TCSANOW 0
57 #endif
58
59 static int tcgetattr(int fd, struct termio *_t)
60 {
61         return ioctl(fd, TCGETA, _t);
62 }
63
64 static int tcsetattr(int fd, int flags, struct termio *_t)
65 {
66         if(flags & TCSAFLUSH)
67                 ioctl(fd, TCFLSH, TCIOFLUSH);
68         return ioctl(fd, TCSETS, _t);
69 }
70
71 #elif !defined(TCSAFLUSH)
72
73 /* BSD TERMIO HANDLING */
74
75 static struct sgttyb t;  
76
77 #define ECHO_IS_ON(t) ((t).sg_flags & ECHO)
78 #define TURN_ECHO_OFF(t) ((t).sg_flags &= ~ECHO)
79 #define TURN_ECHO_ON(t) ((t).sg_flags |= ECHO)
80
81 #define TCSAFLUSH 1
82 #define TCSANOW 0
83
84 static int tcgetattr(int fd, struct sgttyb *_t)
85 {
86         return ioctl(fd, TIOCGETP, (char *)_t);
87 }
88
89 static int tcsetattr(int fd, int flags, struct sgttyb *_t)
90 {
91         return ioctl(fd, TIOCSETP, (char *)_t);
92 }
93
94 #else /* POSIX TERMIO HANDLING */
95 #define ECHO_IS_ON(t) ((t).c_lflag & ECHO)
96 #define TURN_ECHO_OFF(t) ((t).c_lflag &= ~ECHO)
97 #define TURN_ECHO_ON(t) ((t).c_lflag |= ECHO)
98
99 static struct termios t;
100 #endif /* SYSV_TERMIO */
101
102 static void catch_signal(int signum,void (*handler)(int ))
103 {
104 #ifdef HAVE_SIGACTION
105         struct sigaction act;
106         struct sigaction oldact;
107
108         memset(&act, 0, sizeof(act));
109
110         act.sa_handler = handler;
111 #ifdef SA_RESTART
112         /*
113          * We *want* SIGALRM to interrupt a system call.
114          */
115         if(signum != SIGALRM)
116                 act.sa_flags = SA_RESTART;
117 #endif
118         sigemptyset(&act.sa_mask);
119         sigaddset(&act.sa_mask,signum);
120         sigaction(signum,&act,&oldact);
121 #else /* !HAVE_SIGACTION */
122         /* FIXME: need to handle sigvec and systems with broken signal() */
123         signal(signum, handler);
124 #endif
125 }
126
127 static sig_atomic_t gotintr;
128 static int in_fd = -1;
129
130 /***************************************************************
131  Signal function to tell us were ^C'ed.
132 ****************************************************************/
133
134 static void gotintr_sig(void)
135 {
136         gotintr = 1;
137         if (in_fd != -1)
138                 close(in_fd); /* Safe way to force a return. */
139         in_fd = -1;
140 }
141
142 char *rep_getpass(const char *prompt)
143 {
144         FILE *in, *out;
145         int echo_off;
146         static char buf[256];
147         static size_t bufsize = sizeof(buf);
148         size_t nread;
149
150         /* Catch problematic signals */
151         catch_signal(SIGINT, SIGNAL_CAST gotintr_sig);
152
153         /* Try to write to and read from the terminal if we can.
154                 If we can't open the terminal, use stderr and stdin.  */
155
156         in = fopen ("/dev/tty", "w+");
157         if (in == NULL) {
158                 in = stdin;
159                 out = stderr;
160         } else {
161                 out = in;
162         }
163
164         setvbuf(in, NULL, _IONBF, 0);
165
166         /* Turn echoing off if it is on now.  */
167
168         if (tcgetattr (fileno (in), &t) == 0) {
169                 if (ECHO_IS_ON(t)) {
170                         TURN_ECHO_OFF(t);
171                         echo_off = tcsetattr (fileno (in), TCSAFLUSH, &t) == 0;
172                         TURN_ECHO_ON(t);
173                 } else {
174                         echo_off = 0;
175                 }
176         } else {
177                 echo_off = 0;
178         }
179
180         /* Write the prompt.  */
181         fputs(prompt, out);
182         fflush(out);
183
184         /* Read the password.  */
185         buf[0] = 0;
186         if (!gotintr) {
187                 in_fd = fileno(in);
188                 if (fgets(buf, bufsize, in) == NULL) {
189                         buf[0] = 0;
190                 }
191         }
192         nread = strlen(buf);
193         if (nread) {
194                 if (buf[nread - 1] == '\n')
195                         buf[nread - 1] = '\0';
196         }
197
198         /* Restore echoing.  */
199         if (echo_off) {
200                 if (gotintr && in_fd == -1) {
201                         in = fopen ("/dev/tty", "w+");
202                 }
203                 if (in != NULL)
204                         tcsetattr (fileno (in), TCSANOW, &t);
205         }
206
207         fprintf(out, "\n");
208         fflush(out);
209
210         if (in && in != stdin) /* We opened the terminal; now close it.  */
211                 fclose(in);
212
213         /* Catch problematic signals */
214         catch_signal(SIGINT, SIGNAL_CAST SIG_DFL);
215
216         if (gotintr) {
217                 printf("Interupted by signal.\n");
218                 fflush(stdout);
219                 exit(1);
220         }
221         return buf;
222 }