distributed version 0.1.0
[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 #ifdef HAVE_TCGETATTR
27   #include <termios.h>
28 #endif
29 #include <errno.h>
30 #include <ctype.h>
31 #include "util.h"
32 #include "memory.h"
33 #include "ttyio.h"
34
35 static FILE *ttyfp = NULL;
36 static int last_prompt_len;
37
38 static void
39 init_ttyfp()
40 {
41     if( ttyfp )
42         return;
43
44   #if defined(__MINGW32__)
45     ttyfp = stderr; /* fixme */
46   #else
47     ttyfp = fopen("/dev/tty", "r+");
48   #endif
49     if( !ttyfp )
50         log_fatal("cannot open /dev/tty: %s\n", strerror(errno) );
51 }
52
53
54 void
55 tty_printf( const char *fmt, ... )
56 {
57     va_list arg_ptr;
58
59     if( !ttyfp )
60         init_ttyfp();
61
62     va_start( arg_ptr, fmt ) ;
63     last_prompt_len += vfprintf(ttyfp,fmt,arg_ptr) ;
64     va_end(arg_ptr);
65     fflush(ttyfp);
66 }
67
68
69 /****************
70  * Print a string, but filter all control characters out.
71  */
72 void
73 tty_print_string( byte *p, size_t n )
74 {
75     if( !ttyfp )
76         init_ttyfp();
77
78     for( ; n; n--, p++ )
79         if( iscntrl( *p ) ) {
80             putc('\\', ttyfp);
81             if( *p == '\n' )
82                 putc('n', ttyfp);
83             else if( !*p )
84                 putc('0', ttyfp);
85             else
86                 fprintf(ttyfp, "x%02x", *p );
87         }
88         else
89             putc(*p, ttyfp);
90 }
91
92
93
94
95
96 static char *
97 do_get( const char *prompt, int hidden )
98 {
99     char *buf;
100     byte cbuf[1];
101     int c, n, i;
102     FILE *fp;
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