-/* $Id: conio.c,v 1.2 1999/12/30 01:51:42 dwelch Exp $
+/* $Id: conio.c,v 1.24 2001/08/14 22:00:21 ea Exp $
*
* reactos/subsys/csrss/api/conio.c
*
#include <csrss/csrss.h>
#include "api.h"
+#include <ntdll/rtl.h>
+#include <ddk/ntddblue.h>
+#include <debug.h>
+
+#define LOCK RtlEnterCriticalSection(&ActiveConsoleLock)
+#define UNLOCK RtlLeaveCriticalSection(&ActiveConsoleLock)
+
/* GLOBALS *******************************************************************/
-static HANDLE ConsoleDevice;
-static HANDLE KeyboardDevice;
+static HANDLE ConsoleDeviceHandle;
+static HANDLE KeyboardDeviceHandle;
+static PCSRSS_CONSOLE ActiveConsole;
+CRITICAL_SECTION ActiveConsoleLock;
+static COORD PhysicalConsoleSize;
/* FUNCTIONS *****************************************************************/
-NTSTATUS CsrAllocConsole(PCSRSS_PROCESS_DATA ProcessData,
- PCSRSS_API_REQUEST LpcMessage,
- PHANDLE ReturnedHandle)
+CSR_API(CsrAllocConsole)
{
- return(STATUS_NOT_IMPLEMENTED);
+ PCSRSS_CONSOLE Console;
+ HANDLE Process;
+ NTSTATUS Status;
+ CLIENT_ID ClientId;
+
+ Reply->Header.MessageSize = sizeof(CSRSS_API_REPLY);
+ Reply->Header.DataSize = sizeof(CSRSS_API_REPLY) -
+ sizeof(LPC_MESSAGE_HEADER);
+ if( ProcessData->Console )
+ {
+ Reply->Status = STATUS_INVALID_PARAMETER;
+ return STATUS_INVALID_PARAMETER;
+ }
+ Reply->Status = STATUS_SUCCESS;
+ Console = RtlAllocateHeap( CsrssApiHeap, 0, sizeof( CSRSS_CONSOLE ) );
+ if( Console == 0 )
+ {
+ Reply->Status = STATUS_INSUFFICIENT_RESOURCES;
+ return STATUS_INSUFFICIENT_RESOURCES;
+ }
+ Reply->Status = CsrInitConsole( Console );
+ if( !NT_SUCCESS( Reply->Status ) )
+ {
+ RtlFreeHeap( CsrssApiHeap, 0, Console );
+ return Reply->Status;
+ }
+ ProcessData->Console = Console;
+ /* add a reference count because the process is tied to the console */
+ Console->Header.ReferenceCount++;
+ Status = CsrInsertObject( ProcessData, &Reply->Data.AllocConsoleReply.InputHandle, &Console->Header );
+ if( !NT_SUCCESS( Status ) )
+ {
+ CsrDeleteConsole( Console );
+ ProcessData->Console = 0;
+ return Reply->Status = Status;
+ }
+ Status = CsrInsertObject( ProcessData, &Reply->Data.AllocConsoleReply.OutputHandle, &Console->ActiveBuffer->Header );
+ if( !NT_SUCCESS( Status ) )
+ {
+ Console->Header.ReferenceCount--;
+ CsrReleaseObject( ProcessData, Reply->Data.AllocConsoleReply.InputHandle );
+ ProcessData->Console = 0;
+ return Reply->Status = Status;
+ }
+ ClientId.UniqueProcess = (HANDLE)ProcessData->ProcessId;
+ Status = NtOpenProcess( &Process, PROCESS_DUP_HANDLE, 0, &ClientId );
+ if( !NT_SUCCESS( Status ) )
+ {
+ DbgPrint( "CSR: NtOpenProcess() failed for handle duplication\n" );
+ Console->Header.ReferenceCount--;
+ ProcessData->Console = 0;
+ CsrReleaseObject( ProcessData, Reply->Data.AllocConsoleReply.OutputHandle );
+ CsrReleaseObject( ProcessData, Reply->Data.AllocConsoleReply.InputHandle );
+ Reply->Status = Status;
+ return Status;
+ }
+ Status = NtDuplicateObject( NtCurrentProcess(), &ProcessData->Console->ActiveEvent, Process, &ProcessData->ConsoleEvent, SYNCHRONIZE, FALSE, 0 );
+ if( !NT_SUCCESS( Status ) )
+ {
+ DbgPrint( "CSR: NtDuplicateObject() failed: %x\n", Status );
+ NtClose( Process );
+ Console->Header.ReferenceCount--;
+ CsrReleaseObject( ProcessData, Reply->Data.AllocConsoleReply.OutputHandle );
+ CsrReleaseObject( ProcessData, Reply->Data.AllocConsoleReply.InputHandle );
+ ProcessData->Console = 0;
+ Reply->Status = Status;
+ return Status;
+ }
+ NtClose( Process );
+ return STATUS_SUCCESS;
}
-NTSTATUS CsrFreeConsole(PCSRSS_PROCESS_DATA ProcessData,
- PCSRSS_API_REQUEST LpcMessage)
+CSR_API(CsrFreeConsole)
{
+ Reply->Header.MessageSize = sizeof(CSRSS_API_REPLY);
+ Reply->Header.DataSize = sizeof(CSRSS_API_REPLY) -
+ sizeof(LPC_MESSAGE_HEADER);
+
+ Reply->Status = STATUS_NOT_IMPLEMENTED;
+
return(STATUS_NOT_IMPLEMENTED);
}
-NTSTATUS CsrReadConsole(PCSRSS_PROCESS_DATA ProcessData,
- PCSRSS_API_REQUEST LpcMessage,
- PULONG CharCount)
+CSR_API(CsrReadConsole)
{
- return(STATUS_NOT_IMPLEMENTED);
+ ConsoleInput *Input;
+ PCHAR Buffer;
+ int i = 0;
+ ULONG nNumberOfCharsToRead;
+ PCSRSS_CONSOLE Console;
+ NTSTATUS Status;
+
+ /* truncate length to CSRSS_MAX_READ_CONSOLE_REQUEST */
+ nNumberOfCharsToRead = Request->Data.ReadConsoleRequest.NrCharactersToRead > CSRSS_MAX_READ_CONSOLE_REQUEST ? CSRSS_MAX_READ_CONSOLE_REQUEST : Request->Data.ReadConsoleRequest.NrCharactersToRead;
+ Reply->Header.MessageSize = sizeof(CSRSS_API_REPLY);
+ Reply->Header.DataSize = Reply->Header.MessageSize -
+ sizeof(LPC_MESSAGE_HEADER);
+ Buffer = Reply->Data.ReadConsoleReply.Buffer;
+ Reply->Data.ReadConsoleReply.EventHandle = ProcessData->ConsoleEvent;
+ LOCK;
+ Status = CsrGetObject( ProcessData, Request->Data.ReadConsoleRequest.ConsoleHandle, (Object_t **)&Console );
+ if( !NT_SUCCESS( Status ) )
+ {
+ Reply->Status = Status;
+ UNLOCK;
+ return Status;
+ }
+ if( Console->Header.Type != CSRSS_CONSOLE_MAGIC )
+ {
+ Reply->Status = STATUS_INVALID_HANDLE;
+ UNLOCK;
+ return STATUS_INVALID_HANDLE;
+ }
+ for (; i<nNumberOfCharsToRead && Console->InputEvents.Flink != &Console->InputEvents; i++ )
+ {
+ // remove input event from queue
+ Input = (ConsoleInput *)Console->InputEvents.Flink;
+
+ Input->ListEntry.Blink->Flink = Input->ListEntry.Flink;
+ Input->ListEntry.Flink->Blink = Input->ListEntry.Blink;
+ // only pay attention to valid ascii chars, on key down
+ if( Input->InputEvent.EventType == KEY_EVENT &&
+ Input->InputEvent.Event.KeyEvent.bKeyDown == TRUE &&
+ Input->InputEvent.Event.KeyEvent.uChar.AsciiChar )
+ {
+ // backspace handling
+ if( Input->InputEvent.Event.KeyEvent.uChar.AsciiChar == '\b' )
+ {
+ // echo if it has not already been done, and either we or the client has chars to be deleted
+ if( !Input->Echoed && ( i || Request->Data.ReadConsoleRequest.nCharsCanBeDeleted ) )
+ CsrpWriteConsole( Console->ActiveBuffer, &Input->InputEvent.Event.KeyEvent.uChar.AsciiChar, 1, TRUE );
+ if( i )
+ i-=2; // if we already have something to return, just back it up by 2
+ else
+ { // otherwise, return STATUS_NOTIFY_CLEANUP to tell client to back up its buffer
+ Reply->Data.ReadConsoleReply.NrCharactersRead = 0;
+ Reply->Status = STATUS_NOTIFY_CLEANUP;
+ Console->WaitingChars--;
+ RtlFreeHeap( CsrssApiHeap, 0, Input );
+ UNLOCK;
+ return STATUS_NOTIFY_CLEANUP;
+ }
+ Request->Data.ReadConsoleRequest.nCharsCanBeDeleted--;
+ Input->Echoed = TRUE; // mark as echoed so we don't echo it below
+ }
+ // do not copy backspace to buffer
+ else Buffer[i] = Input->InputEvent.Event.KeyEvent.uChar.AsciiChar;
+ // echo to screen if enabled and we did not already echo the char
+ if( Console->Mode & ENABLE_ECHO_INPUT &&
+ !Input->Echoed &&
+ Input->InputEvent.Event.KeyEvent.uChar.AsciiChar != '\r' )
+ CsrpWriteConsole( Console->ActiveBuffer, &Input->InputEvent.Event.KeyEvent.uChar.AsciiChar, 1, TRUE );
+ }
+ else i--;
+ Console->WaitingChars--;
+ RtlFreeHeap( CsrssApiHeap, 0, Input );
+ }
+ Reply->Data.ReadConsoleReply.NrCharactersRead = i;
+ if( !i )
+ Reply->Status = STATUS_PENDING; // we didn't read anything
+ else if( Console->Mode & ENABLE_LINE_INPUT )
+ if( !Console->WaitingLines || Buffer[i-1] != '\n' )
+ {
+ Reply->Status = STATUS_PENDING; // line buffered, didn't get a complete line
+ }
+ else {
+ Console->WaitingLines--;
+ Reply->Status = STATUS_SUCCESS; // line buffered, did get a complete line
+ }
+ else Reply->Status = STATUS_SUCCESS; // not line buffered, did read something
+ if( Reply->Status == STATUS_PENDING )
+ {
+ Console->EchoCount = nNumberOfCharsToRead - i;
+ }
+ else {
+ Console->EchoCount = 0; // if the client is no longer waiting on input, do not echo
+ }
+ Reply->Header.MessageSize += i;
+ UNLOCK;
+ return Reply->Status;
}
+#define SET_CELL_BUFFER(b,o,c,a)\
+(b)->Buffer[(o)++]=(c);\
+(b)->Buffer[(o)++]=(a);
+
+static VOID FASTCALL
+ClearLineBuffer (
+ PCSRSS_SCREEN_BUFFER Buff,
+ DWORD StartX
+ )
+{
+ DWORD Offset = 2 * ((Buff->CurrentY * Buff->MaxX) + StartX);
+
+ for ( ; StartX < Buff->MaxX; StartX ++ )
+ {
+ /* Fill the cell: Offset is incremented by the macro */
+ SET_CELL_BUFFER(Buff,Offset,' ',Buff->DefaultAttrib)
+ }
+}
-NTSTATUS CsrWriteConsole(PCSRSS_PROCESS_DATA ProcessData,
- PCSRSS_API_REQUEST Message,
- PULONG CharCount)
+NTSTATUS STDCALL CsrpWriteConsole( PCSRSS_SCREEN_BUFFER Buff, CHAR *Buffer, DWORD Length, BOOL Attrib )
{
+ IO_STATUS_BLOCK Iosb;
NTSTATUS Status;
- PCSRSS_CONSOLE Console;
+ int i;
+ DWORD Offset;
+
+ for( i = 0; i < Length; i++ )
+ {
+ switch( Buffer[ i ] )
+ {
+ /* --- LF --- */
+ case '\n':
+ Buff->CurrentX = 0;
+ /* slide the viewable screen */
+ if( ((PhysicalConsoleSize.Y + Buff->ShowY) % Buff->MaxY) == (Buff->CurrentY + 1) % Buff->MaxY)
+ if( ++Buff->ShowY == (Buff->MaxY - 1) )
+ Buff->ShowY = 0;
+ if( ++Buff->CurrentY == Buff->MaxY )
+ {
+ Buff->CurrentY = 0;
+ }
+ ClearLineBuffer (Buff, 0);
+ break;
+ /* --- BS --- */
+ case '\b':
+ if( Buff->CurrentX == 0 )
+ {
+ /* slide viewable screen up */
+ if( Buff->ShowY == Buff->CurrentY )
+ {
+ if( Buff->ShowY == 0 )
+ Buff->ShowY = Buff->MaxY;
+ else
+ Buff->ShowY--;
+ }
+ /* slide virtual position up */
+ Buff->CurrentX = Buff->MaxX;
+ if( Buff->CurrentY == 0 )
+ Buff->CurrentY = Buff->MaxY;
+ else
+ Buff->CurrentY--;
+ }
+ else
+ Buff->CurrentX--;
+ Offset = 2 * ((Buff->CurrentY * Buff->MaxX) + Buff->CurrentX);
+ SET_CELL_BUFFER(Buff,Offset,' ',Buff->DefaultAttrib);
+ break;
+ /* --- CR --- */
+ case '\r':
+ Buff->CurrentX = 0;
+ break;
+ /* --- TAB --- */
+ case '\t':
+ CsrpWriteConsole(Buff, " ", (8 - (Buff->CurrentX % 8)), FALSE);
+ break;
+ /* --- */
+ default:
+ Offset = 2 * (((Buff->CurrentY * Buff->MaxX)) + Buff->CurrentX);
+ Buff->Buffer[Offset ++] = Buffer[ i ];
+ if( Attrib )
+ Buff->Buffer[Offset] = Buff->DefaultAttrib;
+ Buff->CurrentX++;
+ if( Buff->CurrentX == Buff->MaxX )
+ {
+ /* if end of line, go to next */
+ Buff->CurrentX = 0;
+ if( ++Buff->CurrentY == Buff->MaxY )
+ {
+ /* if end of buffer, wrap back to beginning */
+ Buff->CurrentY = 0;
+ }
+ /* clear new line */
+ ClearLineBuffer (Buff, 0);
+ /* slide the viewable screen */
+ if( (Buff->CurrentY - Buff->ShowY) == PhysicalConsoleSize.Y )
+ if( ++Buff->ShowY == Buff->MaxY )
+ Buff->ShowY = 0;
+ }
+ }
+ }
+ if( Buff == ActiveConsole->ActiveBuffer )
+ { /* only write to screen if Console is Active, and not scrolled up */
+ if( Attrib )
+ {
+ Status = NtWriteFile( ConsoleDeviceHandle, NULL, NULL, NULL, &Iosb, Buffer, Length, NULL, 0);
+ if (!NT_SUCCESS(Status))
+ DbgPrint("CSR: Write failed\n");
+ }
+ }
+ return(STATUS_SUCCESS);
+}
+
+CSR_API(CsrWriteConsole)
+{
+ BYTE *Buffer = Request->Data.WriteConsoleRequest.Buffer;
+ PCSRSS_SCREEN_BUFFER Buff;
- if (ProcessData->Console == NULL)
+ Reply->Header.MessageSize = sizeof(CSRSS_API_REPLY);
+ Reply->Header.DataSize = sizeof(CSRSS_API_REPLY) -
+ sizeof(LPC_MESSAGE_HEADER);
+
+ LOCK;
+ if( !NT_SUCCESS( CsrGetObject( ProcessData, Request->Data.WriteConsoleRequest.ConsoleHandle, (Object_t **)&Buff ) ) || Buff->Header.Type != CSRSS_SCREEN_BUFFER_MAGIC )
+ {
+ UNLOCK;
+ return Reply->Status = STATUS_INVALID_HANDLE;
+ }
+ CsrpWriteConsole( Buff, Buffer, Request->Data.WriteConsoleRequest.NrCharactersToWrite, TRUE );
+ UNLOCK;
+ return Reply->Status = STATUS_SUCCESS;
+}
+
+
+NTSTATUS STDCALL CsrInitConsoleScreenBuffer( PCSRSS_SCREEN_BUFFER Console )
+{
+ Console->Header.Type = CSRSS_SCREEN_BUFFER_MAGIC;
+ Console->Header.ReferenceCount = 0;
+ Console->MaxX = PhysicalConsoleSize.X;
+ Console->MaxY = PhysicalConsoleSize.Y * 2;
+ Console->ShowX = 0;
+ Console->ShowY = 0;
+ Console->CurrentX = 0;
+ Console->CurrentY = 0;
+ Console->Buffer = RtlAllocateHeap( CsrssApiHeap, 0, Console->MaxX * Console->MaxY * 2 );
+ if( Console->Buffer == 0 )
+ return STATUS_INSUFFICIENT_RESOURCES;
+ Console->DefaultAttrib = 0x17;
+ /* initialize buffer to be empty with default attributes */
+ for( ; Console->CurrentY < Console->MaxY; Console->CurrentY++ )
+ {
+ ClearLineBuffer (Console, 0);
+ }
+ Console->CursorInfo.bVisible = TRUE;
+ Console->CursorInfo.dwSize = 5;
+ Console->Mode = ENABLE_PROCESSED_OUTPUT | ENABLE_WRAP_AT_EOL_OUTPUT;
+ Console->CurrentX = 0;
+ Console->CurrentY = 0;
+ return STATUS_SUCCESS;
+}
+
+VOID STDCALL CsrDeleteScreenBuffer( PCSRSS_SCREEN_BUFFER Buffer )
+{
+ RtlFreeHeap( CsrssApiHeap, 0, Buffer->Buffer );
+ RtlFreeHeap( CsrssApiHeap, 0, Buffer );
+}
+
+NTSTATUS STDCALL CsrInitConsole(PCSRSS_CONSOLE Console)
+{
+ NTSTATUS Status;
+
+ Console->Title.MaximumLength = Console->Title.Length = 0;
+ Console->Title.Buffer = 0;
+
+ RtlCreateUnicodeString( &Console->Title, L"Command Prompt" );
+
+ Console->Header.ReferenceCount = 0;
+ Console->WaitingChars = 0;
+ Console->WaitingLines = 0;
+ Console->EchoCount = 0;
+ Console->Header.Type = CSRSS_CONSOLE_MAGIC;
+ Console->Mode = ENABLE_LINE_INPUT | ENABLE_ECHO_INPUT | ENABLE_PROCESSED_INPUT | ENABLE_MOUSE_INPUT;
+ Console->InputEvents.Flink = Console->InputEvents.Blink = &Console->InputEvents;
+ Status = NtCreateEvent( &Console->ActiveEvent, STANDARD_RIGHTS_ALL, 0, FALSE, FALSE );
+ if( !NT_SUCCESS( Status ) )
+ {
+ return Status;
+ }
+ Console->ActiveBuffer = RtlAllocateHeap( CsrssApiHeap, 0, sizeof( CSRSS_SCREEN_BUFFER ) );
+ if( !Console->ActiveBuffer )
{
- return(STATUS_UNSUCCESSFUL);
+ NtClose( Console->ActiveEvent );
+ return STATUS_INSUFFICIENT_RESOURCES;
}
- Console = ProcessData->Console;
+ Status = CsrInitConsoleScreenBuffer( Console->ActiveBuffer );
+ if( !NT_SUCCESS( Status ) )
+ {
+ NtClose( Console->ActiveEvent );
+ RtlFreeHeap( CsrssApiHeap, 0, Console->ActiveBuffer );
+ return Status;
+ }
+ /* add a reference count because the buffer is tied to the console */
+ Console->ActiveBuffer->Header.ReferenceCount++;
+ /* make console active, and insert into console list */
+ LOCK;
+ if( ActiveConsole )
+ {
+ Console->Prev = ActiveConsole;
+ Console->Next = ActiveConsole->Next;
+ ActiveConsole->Next->Prev = Console;
+ ActiveConsole->Next = Console;
+ }
+ else {
+ Console->Prev = Console;
+ Console->Next = Console;
+ }
+ ActiveConsole = Console;
+ /* copy buffer contents to screen */
+ CsrDrawConsole( Console->ActiveBuffer );
+ UNLOCK;
+ return STATUS_SUCCESS;
+}
+
+/***************************************************************
+ * CsrDrawConsole blasts the console buffer onto the screen *
+ * must be called while holding the active console lock *
+ **************************************************************/
+VOID STDCALL CsrDrawConsole( PCSRSS_SCREEN_BUFFER Buff )
+{
+ IO_STATUS_BLOCK Iosb;
+ NTSTATUS Status;
+ CONSOLE_SCREEN_BUFFER_INFO ScrInfo;
+ CONSOLE_MODE Mode;
+ int i, y;
+
+ /* first set position to 0,0 */
+ ScrInfo.dwCursorPosition.X = 0;
+ ScrInfo.dwCursorPosition.Y = 0;
+ ScrInfo.wAttributes = Buff->DefaultAttrib;
+ Status = NtDeviceIoControlFile( ConsoleDeviceHandle, NULL, NULL, NULL, &Iosb, IOCTL_CONSOLE_SET_SCREEN_BUFFER_INFO, &ScrInfo, sizeof( ScrInfo ), 0, 0 );
+ if( !NT_SUCCESS( Status ) )
+ {
+ DbgPrint( "CSR: Failed to set console info\n" );
+ return;
+ }
+ Mode.dwMode = 0; /* clear ENABLE_PROCESSED_OUTPUT mode */
+ Status = NtDeviceIoControlFile( ConsoleDeviceHandle, NULL, NULL, NULL, &Iosb, IOCTL_CONSOLE_SET_MODE, &Mode, sizeof( Mode ), 0, 0 );
+ if( !NT_SUCCESS( Status ) )
+ {
+ DbgPrint( "CSR: Failed to set console mode\n" );
+ return;
+ }
+ /* blast out buffer */
+ for( i = 0, y = Buff->ShowY; i < PhysicalConsoleSize.Y; i++ )
+ {
+ Status = NtWriteFile( ConsoleDeviceHandle, NULL, NULL, NULL, &Iosb, &Buff->Buffer[ (Buff->ShowX * 2) + (y * Buff->MaxX * 2) ], PhysicalConsoleSize.X * 2, 0, 0 );
+ if( !NT_SUCCESS( Status ) )
+ {
+ DbgPrint( "CSR: Write to console failed\n" );
+ return;
+ }
+ /* wrap back around the end of the buffer */
+ if( ++y == Buff->MaxY )
+ y = 0;
+ }
+ Mode.dwMode = ENABLE_PROCESSED_OUTPUT;
+ Status = NtDeviceIoControlFile( ConsoleDeviceHandle, NULL, NULL, NULL, &Iosb, IOCTL_CONSOLE_SET_MODE, &Mode, sizeof( Mode ), 0, 0 );
+ if( !NT_SUCCESS( Status ) )
+ {
+ DbgPrint( "CSR: Failed to set console mode\n" );
+ return;
+ }
+ ScrInfo.dwCursorPosition.X = Buff->CurrentX - Buff->ShowX;
+ ScrInfo.dwCursorPosition.Y = ((Buff->CurrentY + Buff->MaxY) - Buff->ShowY) % Buff->MaxY;
+ Status = NtDeviceIoControlFile( ConsoleDeviceHandle, NULL, NULL, NULL, &Iosb, IOCTL_CONSOLE_SET_SCREEN_BUFFER_INFO, &ScrInfo, sizeof( ScrInfo ), 0, 0 );
+ if( !NT_SUCCESS( Status ) )
+ {
+ DbgPrint( "CSR: Failed to set console info\n" );
+ return;
+ }
+ Status = NtDeviceIoControlFile( ConsoleDeviceHandle, NULL, NULL, NULL, &Iosb, IOCTL_CONSOLE_SET_CURSOR_INFO, &Buff->CursorInfo, sizeof( Buff->CursorInfo ), 0, 0 );
+ if( !NT_SUCCESS( Status ) )
+ {
+ DbgPrint( "CSR: Failed to set cursor info\n" );
+ return;
+ }
+}
+
+
+VOID STDCALL CsrDeleteConsole( PCSRSS_CONSOLE Console )
+{
+ ConsoleInput *Event;
+ DPRINT1( "CsrDeleteConsole\n" );
+ LOCK;
+ /* Drain input event queue */
+ while( Console->InputEvents.Flink != &Console->InputEvents )
+ {
+ Event = (ConsoleInput *)Console->InputEvents.Flink;
+ Console->InputEvents.Flink = Console->InputEvents.Flink->Flink;
+ Console->InputEvents.Flink->Flink->Blink = &Console->InputEvents;
+ RtlFreeHeap( CsrssApiHeap, 0, Event );
+ }
+ /* Switch to next console */
+ if( ActiveConsole == Console )
+ {
+ if( Console->Next != Console )
+ {
+ ActiveConsole = Console->Next;
+ Console->Prev->Next = Console->Next;
+ Console->Next->Prev = Console->Prev;
+ }
+ else ActiveConsole = 0;
+ }
+ if( ActiveConsole )
+ CsrDrawConsole( ActiveConsole->ActiveBuffer );
+ UNLOCK;
+ if( !--Console->ActiveBuffer->Header.ReferenceCount )
+ CsrDeleteScreenBuffer( Console->ActiveBuffer );
+ NtClose( Console->ActiveEvent );
+ RtlFreeUnicodeString( &Console->Title );
+ RtlFreeHeap( CsrssApiHeap, 0, Console );
+}
+
+VOID STDCALL CsrInitConsoleSupport(VOID)
+{
+ OBJECT_ATTRIBUTES ObjectAttributes;
+ UNICODE_STRING DeviceName;
+ NTSTATUS Status;
+ IO_STATUS_BLOCK Iosb;
+ CONSOLE_SCREEN_BUFFER_INFO ScrInfo;
- Status = NtWaitForSingleObject(Console->LockMutant,
- TRUE,
- NULL);
+ DbgPrint("CSR: CsrInitConsoleSupport()\n");
+
+ RtlInitUnicodeString(&DeviceName, L"\\??\\BlueScreen");
+ InitializeObjectAttributes(&ObjectAttributes,
+ &DeviceName,
+ 0,
+ NULL,
+ NULL);
+ Status = NtOpenFile(&ConsoleDeviceHandle,
+ FILE_ALL_ACCESS,
+ &ObjectAttributes,
+ &Iosb,
+ 0,
+ FILE_SYNCHRONOUS_IO_ALERT);
if (!NT_SUCCESS(Status))
{
- return(Status);
+ DbgPrint("CSR: Failed to open console. Expect problems.\n");
}
- if (Status == STATUS_ALERTED ||
- Status == STATUS_USER_APC)
+// DbgPrint("CSR: ConsoleDeviceHandle %x\n", ConsoleDeviceHandle);
+
+ RtlInitUnicodeString(&DeviceName, L"\\??\\Keyboard");
+ InitializeObjectAttributes(&ObjectAttributes,
+ &DeviceName,
+ 0,
+ NULL,
+ NULL);
+ Status = NtOpenFile(&KeyboardDeviceHandle,
+ FILE_ALL_ACCESS,
+ &ObjectAttributes,
+ &Iosb,
+ 0,
+ 0);
+ if (!NT_SUCCESS(Status))
{
- return(Status);
+ DbgPrint("CSR: Failed to open keyboard. Expect problems.\n");
}
- if (Console->TopLevel == TRUE)
+ ActiveConsole = 0;
+ RtlInitializeCriticalSection( &ActiveConsoleLock );
+ Status = NtDeviceIoControlFile( ConsoleDeviceHandle, NULL, NULL, NULL, &Iosb, IOCTL_CONSOLE_GET_SCREEN_BUFFER_INFO, 0, 0, &ScrInfo, sizeof( ScrInfo ) );
+ if( !NT_SUCCESS( Status ) )
{
- Status = NtReleaseMutant(Console->LockMutant, NULL);
- if (!NT_SUCCESS(Status))
- {
- return(Status);
- }
-
- return(Status);
+ DbgPrint( "CSR: Failed to get console info, expect trouble\n" );
+ return;
+ }
+ PhysicalConsoleSize = ScrInfo.dwSize;
+}
+
+VOID Console_Api( DWORD RefreshEvent )
+{
+ /* keep reading events from the keyboard and stuffing them into the current
+ console's input queue */
+ ConsoleInput *KeyEventRecord;
+ ConsoleInput *TempInput;
+ IO_STATUS_BLOCK Iosb;
+ NTSTATUS Status;
+ HANDLE Events[2]; // 0 = keyboard, 1 = refresh
+ int c;
+ int updown;
+ PCSRSS_CONSOLE SwapConsole = 0; // console we are thinking about swapping with
+
+ Events[0] = 0;
+ Status = NtCreateEvent( &Events[0], STANDARD_RIGHTS_ALL, NULL, FALSE, FALSE );
+ if( !NT_SUCCESS( Status ) )
+ {
+ DbgPrint( "CSR: NtCreateEvent failed: %x\n", Status );
+ NtTerminateProcess( NtCurrentProcess(), Status );
+ }
+ Events[1] = (HANDLE)RefreshEvent;
+ while( 1 )
+ {
+ KeyEventRecord = RtlAllocateHeap(CsrssApiHeap,
+ 0,
+ sizeof(ConsoleInput));
+ if ( KeyEventRecord == 0 )
+ {
+ DbgPrint( "CSR: Memory allocation failure!" );
+ continue;
+ }
+ KeyEventRecord->InputEvent.EventType = KEY_EVENT;
+ Status = NtReadFile( KeyboardDeviceHandle, Events[0], NULL, NULL, &Iosb, &KeyEventRecord->InputEvent.Event.KeyEvent, sizeof( KEY_EVENT_RECORD ), NULL, 0 );
+ if( !NT_SUCCESS( Status ) )
+ {
+ DbgPrint( "CSR: ReadFile on keyboard device failed\n" );
+ RtlFreeHeap( CsrssApiHeap, 0, KeyEventRecord );
+ continue;
+ }
+ if( Status == STATUS_PENDING )
+ {
+ while( 1 )
+ {
+ Status = NtWaitForMultipleObjects( 2, Events, WaitAny, FALSE, NULL );
+ if( Status == STATUS_WAIT_0 + 1 )
+ {
+ LOCK;
+ CsrDrawConsole( ActiveConsole->ActiveBuffer );
+ UNLOCK;
+ continue;
+ }
+ else if( Status != STATUS_WAIT_0 )
+ {
+ DbgPrint( "CSR: NtWaitForMultipleObjects failed: %x, exiting\n", Status );
+ NtTerminateProcess( NtCurrentProcess(), Status );
+ }
+ else break;
+ }
+ }
+ if( KeyEventRecord->InputEvent.Event.KeyEvent.dwControlKeyState & ( RIGHT_ALT_PRESSED | LEFT_ALT_PRESSED )&& KeyEventRecord->InputEvent.Event.KeyEvent.wVirtualKeyCode == VK_TAB )
+ if( KeyEventRecord->InputEvent.Event.KeyEvent.bKeyDown == TRUE )
+ {
+ ANSI_STRING Title;
+ void * Buffer;
+ COORD *pos;
+ unsigned int src, dst;
+
+ /* alt-tab, swap consoles */
+ // move SwapConsole to next console, and print its title
+ LOCK;
+ if( !SwapConsole )
+ SwapConsole = ActiveConsole;
+
+ if( KeyEventRecord->InputEvent.Event.KeyEvent.dwControlKeyState & SHIFT_PRESSED )
+ SwapConsole = SwapConsole->Prev;
+ else SwapConsole = SwapConsole->Next;
+ Title.MaximumLength = RtlUnicodeStringToAnsiSize( &SwapConsole->Title );
+ Title.Length = 0;
+ Buffer = RtlAllocateHeap( CsrssApiHeap,
+ 0,
+ sizeof( COORD ) + Title.MaximumLength );
+ pos = (COORD *)Buffer;
+ Title.Buffer = Buffer + sizeof( COORD );
+
+ /* this does not seem to work
+ RtlUnicodeStringToAnsiString( &Title, &SwapConsole->Title, FALSE ); */
+ // temp hack
+ for( src = 0, dst = 0; src < SwapConsole->Title.Length; src++, dst++ )
+ Title.Buffer[dst] = (char)SwapConsole->Title.Buffer[dst];
+
+ pos->Y = PhysicalConsoleSize.Y / 2;
+ pos->X = ( PhysicalConsoleSize.X - Title.MaximumLength ) / 2;
+ // redraw the console to clear off old title
+ CsrDrawConsole( ActiveConsole->ActiveBuffer );
+ Status = NtDeviceIoControlFile( ConsoleDeviceHandle,
+ NULL,
+ NULL,
+ NULL,
+ &Iosb,
+ IOCTL_CONSOLE_WRITE_OUTPUT_CHARACTER,
+ 0,
+ 0,
+ Buffer,
+ sizeof (COORD) + Title.MaximumLength );
+ if( !NT_SUCCESS( Status ) )
+ {
+ DPRINT1( "Error writing to console\n" );
+ }
+ RtlFreeHeap( CsrssApiHeap, 0, Buffer );
+
+ UNLOCK;
+ RtlFreeHeap( CsrssApiHeap, 0, KeyEventRecord );
+ continue;
+ }
+ else {
+ RtlFreeHeap( CsrssApiHeap, 0, KeyEventRecord );
+ continue;
+ }
+ else if( SwapConsole &&
+ KeyEventRecord->InputEvent.Event.KeyEvent.wVirtualKeyCode == VK_MENU &&
+ KeyEventRecord->InputEvent.Event.KeyEvent.bKeyDown == FALSE )
+ {
+ // alt key released, swap consoles
+ PCSRSS_CONSOLE tmp;
+
+ LOCK;
+ if( SwapConsole != ActiveConsole )
+ {
+ // first remove swapconsole from the list
+ SwapConsole->Prev->Next = SwapConsole->Next;
+ SwapConsole->Next->Prev = SwapConsole->Prev;
+ // now insert before activeconsole
+ SwapConsole->Next = ActiveConsole;
+ SwapConsole->Prev = ActiveConsole->Prev;
+ ActiveConsole->Prev->Next = SwapConsole;
+ ActiveConsole->Prev = SwapConsole;
+ }
+ ActiveConsole = SwapConsole;
+ SwapConsole = 0;
+ CsrDrawConsole( ActiveConsole->ActiveBuffer );
+
+ UNLOCK;
+ }
+ if( KeyEventRecord->InputEvent.Event.KeyEvent.dwControlKeyState & ( RIGHT_ALT_PRESSED | LEFT_ALT_PRESSED ) && (KeyEventRecord->InputEvent.Event.KeyEvent.wVirtualKeyCode == VK_UP || KeyEventRecord->InputEvent.Event.KeyEvent.wVirtualKeyCode == VK_DOWN) )
+ {
+ if( KeyEventRecord->InputEvent.Event.KeyEvent.bKeyDown == TRUE )
+ {
+ /* scroll up or down */
+ LOCK;
+ if( ActiveConsole == 0 )
+ {
+ DbgPrint( "CSR: No Active Console!\n" );
+ UNLOCK;
+ RtlFreeHeap( CsrssApiHeap, 0, KeyEventRecord );
+ continue;
+ }
+ if( KeyEventRecord->InputEvent.Event.KeyEvent.wVirtualKeyCode == VK_UP )
+ {
+ /* only scroll up if there is room to scroll up into */
+ if( ActiveConsole->ActiveBuffer->ShowY != ((ActiveConsole->ActiveBuffer->CurrentY + 1) % ActiveConsole->ActiveBuffer->MaxY) )
+ ActiveConsole->ActiveBuffer->ShowY = (ActiveConsole->ActiveBuffer->ShowY + ActiveConsole->ActiveBuffer->MaxY - 1) % ActiveConsole->ActiveBuffer->MaxY;
+ }
+ else if( ActiveConsole->ActiveBuffer->ShowY != ActiveConsole->ActiveBuffer->CurrentY )
+ /* only scroll down if there is room to scroll down into */
+ if( ActiveConsole->ActiveBuffer->ShowY % ActiveConsole->ActiveBuffer->MaxY != ActiveConsole->ActiveBuffer->CurrentY )
+ if( ((ActiveConsole->ActiveBuffer->CurrentY + 1) % ActiveConsole->ActiveBuffer->MaxY) != (ActiveConsole->ActiveBuffer->ShowY + PhysicalConsoleSize.Y) % ActiveConsole->ActiveBuffer->MaxY )
+ ActiveConsole->ActiveBuffer->ShowY = (ActiveConsole->ActiveBuffer->ShowY + 1) % ActiveConsole->ActiveBuffer->MaxY;
+ CsrDrawConsole( ActiveConsole->ActiveBuffer );
+ UNLOCK;
+ }
+ RtlFreeHeap( CsrssApiHeap, 0, KeyEventRecord );
+ continue;
+ }
+ LOCK;
+ if( ActiveConsole == 0 )
+ {
+ DbgPrint( "CSR: No Active Console!\n" );
+ UNLOCK;
+ RtlFreeHeap( CsrssApiHeap, 0, KeyEventRecord );
+ continue;
+ }
+ // process special keys if enabled
+ if( ActiveConsole->Mode & (ENABLE_PROCESSED_INPUT | ENABLE_LINE_INPUT) )
+ switch( KeyEventRecord->InputEvent.Event.KeyEvent.uChar.AsciiChar )
+ {
+ case '\r':
+ // add a \n to the queue as well
+ // first add the \r
+ updown = KeyEventRecord->InputEvent.Event.KeyEvent.bKeyDown;
+ KeyEventRecord->Echoed = FALSE;
+ KeyEventRecord->InputEvent.Event.KeyEvent.uChar.AsciiChar = '\r';
+ KeyEventRecord->ListEntry.Flink = &ActiveConsole->InputEvents;
+ KeyEventRecord->ListEntry.Blink = ActiveConsole->InputEvents.Blink;
+ ActiveConsole->InputEvents.Blink->Flink = &KeyEventRecord->ListEntry;
+ ActiveConsole->InputEvents.Blink = &KeyEventRecord->ListEntry;
+ ActiveConsole->WaitingChars++;
+ KeyEventRecord = RtlAllocateHeap( CsrssApiHeap, 0, sizeof( ConsoleInput ) );
+ if( !KeyEventRecord )
+ {
+ DbgPrint( "CSR: Failed to allocate KeyEventRecord\n" );
+ UNLOCK;
+ continue;
+ }
+ KeyEventRecord->InputEvent.EventType = KEY_EVENT;
+ KeyEventRecord->InputEvent.Event.KeyEvent.bKeyDown = updown;
+ KeyEventRecord->InputEvent.Event.KeyEvent.wVirtualKeyCode = 0;
+ KeyEventRecord->InputEvent.Event.KeyEvent.wVirtualScanCode = 0;
+ KeyEventRecord->InputEvent.Event.KeyEvent.uChar.AsciiChar = '\n';
+ }
+ // add event to the queue
+ KeyEventRecord->ListEntry.Flink = &ActiveConsole->InputEvents;
+ KeyEventRecord->ListEntry.Blink = ActiveConsole->InputEvents.Blink;
+ ActiveConsole->InputEvents.Blink->Flink = &KeyEventRecord->ListEntry;
+ ActiveConsole->InputEvents.Blink = &KeyEventRecord->ListEntry;
+ // if line input mode is enabled, only wake the client on enter key down
+ if( !(ActiveConsole->Mode & ENABLE_LINE_INPUT ) ||
+ ( KeyEventRecord->InputEvent.Event.KeyEvent.uChar.AsciiChar == '\n' &&
+ KeyEventRecord->InputEvent.Event.KeyEvent.bKeyDown == TRUE ) )
+ {
+ NtSetEvent( ActiveConsole->ActiveEvent, 0 );
+ if( KeyEventRecord->InputEvent.Event.KeyEvent.uChar.AsciiChar == '\n' )
+ ActiveConsole->WaitingLines++;
+ }
+ KeyEventRecord->Echoed = FALSE;
+ if( ActiveConsole->Mode & (ENABLE_PROCESSED_INPUT | ENABLE_LINE_INPUT) &&
+ KeyEventRecord->InputEvent.Event.KeyEvent.uChar.AsciiChar == '\b' &&
+ KeyEventRecord->InputEvent.Event.KeyEvent.bKeyDown )
+ {
+ // walk the input queue looking for a char to backspace
+ for( TempInput = (ConsoleInput *)ActiveConsole->InputEvents.Blink;
+ TempInput != (ConsoleInput *)&ActiveConsole->InputEvents &&
+ (TempInput->InputEvent.EventType != KEY_EVENT ||
+ TempInput->InputEvent.Event.KeyEvent.bKeyDown == FALSE ||
+ TempInput->InputEvent.Event.KeyEvent.uChar.AsciiChar == '\b' );
+ TempInput = (ConsoleInput *)TempInput->ListEntry.Blink );
+ // if we found one, delete it, otherwise, wake the client
+ if( TempInput != (ConsoleInput *)&ActiveConsole->InputEvents )
+ {
+ // delete previous key in queue, maybe echo backspace to screen, and do not place backspace on queue
+ TempInput->ListEntry.Blink->Flink = TempInput->ListEntry.Flink;
+ TempInput->ListEntry.Flink->Blink = TempInput->ListEntry.Blink;
+ if( TempInput->Echoed )
+ CsrpWriteConsole( ActiveConsole->ActiveBuffer, &KeyEventRecord->InputEvent.Event.KeyEvent.uChar.AsciiChar, 1, TRUE );
+ RtlFreeHeap( CsrssApiHeap, 0, TempInput );
+ KeyEventRecord->ListEntry.Blink->Flink = KeyEventRecord->ListEntry.Flink;
+ KeyEventRecord->ListEntry.Flink->Blink = KeyEventRecord->ListEntry.Blink;
+ RtlFreeHeap( CsrssApiHeap, 0, KeyEventRecord );
+ ActiveConsole->WaitingChars -= 2;
+ }
+ else NtSetEvent( ActiveConsole->ActiveEvent, 0 );
+ }
+ else {
+ // echo chars if we are supposed to and client is waiting for some
+ if( ActiveConsole->Mode & ENABLE_ECHO_INPUT && ActiveConsole->EchoCount &&
+ KeyEventRecord->InputEvent.Event.KeyEvent.uChar.AsciiChar &&
+ KeyEventRecord->InputEvent.Event.KeyEvent.bKeyDown == TRUE &&
+ KeyEventRecord->InputEvent.Event.KeyEvent.uChar.AsciiChar != '\r' )
+ {
+ // mark the char as already echoed
+ CsrpWriteConsole( ActiveConsole->ActiveBuffer, &KeyEventRecord->InputEvent.Event.KeyEvent.uChar.AsciiChar, 1, TRUE );
+ ActiveConsole->EchoCount--;
+ KeyEventRecord->Echoed = TRUE;
+ }
+ }
+ ActiveConsole->WaitingChars++;
+ if( !(ActiveConsole->Mode & ENABLE_LINE_INPUT) )
+ NtSetEvent( ActiveConsole->ActiveEvent, 0 );
+ UNLOCK;
+ }
+}
+
+CSR_API(CsrGetScreenBufferInfo)
+{
+ NTSTATUS Status;
+ PCSRSS_SCREEN_BUFFER Buff;
+ PCONSOLE_SCREEN_BUFFER_INFO pInfo;
+ IO_STATUS_BLOCK Iosb;
+
+ Reply->Header.MessageSize = sizeof(CSRSS_API_REPLY);
+ Reply->Header.DataSize = sizeof(CSRSS_API_REPLY) -
+ sizeof(LPC_MESSAGE_HEADER);
+
+ LOCK;
+ if( !NT_SUCCESS( CsrGetObject( ProcessData, Request->Data.ScreenBufferInfoRequest.ConsoleHandle, (Object_t **)&Buff ) ) || Buff->Header.Type != CSRSS_SCREEN_BUFFER_MAGIC )
+ {
+ UNLOCK;
+ return Reply->Status = STATUS_INVALID_HANDLE;
+ }
+ pInfo = &Reply->Data.ScreenBufferInfoReply.Info;
+ if( Buff == ActiveConsole->ActiveBuffer )
+ {
+ Status = NtDeviceIoControlFile( ConsoleDeviceHandle, NULL, NULL, NULL, &Iosb, IOCTL_CONSOLE_GET_SCREEN_BUFFER_INFO, 0, 0, pInfo, sizeof( *pInfo ) );
+ if( !NT_SUCCESS( Status ) )
+ DbgPrint( "CSR: Failed to get console info, expect trouble\n" );
+ Reply->Status = Status;
}
- else
+ else {
+ pInfo->dwSize.X = PhysicalConsoleSize.X;
+ pInfo->dwSize.Y = PhysicalConsoleSize.Y;
+ pInfo->dwCursorPosition.X = Buff->CurrentX - Buff->ShowX;
+ pInfo->dwCursorPosition.Y = (Buff->CurrentY + Buff->MaxY - Buff->ShowY) % Buff->MaxY;
+ pInfo->wAttributes = Buff->DefaultAttrib;
+ pInfo->srWindow.Left = 0;
+ pInfo->srWindow.Right = PhysicalConsoleSize.X - 1;
+ pInfo->srWindow.Top = 0;
+ pInfo->srWindow.Bottom = PhysicalConsoleSize.Y - 1;
+ Reply->Status = STATUS_SUCCESS;
+ }
+ UNLOCK;
+ return Reply->Status;
+}
+
+CSR_API(CsrSetCursor)
+{
+ NTSTATUS Status;
+ PCSRSS_SCREEN_BUFFER Buff;
+ CONSOLE_SCREEN_BUFFER_INFO Info;
+ IO_STATUS_BLOCK Iosb;
+
+ Reply->Header.MessageSize = sizeof(CSRSS_API_REPLY);
+ Reply->Header.DataSize = sizeof(CSRSS_API_REPLY) -
+ sizeof(LPC_MESSAGE_HEADER);
+
+ LOCK;
+ if( !NT_SUCCESS( CsrGetObject( ProcessData, Request->Data.SetCursorRequest.ConsoleHandle, (Object_t **)&Buff ) ) || Buff->Header.Type != CSRSS_SCREEN_BUFFER_MAGIC )
+ {
+ UNLOCK;
+ return Reply->Status = STATUS_INVALID_HANDLE;
+ }
+ Info.dwCursorPosition = Request->Data.SetCursorRequest.Position;
+ Info.wAttributes = Buff->DefaultAttrib;
+ if( Buff == ActiveConsole->ActiveBuffer )
+ {
+ Status = NtDeviceIoControlFile( ConsoleDeviceHandle, NULL, NULL, NULL, &Iosb, IOCTL_CONSOLE_SET_SCREEN_BUFFER_INFO, &Info, sizeof( Info ), 0, 0 );
+ if( !NT_SUCCESS( Status ) )
+ DbgPrint( "CSR: Failed to set console info, expect trouble\n" );
+ }
+ Buff->CurrentX = Info.dwCursorPosition.X + Buff->ShowX;
+ Buff->CurrentY = (Info.dwCursorPosition.Y + Buff->ShowY) % Buff->MaxY;
+ UNLOCK;
+ return Reply->Status = Status;
+}
+
+CSR_API(CsrWriteConsoleOutputChar)
+{
+ BYTE *Buffer = Request->Data.WriteConsoleOutputCharRequest.String;
+ PCSRSS_SCREEN_BUFFER Buff;
+ DWORD X, Y;
+ NTSTATUS Status;
+ IO_STATUS_BLOCK Iosb;
+
+ Reply->Header.MessageSize = sizeof(CSRSS_API_REPLY);
+ Reply->Header.DataSize = sizeof(CSRSS_API_REPLY) -
+ sizeof(LPC_MESSAGE_HEADER);
+ LOCK;
+ if( !NT_SUCCESS( CsrGetObject( ProcessData, Request->Data.WriteConsoleOutputCharRequest.ConsoleHandle, (Object_t **)&Buff ) ) || Buff->Header.Type != CSRSS_SCREEN_BUFFER_MAGIC )
+ {
+ UNLOCK;
+ return Reply->Status = STATUS_INVALID_HANDLE;
+ }
+ X = Buff->CurrentX;
+ Y = Buff->CurrentY;
+ Buff->CurrentX = Request->Data.WriteConsoleOutputCharRequest.Coord.X;
+ Buff->CurrentY = Request->Data.WriteConsoleOutputCharRequest.Coord.Y;
+ Buffer[Request->Data.WriteConsoleOutputCharRequest.Length] = 0;
+ CsrpWriteConsole( Buff, Buffer, Request->Data.WriteConsoleOutputCharRequest.Length, FALSE );
+ if( ActiveConsole->ActiveBuffer == Buff )
{
- Status = NtReleaseMutant(Console->LockMutant, NULL);
- if (!NT_SUCCESS(Status))
- {
- return(Status);
- }
-
- return(Status);
+ Status = NtDeviceIoControlFile( ConsoleDeviceHandle,
+ NULL,
+ NULL,
+ NULL,
+ &Iosb,
+ IOCTL_CONSOLE_WRITE_OUTPUT_CHARACTER,
+ 0,
+ 0,
+ &Request->Data.WriteConsoleOutputCharRequest.Coord,
+ sizeof (COORD) + Request->Data.WriteConsoleOutputCharRequest.Length );
+ if( !NT_SUCCESS( Status ) )
+ DPRINT1( "Failed to write output chars: %x\n", Status );
+ }
+ Reply->Data.WriteConsoleOutputCharReply.EndCoord.X = Buff->CurrentX - Buff->ShowX;
+ Reply->Data.WriteConsoleOutputCharReply.EndCoord.Y = (Buff->CurrentY + Buff->MaxY - Buff->ShowY) % Buff->MaxY;
+ Buff->CurrentY = Y;
+ Buff->CurrentX = X;
+ UNLOCK;
+ return Reply->Status = STATUS_SUCCESS;
+}
+
+CSR_API(CsrFillOutputChar)
+{
+ PCSRSS_SCREEN_BUFFER Buff;
+ DWORD X, Y, i;
+
+ Reply->Header.MessageSize = sizeof(CSRSS_API_REPLY);
+ Reply->Header.DataSize = sizeof(CSRSS_API_REPLY) -
+ sizeof(LPC_MESSAGE_HEADER);
+
+ LOCK;
+ if( !NT_SUCCESS( CsrGetObject( ProcessData, Request->Data.FillOutputRequest.ConsoleHandle, (Object_t **)&Buff ) ) || Buff->Header.Type != CSRSS_SCREEN_BUFFER_MAGIC )
+ {
+ UNLOCK;
+ return Reply->Status = STATUS_INVALID_HANDLE;
+ }
+ X = Request->Data.FillOutputRequest.Position.X + Buff->ShowX;
+ Y = Request->Data.FillOutputRequest.Position.Y + Buff->ShowY;
+ for( i = 0; i < 20000; i++ );
+ for( i = 0; i < Request->Data.FillOutputRequest.Length; i++ )
+ {
+ Buff->Buffer[ (Y * 2 * Buff->MaxX) + (X * 2) ] = Request->Data.FillOutputRequest.Char;
+ if( ++X == Buff->MaxX )
+ {
+ if( ++Y == Buff->MaxY )
+ Y = 0;
+ X = 0;
+ }
+ }
+ if( Buff == ActiveConsole->ActiveBuffer )
+ CsrDrawConsole( Buff );
+ UNLOCK;
+ return Reply->Status;
+}
+
+CSR_API(CsrReadInputEvent)
+{
+ PCSRSS_CONSOLE Console;
+ NTSTATUS Status;
+ ConsoleInput *Input;
+
+ Reply->Header.MessageSize = sizeof(CSRSS_API_REPLY);
+ Reply->Header.DataSize = sizeof(CSRSS_API_REPLY) -
+ sizeof(LPC_MESSAGE_HEADER);
+ Reply->Data.ReadInputReply.Event = ProcessData->ConsoleEvent;
+
+ LOCK;
+ Status = CsrGetObject( ProcessData, Request->Data.ReadInputRequest.ConsoleHandle, (Object_t **)&Console );
+ if( !NT_SUCCESS( Status ) || (Status = Console->Header.Type == CSRSS_CONSOLE_MAGIC ? 0 : STATUS_INVALID_HANDLE))
+ {
+ Reply->Status = Status;
+ UNLOCK;
+ return Status;
+ }
+ // only get input if there is input, and we are not in line input mode, or if we are, if we have a whole line
+ if( Console->InputEvents.Flink != &Console->InputEvents &&
+ ( !Console->Mode & ENABLE_LINE_INPUT || Console->WaitingLines ) )
+ {
+ Input = (ConsoleInput *)Console->InputEvents.Flink;
+ Input->ListEntry.Blink->Flink = Input->ListEntry.Flink;
+ Input->ListEntry.Flink->Blink = Input->ListEntry.Blink;
+ Reply->Data.ReadInputReply.Input = Input->InputEvent;
+ if( Console->Mode & ENABLE_LINE_INPUT &&
+ Input->InputEvent.Event.KeyEvent.bKeyDown == FALSE &&
+ Input->InputEvent.Event.KeyEvent.uChar.AsciiChar == '\n' )
+ Console->WaitingLines--;
+ Console->WaitingChars--;
+ RtlFreeHeap( CsrssApiHeap, 0, Input );
+ Reply->Data.ReadInputReply.MoreEvents = (Console->InputEvents.Flink != &Console->InputEvents) ? TRUE : FALSE;
+ Status = STATUS_SUCCESS;
+ }
+ else Status = STATUS_PENDING;
+ UNLOCK;
+ return Reply->Status = Status;
+}
+
+CSR_API(CsrWriteConsoleOutputAttrib)
+{
+ int c;
+ PCSRSS_SCREEN_BUFFER Buff;
+ NTSTATUS Status;
+ int X, Y;
+ IO_STATUS_BLOCK Iosb;
+
+ Reply->Header.MessageSize = sizeof(CSRSS_API_REPLY);
+ Reply->Header.DataSize = sizeof(CSRSS_API_REPLY) -
+ sizeof(LPC_MESSAGE_HEADER);
+ LOCK;
+ Status = CsrGetObject( ProcessData, Request->Data.WriteConsoleOutputAttribRequest.ConsoleHandle, (Object_t **)&Buff );
+ if( !NT_SUCCESS( Status ) || (Status = Buff->Header.Type == CSRSS_SCREEN_BUFFER_MAGIC ? 0 : STATUS_INVALID_HANDLE ))
+ {
+ Reply->Status = Status;
+ UNLOCK;
+ return Status;
+ }
+ X = Buff->CurrentX;
+ Y = Buff->CurrentY;
+ Buff->CurrentX = Request->Data.WriteConsoleOutputAttribRequest.Coord.X + Buff->ShowX;
+ Buff->CurrentY = (Request->Data.WriteConsoleOutputAttribRequest.Coord.Y + Buff->ShowY) % Buff->MaxY;
+ for( c = 0; c < Request->Data.WriteConsoleOutputAttribRequest.Length; c++ )
+ {
+ Buff->Buffer[(Buff->CurrentY * Buff->MaxX * 2) + (Buff->CurrentX * 2) + 1] = Request->Data.WriteConsoleOutputAttribRequest.String[c];
+ if( ++Buff->CurrentX == Buff->MaxX )
+ {
+ Buff->CurrentX = 0;
+ if( ++Buff->CurrentY == Buff->MaxY )
+ Buff->CurrentY = 0;
+ }
+ }
+ if( Buff == ActiveConsole->ActiveBuffer )
+ {
+ Status = NtDeviceIoControlFile( ConsoleDeviceHandle,
+ NULL,
+ NULL,
+ NULL,
+ &Iosb,
+ IOCTL_CONSOLE_WRITE_OUTPUT_ATTRIBUTE,
+ 0,
+ 0,
+ &Request->Data.WriteConsoleOutputAttribRequest.Coord,
+ Request->Data.WriteConsoleOutputAttribRequest.Length +
+ sizeof (COORD) );
+ if( !NT_SUCCESS( Status ) )
+ DPRINT1( "Failed to write output attributes to console\n" );
+ }
+ Reply->Data.WriteConsoleOutputAttribReply.EndCoord.X = Buff->CurrentX - Buff->ShowX;
+ Reply->Data.WriteConsoleOutputAttribReply.EndCoord.Y = ( Buff->CurrentY + Buff->MaxY - Buff->ShowY ) % Buff->MaxY;
+ Buff->CurrentX = X;
+ Buff->CurrentY = Y;
+ UNLOCK;
+ return Reply->Status = STATUS_SUCCESS;
+}
+
+CSR_API(CsrFillOutputAttrib)
+{
+ int c;
+ PCSRSS_SCREEN_BUFFER Buff;
+ NTSTATUS Status;
+ int X, Y;
+ IO_STATUS_BLOCK Iosb;
+ OUTPUT_ATTRIBUTE Attr;
+
+ Reply->Header.MessageSize = sizeof(CSRSS_API_REPLY);
+ Reply->Header.DataSize = sizeof(CSRSS_API_REPLY) -
+ sizeof(LPC_MESSAGE_HEADER);
+ LOCK;
+ Status = CsrGetObject( ProcessData, Request->Data.FillOutputAttribRequest.ConsoleHandle, (Object_t **)&Buff );
+ if( !NT_SUCCESS( Status ) || (Status = Buff->Header.Type == CSRSS_SCREEN_BUFFER_MAGIC ? 0 : STATUS_INVALID_HANDLE ))
+ {
+ Reply->Status = Status;
+ UNLOCK;
+ return Status;
+ }
+ X = Buff->CurrentX;
+ Y = Buff->CurrentY;
+ Buff->CurrentX = Request->Data.FillOutputAttribRequest.Coord.X + Buff->ShowX;
+ Buff->CurrentY = Request->Data.FillOutputAttribRequest.Coord.Y + Buff->ShowY;
+ for( c = 0; c < Request->Data.FillOutputAttribRequest.Length; c++ )
+ {
+ Buff->Buffer[(Buff->CurrentY * Buff->MaxX * 2) + (Buff->CurrentX * 2) + 1] = Request->Data.FillOutputAttribRequest.Attribute;
+ if( ++Buff->CurrentX == Buff->MaxX )
+ {
+ Buff->CurrentX = 0;
+ if( ++Buff->CurrentY == Buff->MaxY )
+ Buff->CurrentY = 0;
+ }
+ }
+ if( Buff == ActiveConsole->ActiveBuffer )
+ {
+ Attr.wAttribute = Request->Data.FillOutputAttribRequest.Attribute;
+ Attr.nLength = Request->Data.FillOutputAttribRequest.Length;
+ Attr.dwCoord = Request->Data.FillOutputAttribRequest.Coord;
+ Status = NtDeviceIoControlFile( ConsoleDeviceHandle,
+ NULL,
+ NULL,
+ NULL,
+ &Iosb,
+ IOCTL_CONSOLE_FILL_OUTPUT_ATTRIBUTE,
+ &Attr,
+ sizeof (Attr),
+ 0,
+ 0 );
+ if( !NT_SUCCESS( Status ) )
+ DPRINT1( "Failed to fill output attribute\n" );
}
+ Buff->CurrentX = X;
+ Buff->CurrentY = Y;
+ UNLOCK;
+ return Reply->Status = STATUS_SUCCESS;
+}
+
+
+CSR_API(CsrGetCursorInfo)
+{
+ PCSRSS_SCREEN_BUFFER Buff;
+ NTSTATUS Status;
+
+ Reply->Header.MessageSize = sizeof(CSRSS_API_REPLY);
+ Reply->Header.DataSize = sizeof(CSRSS_API_REPLY) -
+ sizeof(LPC_MESSAGE_HEADER);
+ LOCK;
+ Status = CsrGetObject( ProcessData, Request->Data.GetCursorInfoRequest.ConsoleHandle, (Object_t **)&Buff );
+ if( !NT_SUCCESS( Status ) || (Status = Buff->Header.Type == CSRSS_SCREEN_BUFFER_MAGIC ? 0 : STATUS_INVALID_HANDLE ))
+ {
+ Reply->Status = Status;
+ UNLOCK;
+ return Status;
+ }
+ Reply->Data.GetCursorInfoReply.Info = Buff->CursorInfo;
+ UNLOCK;
+ return Reply->Status = STATUS_SUCCESS;
+}
+
+CSR_API(CsrSetCursorInfo)
+{
+ PCSRSS_SCREEN_BUFFER Buff;
+ NTSTATUS Status;
+ IO_STATUS_BLOCK Iosb;
+
+ Reply->Header.MessageSize = sizeof(CSRSS_API_REPLY);
+ Reply->Header.DataSize = sizeof(CSRSS_API_REPLY) -
+ sizeof(LPC_MESSAGE_HEADER);
+ LOCK;
+ Status = CsrGetObject( ProcessData, Request->Data.SetCursorInfoRequest.ConsoleHandle, (Object_t **)&Buff );
+ if( !NT_SUCCESS( Status ) || (Status = Buff->Header.Type == CSRSS_SCREEN_BUFFER_MAGIC ? 0 : STATUS_INVALID_HANDLE ))
+ {
+ Reply->Status = Status;
+ UNLOCK;
+ return Status;
+ }
+ Buff->CursorInfo = Request->Data.SetCursorInfoRequest.Info;
+ if( Buff == ActiveConsole->ActiveBuffer )
+ {
+ Status = NtDeviceIoControlFile( ConsoleDeviceHandle, NULL, NULL, NULL, &Iosb, IOCTL_CONSOLE_SET_CURSOR_INFO, &Buff->CursorInfo, sizeof( Buff->CursorInfo ), 0, 0 );
+ if( !NT_SUCCESS( Status ) )
+ {
+ DbgPrint( "CSR: Failed to set cursor info\n" );
+ return Reply->Status = Status;
+ }
+ }
+ UNLOCK;
+ return Reply->Status = STATUS_SUCCESS;
+}
+
+CSR_API(CsrSetTextAttrib)
+{
+ NTSTATUS Status;
+ CONSOLE_SCREEN_BUFFER_INFO ScrInfo;
+ IO_STATUS_BLOCK Iosb;
+ PCSRSS_SCREEN_BUFFER Buff;
+
+ Reply->Header.MessageSize = sizeof(CSRSS_API_REPLY);
+ Reply->Header.DataSize = sizeof(CSRSS_API_REPLY) -
+ sizeof(LPC_MESSAGE_HEADER);
+ LOCK;
+ Status = CsrGetObject( ProcessData, Request->Data.SetAttribRequest.ConsoleHandle, (Object_t **)&Buff );
+ if( !NT_SUCCESS( Status ) || (Status = Buff->Header.Type == CSRSS_SCREEN_BUFFER_MAGIC ? 0 : STATUS_INVALID_HANDLE ))
+ {
+ Reply->Status = Status;
+ UNLOCK;
+ return Status;
+ }
+ Buff->DefaultAttrib = Request->Data.SetAttribRequest.Attrib;
+ if( Buff == ActiveConsole->ActiveBuffer )
+ {
+ ScrInfo.wAttributes = Buff->DefaultAttrib;
+ ScrInfo.dwCursorPosition.X = Buff->CurrentX - Buff->ShowX;
+ ScrInfo.dwCursorPosition.Y = ((Buff->CurrentY + Buff->MaxY) - Buff->ShowY) % Buff->MaxY;
+ Status = NtDeviceIoControlFile( ConsoleDeviceHandle, NULL, NULL, NULL, &Iosb, IOCTL_CONSOLE_SET_SCREEN_BUFFER_INFO, &ScrInfo, sizeof( ScrInfo ), 0, 0 );
+ if( !NT_SUCCESS( Status ) )
+ {
+ DbgPrint( "CSR: Failed to set console info\n" );
+ UNLOCK;
+ return Reply->Status = Status;
+ }
+ }
+ UNLOCK;
+ return Reply->Status = STATUS_SUCCESS;
+}
+
+CSR_API(CsrSetConsoleMode)
+{
+ NTSTATUS Status;
+ PCSRSS_CONSOLE Console;
+ PCSRSS_SCREEN_BUFFER Buff;
+
+ Reply->Header.MessageSize = sizeof(CSRSS_API_REPLY);
+ Reply->Header.DataSize = sizeof(CSRSS_API_REPLY) - sizeof(LPC_MESSAGE_HEADER);
+ LOCK;
+ Status = CsrGetObject( ProcessData, Request->Data.SetConsoleModeRequest.ConsoleHandle, (Object_t **)&Console );
+ if( !NT_SUCCESS( Status ) )
+ {
+ Reply->Status = Status;
+ UNLOCK;
+ return Status;
+ }
+ Buff = (PCSRSS_SCREEN_BUFFER)Console;
+ if( Console->Header.Type == CSRSS_CONSOLE_MAGIC )
+ Console->Mode = Request->Data.SetConsoleModeRequest.Mode & CONSOLE_INPUT_MODE_VALID;
+ else if( Console->Header.Type == CSRSS_SCREEN_BUFFER_MAGIC )
+ Buff->Mode = Request->Data.SetConsoleModeRequest.Mode & CONSOLE_OUTPUT_MODE_VALID;
+ else {
+ Reply->Status = STATUS_INVALID_HANDLE;
+ UNLOCK;
+ return Status;
+ }
+ UNLOCK;
+ Reply->Status = STATUS_SUCCESS;
+ return Reply->Status;
+}
+
+CSR_API(CsrGetConsoleMode)
+{
+ NTSTATUS Status;
+ PCSRSS_CONSOLE Console;
+ PCSRSS_SCREEN_BUFFER Buff; /* gee, I really wish I could use an anonymous union here */
+
+ Reply->Header.MessageSize = sizeof(CSRSS_API_REPLY);
+ Reply->Header.DataSize = sizeof(CSRSS_API_REPLY) - sizeof(LPC_MESSAGE_HEADER);
+ LOCK;
+ Status = CsrGetObject( ProcessData, Request->Data.GetConsoleModeRequest.ConsoleHandle, (Object_t **)&Console );
+ if( !NT_SUCCESS( Status ) )
+ {
+ Reply->Status = Status;
+ UNLOCK;
+ return Status;
+ }
+ Reply->Status = STATUS_SUCCESS;
+ Buff = (PCSRSS_SCREEN_BUFFER)Console;
+ if( Console->Header.Type = CSRSS_CONSOLE_MAGIC )
+ Reply->Data.GetConsoleModeReply.ConsoleMode = Console->Mode;
+ else if( Buff->Header.Type = CSRSS_SCREEN_BUFFER_MAGIC )
+ Reply->Data.GetConsoleModeReply.ConsoleMode = Buff->Mode;
+ else Status = STATUS_INVALID_HANDLE;
+ UNLOCK;
+ return Reply->Status;
+}
+
+CSR_API(CsrCreateScreenBuffer)
+{
+ PCSRSS_SCREEN_BUFFER Buff = RtlAllocateHeap( CsrssApiHeap, 0, sizeof( CSRSS_SCREEN_BUFFER ) );
+ NTSTATUS Status;
+
+ Reply->Header.MessageSize = sizeof( CSRSS_API_REPLY );
+ Reply->Header.DataSize = sizeof(CSRSS_API_REPLY) - sizeof(LPC_MESSAGE_HEADER);
+ if( !Buff )
+ Reply->Status = STATUS_INSUFFICIENT_RESOURCES;
+ LOCK;
+ Status = CsrInitConsoleScreenBuffer( Buff );
+ if( !NT_SUCCESS( Status ) )
+ Reply->Status = Status;
+ else {
+ Status = CsrInsertObject( ProcessData, &Reply->Data.CreateScreenBufferReply.OutputHandle, &Buff->Header );
+ if( !NT_SUCCESS( Status ) )
+ Reply->Status = Status;
+ else Reply->Status = STATUS_SUCCESS;
+ }
+ UNLOCK;
+ return Reply->Status;
+}
+
+CSR_API(CsrSetScreenBuffer)
+{
+ NTSTATUS Status;
+ PCSRSS_SCREEN_BUFFER Buff;
+
+ Reply->Header.MessageSize = sizeof( CSRSS_API_REPLY );
+ Reply->Header.DataSize = sizeof(CSRSS_API_REPLY) - sizeof(LPC_MESSAGE_HEADER);
+ LOCK;
+ Status = CsrGetObject( ProcessData, Request->Data.SetActiveScreenBufferRequest.OutputHandle, (Object_t **)&Buff );
+ if( !NT_SUCCESS( Status ) )
+ Reply->Status = Status;
+ else {
+ // drop reference to old buffer, maybe delete
+ if( !InterlockedDecrement( &ProcessData->Console->ActiveBuffer->Header.ReferenceCount ) )
+ CsrDeleteScreenBuffer( ProcessData->Console->ActiveBuffer );
+ // tie console to new buffer
+ ProcessData->Console->ActiveBuffer = Buff;
+ // inc ref count on new buffer
+ InterlockedIncrement( &Buff->Header.ReferenceCount );
+ // if the console is active, redraw it
+ if( ActiveConsole == ProcessData->Console )
+ CsrDrawConsole( Buff );
+ Reply->Status = STATUS_SUCCESS;
+ }
+ UNLOCK;
+ return Reply->Status;
+}
+
+CSR_API(CsrSetTitle)
+{
+ NTSTATUS Status;
+ PCSRSS_CONSOLE Console;
+
+ Reply->Header.MessageSize = sizeof( CSRSS_API_REPLY );
+ Reply->Header.DataSize = sizeof(CSRSS_API_REPLY) - sizeof(LPC_MESSAGE_HEADER);
+ LOCK;
+ Status = CsrGetObject( ProcessData, Request->Data.SetTitleRequest.Console, (Object_t **)&Console );
+ if( !NT_SUCCESS( Status ) )
+ Reply->Status = Status;
+ else {
+ // copy title to console
+ RtlFreeUnicodeString( &Console->Title );
+ RtlCreateUnicodeString( &Console->Title, Request->Data.SetTitleRequest.Title );
+ Reply->Status = STATUS_SUCCESS;
+ }
+ UNLOCK;
+ return Reply->Status;
}
-VOID CsrInitConsole(PCSRSS_PROCESS_DATA ProcessData,
- PCSRSS_CONSOLE Console)
+CSR_API(CsrGetTitle)
{
+ NTSTATUS Status;
+ PCSRSS_CONSOLE Console;
+
+ Reply->Header.MessageSize = sizeof (CSRSS_API_REPLY);
+ Reply->Header.DataSize =
+ sizeof (CSRSS_API_REPLY)
+ - sizeof(LPC_MESSAGE_HEADER);
+ LOCK;
+ Status = CsrGetObject (
+ ProcessData,
+ Request->Data.GetTitleRequest.ConsoleHandle,
+ (Object_t **) & Console
+ );
+ if ( !NT_SUCCESS( Status ) )
+ {
+ Reply->Status = Status;
+ }
+ else
+ {
+ HANDLE ConsoleHandle = Request->Data.GetTitleRequest.ConsoleHandle;
+
+ /* Copy title of the console to the user title buffer */
+ RtlZeroMemory (
+ & Reply->Data.GetTitleReply,
+ sizeof (CSRSS_GET_TITLE_REPLY)
+ );
+ Reply->Data.GetTitleReply.ConsoleHandle = ConsoleHandle;
+ Reply->Data.GetTitleReply.Length = Console->Title.Length;
+ wcscpy (Reply->Data.GetTitleReply.Title, Console->Title.Buffer);
+ Reply->Status = STATUS_SUCCESS;
+ }
+ UNLOCK;
+ return Reply->Status;
}
/* EOF */
projects / reactos.git / blobdiff