?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
kernelbase: Move console code page validation to conhost.
[wine.git] / dlls / kernelbase / console.c
blobd387b49c5344a7d515406422fcc300cc452d3073
1 /*
2 * Win32 console functions
3 *
4 * Copyright 1995 Martin von Loewis and Cameron Heide
5 * Copyright 1997 Karl Garrison
6 * Copyright 1998 John Richardson
7 * Copyright 1998 Marcus Meissner
8 * Copyright 2001,2002,2004,2005,2010 Eric Pouech
9 * Copyright 2001 Alexandre Julliard
10 * Copyright 2020 Jacek Caban for CodeWeavers
11 *
12 * This library is free software; you can redistribute it and/or
13 * modify it under the terms of the GNU Lesser General Public
14 * License as published by the Free Software Foundation; either
15 * version 2.1 of the License, or (at your option) any later version.
16 *
17 * This library is distributed in the hope that it will be useful,
18 * but WITHOUT ANY WARRANTY; without even the implied warranty of
19 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
20 * Lesser General Public License for more details.
21 *
22 * You should have received a copy of the GNU Lesser General Public
23 * License along with this library; if not, write to the Free Software
24 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
25 */
26
27 #include <stdarg.h>
28 #include <stdio.h>
29 #include <string.h>
30 #include <limits.h>
31
32 #define NONAMELESSUNION
33 #include "ntstatus.h"
34 #define WIN32_NO_STATUS
35 #include "windef.h"
36 #include "winbase.h"
37 #include "winnls.h"
38 #include "winerror.h"
39 #include "wincon.h"
40 #include "wine/condrv.h"
41 #include "wine/server.h"
42 #include "wine/exception.h"
43 #include "wine/debug.h"
44 #include "kernelbase.h"
45
46 WINE_DEFAULT_DEBUG_CHANNEL(console);
47
48
49 static CRITICAL_SECTION console_section;
50 static CRITICAL_SECTION_DEBUG critsect_debug =
51 {
52 0, 0, &console_section,
53 { &critsect_debug.ProcessLocksList, &critsect_debug.ProcessLocksList },
54 0, 0, { (DWORD_PTR)(__FILE__ ": console_section") }
55 };
56 static CRITICAL_SECTION console_section = { &critsect_debug, -1, 0, 0, 0, 0 };
57
58 static HANDLE console_connection;
59 static HANDLE console_wait_event;
60 static unsigned int console_flags;
61
62 #define CONSOLE_INPUT_HANDLE 0x01
63 #define CONSOLE_OUTPUT_HANDLE 0x02
64 #define CONSOLE_ERROR_HANDLE 0x04
65
66 static WCHAR input_exe[MAX_PATH + 1];
67
68 struct ctrl_handler
69 {
70 PHANDLER_ROUTINE func;
71 struct ctrl_handler *next;
72 };
73
74 static BOOL WINAPI default_ctrl_handler( DWORD type )
75 {
76 FIXME( "Terminating process %x on event %x\n", GetCurrentProcessId(), type );
77 RtlExitUserProcess( 0 );
78 return TRUE;
79 }
80
81 static struct ctrl_handler default_handler = { default_ctrl_handler, NULL };
82 static struct ctrl_handler *ctrl_handlers = &default_handler;
83
84 static BOOL console_ioctl( HANDLE handle, DWORD code, void *in_buff, DWORD in_count,
85 void *out_buff, DWORD out_count, DWORD *read )
86 {
87 IO_STATUS_BLOCK io;
88 NTSTATUS status;
89
90 status = NtDeviceIoControlFile( handle, NULL, NULL, NULL, &io, code, in_buff, in_count,
91 out_buff, out_count );
92 switch( status )
93 {
94 case STATUS_SUCCESS:
95 if (read) *read = io.Information;
96 return TRUE;
97 case STATUS_INVALID_PARAMETER:
98 break;
99 default:
100 status = STATUS_INVALID_HANDLE;
101 break;
102 }
103 if (read) *read = 0;
104 return set_ntstatus( status );
105 }
106
107 /* map input records to ASCII */
108 static void input_records_WtoA( INPUT_RECORD *buffer, int count )
109 {
110 UINT cp = GetConsoleCP();
111 int i;
112 char ch;
113
114 for (i = 0; i < count; i++)
115 {
116 if (buffer[i].EventType != KEY_EVENT) continue;
117 WideCharToMultiByte( cp, 0, &buffer[i].Event.KeyEvent.uChar.UnicodeChar, 1, &ch, 1, NULL, NULL );
118 buffer[i].Event.KeyEvent.uChar.AsciiChar = ch;
119 }
120 }
121
122 /* map input records to Unicode */
123 static void input_records_AtoW( INPUT_RECORD *buffer, int count )
124 {
125 UINT cp = GetConsoleCP();
126 int i;
127 WCHAR ch;
128
129 for (i = 0; i < count; i++)
130 {
131 if (buffer[i].EventType != KEY_EVENT) continue;
132 MultiByteToWideChar( cp, 0, &buffer[i].Event.KeyEvent.uChar.AsciiChar, 1, &ch, 1 );
133 buffer[i].Event.KeyEvent.uChar.UnicodeChar = ch;
134 }
135 }
136
137 /* map char infos to ASCII */
138 static void char_info_WtoA( UINT cp, CHAR_INFO *buffer, int count )
139 {
140 char ch;
141
142 while (count-- > 0)
143 {
144 WideCharToMultiByte( cp, 0, &buffer->Char.UnicodeChar, 1, &ch, 1, NULL, NULL );
145 buffer->Char.AsciiChar = ch;
146 buffer++;
147 }
148 }
149
150 /* map char infos to Unicode */
151 static void char_info_AtoW( CHAR_INFO *buffer, int count )
152 {
153 UINT cp = GetConsoleOutputCP();
154 WCHAR ch;
155
156 while (count-- > 0)
157 {
158 MultiByteToWideChar( cp, 0, &buffer->Char.AsciiChar, 1, &ch, 1 );
159 buffer->Char.UnicodeChar = ch;
160 buffer++;
161 }
162 }
163
164 /* helper function for GetLargestConsoleWindowSize */
165 static COORD get_largest_console_window_size( HANDLE handle )
166 {
167 struct condrv_output_info info;
168 COORD c = { 0, 0 };
169
170 if (!console_ioctl( handle, IOCTL_CONDRV_GET_OUTPUT_INFO, NULL, 0, &info, sizeof(info), NULL ))
171 return c;
172
173 c.X = info.max_width;
174 c.Y = info.max_height;
175 TRACE( "(%p), returning %dx%d\n", handle, c.X, c.Y );
176 return c;
177 }
178
179 static HANDLE create_console_server( void )
180 {
181 OBJECT_ATTRIBUTES attr = {sizeof(attr)};
182 UNICODE_STRING string;
183 IO_STATUS_BLOCK iosb;
184 HANDLE handle;
185 NTSTATUS status;
186
187 RtlInitUnicodeString( &string, L"\\Device\\ConDrv\\Server" );
188 attr.ObjectName = &string;
189 attr.Attributes = OBJ_INHERIT;
190 status = NtCreateFile( &handle, FILE_WRITE_PROPERTIES | FILE_READ_PROPERTIES | SYNCHRONIZE,
191 &attr, &iosb, NULL, FILE_ATTRIBUTE_NORMAL, 0, FILE_OPEN,
192 FILE_NON_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT, NULL, 0 );
193 return set_ntstatus( status ) ? handle : NULL;
194 }
195
196 static HANDLE create_console_reference( HANDLE root )
197 {
198 OBJECT_ATTRIBUTES attr = {sizeof(attr)};
199 UNICODE_STRING string;
200 IO_STATUS_BLOCK iosb;
201 HANDLE handle;
202 NTSTATUS status;
203
204 RtlInitUnicodeString( &string, L"Reference" );
205 attr.RootDirectory = root;
206 attr.ObjectName = &string;
207 status = NtCreateFile( &handle, FILE_READ_DATA | FILE_WRITE_DATA | FILE_WRITE_PROPERTIES |
208 FILE_READ_PROPERTIES | SYNCHRONIZE, &attr, &iosb, NULL, FILE_ATTRIBUTE_NORMAL,
209 0, FILE_OPEN, FILE_NON_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT, NULL, 0 );
210 return set_ntstatus( status ) ? handle : NULL;
211 }
212
213 static BOOL create_console_connection( HANDLE root )
214 {
215 OBJECT_ATTRIBUTES attr = {sizeof(attr)};
216 UNICODE_STRING string;
217 IO_STATUS_BLOCK iosb;
218 NTSTATUS status;
219
220 RtlInitUnicodeString( &string, root ? L"Connection" : L"\\Device\\ConDrv\\Connection" );
221 attr.RootDirectory = root;
222 attr.ObjectName = &string;
223 status = NtCreateFile( &console_connection, FILE_WRITE_PROPERTIES | FILE_READ_PROPERTIES | SYNCHRONIZE, &attr,
224 &iosb, NULL, FILE_ATTRIBUTE_NORMAL, 0, FILE_OPEN, FILE_NON_DIRECTORY_FILE, NULL, 0 );
225 return set_ntstatus( status );
226 }
227
228 static BOOL init_console_std_handles( BOOL override_all )
229 {
230 HANDLE std_out = NULL, std_err = NULL, handle;
231 OBJECT_ATTRIBUTES attr = {sizeof(attr)};
232 IO_STATUS_BLOCK iosb;
233 UNICODE_STRING name;
234 NTSTATUS status;
235
236 attr.ObjectName = &name;
237 attr.Attributes = OBJ_INHERIT;
238
239 if (override_all || !GetStdHandle( STD_INPUT_HANDLE ))
240 {
241 /* FIXME: Use unbound console handle */
242 RtlInitUnicodeString( &name, L"\\Device\\ConDrv\\CurrentIn" );
243 status = NtCreateFile( &handle, FILE_READ_DATA | FILE_WRITE_DATA | SYNCHRONIZE | FILE_READ_ATTRIBUTES |
244 FILE_WRITE_ATTRIBUTES, &attr, &iosb, NULL, FILE_ATTRIBUTE_NORMAL,
245 FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, FILE_CREATE,
246 FILE_NON_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT, NULL, 0 );
247 if (!set_ntstatus( status )) return FALSE;
248 console_flags |= CONSOLE_INPUT_HANDLE;
249 SetStdHandle( STD_INPUT_HANDLE, console_handle_map( handle ));
250 }
251
252 if (!override_all)
253 {
254 std_out = GetStdHandle( STD_OUTPUT_HANDLE );
255 std_err = GetStdHandle( STD_ERROR_HANDLE );
256 if (std_out && std_err) return TRUE;
257 }
258
259 /* FIXME: Use unbound console handle */
260 RtlInitUnicodeString( &name, L"\\Device\\ConDrv\\CurrentOut" );
261 status = NtCreateFile( &handle, FILE_READ_DATA | FILE_WRITE_DATA | SYNCHRONIZE | FILE_READ_ATTRIBUTES |
262 FILE_WRITE_ATTRIBUTES, &attr, &iosb, NULL, FILE_ATTRIBUTE_NORMAL,
263 FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, FILE_CREATE,
264 FILE_NON_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT, NULL, 0 );
265 if (!set_ntstatus( status )) return FALSE;
266 if (!std_out)
267 {
268 console_flags |= CONSOLE_OUTPUT_HANDLE;
269 SetStdHandle( STD_OUTPUT_HANDLE, console_handle_map( handle ));
270 }
271
272 if (!std_err)
273 {
274 if (!std_out && !DuplicateHandle( GetCurrentProcess(), handle, GetCurrentProcess(),
275 &handle, 0, TRUE, DUPLICATE_SAME_ACCESS ))
276 return FALSE;
277 console_flags |= CONSOLE_ERROR_HANDLE;
278 SetStdHandle( STD_ERROR_HANDLE, console_handle_map( handle ));
279 }
280
281 return TRUE;
282 }
283
284 /******************************************************************
285 * AttachConsole (kernelbase.@)
286 */
287 BOOL WINAPI DECLSPEC_HOTPATCH AttachConsole( DWORD pid )
288 {
289 BOOL ret;
290
291 TRACE( "(%x)\n", pid );
292
293 RtlEnterCriticalSection( &console_section );
294
295 if (RtlGetCurrentPeb()->ProcessParameters->ConsoleHandle)
296 {
297 RtlLeaveCriticalSection( &console_section );
298 WARN( "console already attached\n" );
299 SetLastError( ERROR_ACCESS_DENIED );
300 return FALSE;
301 }
302
303 ret = create_console_connection( NULL ) &&
304 console_ioctl( console_connection, IOCTL_CONDRV_BIND_PID, &pid, sizeof(pid), NULL, 0, NULL );
305 if (ret)
306 {
307 RtlGetCurrentPeb()->ProcessParameters->ConsoleHandle = create_console_reference( console_connection );
308 if (RtlGetCurrentPeb()->ProcessParameters->ConsoleHandle)
309 {
310 STARTUPINFOW si;
311 GetStartupInfoW( &si );
312 init_console_std_handles( !(si.dwFlags & STARTF_USESTDHANDLES) );
313 }
314 else ret = FALSE;
315 }
316
317 if (!ret) FreeConsole();
318 RtlLeaveCriticalSection( &console_section );
319 return ret;
320 }
321
322
323 /******************************************************************
324 * AllocConsole (kernelbase.@)
325 */
326 BOOL WINAPI AllocConsole(void)
327 {
328 SECURITY_ATTRIBUTES inheritable_attr = { sizeof(inheritable_attr), NULL, TRUE };
329 STARTUPINFOW app_si, console_si;
330 HANDLE server, console = NULL;
331 WCHAR buffer[1024], cmd[256], conhost_path[MAX_PATH];
332 PROCESS_INFORMATION pi;
333 void *redir;
334 BOOL ret;
335
336 TRACE("()\n");
337
338 RtlEnterCriticalSection( &console_section );
339
340 if (RtlGetCurrentPeb()->ProcessParameters->ConsoleHandle)
341 {
342 /* we already have a console opened on this process, don't create a new one */
343 RtlLeaveCriticalSection( &console_section );
344 SetLastError( ERROR_ACCESS_DENIED );
345 return FALSE;
346 }
347
348 if (!(server = create_console_server()) || !(console = create_console_reference( server ))) goto error;
349
350 GetStartupInfoW(&app_si);
351
352 memset(&console_si, 0, sizeof(console_si));
353 console_si.cb = sizeof(console_si);
354 /* setup a view arguments for wineconsole (it'll use them as default values) */
355 if (app_si.dwFlags & STARTF_USECOUNTCHARS)
356 {
357 console_si.dwFlags |= STARTF_USECOUNTCHARS;
358 console_si.dwXCountChars = app_si.dwXCountChars;
359 console_si.dwYCountChars = app_si.dwYCountChars;
360 }
361 if (app_si.dwFlags & STARTF_USEFILLATTRIBUTE)
362 {
363 console_si.dwFlags |= STARTF_USEFILLATTRIBUTE;
364 console_si.dwFillAttribute = app_si.dwFillAttribute;
365 }
366 if (app_si.dwFlags & STARTF_USESHOWWINDOW)
367 {
368 console_si.dwFlags |= STARTF_USESHOWWINDOW;
369 console_si.wShowWindow = app_si.wShowWindow;
370 }
371 if (app_si.lpTitle)
372 console_si.lpTitle = app_si.lpTitle;
373 else if (GetModuleFileNameW(0, buffer, ARRAY_SIZE(buffer)))
374 {
375 buffer[ARRAY_SIZE(buffer) - 1] = 0;
376 console_si.lpTitle = buffer;
377 }
378
379 swprintf( conhost_path, ARRAY_SIZE(conhost_path), L"%s\\conhost.exe", system_dir );
380 swprintf( cmd, ARRAY_SIZE(cmd), L"\"%s\" --server 0x%x", conhost_path, condrv_handle( server ));
381 Wow64DisableWow64FsRedirection( &redir );
382 ret = CreateProcessW( conhost_path, cmd, NULL, NULL, TRUE, DETACHED_PROCESS, NULL, NULL, &console_si, &pi );
383 Wow64RevertWow64FsRedirection( redir );
384
385 if (!ret || !create_console_connection( console)) goto error;
386 if (!init_console_std_handles( !(app_si.dwFlags & STARTF_USESTDHANDLES) )) goto error;
387
388 RtlGetCurrentPeb()->ProcessParameters->ConsoleHandle = console;
389 TRACE( "Started wineconsole pid=%08x tid=%08x\n", pi.dwProcessId, pi.dwThreadId );
390
391 CloseHandle( server );
392 RtlLeaveCriticalSection( &console_section );
393 SetLastError( ERROR_SUCCESS );
394 return TRUE;
395
396 error:
397 ERR("Can't allocate console\n");
398 NtClose( console );
399 NtClose( server );
400 FreeConsole();
401 RtlLeaveCriticalSection( &console_section );
402 return FALSE;
403 }
404
405
406 /******************************************************************************
407 * CreateConsoleScreenBuffer (kernelbase.@)
408 */
409 HANDLE WINAPI DECLSPEC_HOTPATCH CreateConsoleScreenBuffer( DWORD access, DWORD share,
410 SECURITY_ATTRIBUTES *sa, DWORD flags,
411 void *data )
412 {
413 OBJECT_ATTRIBUTES attr = {sizeof(attr)};
414 IO_STATUS_BLOCK iosb;
415 UNICODE_STRING name;
416 HANDLE handle;
417 NTSTATUS status;
418
419 TRACE( "(%x,%x,%p,%x,%p)\n", access, share, sa, flags, data );
420
421 if (flags != CONSOLE_TEXTMODE_BUFFER || data)
422 {
423 SetLastError( ERROR_INVALID_PARAMETER );
424 return INVALID_HANDLE_VALUE;
425 }
426
427 RtlInitUnicodeString( &name, L"\\Device\\ConDrv\\ScreenBuffer" );
428 attr.ObjectName = &name;
429 attr.SecurityDescriptor = sa ? sa->lpSecurityDescriptor : NULL;
430 if (sa && sa->bInheritHandle) attr.Attributes |= OBJ_INHERIT;
431 status = NtCreateFile( &handle, access, &attr, &iosb, NULL, FILE_ATTRIBUTE_NORMAL, 0, FILE_OPEN,
432 FILE_NON_DIRECTORY_FILE, NULL, 0 );
433 return set_ntstatus( status ) ? handle : INVALID_HANDLE_VALUE;
434 }
435
436
437 /******************************************************************************
438 * CtrlRoutine (kernelbase.@)
439 */
440 DWORD WINAPI CtrlRoutine( void *arg )
441 {
442 DWORD_PTR event = (DWORD_PTR)arg;
443 struct ctrl_handler *handler;
444
445 if (event == CTRL_C_EVENT)
446 {
447 BOOL caught_by_dbg = TRUE;
448 /* First, try to pass the ctrl-C event to the debugger (if any)
449 * If it continues, there's nothing more to do
450 * Otherwise, we need to send the ctrl-C event to the handlers
451 */
452 __TRY
453 {
454 RaiseException( DBG_CONTROL_C, 0, 0, NULL );
455 }
456 __EXCEPT_ALL
457 {
458 caught_by_dbg = FALSE;
459 }
460 __ENDTRY
461 if (caught_by_dbg) return 0;
462 }
463
464 if (NtCurrentTeb()->Peb->ProcessParameters->ConsoleFlags & 1) return 0;
465
466 RtlEnterCriticalSection( &console_section );
467 for (handler = ctrl_handlers; handler; handler = handler->next)
468 {
469 if (handler->func( event )) break;
470 }
471 RtlLeaveCriticalSection( &console_section );
472 return 1;
473 }
474
475
476 static LONG WINAPI handle_ctrl_c( EXCEPTION_POINTERS *eptr )
477 {
478 if (eptr->ExceptionRecord->ExceptionCode != CONTROL_C_EXIT) return EXCEPTION_CONTINUE_SEARCH;
479 if (!RtlGetCurrentPeb()->ProcessParameters->ConsoleHandle) return EXCEPTION_CONTINUE_SEARCH;
480
481 if (!(NtCurrentTeb()->Peb->ProcessParameters->ConsoleFlags & 1))
482 {
483 HANDLE thread = CreateThread( NULL, 0, CtrlRoutine, (void*)CTRL_C_EVENT, 0, NULL );
484 if (thread) CloseHandle( thread );
485 }
486 return EXCEPTION_CONTINUE_EXECUTION;
487 }
488
489
490 /******************************************************************************
491 * FillConsoleOutputAttribute (kernelbase.@)
492 */
493 BOOL WINAPI DECLSPEC_HOTPATCH FillConsoleOutputAttribute( HANDLE handle, WORD attr, DWORD length,
494 COORD coord, DWORD *written )
495 {
496 struct condrv_fill_output_params params;
497
498 TRACE( "(%p,%d,%d,(%dx%d),%p)\n", handle, attr, length, coord.X, coord.Y, written );
499
500 if (!written)
501 {
502 SetLastError( ERROR_INVALID_ACCESS );
503 return FALSE;
504 }
505
506 *written = 0;
507
508 params.mode = CHAR_INFO_MODE_ATTR;
509 params.x = coord.X;
510 params.y = coord.Y;
511 params.count = length;
512 params.wrap = TRUE;
513 params.ch = 0;
514 params.attr = attr;
515 return console_ioctl( handle, IOCTL_CONDRV_FILL_OUTPUT, &params, sizeof(params),
516 written, sizeof(*written), NULL );
517 }
518
519
520 /******************************************************************************
521 * FillConsoleOutputCharacterA (kernelbase.@)
522 */
523 BOOL WINAPI DECLSPEC_HOTPATCH FillConsoleOutputCharacterA( HANDLE handle, CHAR ch, DWORD length,
524 COORD coord, DWORD *written )
525 {
526 WCHAR wch;
527
528 MultiByteToWideChar( GetConsoleOutputCP(), 0, &ch, 1, &wch, 1 );
529 return FillConsoleOutputCharacterW( handle, wch, length, coord, written );
530 }
531
532
533 /******************************************************************************
534 * FillConsoleOutputCharacterW (kernelbase.@)
535 */
536 BOOL WINAPI DECLSPEC_HOTPATCH FillConsoleOutputCharacterW( HANDLE handle, WCHAR ch, DWORD length,
537 COORD coord, DWORD *written )
538 {
539 struct condrv_fill_output_params params;
540
541 TRACE( "(%p,%s,%d,(%dx%d),%p)\n", handle, debugstr_wn(&ch, 1), length, coord.X, coord.Y, written );
542
543 if (!written)
544 {
545 SetLastError( ERROR_INVALID_ACCESS );
546 return FALSE;
547 }
548
549 *written = 0;
550
551 params.mode = CHAR_INFO_MODE_TEXT;
552 params.x = coord.X;
553 params.y = coord.Y;
554 params.count = length;
555 params.wrap = TRUE;
556 params.ch = ch;
557 params.attr = 0;
558 return console_ioctl( handle, IOCTL_CONDRV_FILL_OUTPUT, &params, sizeof(params),
559 written, sizeof(*written), NULL );
560 }
561
562 HANDLE get_console_wait_handle( HANDLE handle )
563 {
564 HANDLE event = 0;
565
566 SERVER_START_REQ( get_console_wait_event )
567 {
568 req->handle = wine_server_obj_handle( console_handle_map( handle ));
569 if (!wine_server_call( req )) event = wine_server_ptr_handle( reply->event );
570 }
571 SERVER_END_REQ;
572 if (event)
573 {
574 if (InterlockedCompareExchangePointer( &console_wait_event, event, 0 )) NtClose( event );
575 handle = console_wait_event;
576 }
577 return handle;
578 }
579
580
581 /***********************************************************************
582 * FreeConsole (kernelbase.@)
583 */
584 BOOL WINAPI DECLSPEC_HOTPATCH FreeConsole(void)
585 {
586 HANDLE event;
587
588 RtlEnterCriticalSection( &console_section );
589
590 NtClose( console_connection );
591 console_connection = NULL;
592
593 NtClose( RtlGetCurrentPeb()->ProcessParameters->ConsoleHandle );
594 RtlGetCurrentPeb()->ProcessParameters->ConsoleHandle = NULL;
595
596 if (console_flags & CONSOLE_INPUT_HANDLE) NtClose( GetStdHandle( STD_INPUT_HANDLE ));
597 if (console_flags & CONSOLE_OUTPUT_HANDLE) NtClose( GetStdHandle( STD_OUTPUT_HANDLE ));
598 if (console_flags & CONSOLE_ERROR_HANDLE) NtClose( GetStdHandle( STD_ERROR_HANDLE ));
599 console_flags = 0;
600
601 if ((event = InterlockedExchangePointer( &console_wait_event, NULL ))) NtClose( event );
602
603 RtlLeaveCriticalSection( &console_section );
604 return TRUE;
605 }
606
607
608 /******************************************************************************
609 * GenerateConsoleCtrlEvent (kernelbase.@)
610 */
611 BOOL WINAPI DECLSPEC_HOTPATCH GenerateConsoleCtrlEvent( DWORD event, DWORD group )
612 {
613 struct condrv_ctrl_event ctrl_event;
614
615 TRACE( "(%d, %x)\n", event, group );
616
617 if (event != CTRL_C_EVENT && event != CTRL_BREAK_EVENT)
618 {
619 ERR( "Invalid event %d for PGID %x\n", event, group );
620 return FALSE;
621 }
622
623 ctrl_event.event = event;
624 ctrl_event.group_id = group;
625 return console_ioctl( RtlGetCurrentPeb()->ProcessParameters->ConsoleHandle,
626 IOCTL_CONDRV_CTRL_EVENT, &ctrl_event, sizeof(ctrl_event), NULL, 0, NULL );
627 }
628
629
630 /******************************************************************************
631 * GetConsoleCP (kernelbase.@)
632 */
633 UINT WINAPI DECLSPEC_HOTPATCH GetConsoleCP(void)
634 {
635 struct condrv_input_info info;
636
637 if (!console_ioctl( RtlGetCurrentPeb()->ProcessParameters->ConsoleHandle,
638 IOCTL_CONDRV_GET_INPUT_INFO, NULL, 0, &info, sizeof(info), NULL ))
639 return 0;
640 return info.input_cp;
641 }
642
643
644 /******************************************************************************
645 * GetConsoleCursorInfo (kernelbase.@)
646 */
647 BOOL WINAPI DECLSPEC_HOTPATCH GetConsoleCursorInfo( HANDLE handle, CONSOLE_CURSOR_INFO *info )
648 {
649 struct condrv_output_info condrv_info;
650
651 if (!console_ioctl( handle, IOCTL_CONDRV_GET_OUTPUT_INFO, NULL, 0, &condrv_info, sizeof(condrv_info), NULL ))
652 return FALSE;
653
654 if (!info)
655 {
656 SetLastError( ERROR_INVALID_ACCESS );
657 return FALSE;
658 }
659
660 info->dwSize = condrv_info.cursor_size;
661 info->bVisible = condrv_info.cursor_visible;
662 TRACE("(%p) returning (%d,%d)\n", handle, info->dwSize, info->bVisible);
663 return TRUE;
664 }
665
666
667 /***********************************************************************
668 * GetConsoleInputExeNameA (kernelbase.@)
669 */
670 BOOL WINAPI DECLSPEC_HOTPATCH GetConsoleInputExeNameA( DWORD len, LPSTR buffer )
671 {
672 RtlEnterCriticalSection( &console_section );
673 if (WideCharToMultiByte( CP_ACP, 0, input_exe, -1, NULL, 0, NULL, NULL ) <= len)
674 WideCharToMultiByte( CP_ACP, 0, input_exe, -1, buffer, len, NULL, NULL );
675 else SetLastError(ERROR_BUFFER_OVERFLOW);
676 RtlLeaveCriticalSection( &console_section );
677 return TRUE;
678 }
679
680
681 /***********************************************************************
682 * GetConsoleInputExeNameW (kernelbase.@)
683 */
684 BOOL WINAPI DECLSPEC_HOTPATCH GetConsoleInputExeNameW( DWORD len, LPWSTR buffer )
685 {
686 RtlEnterCriticalSection( &console_section );
687 if (len > lstrlenW(input_exe)) lstrcpyW( buffer, input_exe );
688 else SetLastError( ERROR_BUFFER_OVERFLOW );
689 RtlLeaveCriticalSection( &console_section );
690 return TRUE;
691 }
692
693
694 /***********************************************************************
695 * GetConsoleMode (kernelbase.@)
696 */
697 BOOL WINAPI DECLSPEC_HOTPATCH GetConsoleMode( HANDLE handle, DWORD *mode )
698 {
699 return console_ioctl( handle, IOCTL_CONDRV_GET_MODE, NULL, 0, mode, sizeof(*mode), NULL );
700 }
701
702
703 /***********************************************************************
704 * GetConsoleOutputCP (kernelbase.@)
705 */
706 UINT WINAPI DECLSPEC_HOTPATCH GetConsoleOutputCP(void)
707 {
708 struct condrv_input_info info;
709
710 if (!console_ioctl( RtlGetCurrentPeb()->ProcessParameters->ConsoleHandle,
711 IOCTL_CONDRV_GET_INPUT_INFO, NULL, 0, &info, sizeof(info), NULL ))
712 return 0;
713 return info.output_cp;
714 }
715
716
717 /***********************************************************************
718 * GetConsoleScreenBufferInfo (kernelbase.@)
719 */
720 BOOL WINAPI DECLSPEC_HOTPATCH GetConsoleScreenBufferInfo( HANDLE handle, CONSOLE_SCREEN_BUFFER_INFO *info )
721 {
722 struct condrv_output_info condrv_info;
723
724 if (!console_ioctl( handle , IOCTL_CONDRV_GET_OUTPUT_INFO, NULL, 0,
725 &condrv_info, sizeof(condrv_info), NULL ))
726 return FALSE;
727
728 info->dwSize.X = condrv_info.width;
729 info->dwSize.Y = condrv_info.height;
730 info->dwCursorPosition.X = condrv_info.cursor_x;
731 info->dwCursorPosition.Y = condrv_info.cursor_y;
732 info->wAttributes = condrv_info.attr;
733 info->srWindow.Left = condrv_info.win_left;
734 info->srWindow.Right = condrv_info.win_right;
735 info->srWindow.Top = condrv_info.win_top;
736 info->srWindow.Bottom = condrv_info.win_bottom;
737 info->dwMaximumWindowSize.X = min(condrv_info.width, condrv_info.max_width);
738 info->dwMaximumWindowSize.Y = min(condrv_info.height, condrv_info.max_height);
739
740 TRACE( "(%p,(%d,%d) (%d,%d) %d (%d,%d-%d,%d) (%d,%d)\n", handle,
741 info->dwSize.X, info->dwSize.Y, info->dwCursorPosition.X, info->dwCursorPosition.Y,
742 info->wAttributes, info->srWindow.Left, info->srWindow.Top, info->srWindow.Right,
743 info->srWindow.Bottom, info->dwMaximumWindowSize.X, info->dwMaximumWindowSize.Y );
744 return TRUE;
745 }
746
747
748 /***********************************************************************
749 * GetConsoleScreenBufferInfoEx (kernelbase.@)
750 */
751 BOOL WINAPI DECLSPEC_HOTPATCH GetConsoleScreenBufferInfoEx( HANDLE handle,
752 CONSOLE_SCREEN_BUFFER_INFOEX *info )
753 {
754 struct condrv_output_info condrv_info;
755
756 if (info->cbSize != sizeof(CONSOLE_SCREEN_BUFFER_INFOEX))
757 {
758 SetLastError( ERROR_INVALID_PARAMETER );
759 return FALSE;
760 }
761
762 if (!console_ioctl( handle, IOCTL_CONDRV_GET_OUTPUT_INFO, NULL, 0, &condrv_info,
763 sizeof(condrv_info), NULL ))
764 return FALSE;
765
766 info->dwSize.X = condrv_info.width;
767 info->dwSize.Y = condrv_info.height;
768 info->dwCursorPosition.X = condrv_info.cursor_x;
769 info->dwCursorPosition.Y = condrv_info.cursor_y;
770 info->wAttributes = condrv_info.attr;
771 info->srWindow.Left = condrv_info.win_left;
772 info->srWindow.Top = condrv_info.win_top;
773 info->srWindow.Right = condrv_info.win_right;
774 info->srWindow.Bottom = condrv_info.win_bottom;
775 info->dwMaximumWindowSize.X = min( condrv_info.width, condrv_info.max_width );
776 info->dwMaximumWindowSize.Y = min( condrv_info.height, condrv_info.max_height );
777 info->wPopupAttributes = condrv_info.popup_attr;
778 info->bFullscreenSupported = FALSE;
779 memcpy( info->ColorTable, condrv_info.color_map, sizeof(info->ColorTable) );
780 return TRUE;
781 }
782
783
784 /******************************************************************************
785 * GetConsoleTitleW (kernelbase.@)
786 */
787 DWORD WINAPI DECLSPEC_HOTPATCH GetConsoleTitleW( LPWSTR title, DWORD size )
788 {
789 if (!size) return 0;
790
791 if (!console_ioctl( RtlGetCurrentPeb()->ProcessParameters->ConsoleHandle, IOCTL_CONDRV_GET_TITLE,
792 NULL, 0, title, (size - 1) * sizeof(WCHAR), &size ))
793 return 0;
794
795 size /= sizeof(WCHAR);
796 title[size] = 0;
797 return size + 1;
798 }
799
800
801 /***********************************************************************
802 * GetLargestConsoleWindowSize (kernelbase.@)
803 */
804 #if defined(__i386__) && !defined(__MINGW32__) && !defined(_MSC_VER)
805 #undef GetLargestConsoleWindowSize
806 DWORD WINAPI DECLSPEC_HOTPATCH GetLargestConsoleWindowSize( HANDLE handle )
807 {
808 union {
809 COORD c;
810 DWORD w;
811 } x;
812 x.c = get_largest_console_window_size( handle );
813 return x.w;
814 }
815
816 #else
817
818 COORD WINAPI DECLSPEC_HOTPATCH GetLargestConsoleWindowSize( HANDLE handle )
819 {
820 return get_largest_console_window_size( handle );
821 }
822
823 #endif /* !defined(__i386__) */
824
825
826 /***********************************************************************
827 * GetNumberOfConsoleInputEvents (kernelbase.@)
828 */
829 BOOL WINAPI DECLSPEC_HOTPATCH GetNumberOfConsoleInputEvents( HANDLE handle, DWORD *count )
830 {
831 struct condrv_input_info info;
832 if (!console_ioctl( handle, IOCTL_CONDRV_GET_INPUT_INFO, NULL, 0, &info, sizeof(info), NULL ))
833 return FALSE;
834 *count = info.input_count;
835 return TRUE;
836 }
837
838
839 /***********************************************************************
840 * PeekConsoleInputA (kernelbase.@)
841 */
842 BOOL WINAPI DECLSPEC_HOTPATCH PeekConsoleInputA( HANDLE handle, INPUT_RECORD *buffer,
843 DWORD length, DWORD *count )
844 {
845 DWORD read;
846
847 if (!PeekConsoleInputW( handle, buffer, length, &read )) return FALSE;
848 input_records_WtoA( buffer, read );
849 if (count) *count = read;
850 return TRUE;
851 }
852
853
854 /***********************************************************************
855 * PeekConsoleInputW (kernelbase.@)
856 */
857 BOOL WINAPI DECLSPEC_HOTPATCH PeekConsoleInputW( HANDLE handle, INPUT_RECORD *buffer,
858 DWORD length, DWORD *count )
859 {
860 DWORD read;
861 if (!console_ioctl( handle, IOCTL_CONDRV_PEEK, NULL, 0, buffer, length * sizeof(*buffer), &read ))
862 return FALSE;
863 if (count) *count = read / sizeof(*buffer);
864 return TRUE;
865 }
866
867
868 /******************************************************************************
869 * ReadConsoleOutputAttribute (kernelbase.@)
870 */
871 BOOL WINAPI DECLSPEC_HOTPATCH ReadConsoleOutputAttribute( HANDLE handle, WORD *attr, DWORD length,
872 COORD coord, DWORD *count )
873 {
874 struct condrv_output_params params;
875 BOOL ret;
876
877 TRACE( "(%p,%p,%d,%dx%d,%p)\n", handle, attr, length, coord.X, coord.Y, count );
878
879 if (!count)
880 {
881 SetLastError( ERROR_INVALID_ACCESS );
882 return FALSE;
883 }
884
885 params.mode = CHAR_INFO_MODE_ATTR;
886 params.x = coord.X;
887 params.y = coord.Y;
888 params.width = 0;
889 ret = console_ioctl( handle, IOCTL_CONDRV_READ_OUTPUT, &params, sizeof(params),
890 attr, length * sizeof(*attr), count );
891 *count /= sizeof(*attr);
892 return ret;
893 }
894
895
896 /******************************************************************************
897 * ReadConsoleOutputCharacterA (kernelbase.@)
898 */
899 BOOL WINAPI DECLSPEC_HOTPATCH ReadConsoleOutputCharacterA( HANDLE handle, LPSTR buffer, DWORD length,
900 COORD coord, DWORD *count )
901 {
902 DWORD read;
903 BOOL ret;
904 LPWSTR wptr;
905
906 if (!count)
907 {
908 SetLastError( ERROR_INVALID_ACCESS );
909 return FALSE;
910 }
911
912 *count = 0;
913 if (!(wptr = HeapAlloc( GetProcessHeap(), 0, length * sizeof(WCHAR) )))
914 {
915 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
916 return FALSE;
917 }
918 if ((ret = ReadConsoleOutputCharacterW( handle, wptr, length, coord, &read )))
919 {
920 read = WideCharToMultiByte( GetConsoleOutputCP(), 0, wptr, read, buffer, length, NULL, NULL);
921 *count = read;
922 }
923 HeapFree( GetProcessHeap(), 0, wptr );
924 return ret;
925 }
926
927
928 /******************************************************************************
929 * ReadConsoleOutputCharacterW (kernelbase.@)
930 */
931 BOOL WINAPI DECLSPEC_HOTPATCH ReadConsoleOutputCharacterW( HANDLE handle, LPWSTR buffer, DWORD length,
932 COORD coord, DWORD *count )
933 {
934 struct condrv_output_params params;
935 BOOL ret;
936
937 TRACE( "(%p,%p,%d,%dx%d,%p)\n", handle, buffer, length, coord.X, coord.Y, count );
938
939 if (!count)
940 {
941 SetLastError( ERROR_INVALID_ACCESS );
942 return FALSE;
943 }
944
945 params.mode = CHAR_INFO_MODE_TEXT;
946 params.x = coord.X;
947 params.y = coord.Y;
948 params.width = 0;
949 ret = console_ioctl( handle, IOCTL_CONDRV_READ_OUTPUT, &params, sizeof(params), buffer,
950 length * sizeof(*buffer), count );
951 *count /= sizeof(*buffer);
952 return ret;
953 }
954
955
956 /******************************************************************************
957 * ReadConsoleOutputA (kernelbase.@)
958 */
959 BOOL WINAPI DECLSPEC_HOTPATCH ReadConsoleOutputA( HANDLE handle, CHAR_INFO *buffer, COORD size,
960 COORD coord, SMALL_RECT *region )
961 {
962 BOOL ret;
963 int y;
964
965 ret = ReadConsoleOutputW( handle, buffer, size, coord, region );
966 if (ret && region->Right >= region->Left)
967 {
968 UINT cp = GetConsoleOutputCP();
969 for (y = 0; y <= region->Bottom - region->Top; y++)
970 char_info_WtoA( cp, &buffer[(coord.Y + y) * size.X + coord.X], region->Right - region->Left + 1 );
971 }
972 return ret;
973 }
974
975
976 /******************************************************************************
977 * ReadConsoleOutputW (kernelbase.@)
978 */
979 BOOL WINAPI DECLSPEC_HOTPATCH ReadConsoleOutputW( HANDLE handle, CHAR_INFO *buffer, COORD size,
980 COORD coord, SMALL_RECT *region )
981 {
982 struct condrv_output_params params;
983 unsigned int width, height, y;
984 SMALL_RECT *result;
985 DWORD count;
986 BOOL ret;
987
988 if (region->Left > region->Right || region->Top > region->Bottom)
989 {
990 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
991 return FALSE;
992 }
993 if (size.X <= coord.X || size.Y <= coord.Y)
994 {
995 region->Right = region->Left - 1;
996 region->Bottom = region->Top - 1;
997 SetLastError( ERROR_INVALID_FUNCTION );
998 return FALSE;
999 }
1000 width = min( region->Right - region->Left + 1, size.X - coord.X );
1001 height = min( region->Bottom - region->Top + 1, size.Y - coord.Y );
1002 region->Right = region->Left + width - 1;
1003 region->Bottom = region->Top + height - 1;
1004
1005 count = sizeof(*result) + width * height * sizeof(*buffer);
1006 if (!(result = HeapAlloc( GetProcessHeap(), 0, count )))
1007 {
1008 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
1009 return FALSE;
1010 }
1011
1012 params.mode = CHAR_INFO_MODE_TEXTATTR;
1013 params.x = region->Left;
1014 params.y = region->Top;
1015 params.width = width;
1016 if ((ret = console_ioctl( handle, IOCTL_CONDRV_READ_OUTPUT, &params, sizeof(params), result, count, &count )) && count)
1017 {
1018 CHAR_INFO *char_info = (CHAR_INFO *)(result + 1);
1019 *region = *result;
1020 width = region->Right - region->Left + 1;
1021 height = region->Bottom - region->Top + 1;
1022 for (y = 0; y < height; y++)
1023 memcpy( &buffer[(y + coord.Y) * size.X + coord.X], &char_info[y * width], width * sizeof(*buffer) );
1024 }
1025 HeapFree( GetProcessHeap(), 0, result );
1026 return ret;
1027 }
1028
1029
1030 /******************************************************************************
1031 * ScrollConsoleScreenBufferA (kernelbase.@)
1032 */
1033 BOOL WINAPI DECLSPEC_HOTPATCH ScrollConsoleScreenBufferA( HANDLE handle, const SMALL_RECT *scroll,
1034 const SMALL_RECT *clip, COORD origin, const CHAR_INFO *fill )
1035 {
1036 CHAR_INFO ciW;
1037
1038 ciW.Attributes = fill->Attributes;
1039 MultiByteToWideChar( GetConsoleOutputCP(), 0, &fill->Char.AsciiChar, 1, &ciW.Char.UnicodeChar, 1 );
1040
1041 return ScrollConsoleScreenBufferW( handle, scroll, clip, origin, &ciW );
1042 }
1043
1044
1045 /******************************************************************************
1046 * ScrollConsoleScreenBufferW (kernelbase.@)
1047 */
1048 BOOL WINAPI DECLSPEC_HOTPATCH ScrollConsoleScreenBufferW( HANDLE handle, const SMALL_RECT *scroll,
1049 const SMALL_RECT *clip_rect, COORD origin,
1050 const CHAR_INFO *fill )
1051 {
1052 struct condrv_scroll_params params;
1053
1054 if (clip_rect)
1055 TRACE( "(%p,(%d,%d-%d,%d),(%d,%d-%d,%d),%d-%d,%p)\n", handle,
1056 scroll->Left, scroll->Top, scroll->Right, scroll->Bottom,
1057 clip_rect->Left, clip_rect->Top, clip_rect->Right, clip_rect->Bottom,
1058 origin.X, origin.Y, fill );
1059 else
1060 TRACE("(%p,(%d,%d-%d,%d),(nil),%d-%d,%p)\n", handle,
1061 scroll->Left, scroll->Top, scroll->Right, scroll->Bottom,
1062 origin.X, origin.Y, fill );
1063
1064 params.scroll = *scroll;
1065 params.origin = origin;
1066 params.fill.ch = fill->Char.UnicodeChar;
1067 params.fill.attr = fill->Attributes;
1068 if (!clip_rect)
1069 {
1070 params.clip.Left = params.clip.Top = 0;
1071 params.clip.Right = params.clip.Bottom = SHRT_MAX;
1072 }
1073 else params.clip = *clip_rect;
1074 return console_ioctl( handle, IOCTL_CONDRV_SCROLL, (void *)&params, sizeof(params), NULL, 0, NULL );
1075 }
1076
1077
1078 /******************************************************************************
1079 * SetConsoleActiveScreenBuffer (kernelbase.@)
1080 */
1081 BOOL WINAPI DECLSPEC_HOTPATCH SetConsoleActiveScreenBuffer( HANDLE handle )
1082 {
1083 TRACE( "(%p)\n", handle );
1084 return console_ioctl( handle, IOCTL_CONDRV_ACTIVATE, NULL, 0, NULL, 0, NULL );
1085 }
1086
1087
1088 /******************************************************************************
1089 * SetConsoleCP (kernelbase.@)
1090 */
1091 BOOL WINAPI DECLSPEC_HOTPATCH SetConsoleCP( UINT cp )
1092 {
1093 struct condrv_input_info_params params = { SET_CONSOLE_INPUT_INFO_INPUT_CODEPAGE };
1094
1095 params.info.input_cp = cp;
1096 return console_ioctl( RtlGetCurrentPeb()->ProcessParameters->ConsoleHandle,
1097 IOCTL_CONDRV_SET_INPUT_INFO, &params, sizeof(params), NULL, 0, NULL );
1098 }
1099
1100
1101 /******************************************************************************
1102 * SetConsoleCtrlHandler (kernelbase.@)
1103 */
1104 BOOL WINAPI DECLSPEC_HOTPATCH SetConsoleCtrlHandler( PHANDLER_ROUTINE func, BOOL add )
1105 {
1106 struct ctrl_handler *handler;
1107 BOOL ret = FALSE;
1108
1109 TRACE( "(%p,%d)\n", func, add );
1110
1111 RtlEnterCriticalSection( &console_section );
1112
1113 if (!func)
1114 {
1115 if (add) NtCurrentTeb()->Peb->ProcessParameters->ConsoleFlags |= 1;
1116 else NtCurrentTeb()->Peb->ProcessParameters->ConsoleFlags &= ~1;
1117 ret = TRUE;
1118 }
1119 else if (add)
1120 {
1121 if ((handler = RtlAllocateHeap( GetProcessHeap(), 0, sizeof(*handler) )))
1122 {
1123 handler->func = func;
1124 handler->next = ctrl_handlers;
1125 ctrl_handlers = handler;
1126 ret = TRUE;
1127 }
1128 }
1129 else
1130 {
1131 struct ctrl_handler **p_handler;
1132
1133 for (p_handler = &ctrl_handlers; *p_handler; p_handler = &(*p_handler)->next)
1134 {
1135 if ((*p_handler)->func == func) break;
1136 }
1137 if (*p_handler && *p_handler != &default_handler)
1138 {
1139 handler = *p_handler;
1140 *p_handler = handler->next;
1141 RtlFreeHeap( GetProcessHeap(), 0, handler );
1142 ret = TRUE;
1143 }
1144 else SetLastError( ERROR_INVALID_PARAMETER );
1145 }
1146
1147 RtlLeaveCriticalSection( &console_section );
1148 return ret;
1149 }
1150
1151
1152 /******************************************************************************
1153 * SetConsoleCursorInfo (kernelbase.@)
1154 */
1155 BOOL WINAPI DECLSPEC_HOTPATCH SetConsoleCursorInfo( HANDLE handle, CONSOLE_CURSOR_INFO *info )
1156 {
1157 struct condrv_output_info_params params = { SET_CONSOLE_OUTPUT_INFO_CURSOR_GEOM };
1158
1159 TRACE( "(%p,%d,%d)\n", handle, info->dwSize, info->bVisible);
1160
1161 params.info.cursor_size = info->dwSize;
1162 params.info.cursor_visible = info->bVisible;
1163 return console_ioctl( handle, IOCTL_CONDRV_SET_OUTPUT_INFO, &params, sizeof(params),
1164 NULL, 0, NULL );
1165 }
1166
1167
1168 /******************************************************************************
1169 * SetConsoleCursorPosition (kernelbase.@)
1170 */
1171 BOOL WINAPI DECLSPEC_HOTPATCH SetConsoleCursorPosition( HANDLE handle, COORD pos )
1172 {
1173 struct condrv_output_info_params params = { SET_CONSOLE_OUTPUT_INFO_CURSOR_POS };
1174
1175 TRACE( "%p %d %d\n", handle, pos.X, pos.Y );
1176
1177 params.info.cursor_x = pos.X;
1178 params.info.cursor_y = pos.Y;
1179 return console_ioctl( handle, IOCTL_CONDRV_SET_OUTPUT_INFO, &params, sizeof(params), NULL, 0, NULL );
1180 }
1181
1182
1183 /******************************************************************************
1184 * SetConsoleInputExeNameA (kernelbase.@)
1185 */
1186 BOOL WINAPI DECLSPEC_HOTPATCH SetConsoleInputExeNameA( LPCSTR name )
1187 {
1188 if (!name || !name[0])
1189 {
1190 SetLastError( ERROR_INVALID_PARAMETER );
1191 return FALSE;
1192 }
1193 RtlEnterCriticalSection( &console_section );
1194 MultiByteToWideChar( CP_ACP, 0, name, -1, input_exe, ARRAY_SIZE(input_exe) );
1195 RtlLeaveCriticalSection( &console_section );
1196 return TRUE;
1197 }
1198
1199
1200 /******************************************************************************
1201 * SetConsoleInputExeNameW (kernelbase.@)
1202 */
1203 BOOL WINAPI DECLSPEC_HOTPATCH SetConsoleInputExeNameW( LPCWSTR name )
1204 {
1205 if (!name || !name[0])
1206 {
1207 SetLastError( ERROR_INVALID_PARAMETER );
1208 return FALSE;
1209 }
1210 RtlEnterCriticalSection( &console_section );
1211 lstrcpynW( input_exe, name, ARRAY_SIZE(input_exe) );
1212 RtlLeaveCriticalSection( &console_section );
1213 return TRUE;
1214 }
1215
1216
1217 /******************************************************************************
1218 * SetConsoleMode (kernelbase.@)
1219 */
1220 BOOL WINAPI DECLSPEC_HOTPATCH SetConsoleMode( HANDLE handle, DWORD mode )
1221 {
1222 TRACE( "(%p,%x)\n", handle, mode );
1223 return console_ioctl( handle, IOCTL_CONDRV_SET_MODE, &mode, sizeof(mode), NULL, 0, NULL );
1224 }
1225
1226
1227 /******************************************************************************
1228 * SetConsoleOutputCP (kernelbase.@)
1229 */
1230 BOOL WINAPI DECLSPEC_HOTPATCH SetConsoleOutputCP( UINT cp )
1231 {
1232 struct condrv_input_info_params params = { SET_CONSOLE_INPUT_INFO_OUTPUT_CODEPAGE };
1233
1234 params.info.output_cp = cp;
1235 return console_ioctl( RtlGetCurrentPeb()->ProcessParameters->ConsoleHandle,
1236 IOCTL_CONDRV_SET_INPUT_INFO, &params, sizeof(params), NULL, 0, NULL );
1237 }
1238
1239
1240 /******************************************************************************
1241 * SetConsoleScreenBufferInfoEx (kernelbase.@)
1242 */
1243 BOOL WINAPI DECLSPEC_HOTPATCH SetConsoleScreenBufferInfoEx( HANDLE handle,
1244 CONSOLE_SCREEN_BUFFER_INFOEX *info )
1245 {
1246 struct condrv_output_info_params params =
1247 { SET_CONSOLE_OUTPUT_INFO_CURSOR_POS | SET_CONSOLE_OUTPUT_INFO_SIZE |
1248 SET_CONSOLE_OUTPUT_INFO_ATTR | SET_CONSOLE_OUTPUT_INFO_POPUP_ATTR |
1249 SET_CONSOLE_OUTPUT_INFO_DISPLAY_WINDOW | SET_CONSOLE_OUTPUT_INFO_MAX_SIZE };
1250
1251 TRACE("(%p, %p)\n", handle, info);
1252
1253 if (info->cbSize != sizeof(CONSOLE_SCREEN_BUFFER_INFOEX))
1254 {
1255 SetLastError(ERROR_INVALID_PARAMETER);
1256 return FALSE;
1257 }
1258
1259 params.info.width = info->dwSize.X;
1260 params.info.height = info->dwSize.Y;
1261 params.info.cursor_x = info->dwCursorPosition.X;
1262 params.info.cursor_y = info->dwCursorPosition.Y;
1263 params.info.attr = info->wAttributes;
1264 params.info.win_left = info->srWindow.Left;
1265 params.info.win_top = info->srWindow.Top;
1266 params.info.win_right = info->srWindow.Right;
1267 params.info.win_bottom = info->srWindow.Bottom;
1268 params.info.popup_attr = info->wPopupAttributes;
1269 params.info.max_width = min( info->dwMaximumWindowSize.X, info->dwSize.X );
1270 params.info.max_height = min( info->dwMaximumWindowSize.Y, info->dwSize.Y );
1271 return console_ioctl( handle, IOCTL_CONDRV_SET_OUTPUT_INFO, &params, sizeof(params), NULL, 0, NULL );
1272 }
1273
1274
1275 /******************************************************************************
1276 * SetConsoleScreenBufferSize (kernelbase.@)
1277 */
1278 BOOL WINAPI DECLSPEC_HOTPATCH SetConsoleScreenBufferSize( HANDLE handle, COORD size )
1279 {
1280 struct condrv_output_info_params params = { SET_CONSOLE_OUTPUT_INFO_SIZE };
1281
1282 TRACE( "(%p,(%d,%d))\n", handle, size.X, size.Y );
1283
1284 params.info.width = size.X;
1285 params.info.height = size.Y;
1286 return console_ioctl( handle, IOCTL_CONDRV_SET_OUTPUT_INFO, &params, sizeof(params), NULL, 0, NULL );
1287 }
1288
1289
1290 /******************************************************************************
1291 * SetConsoleTextAttribute (kernelbase.@)
1292 */
1293 BOOL WINAPI DECLSPEC_HOTPATCH SetConsoleTextAttribute( HANDLE handle, WORD attr )
1294 {
1295 struct condrv_output_info_params params = { SET_CONSOLE_OUTPUT_INFO_ATTR };
1296
1297 TRACE( "(%p,%d)\n", handle, attr );
1298
1299 params.info.attr = attr;
1300 return console_ioctl( handle, IOCTL_CONDRV_SET_OUTPUT_INFO, &params, sizeof(params), NULL, 0, NULL );
1301 }
1302
1303
1304 /******************************************************************************
1305 * SetConsoleTitleW (kernelbase.@)
1306 */
1307 BOOL WINAPI DECLSPEC_HOTPATCH SetConsoleTitleW( LPCWSTR title )
1308 {
1309 TRACE( "%s\n", debugstr_w( title ));
1310
1311 return console_ioctl( RtlGetCurrentPeb()->ProcessParameters->ConsoleHandle, IOCTL_CONDRV_SET_TITLE,
1312 (void *)title, lstrlenW(title) * sizeof(WCHAR), NULL, 0, NULL );
1313 }
1314
1315
1316 /******************************************************************************
1317 * SetConsoleWindowInfo (kernelbase.@)
1318 */
1319 BOOL WINAPI DECLSPEC_HOTPATCH SetConsoleWindowInfo( HANDLE handle, BOOL absolute, SMALL_RECT *window )
1320 {
1321 struct condrv_output_info_params params = { SET_CONSOLE_OUTPUT_INFO_DISPLAY_WINDOW };
1322 SMALL_RECT rect = *window;
1323
1324 TRACE( "(%p,%d,(%d,%d-%d,%d))\n", handle, absolute, rect.Left, rect.Top, rect.Right, rect.Bottom );
1325
1326 if (!absolute)
1327 {
1328 CONSOLE_SCREEN_BUFFER_INFO info;
1329
1330 if (!GetConsoleScreenBufferInfo( handle, &info )) return FALSE;
1331 rect.Left += info.srWindow.Left;
1332 rect.Top += info.srWindow.Top;
1333 rect.Right += info.srWindow.Right;
1334 rect.Bottom += info.srWindow.Bottom;
1335 }
1336
1337 params.info.win_left = rect.Left;
1338 params.info.win_top = rect.Top;
1339 params.info.win_right = rect.Right;
1340 params.info.win_bottom = rect.Bottom;
1341 return console_ioctl( handle, IOCTL_CONDRV_SET_OUTPUT_INFO, &params, sizeof(params), NULL, 0, NULL );
1342 }
1343
1344
1345 /***********************************************************************
1346 * ReadConsoleInputA (kernelbase.@)
1347 */
1348 BOOL WINAPI ReadConsoleInputA( HANDLE handle, INPUT_RECORD *buffer, DWORD length, DWORD *count )
1349 {
1350 DWORD read;
1351
1352 if (!ReadConsoleInputW( handle, buffer, length, &read )) return FALSE;
1353 input_records_WtoA( buffer, read );
1354 if (count) *count = read;
1355 return TRUE;
1356 }
1357
1358
1359 /***********************************************************************
1360 * ReadConsoleInputW (kernelbase.@)
1361 */
1362 BOOL WINAPI ReadConsoleInputW( HANDLE handle, INPUT_RECORD *buffer, DWORD length, DWORD *count )
1363 {
1364 if (!console_ioctl( handle, IOCTL_CONDRV_READ_INPUT, NULL, 0,
1365 buffer, length * sizeof(*buffer), count ))
1366 return FALSE;
1367 *count /= sizeof(*buffer);
1368 return TRUE;
1369 }
1370
1371
1372 /******************************************************************************
1373 * WriteConsoleInputA (kernelbase.@)
1374 */
1375 BOOL WINAPI DECLSPEC_HOTPATCH WriteConsoleInputA( HANDLE handle, const INPUT_RECORD *buffer,
1376 DWORD count, DWORD *written )
1377 {
1378 INPUT_RECORD *recW = NULL;
1379 BOOL ret;
1380
1381 if (count > 0)
1382 {
1383 if (!buffer)
1384 {
1385 SetLastError( ERROR_INVALID_ACCESS );
1386 return FALSE;
1387 }
1388 if (!(recW = HeapAlloc( GetProcessHeap(), 0, count * sizeof(*recW) )))
1389 {
1390 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
1391 return FALSE;
1392 }
1393 memcpy( recW, buffer, count * sizeof(*recW) );
1394 input_records_AtoW( recW, count );
1395 }
1396 ret = WriteConsoleInputW( handle, recW, count, written );
1397 HeapFree( GetProcessHeap(), 0, recW );
1398 return ret;
1399 }
1400
1401
1402 /******************************************************************************
1403 * WriteConsoleInputW (kernelbase.@)
1404 */
1405 BOOL WINAPI DECLSPEC_HOTPATCH WriteConsoleInputW( HANDLE handle, const INPUT_RECORD *buffer,
1406 DWORD count, DWORD *written )
1407 {
1408 TRACE( "(%p,%p,%d,%p)\n", handle, buffer, count, written );
1409
1410 if (count > 0 && !buffer)
1411 {
1412 SetLastError( ERROR_INVALID_ACCESS );
1413 return FALSE;
1414 }
1415
1416 if (!DeviceIoControl( handle, IOCTL_CONDRV_WRITE_INPUT, (void *)buffer, count * sizeof(*buffer), NULL, 0, NULL, NULL ))
1417 return FALSE;
1418
1419 if (!written)
1420 {
1421 SetLastError( ERROR_INVALID_ACCESS );
1422 return FALSE;
1423 }
1424 *written = count;
1425 return TRUE;
1426 }
1427
1428
1429 /***********************************************************************
1430 * WriteConsoleOutputA (kernelbase.@)
1431 */
1432 BOOL WINAPI DECLSPEC_HOTPATCH WriteConsoleOutputA( HANDLE handle, const CHAR_INFO *buffer,
1433 COORD size, COORD coord, SMALL_RECT *region )
1434 {
1435 int y;
1436 BOOL ret;
1437 COORD new_size, new_coord;
1438 CHAR_INFO *ciW;
1439
1440 new_size.X = min( region->Right - region->Left + 1, size.X - coord.X );
1441 new_size.Y = min( region->Bottom - region->Top + 1, size.Y - coord.Y );
1442
1443 if (new_size.X <= 0 || new_size.Y <= 0)
1444 {
1445 region->Bottom = region->Top + new_size.Y - 1;
1446 region->Right = region->Left + new_size.X - 1;
1447 return TRUE;
1448 }
1449
1450 /* only copy the useful rectangle */
1451 if (!(ciW = HeapAlloc( GetProcessHeap(), 0, sizeof(CHAR_INFO) * new_size.X * new_size.Y )))
1452 return FALSE;
1453 for (y = 0; y < new_size.Y; y++)
1454 memcpy( &ciW[y * new_size.X], &buffer[(y + coord.Y) * size.X + coord.X],
1455 new_size.X * sizeof(CHAR_INFO) );
1456 char_info_AtoW( ciW, new_size.X * new_size.Y );
1457 new_coord.X = new_coord.Y = 0;
1458 ret = WriteConsoleOutputW( handle, ciW, new_size, new_coord, region );
1459 HeapFree( GetProcessHeap(), 0, ciW );
1460 return ret;
1461 }
1462
1463
1464 /***********************************************************************
1465 * WriteConsoleOutputW (kernelbase.@)
1466 */
1467 BOOL WINAPI DECLSPEC_HOTPATCH WriteConsoleOutputW( HANDLE handle, const CHAR_INFO *buffer,
1468 COORD size, COORD coord, SMALL_RECT *region )
1469 {
1470 struct condrv_output_params *params;
1471 unsigned int width, height, y;
1472 size_t params_size;
1473 BOOL ret;
1474
1475 TRACE( "(%p,%p,(%d,%d),(%d,%d),(%d,%dx%d,%d)\n",
1476 handle, buffer, size.X, size.Y, coord.X, coord.Y,
1477 region->Left, region->Top, region->Right, region->Bottom );
1478
1479 if (region->Left > region->Right || region->Top > region->Bottom || size.X <= coord.X || size.Y <= coord.Y)
1480 {
1481 SetLastError( ERROR_INVALID_PARAMETER );
1482 return FALSE;
1483 }
1484
1485 width = min( region->Right - region->Left + 1, size.X - coord.X );
1486 height = min( region->Bottom - region->Top + 1, size.Y - coord.Y );
1487 region->Right = region->Left + width - 1;
1488 region->Bottom = region->Top + height - 1;
1489
1490 params_size = sizeof(*params) + width * height * sizeof(*buffer);
1491 if (!(params = HeapAlloc( GetProcessHeap(), 0, params_size )))
1492 {
1493 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
1494 return FALSE;
1495 }
1496
1497 params->mode = CHAR_INFO_MODE_TEXTATTR;
1498 params->x = region->Left;
1499 params->y = region->Top;
1500 params->width = width;
1501
1502 for (y = 0; y < height; y++)
1503 memcpy( &((CHAR_INFO *)(params + 1))[y * width], &buffer[(y + coord.Y) * size.X + coord.X], width * sizeof(CHAR_INFO) );
1504
1505 ret = console_ioctl( handle, IOCTL_CONDRV_WRITE_OUTPUT, params, params_size, region, sizeof(*region), NULL );
1506 HeapFree( GetProcessHeap(), 0, params );
1507 return ret;
1508 }
1509
1510
1511 /******************************************************************************
1512 * WriteConsoleOutputAttribute (kernelbase.@)
1513 */
1514 BOOL WINAPI DECLSPEC_HOTPATCH WriteConsoleOutputAttribute( HANDLE handle, const WORD *attr, DWORD length,
1515 COORD coord, DWORD *written )
1516 {
1517 struct condrv_output_params *params;
1518 size_t size;
1519 BOOL ret;
1520
1521 TRACE( "(%p,%p,%d,%dx%d,%p)\n", handle, attr, length, coord.X, coord.Y, written );
1522
1523 if ((length > 0 && !attr) || !written)
1524 {
1525 SetLastError( ERROR_INVALID_ACCESS );
1526 return FALSE;
1527 }
1528
1529 *written = 0;
1530 size = sizeof(*params) + length * sizeof(WORD);
1531 if (!(params = HeapAlloc( GetProcessHeap(), 0, size ))) return FALSE;
1532 params->mode = CHAR_INFO_MODE_ATTR;
1533 params->x = coord.X;
1534 params->y = coord.Y;
1535 params->width = 0;
1536 memcpy( params + 1, attr, length * sizeof(*attr) );
1537 ret = console_ioctl( handle, IOCTL_CONDRV_WRITE_OUTPUT, params, size, written, sizeof(*written), NULL );
1538 HeapFree( GetProcessHeap(), 0, params );
1539 return ret;
1540 }
1541
1542
1543 /******************************************************************************
1544 * WriteConsoleOutputCharacterA (kernelbase.@)
1545 */
1546 BOOL WINAPI DECLSPEC_HOTPATCH WriteConsoleOutputCharacterA( HANDLE handle, LPCSTR str, DWORD length,
1547 COORD coord, DWORD *written )
1548 {
1549 BOOL ret;
1550 LPWSTR strW = NULL;
1551 DWORD lenW = 0;
1552
1553 TRACE( "(%p,%s,%d,%dx%d,%p)\n", handle, debugstr_an(str, length), length, coord.X, coord.Y, written );
1554
1555 if (length > 0)
1556 {
1557 UINT cp = GetConsoleOutputCP();
1558 if (!str)
1559 {
1560 SetLastError( ERROR_INVALID_ACCESS );
1561 return FALSE;
1562 }
1563 lenW = MultiByteToWideChar( cp, 0, str, length, NULL, 0 );
1564
1565 if (!(strW = HeapAlloc( GetProcessHeap(), 0, lenW * sizeof(WCHAR) )))
1566 {
1567 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
1568 return FALSE;
1569 }
1570 MultiByteToWideChar( cp, 0, str, length, strW, lenW );
1571 }
1572 ret = WriteConsoleOutputCharacterW( handle, strW, lenW, coord, written );
1573 HeapFree( GetProcessHeap(), 0, strW );
1574 return ret;
1575 }
1576
1577
1578 /******************************************************************************
1579 * WriteConsoleOutputCharacterW (kernelbase.@)
1580 */
1581 BOOL WINAPI DECLSPEC_HOTPATCH WriteConsoleOutputCharacterW( HANDLE handle, LPCWSTR str, DWORD length,
1582 COORD coord, DWORD *written )
1583 {
1584 struct condrv_output_params *params;
1585 size_t size;
1586 BOOL ret;
1587
1588 TRACE( "(%p,%s,%d,%dx%d,%p)\n", handle, debugstr_wn(str, length), length, coord.X, coord.Y, written );
1589
1590 if ((length > 0 && !str) || !written)
1591 {
1592 SetLastError( ERROR_INVALID_ACCESS );
1593 return FALSE;
1594 }
1595
1596 *written = 0;
1597 size = sizeof(*params) + length * sizeof(WCHAR);
1598 if (!(params = HeapAlloc( GetProcessHeap(), 0, size ))) return FALSE;
1599 params->mode = CHAR_INFO_MODE_TEXT;
1600 params->x = coord.X;
1601 params->y = coord.Y;
1602 params->width = 0;
1603 memcpy( params + 1, str, length * sizeof(*str) );
1604 ret = console_ioctl( handle, IOCTL_CONDRV_WRITE_OUTPUT, params, size, written, sizeof(*written), NULL );
1605 HeapFree( GetProcessHeap(), 0, params );
1606 return ret;
1607 }
1608
1609
1610 /***********************************************************************
1611 * ReadConsoleA (kernelbase.@)
1612 */
1613 BOOL WINAPI ReadConsoleA( HANDLE handle, void *buffer, DWORD length, DWORD *ret_count, void *reserved )
1614 {
1615 LPWSTR strW = HeapAlloc( GetProcessHeap(), 0, length * sizeof(WCHAR) );
1616 DWORD count = 0;
1617 BOOL ret;
1618
1619 if (!strW)
1620 {
1621 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
1622 return FALSE;
1623 }
1624 if ((ret = ReadConsoleW( handle, strW, length, &count, NULL )))
1625 {
1626 count = WideCharToMultiByte( GetConsoleCP(), 0, strW, count, buffer, length, NULL, NULL );
1627 if (ret_count) *ret_count = count;
1628 }
1629 HeapFree( GetProcessHeap(), 0, strW );
1630 return ret;
1631 }
1632
1633
1634 /***********************************************************************
1635 * ReadConsoleW (kernelbase.@)
1636 */
1637 BOOL WINAPI ReadConsoleW( HANDLE handle, void *buffer, DWORD length, DWORD *count, void *reserved )
1638 {
1639 BOOL ret;
1640
1641 TRACE( "(%p,%p,%d,%p,%p)\n", handle, buffer, length, count, reserved );
1642
1643 if (length > INT_MAX)
1644 {
1645 SetLastError(ERROR_NOT_ENOUGH_MEMORY);
1646 return FALSE;
1647 }
1648
1649 ret = console_ioctl( handle, IOCTL_CONDRV_READ_CONSOLE, NULL, 0, buffer,
1650 length * sizeof(WCHAR), count );
1651 if (ret) *count /= sizeof(WCHAR);
1652 return ret;
1653 }
1654
1655
1656 /***********************************************************************
1657 * WriteConsoleA (kernelbase.@)
1658 */
1659 BOOL WINAPI DECLSPEC_HOTPATCH WriteConsoleA( HANDLE handle, const void *buffer, DWORD length,
1660 DWORD *written, void *reserved )
1661 {
1662 UINT cp = GetConsoleOutputCP();
1663 LPWSTR strW;
1664 DWORD lenW;
1665 BOOL ret;
1666
1667 if (written) *written = 0;
1668 lenW = MultiByteToWideChar( cp, 0, buffer, length, NULL, 0 );
1669 if (!(strW = HeapAlloc( GetProcessHeap(), 0, lenW * sizeof(WCHAR) ))) return FALSE;
1670 MultiByteToWideChar( cp, 0, buffer, length, strW, lenW );
1671 ret = WriteConsoleW( handle, strW, lenW, written, 0 );
1672 HeapFree( GetProcessHeap(), 0, strW );
1673 return ret;
1674 }
1675
1676
1677 /***********************************************************************
1678 * WriteConsoleW (kernelbase.@)
1679 */
1680 BOOL WINAPI DECLSPEC_HOTPATCH WriteConsoleW( HANDLE handle, const void *buffer, DWORD length,
1681 DWORD *written, void *reserved )
1682 {
1683 BOOL ret;
1684
1685 TRACE( "(%p,%s,%d,%p,%p)\n", handle, debugstr_wn(buffer, length), length, written, reserved );
1686
1687 ret = console_ioctl( handle, IOCTL_CONDRV_WRITE_CONSOLE, (void *)buffer,
1688 length * sizeof(WCHAR), NULL, 0, NULL );
1689 if (written) *written = ret ? length : 0;
1690 return ret;
1691 }
1692
1693
1694 /***********************************************************************
1695 * FlushConsoleInputBuffer (kernelbase.@)
1696 */
1697 BOOL WINAPI FlushConsoleInputBuffer( HANDLE handle )
1698 {
1699 return console_ioctl( handle, IOCTL_CONDRV_FLUSH, NULL, 0, NULL, 0, NULL );
1700 }
1701
1702
1703 /***********************************************************************
1704 * Beep (kernelbase.@)
1705 */
1706 BOOL WINAPI Beep( DWORD frequency, DWORD duration )
1707 {
1708 /* FIXME: we should not require a console to be attached */
1709 console_ioctl( RtlGetCurrentPeb()->ProcessParameters->ConsoleHandle,
1710 IOCTL_CONDRV_BEEP, NULL, 0, NULL, 0, NULL );
1711 return TRUE;
1712 }
1713
1714
1715 static HANDLE create_pseudo_console( COORD size, HANDLE input, HANDLE output, HANDLE signal,
1716 DWORD flags, HANDLE *process )
1717 {
1718 WCHAR cmd[MAX_PATH], conhost_path[MAX_PATH];
1719 PROCESS_INFORMATION pi;
1720 HANDLE server, console;
1721 STARTUPINFOEXW si;
1722 void *redir;
1723 BOOL res;
1724
1725 if (!(server = create_console_server())) return NULL;
1726
1727 console = create_console_reference( server );
1728 if (!console)
1729 {
1730 NtClose( server );
1731 return NULL;
1732 }
1733
1734 memset( &si, 0, sizeof(si) );
1735 si.StartupInfo.cb = sizeof(STARTUPINFOEXW);
1736 si.StartupInfo.hStdInput = input;
1737 si.StartupInfo.hStdOutput = output;
1738 si.StartupInfo.hStdError = output;
1739 si.StartupInfo.dwFlags = STARTF_USESTDHANDLES;
1740 swprintf( conhost_path, ARRAY_SIZE(conhost_path), L"%s\\conhost.exe", system_dir );
1741 if (signal)
1742 {
1743 swprintf( cmd, ARRAY_SIZE(cmd),
1744 L"\"%s\" --headless %s--width %u --height %u --signal 0x%x --server 0x%x",
1745 conhost_path, (flags & PSEUDOCONSOLE_INHERIT_CURSOR) ? L"--inheritcursor " : L"",
1746 size.X, size.Y, signal, server );
1747 }
1748 else
1749 {
1750 swprintf( cmd, ARRAY_SIZE(cmd), L"\"%s\" --unix --width %u --height %u --server 0x%x",
1751 conhost_path, size.X, size.Y, server );
1752 }
1753 Wow64DisableWow64FsRedirection( &redir );
1754 res = CreateProcessW( conhost_path, cmd, NULL, NULL, TRUE, DETACHED_PROCESS, NULL, NULL,
1755 &si.StartupInfo, &pi );
1756 Wow64RevertWow64FsRedirection( redir );
1757 NtClose( server );
1758 if (!res)
1759 {
1760 NtClose( console );
1761 return NULL;
1762 }
1763
1764 NtClose( pi.hThread );
1765 *process = pi.hProcess;
1766 return console;
1767 }
1768
1769 /******************************************************************************
1770 * CreatePseudoConsole (kernelbase.@)
1771 */
1772 HRESULT WINAPI CreatePseudoConsole( COORD size, HANDLE input, HANDLE output, DWORD flags, HPCON *ret )
1773 {
1774 SECURITY_ATTRIBUTES inherit_attr = { sizeof(inherit_attr), NULL, TRUE };
1775 struct pseudo_console *pseudo_console;
1776 HANDLE tty_input = NULL, tty_output;
1777 HANDLE signal = NULL;
1778 WCHAR pipe_name[64];
1779
1780 TRACE( "(%u,%u) %p %p %x %p\n", size.X, size.Y, input, output, flags, ret );
1781
1782 if (!size.X || !size.Y || !ret) return E_INVALIDARG;
1783
1784 if (!(pseudo_console = HeapAlloc( GetProcessHeap(), 0, HEAP_ZERO_MEMORY ))) return E_OUTOFMEMORY;
1785
1786 swprintf( pipe_name, ARRAY_SIZE(pipe_name), L"\\\\.\\pipe\\wine_pty_signal_pipe%x",
1787 GetCurrentThreadId() );
1788 signal = CreateNamedPipeW( pipe_name, PIPE_ACCESS_INBOUND | FILE_FLAG_OVERLAPPED, PIPE_TYPE_BYTE,
1789 PIPE_UNLIMITED_INSTANCES, 4096, 4096, NMPWAIT_USE_DEFAULT_WAIT, &inherit_attr );
1790 if (signal == INVALID_HANDLE_VALUE)
1791 {
1792 HeapFree( GetProcessHeap(), 0, pseudo_console );
1793 return HRESULT_FROM_WIN32( GetLastError() );
1794 }
1795 pseudo_console->signal = CreateFileW( pipe_name, GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, NULL );
1796 if (pseudo_console->signal != INVALID_HANDLE_VALUE &&
1797 DuplicateHandle( GetCurrentProcess(), input, GetCurrentProcess(), &tty_input, 0, TRUE, DUPLICATE_SAME_ACCESS) &&
1798 DuplicateHandle( GetCurrentProcess(), output, GetCurrentProcess(), &tty_output, 0, TRUE, DUPLICATE_SAME_ACCESS))
1799 {
1800 pseudo_console->reference = create_pseudo_console( size, tty_input, tty_output, signal, flags,
1801 &pseudo_console->process );
1802 NtClose( tty_output );
1803 }
1804 NtClose( tty_input );
1805 NtClose( signal );
1806 if (!pseudo_console->reference)
1807 {
1808 ClosePseudoConsole( pseudo_console );
1809 return HRESULT_FROM_WIN32( GetLastError() );
1810 }
1811
1812 *ret = pseudo_console;
1813 return S_OK;
1814 }
1815
1816 /******************************************************************************
1817 * ClosePseudoConsole (kernelbase.@)
1818 */
1819 void WINAPI ClosePseudoConsole( HPCON handle )
1820 {
1821 struct pseudo_console *pseudo_console = handle;
1822
1823 TRACE( "%p\n", handle );
1824
1825 if (!pseudo_console) return;
1826 if (pseudo_console->signal) CloseHandle( pseudo_console->signal );
1827 if (pseudo_console->process)
1828 {
1829 WaitForSingleObject( pseudo_console->process, INFINITE );
1830 CloseHandle( pseudo_console->process );
1831 }
1832 if (pseudo_console->reference) CloseHandle( pseudo_console->reference );
1833 }
1834
1835 /******************************************************************************
1836 * ResizePseudoConsole (kernelbase.@)
1837 */
1838 HRESULT WINAPI ResizePseudoConsole( HPCON handle, COORD size )
1839 {
1840 FIXME( "%p (%u,%u)\n", handle, size.X, size.Y );
1841 return E_NOTIMPL;
1842 }
1843
1844 static BOOL is_tty_handle( HANDLE handle )
1845 {
1846 return ((UINT_PTR)handle & 3) == 1;
1847 }
1848
1849 void init_console( void )
1850 {
1851 RTL_USER_PROCESS_PARAMETERS *params = RtlGetCurrentPeb()->ProcessParameters;
1852
1853 if (params->ConsoleHandle == CONSOLE_HANDLE_SHELL)
1854 {
1855 HANDLE tty_in = NULL, tty_out = NULL, process = NULL;
1856 COORD size;
1857
1858 if (is_tty_handle( params->hStdInput ))
1859 {
1860 tty_in = params->hStdInput;
1861 params->hStdInput = NULL;
1862 }
1863 if (is_tty_handle( params->hStdOutput ))
1864 {
1865 tty_out = params->hStdOutput;
1866 params->hStdOutput = NULL;
1867 }
1868 if (is_tty_handle( params->hStdError ))
1869 {
1870 if (tty_out) CloseHandle( params->hStdError );
1871 else tty_out = params->hStdError;
1872 params->hStdError = NULL;
1873 }
1874
1875 size.X = params->dwXCountChars;
1876 size.Y = params->dwYCountChars;
1877 TRACE( "creating unix console (size %u %u)\n", size.X, size.Y );
1878 params->ConsoleHandle = create_pseudo_console( size, tty_in, tty_out, NULL, 0, &process );
1879 CloseHandle( process );
1880 CloseHandle( tty_in );
1881 CloseHandle( tty_out );
1882
1883 if (params->ConsoleHandle && create_console_connection( params->ConsoleHandle ))
1884 {
1885 init_console_std_handles( FALSE );
1886 console_flags = 0;
1887 }
1888 }
1889 else if (params->ConsoleHandle == CONSOLE_HANDLE_ALLOC)
1890 {
1891 HMODULE mod = GetModuleHandleW( NULL );
1892 params->ConsoleHandle = NULL;
1893 if (RtlImageNtHeader( mod )->OptionalHeader.Subsystem == IMAGE_SUBSYSTEM_WINDOWS_CUI)
1894 AllocConsole();
1895 }
1896 else if (params->ConsoleHandle) create_console_connection( params->ConsoleHandle );
1897
1898 RtlAddVectoredExceptionHandler( FALSE, handle_ctrl_c );
1899 }
Windows API implementation