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