New test.
[mono.git] / mono / metadata / console-io.c
1 /*
2  * console-io.c: ConsoleDriver internal calls
3  *
4  * Author:
5  *      Gonzalo Paniagua Javier (gonzalo@ximian.com)
6  *
7  * Copyright (C) 2005 Novell, Inc. (http://www.novell.com)
8  */
9
10 #include <config.h>
11 #include <glib.h>
12 #include <stdio.h>
13 #include <string.h>
14 #include <errno.h>
15 #include <signal.h>
16 #include <sys/time.h>
17 #include <sys/types.h>
18 #include <unistd.h>
19 #include <mono/metadata/appdomain.h>
20 #include <mono/metadata/object-internals.h>
21 #include <mono/metadata/class-internals.h>
22 #include <mono/metadata/domain-internals.h>
23 #include <mono/metadata/metadata.h>
24 #include <mono/metadata/threadpool.h>
25 /* On solaris, curses.h must come before both termios.h and term.h */
26 #ifdef HAVE_CURSES_H
27 #include <curses.h>
28 #endif
29 #ifdef HAVE_TERMIOS_H
30 #include <termios.h>
31 #endif
32 #ifdef HAVE_TERM_H
33 #include <term.h>
34 #endif
35 /* Needed for FIONREAD under solaris */
36 #ifdef HAVE_SYS_FILIO_H
37 #include <sys/filio.h>
38 #endif
39 #ifndef PLATFORM_WIN32
40 #ifndef TIOCGWINSZ
41 #include <sys/ioctl.h>
42 #endif
43 #endif
44
45 #include <mono/metadata/console-io.h>
46 #include <mono/metadata/exception.h>
47
48 static gboolean setup_finished;
49 static gboolean atexit_called;
50 static gchar *teardown_str;
51
52 #ifdef PLATFORM_WIN32
53 MonoBoolean
54 ves_icall_System_ConsoleDriver_Isatty (HANDLE handle)
55 {
56         MONO_ARCH_SAVE_REGS;
57
58         return (GetFileType (handle) == FILE_TYPE_CHAR);
59 }
60
61 MonoBoolean
62 ves_icall_System_ConsoleDriver_SetEcho (MonoBoolean want_echo)
63 {
64         return FALSE;
65 }
66
67 MonoBoolean
68 ves_icall_System_ConsoleDriver_SetBreak (MonoBoolean want_break)
69 {
70         return FALSE;
71 }
72
73 gint32
74 ves_icall_System_ConsoleDriver_InternalKeyAvailable (gint32 timeout)
75 {
76         return FALSE;
77 }
78
79 MonoBoolean
80 ves_icall_System_ConsoleDriver_TtySetup (MonoString *teardown, char *verase, char *vsusp, char *intr)
81 {
82         return FALSE;
83 }
84
85 MonoBoolean
86 ves_icall_System_ConsoleDriver_GetTtySize (HANDLE handle, gint32 *width, gint32 *height)
87 {
88         return FALSE;
89 }
90
91 #else
92 static struct termios initial_attr;
93
94 MonoBoolean
95 ves_icall_System_ConsoleDriver_Isatty (HANDLE handle)
96 {
97         MONO_ARCH_SAVE_REGS;
98
99         return isatty (GPOINTER_TO_INT (handle));
100 }
101
102 static MonoBoolean
103 set_property (gint property, gboolean value)
104 {
105         struct termios attr;
106         gboolean callset = FALSE;
107         gboolean check;
108         
109         MONO_ARCH_SAVE_REGS;
110
111         if (tcgetattr (STDIN_FILENO, &attr) == -1)
112                 return FALSE;
113
114         check = (attr.c_lflag & property) != 0;
115         if ((value || check) && !(value && check)) {
116                 callset = TRUE;
117                 if (value)
118                         attr.c_lflag |= property;
119                 else
120                         attr.c_lflag &= ~property;
121         }
122
123         if (!callset)
124                 return TRUE;
125
126         if (tcsetattr (STDIN_FILENO, TCSANOW, &attr) == -1)
127                 return FALSE;
128
129         return TRUE;
130 }
131
132 MonoBoolean
133 ves_icall_System_ConsoleDriver_SetEcho (MonoBoolean want_echo)
134 {
135         return set_property (ECHO, want_echo);
136 }
137
138 MonoBoolean
139 ves_icall_System_ConsoleDriver_SetBreak (MonoBoolean want_break)
140 {
141         return set_property (IGNBRK, !want_break);
142 }
143
144 gint32
145 ves_icall_System_ConsoleDriver_InternalKeyAvailable (gint32 timeout)
146 {
147         fd_set rfds;
148         struct timeval tv;
149         struct timeval *tvptr;
150         div_t divvy;
151         int ret, nbytes;
152
153         MONO_ARCH_SAVE_REGS;
154
155         do {
156                 FD_ZERO (&rfds);
157                 FD_SET (STDIN_FILENO, &rfds);
158                 if (timeout >= 0) {
159                         divvy = div (timeout, 1000);
160                         tv.tv_sec = divvy.quot;
161                         tv.tv_usec = divvy.rem;
162                         tvptr = &tv;
163                 } else {
164                         tvptr = NULL;
165                 }
166                 ret = select (STDIN_FILENO + 1, &rfds, NULL, NULL, tvptr);
167         } while (ret == -1 && errno == EINTR);
168
169         if (ret > 0) {
170                 nbytes = 0;
171                 ret = ioctl (STDIN_FILENO, FIONREAD, &nbytes);
172                 if (ret >= 0)
173                         ret = nbytes;
174         }
175
176         return (ret > 0) ? ret : 0;
177 }
178
179 static void
180 tty_teardown (void)
181 {
182         MONO_ARCH_SAVE_REGS;
183
184         if (!setup_finished)
185                 return;
186
187         if (teardown_str != NULL) {
188                 write (STDOUT_FILENO, teardown_str, strlen (teardown_str));
189                 g_free (teardown_str);
190                 teardown_str = NULL;
191         }
192
193         tcflush (STDIN_FILENO, TCIFLUSH);
194         tcsetattr (STDIN_FILENO, TCSANOW, &initial_attr);
195         set_property (ECHO, TRUE);
196         setup_finished = FALSE;
197 }
198
199 static void
200 do_console_cancel_event (void)
201 {
202         static MonoClassField *cancel_handler_field;
203         MonoDomain *domain = mono_domain_get ();
204         MonoClass *klass;
205         MonoDelegate *load_value;
206         MonoMethod *method;
207         MonoMethodMessage *msg;
208         MonoMethod *im;
209
210         if (!domain->domain)
211                 return;
212
213         klass = mono_class_from_name (mono_defaults.corlib, "System", "Console");
214         if (klass == NULL)
215                 return;
216
217         if (cancel_handler_field == NULL) {
218                 cancel_handler_field = mono_class_get_field_from_name (klass, "cancel_handler");
219                 g_assert (cancel_handler_field);
220         }
221
222         mono_field_static_get_value (mono_class_vtable (domain, klass), cancel_handler_field, &load_value);
223         if (load_value == NULL)
224                 return;
225
226         klass = load_value->object.vtable->klass;
227         method = mono_class_get_method_from_name (klass, "BeginInvoke", -1);
228         g_assert (method != NULL);
229         im = mono_get_delegate_invoke (method->klass);
230         msg = mono_method_call_message_new (method, NULL, im, NULL, NULL);
231         mono_thread_pool_add ((MonoObject *) load_value, msg, NULL, NULL);
232 }
233
234 static gboolean in_sigint;
235 static void
236 sigint_handler (int signo)
237 {
238         MONO_ARCH_SAVE_REGS;
239
240         if (in_sigint)
241                 return;
242
243         in_sigint = TRUE;
244         do_console_cancel_event ();
245         in_sigint = FALSE;
246 }
247
248 MonoBoolean
249 ves_icall_System_ConsoleDriver_TtySetup (MonoString *teardown, char *verase, char *vsusp, char*intr)
250 {
251         struct termios attr;
252         
253         MONO_ARCH_SAVE_REGS;
254
255
256         *verase = '\0';
257         *vsusp = '\0';
258         *intr = '\0';
259         if (tcgetattr (STDIN_FILENO, &initial_attr) == -1)
260                 return FALSE;
261
262         /* TODO: handle SIGTSTP - Ctrl-Z */
263         attr = initial_attr;
264         attr.c_lflag &= ~ICANON;
265         attr.c_cc [VMIN] = 1;
266         attr.c_cc [VTIME] = 0;
267         if (tcsetattr (STDIN_FILENO, TCSANOW, &attr) == -1)
268                 return FALSE;
269
270         *verase = initial_attr.c_cc [VERASE];
271         *vsusp = initial_attr.c_cc [VSUSP];
272         *intr = initial_attr.c_cc [VINTR];
273         /* If initialized from another appdomain... */
274         if (setup_finished)
275                 return TRUE;
276
277         setup_finished = TRUE;
278         if (!atexit_called) {
279                 if (teardown != NULL)
280                         teardown_str = mono_string_to_utf8 (teardown);
281
282                 atexit (tty_teardown);
283         }
284
285         return TRUE;
286 }
287
288 MonoBoolean
289 ves_icall_System_ConsoleDriver_GetTtySize (HANDLE handle, gint32 *width, gint32 *height)
290 {
291 #ifdef TIOCGWINSZ
292         struct winsize ws;
293         int res;
294
295         MONO_ARCH_SAVE_REGS;
296
297         res = ioctl (GPOINTER_TO_INT (handle), TIOCGWINSZ, &ws);
298
299         if (!res) {
300                 *width = ws.ws_col;
301                 *height = ws.ws_row;
302                 return TRUE;
303         }
304         else
305                 return FALSE;
306 #else
307         return FALSE;
308 #endif
309 }
310
311 #endif /* !PLATFORM_WIN32 */