1 /*
2 * Copyright (c) 1997 - 2000, 2005 Kungliga Tekniska Högskolan
3 * (Royal Institute of Technology, Stockholm, Sweden).
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 *
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 *
13 * 2. Redistributions in binary form must reproduce the above copyright
14 * notice, this list of conditions and the following disclaimer in the
15 * documentation and/or other materials provided with the distribution.
16 *
17 * 3. Neither the name of the Institute nor the names of its contributors
18 * may be used to endorse or promote products derived from this software
19 * without specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31 * SUCH DAMAGE.
32 */
33
34 #include <config.h>
35
36 #include <stdio.h>
37 #include <stdlib.h>
38 #include <string.h>
39 #include <signal.h>
40 #ifdef HAVE_TERMIOS_H
41 #include <termios.h>
42 #endif
43 #include <roken.h>
44
45 #include <ui.h>
46 #ifdef HAVE_CONIO_H
47 #include <conio.h>
48 #endif
49
50 static sig_atomic_t intr_flag;
51
52 static void
intr(int sig)53 intr(int sig)
54 {
55 intr_flag++;
56 }
57
58 #ifdef HAVE_CONIO_H
59
60 /*
61 * Windows does console slightly different then then unix case.
62 */
63
64 static int
read_string(const char * preprompt,const char * prompt,char * buf,size_t len,int echo)65 read_string(const char *preprompt, const char *prompt,
66 char *buf, size_t len, int echo)
67 {
68 int of = 0;
69 int c;
70 char *p;
71 void (*oldsigintr)(int);
72
73 _cprintf("%s%s", preprompt, prompt);
74
75 oldsigintr = signal(SIGINT, intr);
76
77 p = buf;
78 while(intr_flag == 0){
79 c = ((echo)? _getche(): _getch());
80 if(c == '\n' || c == '\r')
81 break;
82 if(of == 0)
83 *p++ = c;
84 of = (p == buf + len);
85 }
86 if(of)
87 p--;
88 *p = 0;
89
90 if(echo == 0){
91 printf("\n");
92 }
93
94 signal(SIGINT, oldsigintr);
95
96 if(intr_flag)
97 return -2;
98 if(of)
99 return -1;
100 return 0;
101 }
102
103 #else /* !HAVE_CONIO_H */
104
105 #ifndef NSIG
106 #define NSIG 47
107 #endif
108
109 static int
read_string(const char * preprompt,const char * prompt,char * buf,size_t len,int echo)110 read_string(const char *preprompt, const char *prompt,
111 char *buf, size_t len, int echo)
112 {
113 struct sigaction sigs[NSIG];
114 int oksigs[NSIG];
115 struct sigaction sa;
116 FILE *tty;
117 int ret = 0;
118 int of = 0;
119 int i;
120 int c;
121 char *p;
122
123 struct termios t_new, t_old;
124
125 memset(&oksigs, 0, sizeof(oksigs));
126
127 memset(&sa, 0, sizeof(sa));
128 sa.sa_handler = intr;
129 sigemptyset(&sa.sa_mask);
130 sa.sa_flags = 0;
131 for(i = 1; i < sizeof(sigs) / sizeof(sigs[0]); i++)
132 if (i != SIGALRM)
133 if (sigaction(i, &sa, &sigs[i]) == 0)
134 oksigs[i] = 1;
135
136 if((tty = fopen("/dev/tty", "r")) != NULL)
137 rk_cloexec_file(tty);
138 else
139 tty = stdin;
140
141 fprintf(stderr, "%s%s", preprompt, prompt);
142 fflush(stderr);
143
144 if(echo == 0){
145 tcgetattr(fileno(tty), &t_old);
146 memcpy(&t_new, &t_old, sizeof(t_new));
147 t_new.c_lflag &= ~ECHO;
148 tcsetattr(fileno(tty), TCSANOW, &t_new);
149 }
150 intr_flag = 0;
151 p = buf;
152 while(intr_flag == 0){
153 c = getc(tty);
154 if(c == EOF){
155 if(!ferror(tty))
156 ret = 1;
157 break;
158 }
159 if(c == '\n')
160 break;
161 if(of == 0)
162 *p++ = c;
163 of = (p == buf + len);
164 }
165 if(of)
166 p--;
167 *p = 0;
168
169 if(echo == 0){
170 fprintf(stderr, "\n");
171 tcsetattr(fileno(tty), TCSANOW, &t_old);
172 }
173
174 if(tty != stdin)
175 fclose(tty);
176
177 for(i = 1; i < sizeof(sigs) / sizeof(sigs[0]); i++)
178 if (oksigs[i])
179 sigaction(i, &sigs[i], NULL);
180
181 if(ret)
182 return -3;
183 if(intr_flag)
184 return -2;
185 if(of)
186 return -1;
187 return 0;
188 }
189
190 #endif /* HAVE_CONIO_H */
191
192 int
UI_UTIL_read_pw_string(char * buf,int length,const char * prompt,int verify)193 UI_UTIL_read_pw_string(char *buf, int length, const char *prompt, int verify)
194 {
195 int ret;
196
197 ret = read_string("", prompt, buf, length, 0);
198 if (ret)
199 return ret;
200
201 if (verify) {
202 char *buf2;
203 buf2 = malloc(length);
204 if (buf2 == NULL)
205 return 1;
206
207 ret = read_string("Verify password - ", prompt, buf2, length, 0);
208 if (ret) {
209 free(buf2);
210 return ret;
211 }
212 if (strcmp(buf2, buf) != 0)
213 ret = 1;
214 free(buf2);
215 }
216 return ret;
217 }
218