2 * COPYRIGHT: See COPYING in the top level directory
3 * PROJECT: ReactOS Console Server DLL
4 * FILE: win32ss/user/winsrv/consrv/console.c
5 * PURPOSE: Console Management Functions
6 * PROGRAMMERS: Gé van Geldorp
8 * Hermes Belusca-Maito (hermes.belusca@sfr.fr)
11 /* INCLUDES *******************************************************************/
15 #include <ndk/psfuncs.h>
24 // FIXME: Add this prototype to winternl.h / rtlfuncs.h / ...
25 NTSTATUS NTAPI
RtlGetLastNtStatus(VOID
);
27 /* GLOBALS ********************************************************************/
29 static ULONG ConsoleListSize
;
30 static PCONSRV_CONSOLE
* ConsoleList
; /* The list of the ConSrv consoles */
31 static RTL_RESOURCE ListLock
;
33 #define ConSrvLockConsoleListExclusive() \
34 RtlAcquireResourceExclusive(&ListLock, TRUE)
36 #define ConSrvLockConsoleListShared() \
37 RtlAcquireResourceShared(&ListLock, TRUE)
39 #define ConSrvUnlockConsoleList() \
40 RtlReleaseResource(&ListLock)
44 InsertConsole(OUT PHANDLE Handle
,
45 IN PCONSRV_CONSOLE Console
)
47 #define CONSOLE_HANDLES_INCREMENT 2 * 3
49 NTSTATUS Status
= STATUS_SUCCESS
;
51 PCONSRV_CONSOLE
* Block
;
53 ASSERT( (ConsoleList
== NULL
&& ConsoleListSize
== 0) ||
54 (ConsoleList
!= NULL
&& ConsoleListSize
!= 0) );
56 /* All went right, so add the console to the list */
57 ConSrvLockConsoleListExclusive();
58 DPRINT1("Insert in the list\n");
62 for (i
= 0; i
< ConsoleListSize
; i
++)
64 if (ConsoleList
[i
] == NULL
) break;
68 if (i
>= ConsoleListSize
)
70 DPRINT1("Creation of a new handles table\n");
71 /* Allocate a new handles table */
72 Block
= ConsoleAllocHeap(HEAP_ZERO_MEMORY
,
74 CONSOLE_HANDLES_INCREMENT
) * sizeof(PCONSRV_CONSOLE
));
77 Status
= STATUS_UNSUCCESSFUL
;
81 /* If we previously had a handles table, free it and use the new one */
84 /* Copy the handles from the old table to the new one */
87 ConsoleListSize
* sizeof(PCONSRV_CONSOLE
));
88 ConsoleFreeHeap(ConsoleList
);
91 ConsoleListSize
+= CONSOLE_HANDLES_INCREMENT
;
94 ConsoleList
[i
] = Console
;
95 *Handle
= ULongToHandle((i
<< 2) | 0x3);
98 /* Unlock the console list and return status */
99 ConSrvUnlockConsoleList();
106 RemoveConsoleByHandle(IN HANDLE Handle
)
108 NTSTATUS Status
= STATUS_SUCCESS
;
109 PCONSRV_CONSOLE Console
;
111 BOOLEAN ValidHandle
= ((HandleToULong(Handle
) & 0x3) == 0x3);
112 ULONG Index
= HandleToULong(Handle
) >> 2;
114 if (!ValidHandle
) return STATUS_INVALID_HANDLE
;
116 ASSERT( (ConsoleList
== NULL
&& ConsoleListSize
== 0) ||
117 (ConsoleList
!= NULL
&& ConsoleListSize
!= 0) );
119 /* Remove the console from the list */
120 ConSrvLockConsoleListExclusive();
122 if (Index
>= ConsoleListSize
||
123 (Console
= ConsoleList
[Index
]) == NULL
)
125 Status
= STATUS_INVALID_HANDLE
;
129 ConsoleList
[Index
] = NULL
;
132 /* Unlock the console list and return status */
133 ConSrvUnlockConsoleList();
139 RemoveConsoleByPointer(IN PCONSRV_CONSOLE Console
)
143 if (!Console
) return STATUS_INVALID_PARAMETER
;
145 ASSERT( (ConsoleList
== NULL
&& ConsoleListSize
== 0) ||
146 (ConsoleList
!= NULL
&& ConsoleListSize
!= 0) );
148 /* Remove the console from the list */
149 ConSrvLockConsoleListExclusive();
153 for (i
= 0; i
< ConsoleListSize
; i
++)
155 if (ConsoleList
[i
] == Console
) ConsoleList
[i
] = NULL
;
159 /* Unlock the console list */
160 ConSrvUnlockConsoleList();
162 return STATUS_SUCCESS
;
166 ConSrvValidateConsole(OUT PCONSRV_CONSOLE
* Console
,
167 IN HANDLE ConsoleHandle
,
168 IN CONSOLE_STATE ExpectedState
,
169 IN BOOLEAN LockConsole
)
171 BOOLEAN RetVal
= FALSE
;
172 PCONSRV_CONSOLE ValidatedConsole
;
174 BOOLEAN ValidHandle
= ((HandleToULong(ConsoleHandle
) & 0x3) == 0x3);
175 ULONG Index
= HandleToULong(ConsoleHandle
) >> 2;
177 if (!ValidHandle
) return FALSE
;
179 if (!Console
) return FALSE
;
183 * Forbid creation or deletion of consoles when
184 * checking for the existence of a console.
186 ConSrvLockConsoleListShared();
188 if (Index
>= ConsoleListSize
||
189 (ValidatedConsole
= ConsoleList
[Index
]) == NULL
)
191 /* Unlock the console list */
192 ConSrvUnlockConsoleList();
197 ValidatedConsole
= ConsoleList
[Index
];
199 /* Unlock the console list and return */
200 ConSrvUnlockConsoleList();
202 RetVal
= ConDrvValidateConsoleUnsafe(ValidatedConsole
,
205 if (RetVal
) *Console
= ValidatedConsole
;
211 /* PRIVATE FUNCTIONS **********************************************************/
214 ConioPause(PCONSRV_CONSOLE Console
, UINT Flags
)
216 Console
->PauseFlags
|= Flags
;
217 ConDrvPause(Console
);
221 ConioUnpause(PCONSRV_CONSOLE Console
, UINT Flags
)
223 Console
->PauseFlags
&= ~Flags
;
225 // if ((Console->PauseFlags & (PAUSED_FROM_KEYBOARD | PAUSED_FROM_SCROLLBAR | PAUSED_FROM_SELECTION)) == 0)
226 if (Console
->PauseFlags
== 0)
228 ConDrvUnpause(Console
);
230 CsrNotifyWait(&Console
->WriteWaitQueue
,
234 if (!IsListEmpty(&Console
->WriteWaitQueue
))
236 CsrDereferenceWait(&Console
->WriteWaitQueue
);
242 ConSrvGetConsole(IN PCONSOLE_PROCESS_DATA ProcessData
,
243 OUT PCONSRV_CONSOLE
* Console
,
244 IN BOOLEAN LockConsole
)
246 NTSTATUS Status
= STATUS_INVALID_HANDLE
;
247 PCONSRV_CONSOLE GrabConsole
;
249 // if (Console == NULL) return STATUS_INVALID_PARAMETER;
253 // RtlEnterCriticalSection(&ProcessData->HandleTableLock);
255 if (ConSrvValidateConsole(&GrabConsole
,
256 ProcessData
->ConsoleHandle
,
260 InterlockedIncrement(&GrabConsole
->ReferenceCount
);
261 *Console
= GrabConsole
;
262 Status
= STATUS_SUCCESS
;
265 // RtlLeaveCriticalSection(&ProcessData->HandleTableLock);
270 ConSrvReleaseConsole(IN PCONSRV_CONSOLE Console
,
271 IN BOOLEAN WasConsoleLocked
)
275 if (!Console
) return;
276 // if (Console->ReferenceCount == 0) return; // This shouldn't happen
277 ASSERT(Console
->ReferenceCount
> 0);
279 /* The console must be locked */
280 // ASSERT(Console_locked);
283 * Decrement the reference count. Save the new value too,
284 * because Console->ReferenceCount might be modified after
285 * the console gets unlocked but before we check whether we
288 RefCount
= _InterlockedDecrement(&Console
->ReferenceCount
);
290 /* Unlock the console if needed */
291 if (WasConsoleLocked
) LeaveCriticalSection(&Console
->Lock
);
293 /* Delete the console if needed */
294 if (RefCount
<= 0) ConSrvDeleteConsole(Console
);
298 /* CONSOLE INITIALIZATION FUNCTIONS *******************************************/
301 ConSrvInitConsoleSupport(VOID
)
303 DPRINT("CONSRV: ConSrvInitConsoleSupport()\n");
305 /* Initialize the console list and its lock */
308 RtlInitializeResource(&ListLock
);
310 /* Should call LoadKeyboardLayout */
314 ConSrvInitTerminal(IN OUT PTERMINAL Terminal
,
315 IN OUT PCONSOLE_INFO ConsoleInfo
,
316 IN OUT PVOID ExtraConsoleInfo
,
319 ConSrvDeinitTerminal(IN OUT PTERMINAL Terminal
);
322 ConSrvInitConsole(OUT PHANDLE NewConsoleHandle
,
323 OUT PCONSRV_CONSOLE
* NewConsole
,
324 IN OUT PCONSOLE_START_INFO ConsoleStartInfo
,
325 IN ULONG ConsoleLeaderProcessId
)
328 HANDLE ConsoleHandle
;
329 PCONSRV_CONSOLE Console
;
330 CONSOLE_INFO ConsoleInfo
;
333 TERMINAL Terminal
; /* The ConSrv terminal for this console */
335 if (NewConsole
== NULL
|| ConsoleStartInfo
== NULL
)
336 return STATUS_INVALID_PARAMETER
;
341 * Load the console settings
344 /* 1. Load the default settings */
345 ConSrvGetDefaultSettings(&ConsoleInfo
, ConsoleLeaderProcessId
);
347 /* 2. Get the title of the console (initialize ConsoleInfo.ConsoleTitle) */
348 Length
= min(wcslen(ConsoleStartInfo
->ConsoleTitle
),
349 sizeof(ConsoleInfo
.ConsoleTitle
) / sizeof(ConsoleInfo
.ConsoleTitle
[0]) - 1);
350 wcsncpy(ConsoleInfo
.ConsoleTitle
, ConsoleStartInfo
->ConsoleTitle
, Length
);
351 ConsoleInfo
.ConsoleTitle
[Length
] = L
'\0';
353 /* 3. Initialize the ConSrv terminal */
354 Status
= ConSrvInitTerminal(&Terminal
,
357 ConsoleLeaderProcessId
);
358 if (!NT_SUCCESS(Status
))
360 DPRINT1("CONSRV: Failed to initialize a terminal, Status = 0x%08lx\n", Status
);
363 DPRINT("CONSRV: Terminal initialized\n");
366 * 4. Load the remaining console settings via the registry.
368 if ((ConsoleStartInfo
->dwStartupFlags
& STARTF_TITLEISLINKNAME
) == 0)
371 * Either we weren't created by an app launched via a shell-link,
372 * or we failed to load shell-link console properties.
373 * Therefore, load the console infos for the application from the registry.
375 ConSrvReadUserSettings(&ConsoleInfo
, ConsoleLeaderProcessId
);
378 * Now, update them with the properties the user might gave to us
379 * via the STARTUPINFO structure before calling CreateProcess
380 * (and which was transmitted via the ConsoleStartInfo structure).
381 * We therefore overwrite the values read in the registry.
383 if (ConsoleStartInfo
->dwStartupFlags
& STARTF_USEFILLATTRIBUTE
)
385 ConsoleInfo
.ScreenAttrib
= (USHORT
)ConsoleStartInfo
->wFillAttribute
;
387 if (ConsoleStartInfo
->dwStartupFlags
& STARTF_USECOUNTCHARS
)
389 ConsoleInfo
.ScreenBufferSize
= ConsoleStartInfo
->dwScreenBufferSize
;
391 if (ConsoleStartInfo
->dwStartupFlags
& STARTF_USESIZE
)
393 ConsoleInfo
.ConsoleSize
= ConsoleStartInfo
->dwWindowSize
;
397 /* Set-up the code page */
398 ConsoleInfo
.CodePage
= GetOEMCP();
400 /* Initialize a new console via the driver */
401 Status
= ConDrvInitConsole(&Console
, &ConsoleInfo
);
402 if (!NT_SUCCESS(Status
))
404 DPRINT1("Creating a new console failed, Status = 0x%08lx\n", Status
);
405 ConSrvDeinitTerminal(&Terminal
);
410 DPRINT("Console initialized\n");
412 /*** Register ConSrv features ***/
414 /* Initialize process support */
415 InitializeListHead(&Console
->ProcessList
);
416 Console
->NotifiedLastCloseProcess
= NULL
;
417 Console
->NotifyLastClose
= FALSE
;
419 /* Initialize pausing support */
420 Console
->PauseFlags
= 0;
421 InitializeListHead(&Console
->ReadWaitQueue
);
422 InitializeListHead(&Console
->WriteWaitQueue
);
424 /* Initialize the alias and history buffers */
425 Console
->Aliases
= NULL
;
426 InitializeListHead(&Console
->HistoryBuffers
);
427 Console
->HistoryBufferSize
= ConsoleInfo
.HistoryBufferSize
;
428 Console
->NumberOfHistoryBuffers
= ConsoleInfo
.NumberOfHistoryBuffers
;
429 Console
->HistoryNoDup
= ConsoleInfo
.HistoryNoDup
;
431 /* Initialize the Input Line Discipline */
432 Console
->LineBuffer
= NULL
;
433 Console
->LinePos
= Console
->LineMaxSize
= Console
->LineSize
= 0;
434 Console
->LineComplete
= Console
->LineUpPressed
= FALSE
;
436 Console
->LineInsertToggle
=
437 Console
->InsertMode
= ConsoleInfo
.InsertMode
;
438 Console
->QuickEdit
= ConsoleInfo
.QuickEdit
;
441 InitializeListHead(&Console
->PopupWindows
);
444 memcpy(Console
->Colors
, ConsoleInfo
.Colors
, sizeof(ConsoleInfo
.Colors
));
446 /* Attach the ConSrv terminal to the console */
447 Status
= ConDrvRegisterTerminal(Console
, &Terminal
);
448 if (!NT_SUCCESS(Status
))
450 DPRINT1("Failed to register terminal to the given console, Status = 0x%08lx\n", Status
);
451 ConDrvDeleteConsole(Console
);
452 ConSrvDeinitTerminal(&Terminal
);
455 DPRINT("Terminal registered\n");
457 /* All went right, so add the console to the list */
458 Status
= InsertConsole(&ConsoleHandle
, Console
);
460 /* Return the newly created console to the caller and a success code too */
461 *NewConsoleHandle
= ConsoleHandle
;
462 *NewConsole
= Console
;
463 return STATUS_SUCCESS
;
467 ConSrvDeleteConsole(PCONSRV_CONSOLE Console
)
469 DPRINT("ConSrvDeleteConsole\n");
471 // FIXME: Send a terminate message to all the processes owning this console
473 /* Remove the console from the list */
474 RemoveConsoleByPointer(Console
);
476 /* Clean aliases and history */
477 IntDeleteAllAliases(Console
);
478 HistoryDeleteBuffers(Console
);
480 /* Now, call the driver. ConDrvDeregisterTerminal is called on-demand. */
481 ConDrvDeleteConsole(Console
);
490 ConSrvConsoleCtrlEventTimeout(IN ULONG CtrlEvent
,
491 IN PCONSOLE_PROCESS_DATA ProcessData
,
494 NTSTATUS Status
= STATUS_SUCCESS
;
496 DPRINT("ConSrvConsoleCtrlEventTimeout Parent ProcessId = %x\n", ProcessData
->Process
->ClientId
.UniqueProcess
);
498 if (ProcessData
->CtrlDispatcher
)
502 HANDLE Thread
= NULL
;
506 Thread
= CreateRemoteThread(ProcessData
->Process
->ProcessHandle
, NULL
, 0,
507 ProcessData
->CtrlDispatcher
,
508 UlongToPtr(CtrlEvent
), 0, NULL
);
511 Status
= RtlGetLastNtStatus();
512 DPRINT1("Failed thread creation, Status = 0x%08lx\n", Status
);
516 DPRINT("ProcessData->CtrlDispatcher remote thread creation succeeded, ProcessId = %x, Process = 0x%p\n", ProcessData
->Process
->ClientId
.UniqueProcess
, ProcessData
->Process
);
517 WaitForSingleObject(Thread
, Timeout
);
526 _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER
)
528 Status
= _SEH2_GetExceptionCode();
529 DPRINT1("ConSrvConsoleCtrlEventTimeout - Caught an exception, Status = 0x%08lx\n", Status
);
538 ConSrvConsoleCtrlEvent(IN ULONG CtrlEvent
,
539 IN PCONSOLE_PROCESS_DATA ProcessData
)
541 return ConSrvConsoleCtrlEventTimeout(CtrlEvent
, ProcessData
, 0);
544 PCONSOLE_PROCESS_DATA NTAPI
545 ConSrvGetConsoleLeaderProcess(IN PCONSRV_CONSOLE Console
)
547 if (Console
== NULL
) return NULL
;
549 return CONTAINING_RECORD(Console
->ProcessList
.Blink
,
550 CONSOLE_PROCESS_DATA
,
555 ConSrvGetConsoleProcessList(IN PCONSRV_CONSOLE Console
,
556 IN OUT PULONG ProcessIdsList
,
557 IN ULONG MaxIdListItems
,
558 OUT PULONG ProcessIdsTotal
)
560 PCONSOLE_PROCESS_DATA current
;
561 PLIST_ENTRY current_entry
;
563 if (Console
== NULL
|| ProcessIdsList
== NULL
|| ProcessIdsTotal
== NULL
)
564 return STATUS_INVALID_PARAMETER
;
566 *ProcessIdsTotal
= 0;
568 for (current_entry
= Console
->ProcessList
.Flink
;
569 current_entry
!= &Console
->ProcessList
;
570 current_entry
= current_entry
->Flink
)
572 current
= CONTAINING_RECORD(current_entry
, CONSOLE_PROCESS_DATA
, ConsoleLink
);
573 if (++(*ProcessIdsTotal
) <= MaxIdListItems
)
575 *ProcessIdsList
++ = HandleToUlong(current
->Process
->ClientId
.UniqueProcess
);
579 return STATUS_SUCCESS
;
582 // ConSrvGenerateConsoleCtrlEvent
584 ConSrvConsoleProcessCtrlEvent(IN PCONSRV_CONSOLE Console
,
585 IN ULONG ProcessGroupId
,
588 NTSTATUS Status
= STATUS_SUCCESS
;
589 PLIST_ENTRY current_entry
;
590 PCONSOLE_PROCESS_DATA current
;
592 /* If the console is already being destroyed, just return */
593 if (!ConDrvValidateConsoleState(Console
, CONSOLE_RUNNING
))
594 return STATUS_UNSUCCESSFUL
;
597 * Loop through the process list, from the most recent process
598 * (the active one) to the oldest one (the first created, i.e.
599 * the console leader process), and for each, send an event
600 * (new processes are inserted at the head of the console process list).
602 current_entry
= Console
->ProcessList
.Flink
;
603 while (current_entry
!= &Console
->ProcessList
)
605 current
= CONTAINING_RECORD(current_entry
, CONSOLE_PROCESS_DATA
, ConsoleLink
);
606 current_entry
= current_entry
->Flink
;
609 * Only processes belonging to the same process group are signaled.
610 * If the process group ID is zero, then all the processes are signaled.
612 if (ProcessGroupId
== 0 || current
->Process
->ProcessGroupId
== ProcessGroupId
)
614 Status
= ConSrvConsoleCtrlEvent(CtrlEvent
, current
);
625 /* PUBLIC SERVER APIS *********************************************************/
627 CSR_API(SrvAllocConsole
)
629 NTSTATUS Status
= STATUS_SUCCESS
;
630 PCONSOLE_ALLOCCONSOLE AllocConsoleRequest
= &((PCONSOLE_API_MESSAGE
)ApiMessage
)->Data
.AllocConsoleRequest
;
631 PCSR_PROCESS CsrProcess
= CsrGetClientThread()->Process
;
632 PCONSOLE_PROCESS_DATA ProcessData
= ConsoleGetPerProcessData(CsrProcess
);
634 if (ProcessData
->ConsoleHandle
!= NULL
)
636 DPRINT1("Process already has a console\n");
637 return STATUS_ACCESS_DENIED
;
640 if (!CsrValidateMessageBuffer(ApiMessage
,
641 (PVOID
*)&AllocConsoleRequest
->ConsoleStartInfo
,
643 sizeof(CONSOLE_START_INFO
)))
645 return STATUS_INVALID_PARAMETER
;
648 /* Initialize a new Console owned by the Console Leader Process */
649 Status
= ConSrvAllocateConsole(ProcessData
,
650 &AllocConsoleRequest
->InputHandle
,
651 &AllocConsoleRequest
->OutputHandle
,
652 &AllocConsoleRequest
->ErrorHandle
,
653 AllocConsoleRequest
->ConsoleStartInfo
);
654 if (!NT_SUCCESS(Status
))
656 DPRINT1("Console allocation failed\n");
660 /* Return the console handle and the input wait handle to the caller */
661 AllocConsoleRequest
->ConsoleHandle
= ProcessData
->ConsoleHandle
;
662 AllocConsoleRequest
->InputWaitHandle
= ProcessData
->InputWaitHandle
;
664 /* Set the Property-Dialog and Control-Dispatcher handlers */
665 ProcessData
->PropDispatcher
= AllocConsoleRequest
->PropDispatcher
;
666 ProcessData
->CtrlDispatcher
= AllocConsoleRequest
->CtrlDispatcher
;
668 return STATUS_SUCCESS
;
671 CSR_API(SrvAttachConsole
)
673 NTSTATUS Status
= STATUS_SUCCESS
;
674 PCONSOLE_ATTACHCONSOLE AttachConsoleRequest
= &((PCONSOLE_API_MESSAGE
)ApiMessage
)->Data
.AttachConsoleRequest
;
675 PCSR_PROCESS SourceProcess
= NULL
; // The parent process.
676 PCSR_PROCESS TargetProcess
= CsrGetClientThread()->Process
; // Ourselves.
677 HANDLE ProcessId
= ULongToHandle(AttachConsoleRequest
->ProcessId
);
678 PCONSOLE_PROCESS_DATA SourceProcessData
, TargetProcessData
;
680 TargetProcessData
= ConsoleGetPerProcessData(TargetProcess
);
682 if (TargetProcessData
->ConsoleHandle
!= NULL
)
684 DPRINT1("Process already has a console\n");
685 return STATUS_ACCESS_DENIED
;
688 /* Check whether we try to attach to the parent's console */
689 if (ProcessId
== ULongToHandle(ATTACH_PARENT_PROCESS
))
691 PROCESS_BASIC_INFORMATION ProcessInfo
;
692 ULONG Length
= sizeof(ProcessInfo
);
694 /* Get the real parent's ID */
696 Status
= NtQueryInformationProcess(TargetProcess
->ProcessHandle
,
697 ProcessBasicInformation
,
700 if (!NT_SUCCESS(Status
))
702 DPRINT1("SrvAttachConsole - Cannot retrieve basic process info, Status = %lu\n", Status
);
706 ProcessId
= ULongToHandle(ProcessInfo
.InheritedFromUniqueProcessId
);
709 /* Lock the source process via its PID */
710 Status
= CsrLockProcessByClientId(ProcessId
, &SourceProcess
);
711 if (!NT_SUCCESS(Status
)) return Status
;
713 SourceProcessData
= ConsoleGetPerProcessData(SourceProcess
);
715 if (SourceProcessData
->ConsoleHandle
== NULL
)
717 Status
= STATUS_INVALID_HANDLE
;
722 * Inherit the console from the parent,
723 * if any, otherwise return an error.
725 Status
= ConSrvInheritConsole(TargetProcessData
,
726 SourceProcessData
->ConsoleHandle
,
728 &AttachConsoleRequest
->InputHandle
,
729 &AttachConsoleRequest
->OutputHandle
,
730 &AttachConsoleRequest
->ErrorHandle
);
731 if (!NT_SUCCESS(Status
))
733 DPRINT1("Console inheritance failed\n");
737 /* Return the console handle and the input wait handle to the caller */
738 AttachConsoleRequest
->ConsoleHandle
= TargetProcessData
->ConsoleHandle
;
739 AttachConsoleRequest
->InputWaitHandle
= TargetProcessData
->InputWaitHandle
;
741 /* Set the Property-Dialog and Control-Dispatcher handlers */
742 TargetProcessData
->PropDispatcher
= AttachConsoleRequest
->PropDispatcher
;
743 TargetProcessData
->CtrlDispatcher
= AttachConsoleRequest
->CtrlDispatcher
;
745 Status
= STATUS_SUCCESS
;
748 /* Unlock the "source" process and exit */
749 CsrUnlockProcess(SourceProcess
);
753 CSR_API(SrvFreeConsole
)
755 ConSrvRemoveConsole(ConsoleGetPerProcessData(CsrGetClientThread()->Process
));
756 return STATUS_SUCCESS
;
760 ConDrvGetConsoleMode(IN PCONSOLE Console
,
761 IN PCONSOLE_IO_OBJECT Object
,
762 OUT PULONG ConsoleMode
);
763 CSR_API(SrvGetConsoleMode
)
766 PCONSOLE_GETSETCONSOLEMODE ConsoleModeRequest
= &((PCONSOLE_API_MESSAGE
)ApiMessage
)->Data
.ConsoleModeRequest
;
767 PCONSOLE_IO_OBJECT Object
;
769 PULONG ConsoleMode
= &ConsoleModeRequest
->Mode
;
771 Status
= ConSrvGetObject(ConsoleGetPerProcessData(CsrGetClientThread()->Process
),
772 ConsoleModeRequest
->Handle
,
773 &Object
, NULL
, GENERIC_READ
, TRUE
, 0);
774 if (!NT_SUCCESS(Status
)) return Status
;
776 /* Get the standard console modes */
777 Status
= ConDrvGetConsoleMode(Object
->Console
, Object
,
779 if (NT_SUCCESS(Status
))
782 * If getting the console modes succeeds, then retrieve
783 * the extended CONSRV-specific input modes.
785 if (INPUT_BUFFER
== Object
->Type
)
787 if (Object
->Console
->InsertMode
|| Object
->Console
->QuickEdit
)
789 /* Windows does this, even if it is not documented on MSDN */
790 *ConsoleMode
|= ENABLE_EXTENDED_FLAGS
;
792 if (Object
->Console
->InsertMode
) *ConsoleMode
|= ENABLE_INSERT_MODE
;
793 if (Object
->Console
->QuickEdit
) *ConsoleMode
|= ENABLE_QUICK_EDIT_MODE
;
798 ConSrvReleaseObject(Object
, TRUE
);
803 ConDrvSetConsoleMode(IN PCONSOLE Console
,
804 IN PCONSOLE_IO_OBJECT Object
,
805 IN ULONG ConsoleMode
);
806 CSR_API(SrvSetConsoleMode
)
808 #define CONSOLE_VALID_CONTROL_MODES ( ENABLE_EXTENDED_FLAGS | \
809 ENABLE_INSERT_MODE | ENABLE_QUICK_EDIT_MODE )
812 PCONSOLE_GETSETCONSOLEMODE ConsoleModeRequest
= &((PCONSOLE_API_MESSAGE
)ApiMessage
)->Data
.ConsoleModeRequest
;
813 PCONSOLE_IO_OBJECT Object
;
815 ULONG ConsoleMode
= ConsoleModeRequest
->Mode
;
817 Status
= ConSrvGetObject(ConsoleGetPerProcessData(CsrGetClientThread()->Process
),
818 ConsoleModeRequest
->Handle
,
819 &Object
, NULL
, GENERIC_WRITE
, TRUE
, 0);
820 if (!NT_SUCCESS(Status
)) return Status
;
822 /* Set the standard console modes (without the CONSRV-specific input modes) */
823 ConsoleMode
&= ~CONSOLE_VALID_CONTROL_MODES
; // Remove CONSRV-specific input modes.
824 Status
= ConDrvSetConsoleMode(Object
->Console
, Object
,
826 if (NT_SUCCESS(Status
))
829 * If setting the console modes succeeds, then set
830 * the extended CONSRV-specific input modes.
832 if (INPUT_BUFFER
== Object
->Type
)
834 ConsoleMode
= ConsoleModeRequest
->Mode
;
836 if (ConsoleMode
& CONSOLE_VALID_CONTROL_MODES
)
839 * If we use control mode flags without ENABLE_EXTENDED_FLAGS,
840 * then consider the flags invalid.
842 if ((ConsoleMode
& ENABLE_EXTENDED_FLAGS
) == 0)
844 Status
= STATUS_INVALID_PARAMETER
;
848 Object
->Console
->InsertMode
= !!(ConsoleMode
& ENABLE_INSERT_MODE
);
849 Object
->Console
->QuickEdit
= !!(ConsoleMode
& ENABLE_QUICK_EDIT_MODE
);
855 ConSrvReleaseObject(Object
, TRUE
);
860 ConDrvGetConsoleTitle(IN PCONSOLE Console
,
862 IN OUT PVOID TitleBuffer
,
863 IN OUT PULONG BufLength
);
864 CSR_API(SrvGetConsoleTitle
)
867 PCONSOLE_GETSETCONSOLETITLE TitleRequest
= &((PCONSOLE_API_MESSAGE
)ApiMessage
)->Data
.TitleRequest
;
868 PCONSRV_CONSOLE Console
;
870 if (!CsrValidateMessageBuffer(ApiMessage
,
871 (PVOID
)&TitleRequest
->Title
,
872 TitleRequest
->Length
,
875 return STATUS_INVALID_PARAMETER
;
878 Status
= ConSrvGetConsole(ConsoleGetPerProcessData(CsrGetClientThread()->Process
), &Console
, TRUE
);
879 if (!NT_SUCCESS(Status
))
881 DPRINT1("Can't get console\n");
885 Status
= ConDrvGetConsoleTitle(Console
,
886 TitleRequest
->Unicode
,
888 &TitleRequest
->Length
);
890 ConSrvReleaseConsole(Console
, TRUE
);
895 ConDrvSetConsoleTitle(IN PCONSOLE Console
,
897 IN PVOID TitleBuffer
,
899 CSR_API(SrvSetConsoleTitle
)
902 PCONSOLE_GETSETCONSOLETITLE TitleRequest
= &((PCONSOLE_API_MESSAGE
)ApiMessage
)->Data
.TitleRequest
;
903 PCONSRV_CONSOLE Console
;
905 if (!CsrValidateMessageBuffer(ApiMessage
,
906 (PVOID
)&TitleRequest
->Title
,
907 TitleRequest
->Length
,
910 return STATUS_INVALID_PARAMETER
;
913 Status
= ConSrvGetConsole(ConsoleGetPerProcessData(CsrGetClientThread()->Process
), &Console
, TRUE
);
914 if (!NT_SUCCESS(Status
))
916 DPRINT1("Can't get console\n");
920 Status
= ConDrvSetConsoleTitle(Console
,
921 TitleRequest
->Unicode
,
923 TitleRequest
->Length
);
924 if (NT_SUCCESS(Status
)) TermChangeTitle(Console
);
926 ConSrvReleaseConsole(Console
, TRUE
);
931 ConDrvGetConsoleCP(IN PCONSOLE Console
,
933 IN BOOLEAN OutputCP
);
934 CSR_API(SrvGetConsoleCP
)
937 PCONSOLE_GETINPUTOUTPUTCP GetConsoleCPRequest
= &((PCONSOLE_API_MESSAGE
)ApiMessage
)->Data
.GetConsoleCPRequest
;
938 PCONSRV_CONSOLE Console
;
940 DPRINT("SrvGetConsoleCP, getting %s Code Page\n",
941 GetConsoleCPRequest
->OutputCP
? "Output" : "Input");
943 Status
= ConSrvGetConsole(ConsoleGetPerProcessData(CsrGetClientThread()->Process
), &Console
, TRUE
);
944 if (!NT_SUCCESS(Status
)) return Status
;
946 Status
= ConDrvGetConsoleCP(Console
,
947 &GetConsoleCPRequest
->CodePage
,
948 GetConsoleCPRequest
->OutputCP
);
950 ConSrvReleaseConsole(Console
, TRUE
);
955 ConDrvSetConsoleCP(IN PCONSOLE Console
,
957 IN BOOLEAN OutputCP
);
958 CSR_API(SrvSetConsoleCP
)
960 NTSTATUS Status
= STATUS_INVALID_PARAMETER
;
961 PCONSOLE_SETINPUTOUTPUTCP SetConsoleCPRequest
= &((PCONSOLE_API_MESSAGE
)ApiMessage
)->Data
.SetConsoleCPRequest
;
962 PCONSRV_CONSOLE Console
;
964 DPRINT("SrvSetConsoleCP, setting %s Code Page\n",
965 SetConsoleCPRequest
->OutputCP
? "Output" : "Input");
967 Status
= ConSrvGetConsole(ConsoleGetPerProcessData(CsrGetClientThread()->Process
), &Console
, TRUE
);
968 if (!NT_SUCCESS(Status
)) return Status
;
970 Status
= ConDrvSetConsoleCP(Console
,
971 SetConsoleCPRequest
->CodePage
,
972 SetConsoleCPRequest
->OutputCP
);
974 ConSrvReleaseConsole(Console
, TRUE
);
978 CSR_API(SrvGetConsoleProcessList
)
981 PCONSOLE_GETPROCESSLIST GetProcessListRequest
= &((PCONSOLE_API_MESSAGE
)ApiMessage
)->Data
.GetProcessListRequest
;
982 PCONSRV_CONSOLE Console
;
984 if (!CsrValidateMessageBuffer(ApiMessage
,
985 (PVOID
)&GetProcessListRequest
->ProcessIdsList
,
986 GetProcessListRequest
->ProcessCount
,
989 return STATUS_INVALID_PARAMETER
;
992 Status
= ConSrvGetConsole(ConsoleGetPerProcessData(CsrGetClientThread()->Process
), &Console
, TRUE
);
993 if (!NT_SUCCESS(Status
)) return Status
;
995 Status
= ConSrvGetConsoleProcessList(Console
,
996 GetProcessListRequest
->ProcessIdsList
,
997 GetProcessListRequest
->ProcessCount
,
998 &GetProcessListRequest
->ProcessCount
);
1000 ConSrvReleaseConsole(Console
, TRUE
);
1004 CSR_API(SrvGenerateConsoleCtrlEvent
)
1007 PCONSOLE_GENERATECTRLEVENT GenerateCtrlEventRequest
= &((PCONSOLE_API_MESSAGE
)ApiMessage
)->Data
.GenerateCtrlEventRequest
;
1008 PCONSRV_CONSOLE Console
;
1010 Status
= ConSrvGetConsole(ConsoleGetPerProcessData(CsrGetClientThread()->Process
), &Console
, TRUE
);
1011 if (!NT_SUCCESS(Status
)) return Status
;
1013 Status
= ConSrvConsoleProcessCtrlEvent(Console
,
1014 GenerateCtrlEventRequest
->ProcessGroupId
,
1015 GenerateCtrlEventRequest
->CtrlEvent
);
1017 ConSrvReleaseConsole(Console
, TRUE
);
1021 CSR_API(SrvConsoleNotifyLastClose
)
1024 PCONSOLE_PROCESS_DATA ProcessData
= ConsoleGetPerProcessData(CsrGetClientThread()->Process
);
1025 PCONSRV_CONSOLE Console
;
1027 Status
= ConSrvGetConsole(ProcessData
, &Console
, TRUE
);
1028 if (!NT_SUCCESS(Status
)) return Status
;
1030 /* Only one process is allowed to be registered for last close notification */
1031 if (!Console
->NotifyLastClose
)
1033 Console
->NotifyLastClose
= TRUE
;
1034 Console
->NotifiedLastCloseProcess
= ProcessData
;
1035 Status
= STATUS_SUCCESS
;
1039 Status
= STATUS_ACCESS_DENIED
;
1042 ConSrvReleaseConsole(Console
, TRUE
);
1048 CSR_API(SrvGetConsoleMouseInfo
)
1051 PCONSOLE_GETMOUSEINFO GetMouseInfoRequest
= &((PCONSOLE_API_MESSAGE
)ApiMessage
)->Data
.GetMouseInfoRequest
;
1052 PCONSRV_CONSOLE Console
;
1054 Status
= ConSrvGetConsole(ConsoleGetPerProcessData(CsrGetClientThread()->Process
), &Console
, TRUE
);
1055 if (!NT_SUCCESS(Status
)) return Status
;
1057 /* Just retrieve the number of buttons of the mouse attached to this console */
1058 GetMouseInfoRequest
->NumButtons
= GetSystemMetrics(SM_CMOUSEBUTTONS
);
1060 ConSrvReleaseConsole(Console
, TRUE
);
1061 return STATUS_SUCCESS
;
1064 CSR_API(SrvSetConsoleKeyShortcuts
)
1066 DPRINT1("%s not yet implemented\n", __FUNCTION__
);
1067 return STATUS_NOT_IMPLEMENTED
;
1070 CSR_API(SrvGetConsoleKeyboardLayoutName
)
1073 PCONSOLE_GETKBDLAYOUTNAME GetKbdLayoutNameRequest
= &((PCONSOLE_API_MESSAGE
)ApiMessage
)->Data
.GetKbdLayoutNameRequest
;
1074 PCONSRV_CONSOLE Console
;
1076 Status
= ConSrvGetConsole(ConsoleGetPerProcessData(CsrGetClientThread()->Process
), &Console
, TRUE
);
1077 if (!NT_SUCCESS(Status
)) return Status
;
1079 /* Retrieve the keyboard layout name of the system */
1080 if (GetKbdLayoutNameRequest
->Ansi
)
1081 GetKeyboardLayoutNameA((PCHAR
)GetKbdLayoutNameRequest
->LayoutBuffer
);
1083 GetKeyboardLayoutNameW((PWCHAR
)GetKbdLayoutNameRequest
->LayoutBuffer
);
1085 ConSrvReleaseConsole(Console
, TRUE
);
1086 return STATUS_SUCCESS
;
1089 CSR_API(SrvGetConsoleCharType
)
1091 DPRINT1("%s not yet implemented\n", __FUNCTION__
);
1092 return STATUS_NOT_IMPLEMENTED
;
1095 CSR_API(SrvSetConsoleLocalEUDC
)
1097 DPRINT1("%s not yet implemented\n", __FUNCTION__
);
1098 return STATUS_NOT_IMPLEMENTED
;
1101 CSR_API(SrvSetConsoleCursorMode
)
1103 DPRINT1("%s not yet implemented\n", __FUNCTION__
);
1104 return STATUS_NOT_IMPLEMENTED
;
1107 CSR_API(SrvGetConsoleCursorMode
)
1109 DPRINT1("%s not yet implemented\n", __FUNCTION__
);
1110 return STATUS_NOT_IMPLEMENTED
;
1113 CSR_API(SrvGetConsoleNlsMode
)
1115 DPRINT1("%s not yet implemented\n", __FUNCTION__
);
1116 return STATUS_NOT_IMPLEMENTED
;
1119 CSR_API(SrvSetConsoleNlsMode
)
1121 DPRINT1("%s not yet implemented\n", __FUNCTION__
);
1122 return STATUS_NOT_IMPLEMENTED
;
1125 CSR_API(SrvGetConsoleLangId
)
1127 DPRINT1("%s not yet implemented\n", __FUNCTION__
);
1128 return STATUS_NOT_IMPLEMENTED
;