added some trust model stuff
[gnupg.git] / util / ttyio.c
1 /* ttyio.c -  tty i/O functions
2  *      Copyright (c) 1997 by Werner Koch (dd9jn)
3  *
4  * This file is part of G10.
5  *
6  * G10 is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  *
11  * G10 is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
19  */
20
21 #include <config.h>
22 #include <stdio.h>
23 #include <stdlib.h>
24 #include <string.h>
25 #include <stdarg.h>
26 #include <unistd.h>
27 #ifdef HAVE_TCGETATTR
28   #include <termios.h>
29 #endif
30 #include <errno.h>
31 #include <ctype.h>
32 #include "util.h"
33 #include "memory.h"
34 #include "ttyio.h"
35
36 static FILE *ttyfp = NULL;
37 static int last_prompt_len;
38
39 static void
40 init_ttyfp()
41 {
42     if( ttyfp )
43         return;
44
45   #if defined(__MINGW32__)
46     ttyfp = stderr; /* fixme */
47   #else
48     ttyfp = fopen("/dev/tty", "r+");
49   #endif
50     if( !ttyfp )
51         log_fatal("cannot open /dev/tty: %s\n", strerror(errno) );
52 }
53
54
55 void
56 tty_printf( const char *fmt, ... )
57 {
58     va_list arg_ptr;
59
60     if( !ttyfp )
61         init_ttyfp();
62
63     va_start( arg_ptr, fmt ) ;
64     last_prompt_len += vfprintf(ttyfp,fmt,arg_ptr) ;
65     va_end(arg_ptr);
66     fflush(ttyfp);
67 }
68
69
70 /****************
71  * Print a string, but filter all control characters out.
72  */
73 void
74 tty_print_string( byte *p, size_t n )
75 {
76     if( !ttyfp )
77         init_ttyfp();
78
79     for( ; n; n--, p++ )
80         if( iscntrl( *p ) ) {
81             putc('\\', ttyfp);
82             if( *p == '\n' )
83                 putc('n', ttyfp);
84             else if( !*p )
85                 putc('0', ttyfp);
86             else
87                 fprintf(ttyfp, "x%02x", *p );
88         }
89         else
90             putc(*p, ttyfp);
91 }
92
93
94
95
96
97 static char *
98 do_get( const char *prompt, int hidden )
99 {
100     char *buf;
101     byte cbuf[1];
102     int c, n, i;
103   #ifdef HAVE_TCGETATTR
104     struct termios termsave;
105   #endif
106
107     if( !ttyfp )
108         init_ttyfp();
109
110     last_prompt_len = 0;
111     tty_printf( prompt );
112     buf = m_alloc(n=50);
113     i = 0;
114
115     if( hidden ) {
116       #ifdef HAVE_TCGETATTR
117         struct termios term;
118
119         if( tcgetattr(fileno(ttyfp), &termsave) )
120             log_fatal("tcgetattr() failed: %s\n", strerror(errno) );
121         term = termsave;
122         term.c_lflag &= ~(ECHO | ECHOE | ECHOK | ECHONL);
123         if( tcsetattr( fileno(ttyfp), TCSAFLUSH, &term ) )
124             log_fatal("tcsetattr() failed: %s\n", strerror(errno) );
125       #endif
126     }
127
128     /* fixme: How can we avoid that the \n is echoed w/o disabling
129      * canonical mode - w/o this kill_prompt can't work */
130     while( read(fileno(ttyfp), cbuf, 1) == 1 && *cbuf != '\n' ) {
131         if( !hidden )
132             last_prompt_len++;
133         c = *cbuf;
134         if( c == '\t' )
135             c = ' ';
136         else if( iscntrl(c) )
137             continue;
138         if( !(i < n-1) ) {
139             n += 50;
140             buf = m_realloc( buf, n );
141         }
142         buf[i++] = c;
143     }
144
145
146     if( hidden ) {
147       #ifdef HAVE_TCGETATTR
148         if( tcsetattr(fileno(ttyfp), TCSAFLUSH, &termsave) )
149             log_error("tcsetattr() failed: %s\n", strerror(errno) );
150       #endif
151     }
152     buf[i] = 0;
153     return buf;
154 }
155
156
157 char *
158 tty_get( const char *prompt )
159 {
160     return do_get( prompt, 0 );
161 }
162
163 char *
164 tty_get_hidden( const char *prompt )
165 {
166     return do_get( prompt, 1 );
167 }
168
169
170 void
171 tty_kill_prompt()
172 {
173     int i;
174
175     if( !ttyfp )
176         init_ttyfp();
177     if( !last_prompt_len )
178         return;
179     fputc('\r', ttyfp);
180     for(i=0; i < last_prompt_len; i ++ )
181         fputc(' ', ttyfp);
182     fputc('\r', ttyfp);
183     last_prompt_len = 0;
184     fflush(ttyfp);
185 }
186