Implemented server main loop and started with import command.
[gnupg.git] / util / ttyio.c
index 38143cb..6aaff00 100644 (file)
@@ -1,14 +1,14 @@
 /* ttyio.c -  tty i/O functions
- *     Copyright (C) 1998 Free Software Foundation, Inc.
+ *     Copyright (C) 1998, 2000 Free Software Foundation, Inc.
  *
- * This file is part of GNUPG.
+ * This file is part of GnuPG.
  *
- * GNUPG is free software; you can redistribute it and/or modify
+ * GnuPG is free software; you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
  * the Free Software Foundation; either version 2 of the License, or
  * (at your option) any later version.
  *
- * GNUPG is distributed in the hope that it will be useful,
+ * GnuPG is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  * GNU General Public License for more details.
 #include <unistd.h>
 #ifdef HAVE_TCGETATTR
   #include <termios.h>
+#else
+  #ifdef HAVE_TERMIO_H
+    /* simulate termios with termio */
+    #include <termio.h>
+    #define termios termio
+    #define tcsetattr ioctl
+    #define TCSAFLUSH TCSETAF
+    #define tcgetattr(A,B) ioctl(A,TCGETA,B)
+    #define HAVE_TCGETATTR
+  #endif
 #endif
 #ifdef __MINGW32__ /* use the odd Win32 functions */
   #include <windows.h>
 #endif
 #include <errno.h>
 #include <ctype.h>
+#include <gcrypt.h>
 #include "util.h"
 #include "memory.h"
 #include "ttyio.h"
 
+#define CONTROL_D ('D' - 'A' + 1)
+
 
 #ifdef __MINGW32__ /* use the odd Win32 functions */
 static struct {
@@ -55,6 +68,8 @@ static FILE *ttyfp = NULL;
 
 static int initialized;
 static int last_prompt_len;
+static int batchmode;
+static int no_terminal;
 
 #ifdef HAVE_TCGETATTR
 static struct termios termsave;
@@ -74,7 +89,7 @@ cleanup(void)
 #endif
 
 static void
-init_ttyfp()
+init_ttyfp(void)
 {
     if( initialized )
        return;
@@ -103,10 +118,14 @@ init_ttyfp()
     SetConsoleMode(con.in, DEF_INPMODE );
     SetConsoleMode(con.out, DEF_OUTMODE );
 
+  #elif defined(__EMX__)
+    ttyfp = stdout; /* Fixme: replace by the real functions: see wklib */
   #else
-    ttyfp = fopen("/dev/tty", "r+");
-    if( !ttyfp )
-       log_fatal("cannot open /dev/tty: %s\n", strerror(errno) );
+    ttyfp = batchmode? stderr : fopen("/dev/tty", "r+");
+    if( !ttyfp ) {
+       log_error("cannot open /dev/tty: %s\n", strerror(errno) );
+       exit(2);
+    }
   #endif
   #ifdef HAVE_TCGETATTR
     atexit( cleanup );
@@ -114,12 +133,31 @@ init_ttyfp()
     initialized = 1;
 }
 
+int
+tty_batchmode( int onoff )
+{
+    int old = batchmode;
+    if( onoff != -1 )
+       batchmode = onoff;
+    return old;
+}
+
+int
+tty_no_terminal(int onoff)
+{
+    int old = no_terminal;
+    no_terminal = onoff ? 1 : 0;
+    return old;
+}
 
 void
 tty_printf( const char *fmt, ... )
 {
     va_list arg_ptr;
 
+    if (no_terminal)
+       return;
+
     if( !initialized )
        init_ttyfp();
 
@@ -134,19 +172,19 @@ tty_printf( const char *fmt, ... )
             * it, so we use a static buffer for now */
        do {
            if( n == -1 || !buf ) {
-               m_free(buf);
+               gcry_free(buf);
                bufsize += 200;
                /* better check the new size; (we use M$ functions) */
                if( bufsize > 50000 )
                    log_bug("vsnprintf probably failed\n");
-               buf = m_alloc( bufsize );
+               buf = gcry_xmalloc( bufsize );
            }
            n = _vsnprintf(buf, bufsize-1, fmt, arg_ptr);
        } while( n == -1 );
       #else
        if( !buf ) {
            bufsize += 1000;
-           buf = m_alloc( bufsize );
+           buf = gcry_xmalloc( bufsize );
        }
        n = vsprintf(buf, fmt, arg_ptr);
        if( n == -1 )
@@ -173,6 +211,9 @@ tty_printf( const char *fmt, ... )
 void
 tty_print_string( byte *p, size_t n )
 {
+    if (no_terminal)
+       return;
+
     if( !initialized )
        init_ttyfp();
 
@@ -205,6 +246,43 @@ tty_print_string( byte *p, size_t n )
   #endif
 }
 
+void
+tty_print_utf8_string2( byte *p, size_t n, size_t max_n )
+{
+    size_t i;
+    char *buf;
+
+    if (no_terminal)
+       return;
+
+    /* we can handle plain ascii simpler, so check for it first */
+    for(i=0; i < n; i++ ) {
+       if( p[i] & 0x80 )
+           break;
+    }
+    if( i < n ) {
+       buf = utf8_to_native( p, n );
+       if( strlen( buf ) > max_n ) {
+           buf[max_n] = 0;
+       }
+       /*(utf8 conversion already does the control character quoting)*/
+       tty_printf("%s", buf );
+       gcry_free( buf );
+    }
+    else {
+       if( n > max_n ) {
+           n = max_n;
+       }
+       tty_print_string( p, n );
+    }
+}
+
+
+void
+tty_print_utf8_string( byte *p, size_t n )
+{
+    tty_print_utf8_string2( p, n, n );
+}
 
 
 
@@ -216,15 +294,25 @@ do_get( const char *prompt, int hidden )
     byte cbuf[1];
     int c, n, i;
 
+    if( batchmode ) {
+       log_error("Sorry, we are in batchmode - can't get input\n");
+       exit(2);
+    }
+
+    if (no_terminal) {
+       log_error("Sorry, no terminal at all requested - can't get input\n");
+       exit(2);
+    }
+
     if( !initialized )
        init_ttyfp();
 
     last_prompt_len = 0;
     tty_printf( prompt );
-    buf = m_alloc(n=50);
+    buf = gcry_xmalloc(n=50);
     i = 0;
 
-  #if __MINGW32__ /* windoze version */
+  #ifdef __MINGW32__ /* windoze version */
     if( hidden )
        SetConsoleMode(con.in, HID_INPMODE );
 
@@ -250,7 +338,7 @@ do_get( const char *prompt, int hidden )
            continue;
        if( !(i < n-1) ) {
            n += 50;
-           buf = m_realloc( buf, n );
+           buf = gcry_xrealloc( buf, n );
        }
        buf[i++] = c;
     }
@@ -279,6 +367,8 @@ do_get( const char *prompt, int hidden )
        if( !hidden )
            last_prompt_len++;
        c = *cbuf;
+       if( c == CONTROL_D )
+           log_info("control d found\n");
        if( c == '\t' )
            c = ' ';
        else if( c > 0xa0 )
@@ -288,10 +378,14 @@ do_get( const char *prompt, int hidden )
            continue;
        if( !(i < n-1) ) {
            n += 50;
-           buf = m_realloc( buf, n );
+           buf = gcry_xrealloc( buf, n );
        }
        buf[i++] = c;
     }
+    if( *cbuf != '\n' ) {
+       buf[0] = CONTROL_D;
+       i = 1;
+    }
 
 
     if( hidden ) {
@@ -323,12 +417,17 @@ tty_get_hidden( const char *prompt )
 void
 tty_kill_prompt()
 {
+    if ( no_terminal )
+       return;
 
     if( !initialized )
        init_ttyfp();
+
+    if( batchmode )
+       last_prompt_len = 0;
     if( !last_prompt_len )
        return;
-  #if __MINGW32__
+  #ifdef __MINGW32__
     tty_printf("\r%*s\r", last_prompt_len, "");
   #else
     {
@@ -351,7 +450,7 @@ tty_get_answer_is_yes( const char *prompt )
     char *p = tty_get( prompt );
     tty_kill_prompt();
     yes = answer_is_yes(p);
-    m_free(p);
+    gcry_free(p);
     return yes;
 }