2009-03-02 Zoltan Varga <vargaz@gmail.com>
[mono.git] / mono / metadata / console-io.c
index 6e374674bcca8a7b3ebb5c22e8270547263d6e98..42bd547d7d869665834270f700d99f8ce6e486b5 100644 (file)
@@ -4,7 +4,7 @@
  * Author:
  *     Gonzalo Paniagua Javier (gonzalo@ximian.com)
  *
- * Copyright (C) 2005 Novell, Inc. (http://www.novell.com)
+ * Copyright (C) 2005-2008 Novell, Inc. (http://www.novell.com)
  */
 
 #include <config.h>
 #include <string.h>
 #include <errno.h>
 #include <signal.h>
+#ifdef HAVE_SYS_TIME_H
 #include <sys/time.h>
+#endif
 #include <sys/types.h>
+#ifdef HAVE_UNISTD_H
 #include <unistd.h>
+#endif
 #include <mono/metadata/appdomain.h>
 #include <mono/metadata/object-internals.h>
 #include <mono/metadata/class-internals.h>
@@ -36,8 +40,8 @@
 #ifdef HAVE_SYS_FILIO_H
 #include <sys/filio.h>
 #endif
-#ifndef PLATFORM_WIN32
 #ifndef TIOCGWINSZ
+#ifdef HAVE_SYS_IOCTL_H
 #include <sys/ioctl.h>
 #endif
 #endif
 
 static gboolean setup_finished;
 static gboolean atexit_called;
+
+/* The string used to return the terminal to its previous state */
 static gchar *teardown_str;
 
-#ifdef PLATFORM_WIN32
+/* The string used to set the terminal into keypad xmit mode after SIGCONT is received */
+static gchar *keypad_xmit_str;
+
+#ifdef HAVE_TERMIOS_H
+/* This is the last state used by Mono, used after a CONT signal is received */
+static struct termios mono_attr;
+#endif
+
+#if defined (PLATFORM_WIN32) || defined (MONO_NULL_TTYDRIVER)
 MonoBoolean
 ves_icall_System_ConsoleDriver_Isatty (HANDLE handle)
 {
@@ -77,13 +91,7 @@ ves_icall_System_ConsoleDriver_InternalKeyAvailable (gint32 timeout)
 }
 
 MonoBoolean
-ves_icall_System_ConsoleDriver_TtySetup (MonoString *teardown, char *verase, char *vsusp, char *intr)
-{
-       return FALSE;
-}
-
-MonoBoolean
-ves_icall_System_ConsoleDriver_GetTtySize (HANDLE handle, gint32 *width, gint32 *height)
+ves_icall_System_ConsoleDriver_TtySetup (MonoString *keypad, MonoString *teardown, char *verase, char *vsusp, char*intr, int **size)
 {
        return FALSE;
 }
@@ -126,12 +134,14 @@ set_property (gint property, gboolean value)
        if (tcsetattr (STDIN_FILENO, TCSANOW, &attr) == -1)
                return FALSE;
 
+       mono_attr = attr;
        return TRUE;
 }
 
 MonoBoolean
 ves_icall_System_ConsoleDriver_SetEcho (MonoBoolean want_echo)
 {
+       
        return set_property (ECHO, want_echo);
 }
 
@@ -176,6 +186,27 @@ ves_icall_System_ConsoleDriver_InternalKeyAvailable (gint32 timeout)
        return (ret > 0) ? ret : 0;
 }
 
+static gint32 cols_and_lines;
+
+#ifdef TIOCGWINSZ
+static int
+terminal_get_dimensions (void)
+{
+       struct winsize ws;
+
+       if (ioctl (STDIN_FILENO, TIOCGWINSZ, &ws) == 0)
+               return (ws.ws_col << 16) | ws.ws_row;
+
+       return -1;
+}
+#else
+static int
+terminal_get_dimensions (void)
+{
+       return -1;
+}
+#endif
+
 static void
 tty_teardown (void)
 {
@@ -245,26 +276,118 @@ sigint_handler (int signo)
        in_sigint = FALSE;
 }
 
-MonoBoolean
-ves_icall_System_ConsoleDriver_TtySetup (MonoString *teardown, char *verase, char *vsusp, char*intr)
+static struct sigaction save_sigcont, save_sigint, save_sigwinch;
+
+static void
+sigcont_handler (int signo, void *the_siginfo, void *data)
 {
-       struct termios attr;
+       // Ignore error, there is not much we can do in the sigcont handler.
+       tcsetattr (STDIN_FILENO, TCSANOW, &mono_attr);
+
+       if (keypad_xmit_str != NULL)
+               write (STDOUT_FILENO, keypad_xmit_str, strlen (keypad_xmit_str));
+
+       // Call previous handler
+       if (save_sigcont.sa_sigaction != NULL &&
+           save_sigcont.sa_sigaction != (void *)SIG_DFL &&
+           save_sigcont.sa_sigaction != (void *)SIG_IGN)
+               (*save_sigcont.sa_sigaction) (signo, the_siginfo, data);
+}
+
+static void
+sigwinch_handler (int signo, void *the_siginfo, void *data)
+{
+       int dims = terminal_get_dimensions ();
+       if (dims != -1)
+               cols_and_lines = dims;
        
-       MONO_ARCH_SAVE_REGS;
+       // Call previous handler
+       if (save_sigwinch.sa_sigaction != NULL &&
+           save_sigwinch.sa_sigaction != (void *)SIG_DFL &&
+           save_sigwinch.sa_sigaction != (void *)SIG_IGN)
+               (*save_sigwinch.sa_sigaction) (signo, the_siginfo, data);
+}
+
+void
+console_set_signal_handlers ()
+{
+       struct sigaction sigcont, sigint, sigwinch;
+
+       memset (&sigcont, 0, sizeof (struct sigaction));
+       memset (&sigint, 0, sizeof (struct sigaction));
+       memset (&sigwinch, 0, sizeof (struct sigaction));
+       
+       // Continuing
+       sigcont.sa_handler = (void *) sigcont_handler;
+       sigcont.sa_flags = 0;
+       sigemptyset (&sigcont.sa_mask);
+       sigaction (SIGCONT, &sigcont, &save_sigcont);
+       
+       // Interrupt handler
+       sigint.sa_handler = sigint_handler;
+       sigint.sa_flags = 0;
+       sigemptyset (&sigint.sa_mask);
+       sigaction (SIGINT, &sigint, &save_sigint);
+
+       // Window size changed
+       sigwinch.sa_handler = (void *) sigwinch_handler;
+       sigwinch.sa_flags = 0;
+       sigemptyset (&sigwinch.sa_mask);
+       sigaction (SIGWINCH, &sigwinch, &save_sigwinch);
+}
+
+//
+// Currently unused, should we ever call the restore handler?
+// Perhaps before calling into Process.Start?
+//
+void
+console_restore_signal_handlers ()
+{
+       sigaction (SIGCONT, &save_sigcont, NULL);
+       sigaction (SIGINT, &save_sigint, NULL);
+       sigaction (SIGWINCH, &save_sigwinch, NULL);
+}
 
+MonoBoolean
+ves_icall_System_ConsoleDriver_TtySetup (MonoString *keypad, MonoString *teardown, char *verase, char *vsusp, char*intr, int **size)
+{
+       int dims;
+
+       MONO_ARCH_SAVE_REGS;
 
+       dims = terminal_get_dimensions ();
+       if (dims == -1){
+               int cols = 0, rows = 0;
+                                     
+               char *str = getenv ("COLUMNS");
+               if (str != NULL)
+                       cols = atoi (str);
+               str = getenv ("LINES");
+               if (str != NULL)
+                       rows = atoi (str);
+
+               if (cols != 0 && rows != 0)
+                       cols_and_lines = (cols << 16) | rows;
+               else
+                       cols_and_lines = -1;
+       } else {
+               cols_and_lines = dims;
+       }
+       
+       *size = &cols_and_lines;
+       
        *verase = '\0';
        *vsusp = '\0';
        *intr = '\0';
        if (tcgetattr (STDIN_FILENO, &initial_attr) == -1)
                return FALSE;
 
-       /* TODO: handle SIGTSTP - Ctrl-Z */
-       attr = initial_attr;
-       attr.c_lflag &= ~ICANON;
-       attr.c_cc [VMIN] = 1;
-       attr.c_cc [VTIME] = 0;
-       if (tcsetattr (STDIN_FILENO, TCSANOW, &attr) == -1)
+       mono_attr = initial_attr;
+       mono_attr.c_lflag &= ~(ICANON);
+       mono_attr.c_iflag &= ~(IXON|IXOFF);
+       mono_attr.c_cc [VMIN] = 1;
+       mono_attr.c_cc [VTIME] = 0;
+       if (tcsetattr (STDIN_FILENO, TCSANOW, &mono_attr) == -1)
                return FALSE;
 
        *verase = initial_attr.c_cc [VERASE];
@@ -274,6 +397,9 @@ ves_icall_System_ConsoleDriver_TtySetup (MonoString *teardown, char *verase, cha
        if (setup_finished)
                return TRUE;
 
+       keypad_xmit_str = keypad != NULL ? mono_string_to_utf8 (keypad) : NULL;
+       
+       console_set_signal_handlers ();
        setup_finished = TRUE;
        if (!atexit_called) {
                if (teardown != NULL)
@@ -285,27 +411,4 @@ ves_icall_System_ConsoleDriver_TtySetup (MonoString *teardown, char *verase, cha
        return TRUE;
 }
 
-MonoBoolean
-ves_icall_System_ConsoleDriver_GetTtySize (HANDLE handle, gint32 *width, gint32 *height)
-{
-#ifdef TIOCGWINSZ
-       struct winsize ws;
-       int res;
-
-       MONO_ARCH_SAVE_REGS;
-
-       res = ioctl (GPOINTER_TO_INT (handle), TIOCGWINSZ, &ws);
-
-       if (!res) {
-               *width = ws.ws_col;
-               *height = ws.ws_row;
-               return TRUE;
-       }
-       else
-               return FALSE;
-#else
-       return FALSE;
-#endif
-}
-
 #endif /* !PLATFORM_WIN32 */