--- gforth/engine/io.c	2009/06/13 10:22:24	1.35
+++ gforth/engine/io.c	2012/04/06 16:34:07	1.43
@@ -1,6 +1,6 @@
 /* direct key io driver
 
-  Copyright (C) 1995,1996,1997,1998,1999,2002,2003,2006,2007,2008 Free Software Foundation, Inc.
+  Copyright (C) 1995,1996,1997,1998,1999,2002,2003,2006,2007,2008,2009,2010 Free Software Foundation, Inc.
 
   This file is part of Gforth.
 
@@ -39,6 +39,7 @@
 typedef unsigned int uint32_t;
 #endif
 
+#include <stdlib.h>
 #include <stdio.h>
 #include <signal.h>
 #include <string.h>
@@ -54,6 +55,7 @@ typedef unsigned int uint32_t;
 #include "io.h"
 
 #ifndef MSDOS
+#include <poll.h>
 #if defined (__GNUC__)
 #  define alloca __builtin_alloca
 #else
@@ -446,8 +448,6 @@ void deprep_terminal ()
 #define VTIME VEOL
 #endif
 
-#include <locale.h>
-
 void prep_terminal ()
 {
   int tty = fileno (stdin);
@@ -473,9 +473,6 @@ void prep_terminal ()
     return;      /* added by MdG */
   }      /* added by MdG */
    
-  setlocale(LC_ALL, "");
-  setlocale(LC_NUMERIC, "C");
-
   /* Try to keep this function from being INTerrupted.  We can do it
      on POSIX and systems with BSD-like signal handling. */
 #if defined (HAVE_POSIX_SIGNALS)
@@ -617,20 +614,80 @@ void deprep_terminal ()
 }
 #endif  /* NEW_TTY_DRIVER */
 
-long key_avail (FILE * stream)
+/* an ungetc() variant where we can know that there is a character waiting:
+   gf_ungetc: like ungetc
+   gf_regetc: call when reading a char, but does not get the character
+   gf_ungottenc: true if there is a char waiting
+
+   Implementation: just an array containing all the FILEs with ungotten chars.
+   Sequential search for membership checking (there should not be many elements)
+*/
+
+static FILE **ungotten_files = NULL;
+static size_t n_ungotten = 0; /* actual number */
+static size_t max_ungotten = 0; /* buffer size */
+
+int gf_ungetc(int c, FILE *stream)
+/* like ungetc, but works with the others */
+{
+  if (n_ungotten>=max_ungotten) {
+    max_ungotten = max_ungotten*2+1;
+    ungotten_files = realloc(ungotten_files, max_ungotten*sizeof(FILE *));
+  }
+  ungotten_files[n_ungotten++] = stream;
+  return ungetc(c, stream);
+}
+
+static long search_ungotten(FILE *stream)
+{
+  long i;
+  for (i=0; i<n_ungotten; i++)
+    if (ungotten_files[i] == stream)
+      return i;
+  return -1;
+}
+
+void gf_regetc(FILE *stream)
+/* remove STREAM from ungotten_files */
+{
+  long i = search_ungotten(stream);
+  if (i>=0)
+    ungotten_files[i] = ungotten_files[--n_ungotten];
+}
+
+int gf_ungottenc(FILE *stream)
+/* true if stream has been ungotten */
+{
+  return search_ungotten(stream)>=0;
+}
+
+long key_avail (FILE *stream)
 {
   int tty = fileno (stream);
-  fd_set selin;
-  static struct timeval now = { 0 , 0 };
+  struct pollfd fds = { tty, POLLIN, 0 };
   int chars_avail;
 
+  if (gf_ungottenc(stream))
+    return 1;
   setvbuf(stream, NULL, _IONBF, 0);
   if(!terminal_prepped && stream == stdin)
     prep_terminal();
 
-  FD_ZERO(&selin);
-  FD_SET(tty, &selin);
-  chars_avail = select(1, &selin, NULL, NULL, &now);
+#if defined(FIONREAD) && !defined(_WIN32)
+  if(isatty (tty)) {
+    int result = ioctl (tty, FIONREAD, &chars_avail);
+  } else
+#endif
+  {
+    chars_avail = poll(&fds, 1, 0);
+  }
+  if (chars_avail > 0) {
+    /* getc won't block */
+    int c = getc(stream);
+    if (c==EOF)
+      return 0;
+    gf_ungetc(c, stream);
+  }
   return (chars_avail == -1) ? 0 : chars_avail;
 }
 
@@ -646,11 +703,14 @@ Cell getkey(FILE * stream)
   Cell result;
   unsigned char c;
 
-  setvbuf(stream, NULL, _IONBF, 0);
+  if (!gf_ungottenc(stream))
+    setvbuf(stream, NULL, _IONBF, 0);
   if(!terminal_prepped && stream == stdin)
     prep_terminal();
 
   result = fread(&c, sizeof(c), 1, stream);
+  if (result>0)
+    gf_regetc(stream);
   return result==0 ? (errno == EINTR ? 12 : 4) : c;
 }