* http.c (connect_server): [_WIN32] actually fill in the sin_addr so we
[gnupg.git] / util / ttyio.c
index e5b2b58..add7467 100644 (file)
@@ -1,5 +1,6 @@
 /* ttyio.c -  tty i/O functions
- * Copyright (C) 1998, 1999, 2000, 2001, 2002 Free Software Foundation, Inc.
+ * Copyright (C) 1998, 1999, 2000, 2001, 2002,
+ *               2004 Free Software Foundation, Inc.
  *
  * This file is part of GnuPG.
  *
 #include <stdarg.h>
 #include <unistd.h>
 #ifdef HAVE_TCGETATTR
-  #include <termios.h>
+#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
+#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 _WIN32 /* use the odd Win32 functions */
+#include <windows.h>
+#ifdef HAVE_TCGETATTR
+#error windows and termios
 #endif
-#ifdef __MINGW32__ /* use the odd Win32 functions */
-  #include <windows.h>
-  #ifdef HAVE_TCGETATTR
-     #error mingw32 and termios
-  #endif
 #endif
 #include <errno.h>
 #include <ctype.h>
+#ifdef HAVE_READLINE_READLINE_H
+#include <readline/readline.h>
+#include <readline/history.h>
+#endif
+
 #include "util.h"
 #include "memory.h"
 #include "ttyio.h"
 
 #define CONTROL_D ('D' - 'A' + 1)
 
-#ifdef __MINGW32__ /* use the odd Win32 functions */
+#ifdef _WIN32 /* use the odd Win32 functions */
 static struct {
     HANDLE in, out;
 } con;
@@ -99,8 +105,8 @@ tty_get_ttyname (void)
       got_name = 1;
     }
 #endif
-  /* Assume the staandrd tty on memory error or when tehre is no
-     certmid. */
+  /* Assume the standard tty on memory error or when there is no
+     ctermid. */
   return name? name : "/dev/tty";
 }
 
@@ -123,7 +129,7 @@ init_ttyfp(void)
     if( initialized )
        return;
 
-#if defined(__MINGW32__)
+#if defined(_WIN32)
     {
        SECURITY_ATTRIBUTES sa;
 
@@ -156,6 +162,10 @@ init_ttyfp(void)
                   tty_get_ttyname (), strerror(errno) );
        exit(2);
     }
+#if defined(HAVE_READLINE_READLINE_H) && defined(HAVE_LIBREADLINE)
+    rl_catch_signals = 0;
+    rl_instream = rl_outstream = ttyfp;
+#endif
 #endif
 #ifdef HAVE_TCGETATTR
     atexit( cleanup );
@@ -192,7 +202,7 @@ tty_printf( const char *fmt, ... )
        init_ttyfp();
 
     va_start( arg_ptr, fmt ) ;
-  #ifdef __MINGW32__
+#ifdef _WIN32
     {   
         char *buf = NULL;
         int n;
@@ -209,10 +219,57 @@ tty_printf( const char *fmt, ... )
        last_prompt_len += n;
         m_free (buf);
     }
-  #else
+#else
+    last_prompt_len += vfprintf(ttyfp,fmt,arg_ptr) ;
+    fflush(ttyfp);
+#endif
+    va_end(arg_ptr);
+}
+
+
+/* Same as tty_printf but if FP is not NULL, behave like the standard
+   fprintf. */
+void
+tty_fprintf (FILE *fp, const char *fmt, ... )
+{
+  va_list arg_ptr;
+
+  if (fp)
+    {
+      va_start (arg_ptr, fmt) ;
+      vfprintf (fp, fmt, arg_ptr );
+      va_end (arg_ptr);
+      return;
+    }
+
+  if (no_terminal)
+    return;
+
+  if( !initialized )
+    init_ttyfp();
+
+    va_start( arg_ptr, fmt ) ;
+#ifdef _WIN32
+    {   
+        char *buf = NULL;
+        int n;
+       DWORD nwritten;
+
+       n = vasprintf(&buf, fmt, arg_ptr);
+       if( !buf )
+           log_bug("vasprintf() failed\n");
+        
+       if( !WriteConsoleA( con.out, buf, n, &nwritten, NULL ) )
+           log_fatal("WriteConsole failed: rc=%d", (int)GetLastError() );
+       if( n != nwritten )
+           log_fatal("WriteConsole failed: %d != %d\n", n, (int)nwritten );
+       last_prompt_len += n;
+        xfree (buf);
+    }
+#else
     last_prompt_len += vfprintf(ttyfp,fmt,arg_ptr) ;
     fflush(ttyfp);
-  #endif
+#endif
     va_end(arg_ptr);
 }
 
@@ -221,7 +278,7 @@ tty_printf( const char *fmt, ... )
  * Print a string, but filter all control characters out.
  */
 void
-tty_print_string( byte *p, size_t n )
+tty_print_string( const byte *p, size_t n )
 {
     if (no_terminal)
        return;
@@ -229,7 +286,7 @@ tty_print_string( byte *p, size_t n )
     if( !initialized )
        init_ttyfp();
 
-  #ifdef __MINGW32__
+#ifdef _WIN32
     /* not so effective, change it if you want */
     for( ; n; n--, p++ )
        if( iscntrl( *p ) ) {
@@ -242,7 +299,7 @@ tty_print_string( byte *p, size_t n )
        }
        else
            tty_printf("%c", *p);
-  #else
+#else
     for( ; n; n--, p++ )
        if( iscntrl( *p ) ) {
            putc('\\', ttyfp);
@@ -255,11 +312,11 @@ tty_print_string( byte *p, size_t n )
        }
        else
            putc(*p, ttyfp);
-  #endif
+#endif
 }
 
 void
-tty_print_utf8_string2( byte *p, size_t n, size_t max_n )
+tty_print_utf8_string2 (const byte *p, size_t n, size_t max_n )
 {
     size_t i;
     char *buf;
@@ -290,7 +347,7 @@ tty_print_utf8_string2( byte *p, size_t n, size_t max_n )
 }
 
 void
-tty_print_utf8_string( byte *p, size_t n )
+tty_print_utf8_string( const byte *p, size_t n )
 {
     tty_print_utf8_string2( p, n, 0 );
 }
@@ -300,9 +357,9 @@ static char *
 do_get( const char *prompt, int hidden )
 {
     char *buf;
-  #ifndef __riscos__
+#ifndef __riscos__
     byte cbuf[1];
-  #endif
+#endif
     int c, n, i;
 
     if( batchmode ) {
@@ -323,7 +380,7 @@ do_get( const char *prompt, int hidden )
     buf = m_alloc(n=50);
     i = 0;
 
-  #ifdef __MINGW32__ /* windoze version */
+#ifdef _WIN32 /* windoze version */
     if( hidden )
        SetConsoleMode(con.in, HID_INPMODE );
 
@@ -357,7 +414,7 @@ do_get( const char *prompt, int hidden )
     if( hidden )
        SetConsoleMode(con.in, DEF_INPMODE );
 
-  #elif defined(__riscos__)
+#elif defined(__riscos__)
     do {
         c = riscos_getchar();
         if (c == 0xa || c == 0xd) { /* Return || Enter */
@@ -397,9 +454,9 @@ do_get( const char *prompt, int hidden )
         }
     } while (c != '\n');
     i = (i>0) ? i-1 : 0;
-  #else /* unix version */
+#else /* unix version */
     if( hidden ) {
-      #ifdef HAVE_TCGETATTR
+#ifdef HAVE_TCGETATTR
        struct termios term;
 
        if( tcgetattr(fileno(ttyfp), &termsave) )
@@ -409,7 +466,7 @@ do_get( const char *prompt, int hidden )
        term.c_lflag &= ~(ECHO | ECHOE | ECHOK | ECHONL);
        if( tcsetattr( fileno(ttyfp), TCSAFLUSH, &term ) )
            log_fatal("tcsetattr() failed: %s\n", strerror(errno) );
-      #endif
+#endif
     }
 
     /* fixme: How can we avoid that the \n is echoed w/o disabling
@@ -440,13 +497,13 @@ do_get( const char *prompt, int hidden )
 
 
     if( hidden ) {
-      #ifdef HAVE_TCGETATTR
+#ifdef HAVE_TCGETATTR
        if( tcsetattr(fileno(ttyfp), TCSAFLUSH, &termsave) )
            log_error("tcsetattr() failed: %s\n", strerror(errno) );
        restore_termios = 0;
-      #endif
+#endif
     }
-  #endif /* end unix version */
+#endif /* end unix version */
     buf[i] = 0;
     return buf;
 }
@@ -455,7 +512,40 @@ do_get( const char *prompt, int hidden )
 char *
 tty_get( const char *prompt )
 {
-    return do_get( prompt, 0 );
+#if defined(HAVE_READLINE_READLINE_H) && defined(HAVE_LIBREADLINE)
+    if (!batchmode && !no_terminal) {
+      char *line;
+      char *buf;
+
+      if( !initialized )
+       init_ttyfp();
+
+      last_prompt_len = 0;
+
+      line = readline (prompt?prompt:"");
+
+      /* We need to copy it to memory controlled by our malloc
+         implementations; further we need to convert an EOF to our
+         convention. */
+      buf = m_alloc(line? strlen(line)+1:2);
+      if (line)
+        {
+          strcpy (buf, line);
+          trim_spaces (buf);
+          if (strlen (buf) > 2 )
+            add_history (line); /* Note that we test BUF but add LINE. */
+          free (line);
+        }
+      else
+        {
+          buf[0] = CONTROL_D;
+          buf[1] = 0;
+        }
+      return buf;
+    }
+    else
+#endif /* HAVE_READLINE_READLINE_H && HAVE_LIBREADLINE */
+      return do_get( prompt, 0 );
 }
 
 char *
@@ -478,9 +568,9 @@ tty_kill_prompt()
        last_prompt_len = 0;
     if( !last_prompt_len )
        return;
-  #ifdef __MINGW32__
+#ifdef _WIN32
     tty_printf("\r%*s\r", last_prompt_len, "");
-  #else
+#else
     {
        int i;
        putc('\r', ttyfp);
@@ -489,7 +579,7 @@ tty_kill_prompt()
        putc('\r', ttyfp);
        fflush(ttyfp);
     }
-  #endif
+#endif
     last_prompt_len = 0;
 }
 
@@ -504,4 +594,3 @@ tty_get_answer_is_yes( const char *prompt )
     m_free(p);
     return yes;
 }
-