mirror of
https://github.com/reactos/reactos.git
synced 2024-09-11 05:12:43 +00:00
645 lines
22 KiB
C
645 lines
22 KiB
C
/*
|
|
* LICENSE: GPL - See COPYING in the top level directory
|
|
* PROJECT: ReactOS Console Server DLL
|
|
* FILE: win32ss/user/winsrv/consrv_new/lineinput.c
|
|
* PURPOSE: Console line input functions
|
|
* PROGRAMMERS: Jeffrey Morlan
|
|
*/
|
|
|
|
/* INCLUDES *******************************************************************/
|
|
|
|
#include "consrv.h"
|
|
#include "console.h"
|
|
#include "include/conio.h"
|
|
#include "include/conio2.h"
|
|
|
|
#define NDEBUG
|
|
#include <debug.h>
|
|
|
|
typedef struct _HISTORY_BUFFER
|
|
{
|
|
LIST_ENTRY ListEntry;
|
|
UINT Position;
|
|
UINT MaxEntries;
|
|
UINT NumEntries;
|
|
PUNICODE_STRING Entries;
|
|
UNICODE_STRING ExeName;
|
|
} HISTORY_BUFFER, *PHISTORY_BUFFER;
|
|
|
|
|
|
/* PRIVATE FUNCTIONS **********************************************************/
|
|
|
|
static PHISTORY_BUFFER
|
|
HistoryCurrentBuffer(PCONSOLE Console)
|
|
{
|
|
/* TODO: use actual EXE name sent from process that called ReadConsole */
|
|
UNICODE_STRING ExeName = { 14, 14, L"cmd.exe" };
|
|
PLIST_ENTRY Entry = Console->HistoryBuffers.Flink;
|
|
PHISTORY_BUFFER Hist;
|
|
|
|
for (; Entry != &Console->HistoryBuffers; Entry = Entry->Flink)
|
|
{
|
|
Hist = CONTAINING_RECORD(Entry, HISTORY_BUFFER, ListEntry);
|
|
if (RtlEqualUnicodeString(&ExeName, &Hist->ExeName, FALSE))
|
|
return Hist;
|
|
}
|
|
|
|
/* Couldn't find the buffer, create a new one */
|
|
Hist = ConsoleAllocHeap(0, sizeof(HISTORY_BUFFER) + ExeName.Length);
|
|
if (!Hist) return NULL;
|
|
Hist->MaxEntries = Console->HistoryBufferSize;
|
|
Hist->NumEntries = 0;
|
|
Hist->Entries = ConsoleAllocHeap(0, Hist->MaxEntries * sizeof(UNICODE_STRING));
|
|
if (!Hist->Entries)
|
|
{
|
|
ConsoleFreeHeap(Hist);
|
|
return NULL;
|
|
}
|
|
Hist->ExeName.Length = Hist->ExeName.MaximumLength = ExeName.Length;
|
|
Hist->ExeName.Buffer = (PWCHAR)(Hist + 1);
|
|
memcpy(Hist->ExeName.Buffer, ExeName.Buffer, ExeName.Length);
|
|
InsertHeadList(&Console->HistoryBuffers, &Hist->ListEntry);
|
|
return Hist;
|
|
}
|
|
|
|
static VOID
|
|
HistoryAddEntry(PCONSOLE Console)
|
|
{
|
|
UNICODE_STRING NewEntry;
|
|
PHISTORY_BUFFER Hist = HistoryCurrentBuffer(Console);
|
|
INT i;
|
|
|
|
if (!Hist) return;
|
|
|
|
NewEntry.Length = NewEntry.MaximumLength = Console->LineSize * sizeof(WCHAR);
|
|
NewEntry.Buffer = Console->LineBuffer;
|
|
|
|
/* Don't add blank or duplicate entries */
|
|
if (NewEntry.Length == 0 || Hist->MaxEntries == 0 ||
|
|
(Hist->NumEntries > 0 &&
|
|
RtlEqualUnicodeString(&Hist->Entries[Hist->NumEntries - 1], &NewEntry, FALSE)))
|
|
{
|
|
return;
|
|
}
|
|
|
|
if (Console->HistoryNoDup)
|
|
{
|
|
/* Check if this line has been entered before */
|
|
for (i = Hist->NumEntries - 1; i >= 0; i--)
|
|
{
|
|
if (RtlEqualUnicodeString(&Hist->Entries[i], &NewEntry, FALSE))
|
|
{
|
|
/* Just rotate the list to bring this entry to the end */
|
|
NewEntry = Hist->Entries[i];
|
|
memmove(&Hist->Entries[i], &Hist->Entries[i + 1],
|
|
(Hist->NumEntries - (i + 1)) * sizeof(UNICODE_STRING));
|
|
Hist->Entries[Hist->NumEntries - 1] = NewEntry;
|
|
Hist->Position = Hist->NumEntries - 1;
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (Hist->NumEntries == Hist->MaxEntries)
|
|
{
|
|
/* List is full, remove oldest entry */
|
|
RtlFreeUnicodeString(&Hist->Entries[0]);
|
|
memmove(&Hist->Entries[0], &Hist->Entries[1],
|
|
--Hist->NumEntries * sizeof(UNICODE_STRING));
|
|
}
|
|
|
|
if (NT_SUCCESS(RtlDuplicateUnicodeString(0, &NewEntry, &Hist->Entries[Hist->NumEntries])))
|
|
Hist->NumEntries++;
|
|
Hist->Position = Hist->NumEntries - 1;
|
|
}
|
|
|
|
static VOID
|
|
HistoryGetCurrentEntry(PCONSOLE Console, PUNICODE_STRING Entry)
|
|
{
|
|
PHISTORY_BUFFER Hist = HistoryCurrentBuffer(Console);
|
|
|
|
if (!Hist || Hist->NumEntries == 0)
|
|
Entry->Length = 0;
|
|
else
|
|
*Entry = Hist->Entries[Hist->Position];
|
|
}
|
|
|
|
static PHISTORY_BUFFER
|
|
HistoryFindBuffer(PCONSOLE Console, PUNICODE_STRING ExeName)
|
|
{
|
|
PLIST_ENTRY Entry = Console->HistoryBuffers.Flink;
|
|
while (Entry != &Console->HistoryBuffers)
|
|
{
|
|
/* For the history APIs, the caller is allowed to give only part of the name */
|
|
PHISTORY_BUFFER Hist = CONTAINING_RECORD(Entry, HISTORY_BUFFER, ListEntry);
|
|
if (RtlPrefixUnicodeString(ExeName, &Hist->ExeName, TRUE))
|
|
return Hist;
|
|
Entry = Entry->Flink;
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
static VOID
|
|
HistoryDeleteBuffer(PHISTORY_BUFFER Hist)
|
|
{
|
|
if (!Hist) return;
|
|
|
|
while (Hist->NumEntries != 0)
|
|
RtlFreeUnicodeString(&Hist->Entries[--Hist->NumEntries]);
|
|
|
|
ConsoleFreeHeap(Hist->Entries);
|
|
RemoveEntryList(&Hist->ListEntry);
|
|
ConsoleFreeHeap(Hist);
|
|
}
|
|
|
|
VOID FASTCALL
|
|
HistoryDeleteBuffers(PCONSOLE Console)
|
|
{
|
|
PLIST_ENTRY CurrentEntry;
|
|
PHISTORY_BUFFER HistoryBuffer;
|
|
|
|
while (!IsListEmpty(&Console->HistoryBuffers))
|
|
{
|
|
CurrentEntry = RemoveHeadList(&Console->HistoryBuffers);
|
|
HistoryBuffer = CONTAINING_RECORD(CurrentEntry, HISTORY_BUFFER, ListEntry);
|
|
HistoryDeleteBuffer(HistoryBuffer);
|
|
}
|
|
}
|
|
|
|
static VOID
|
|
LineInputSetPos(PCONSOLE Console, UINT Pos)
|
|
{
|
|
if (Pos != Console->LinePos && Console->InputBuffer.Mode & ENABLE_ECHO_INPUT)
|
|
{
|
|
PCONSOLE_SCREEN_BUFFER Buffer = Console->ActiveBuffer;
|
|
SHORT OldCursorX = Buffer->CursorPosition.X;
|
|
SHORT OldCursorY = Buffer->CursorPosition.Y;
|
|
INT XY = OldCursorY * Buffer->ScreenBufferSize.X + OldCursorX;
|
|
|
|
XY += (Pos - Console->LinePos);
|
|
if (XY < 0)
|
|
XY = 0;
|
|
else if (XY >= Buffer->ScreenBufferSize.Y * Buffer->ScreenBufferSize.X)
|
|
XY = Buffer->ScreenBufferSize.Y * Buffer->ScreenBufferSize.X - 1;
|
|
|
|
Buffer->CursorPosition.X = XY % Buffer->ScreenBufferSize.X;
|
|
Buffer->CursorPosition.Y = XY / Buffer->ScreenBufferSize.X;
|
|
ConioSetScreenInfo(Console, Buffer, OldCursorX, OldCursorY);
|
|
}
|
|
|
|
Console->LinePos = Pos;
|
|
}
|
|
|
|
static VOID
|
|
LineInputEdit(PCONSOLE Console, UINT NumToDelete, UINT NumToInsert, WCHAR *Insertion)
|
|
{
|
|
PTEXTMODE_SCREEN_BUFFER ActiveBuffer;
|
|
UINT Pos = Console->LinePos;
|
|
UINT NewSize = Console->LineSize - NumToDelete + NumToInsert;
|
|
UINT i;
|
|
|
|
if (GetType(Console->ActiveBuffer) != TEXTMODE_BUFFER) return;
|
|
ActiveBuffer = (PTEXTMODE_SCREEN_BUFFER)Console->ActiveBuffer;
|
|
|
|
/* Make sure there's always enough room for ending \r\n */
|
|
if (NewSize + 2 > Console->LineMaxSize)
|
|
return;
|
|
|
|
memmove(&Console->LineBuffer[Pos + NumToInsert],
|
|
&Console->LineBuffer[Pos + NumToDelete],
|
|
(Console->LineSize - (Pos + NumToDelete)) * sizeof(WCHAR));
|
|
memcpy(&Console->LineBuffer[Pos], Insertion, NumToInsert * sizeof(WCHAR));
|
|
|
|
if (Console->InputBuffer.Mode & ENABLE_ECHO_INPUT)
|
|
{
|
|
for (i = Pos; i < NewSize; i++)
|
|
{
|
|
ConioWriteConsole(Console, ActiveBuffer, &Console->LineBuffer[i], 1, TRUE);
|
|
}
|
|
for (; i < Console->LineSize; i++)
|
|
{
|
|
ConioWriteConsole(Console, ActiveBuffer, L" ", 1, TRUE);
|
|
}
|
|
Console->LinePos = i;
|
|
}
|
|
|
|
Console->LineSize = NewSize;
|
|
LineInputSetPos(Console, Pos + NumToInsert);
|
|
}
|
|
|
|
static VOID
|
|
LineInputRecallHistory(PCONSOLE Console, INT Offset)
|
|
{
|
|
PHISTORY_BUFFER Hist = HistoryCurrentBuffer(Console);
|
|
UINT Position = 0;
|
|
|
|
if (!Hist || Hist->NumEntries == 0) return;
|
|
|
|
Position = Hist->Position + Offset;
|
|
Position = min(max(Position, 0), Hist->NumEntries - 1);
|
|
Hist->Position = Position;
|
|
|
|
LineInputSetPos(Console, 0);
|
|
LineInputEdit(Console, Console->LineSize,
|
|
Hist->Entries[Hist->Position].Length / sizeof(WCHAR),
|
|
Hist->Entries[Hist->Position].Buffer);
|
|
}
|
|
|
|
VOID FASTCALL
|
|
LineInputKeyDown(PCONSOLE Console, KEY_EVENT_RECORD *KeyEvent)
|
|
{
|
|
UINT Pos = Console->LinePos;
|
|
PHISTORY_BUFFER Hist;
|
|
UNICODE_STRING Entry;
|
|
INT HistPos;
|
|
|
|
switch (KeyEvent->wVirtualKeyCode)
|
|
{
|
|
case VK_ESCAPE:
|
|
/* Clear entire line */
|
|
LineInputSetPos(Console, 0);
|
|
LineInputEdit(Console, Console->LineSize, 0, NULL);
|
|
return;
|
|
case VK_HOME:
|
|
/* Move to start of line. With ctrl, erase everything left of cursor */
|
|
LineInputSetPos(Console, 0);
|
|
if (KeyEvent->dwControlKeyState & (LEFT_CTRL_PRESSED | RIGHT_CTRL_PRESSED))
|
|
LineInputEdit(Console, Pos, 0, NULL);
|
|
return;
|
|
case VK_END:
|
|
/* Move to end of line. With ctrl, erase everything right of cursor */
|
|
if (KeyEvent->dwControlKeyState & (LEFT_CTRL_PRESSED | RIGHT_CTRL_PRESSED))
|
|
LineInputEdit(Console, Console->LineSize - Pos, 0, NULL);
|
|
else
|
|
LineInputSetPos(Console, Console->LineSize);
|
|
return;
|
|
case VK_LEFT:
|
|
/* Move left. With ctrl, move to beginning of previous word */
|
|
if (KeyEvent->dwControlKeyState & (LEFT_CTRL_PRESSED | RIGHT_CTRL_PRESSED))
|
|
{
|
|
while (Pos > 0 && Console->LineBuffer[Pos - 1] == L' ') Pos--;
|
|
while (Pos > 0 && Console->LineBuffer[Pos - 1] != L' ') Pos--;
|
|
}
|
|
else
|
|
{
|
|
Pos -= (Pos > 0);
|
|
}
|
|
LineInputSetPos(Console, Pos);
|
|
return;
|
|
case VK_RIGHT:
|
|
case VK_F1:
|
|
/* Move right. With ctrl, move to beginning of next word */
|
|
if (KeyEvent->dwControlKeyState & (LEFT_CTRL_PRESSED | RIGHT_CTRL_PRESSED))
|
|
{
|
|
while (Pos < Console->LineSize && Console->LineBuffer[Pos] != L' ') Pos++;
|
|
while (Pos < Console->LineSize && Console->LineBuffer[Pos] == L' ') Pos++;
|
|
LineInputSetPos(Console, Pos);
|
|
return;
|
|
}
|
|
else
|
|
{
|
|
/* Recall one character (but don't overwrite current line) */
|
|
HistoryGetCurrentEntry(Console, &Entry);
|
|
if (Pos < Console->LineSize)
|
|
LineInputSetPos(Console, Pos + 1);
|
|
else if (Pos * sizeof(WCHAR) < Entry.Length)
|
|
LineInputEdit(Console, 0, 1, &Entry.Buffer[Pos]);
|
|
}
|
|
return;
|
|
case VK_INSERT:
|
|
/* Toggle between insert and overstrike */
|
|
Console->LineInsertToggle = !Console->LineInsertToggle;
|
|
ConioSetCursorInfo(Console, Console->ActiveBuffer);
|
|
return;
|
|
case VK_DELETE:
|
|
/* Remove character to right of cursor */
|
|
if (Pos != Console->LineSize)
|
|
LineInputEdit(Console, 1, 0, NULL);
|
|
return;
|
|
case VK_PRIOR:
|
|
/* Recall first history entry */
|
|
LineInputRecallHistory(Console, -((WORD)-1));
|
|
return;
|
|
case VK_NEXT:
|
|
/* Recall last history entry */
|
|
LineInputRecallHistory(Console, +((WORD)-1));
|
|
return;
|
|
case VK_UP:
|
|
case VK_F5:
|
|
/* Recall previous history entry. On first time, actually recall the
|
|
* current (usually last) entry; on subsequent times go back. */
|
|
LineInputRecallHistory(Console, Console->LineUpPressed ? -1 : 0);
|
|
Console->LineUpPressed = TRUE;
|
|
return;
|
|
case VK_DOWN:
|
|
/* Recall next history entry */
|
|
LineInputRecallHistory(Console, +1);
|
|
return;
|
|
case VK_F3:
|
|
/* Recall remainder of current history entry */
|
|
HistoryGetCurrentEntry(Console, &Entry);
|
|
if (Pos * sizeof(WCHAR) < Entry.Length)
|
|
{
|
|
UINT InsertSize = (Entry.Length / sizeof(WCHAR) - Pos);
|
|
UINT DeleteSize = min(Console->LineSize - Pos, InsertSize);
|
|
LineInputEdit(Console, DeleteSize, InsertSize, &Entry.Buffer[Pos]);
|
|
}
|
|
return;
|
|
case VK_F6:
|
|
/* Insert a ^Z character */
|
|
KeyEvent->uChar.UnicodeChar = 26;
|
|
break;
|
|
case VK_F7:
|
|
if (KeyEvent->dwControlKeyState & (LEFT_ALT_PRESSED | RIGHT_ALT_PRESSED))
|
|
HistoryDeleteBuffer(HistoryCurrentBuffer(Console));
|
|
return;
|
|
case VK_F8:
|
|
/* Search for history entries starting with input. */
|
|
Hist = HistoryCurrentBuffer(Console);
|
|
if (!Hist || Hist->NumEntries == 0) return;
|
|
|
|
/* Like Up/F5, on first time start from current (usually last) entry,
|
|
* but on subsequent times start at previous entry. */
|
|
if (Console->LineUpPressed)
|
|
Hist->Position = (Hist->Position ? Hist->Position : Hist->NumEntries) - 1;
|
|
Console->LineUpPressed = TRUE;
|
|
|
|
Entry.Length = Console->LinePos * sizeof(WCHAR);
|
|
Entry.Buffer = Console->LineBuffer;
|
|
|
|
/* Keep going backwards, even wrapping around to the end,
|
|
* until we get back to starting point */
|
|
HistPos = Hist->Position;
|
|
do
|
|
{
|
|
if (RtlPrefixUnicodeString(&Entry, &Hist->Entries[HistPos], FALSE))
|
|
{
|
|
Hist->Position = HistPos;
|
|
LineInputEdit(Console, Console->LineSize - Pos,
|
|
Hist->Entries[HistPos].Length / sizeof(WCHAR) - Pos,
|
|
&Hist->Entries[HistPos].Buffer[Pos]);
|
|
/* Cursor stays where it was */
|
|
LineInputSetPos(Console, Pos);
|
|
return;
|
|
}
|
|
if (--HistPos < 0) HistPos += Hist->NumEntries;
|
|
} while (HistPos != Hist->Position);
|
|
return;
|
|
}
|
|
|
|
if (KeyEvent->uChar.UnicodeChar == L'\b' && Console->InputBuffer.Mode & ENABLE_PROCESSED_INPUT)
|
|
{
|
|
/* backspace handling - if processed input enabled then we handle it here
|
|
* otherwise we treat it like a normal char. */
|
|
if (Pos > 0)
|
|
{
|
|
LineInputSetPos(Console, Pos - 1);
|
|
LineInputEdit(Console, 1, 0, NULL);
|
|
}
|
|
}
|
|
else if (KeyEvent->uChar.UnicodeChar == L'\r')
|
|
{
|
|
HistoryAddEntry(Console);
|
|
|
|
/* TODO: Expand aliases */
|
|
|
|
LineInputSetPos(Console, Console->LineSize);
|
|
Console->LineBuffer[Console->LineSize++] = L'\r';
|
|
if (Console->InputBuffer.Mode & ENABLE_ECHO_INPUT)
|
|
{
|
|
if (GetType(Console->ActiveBuffer) == TEXTMODE_BUFFER)
|
|
{
|
|
ConioWriteConsole(Console, (PTEXTMODE_SCREEN_BUFFER)(Console->ActiveBuffer), L"\r", 1, TRUE);
|
|
}
|
|
}
|
|
|
|
/* Add \n if processed input. There should usually be room for it,
|
|
* but an exception to the rule exists: the buffer could have been
|
|
* pre-filled with LineMaxSize - 1 characters. */
|
|
if (Console->InputBuffer.Mode & ENABLE_PROCESSED_INPUT &&
|
|
Console->LineSize < Console->LineMaxSize)
|
|
{
|
|
Console->LineBuffer[Console->LineSize++] = L'\n';
|
|
if (Console->InputBuffer.Mode & ENABLE_ECHO_INPUT)
|
|
{
|
|
if (GetType(Console->ActiveBuffer) == TEXTMODE_BUFFER)
|
|
{
|
|
ConioWriteConsole(Console, (PTEXTMODE_SCREEN_BUFFER)(Console->ActiveBuffer), L"\n", 1, TRUE);
|
|
}
|
|
}
|
|
}
|
|
Console->LineComplete = TRUE;
|
|
Console->LinePos = 0;
|
|
}
|
|
else if (KeyEvent->uChar.UnicodeChar != L'\0')
|
|
{
|
|
if (KeyEvent->uChar.UnicodeChar < 0x20 &&
|
|
Console->LineWakeupMask & (1 << KeyEvent->uChar.UnicodeChar))
|
|
{
|
|
/* Control key client wants to handle itself (e.g. for tab completion) */
|
|
Console->LineBuffer[Console->LineSize++] = L' ';
|
|
Console->LineBuffer[Console->LinePos] = KeyEvent->uChar.UnicodeChar;
|
|
Console->LineComplete = TRUE;
|
|
Console->LinePos = 0;
|
|
}
|
|
else
|
|
{
|
|
/* Normal character */
|
|
BOOL Overstrike = Console->LineInsertToggle && Console->LinePos != Console->LineSize;
|
|
LineInputEdit(Console, Overstrike, 1, &KeyEvent->uChar.UnicodeChar);
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
/* PUBLIC SERVER APIS *********************************************************/
|
|
|
|
CSR_API(SrvGetConsoleCommandHistory)
|
|
{
|
|
PCONSOLE_GETCOMMANDHISTORY GetCommandHistoryRequest = &((PCONSOLE_API_MESSAGE)ApiMessage)->Data.GetCommandHistoryRequest;
|
|
PCONSOLE_PROCESS_DATA ProcessData = ConsoleGetPerProcessData(CsrGetClientThread()->Process);
|
|
PCONSOLE Console;
|
|
NTSTATUS Status;
|
|
PHISTORY_BUFFER Hist;
|
|
PBYTE Buffer = (PBYTE)GetCommandHistoryRequest->History;
|
|
ULONG BufferSize = GetCommandHistoryRequest->Length;
|
|
UINT i;
|
|
|
|
if ( !CsrValidateMessageBuffer(ApiMessage,
|
|
(PVOID*)&GetCommandHistoryRequest->History,
|
|
GetCommandHistoryRequest->Length,
|
|
sizeof(BYTE)) ||
|
|
!CsrValidateMessageBuffer(ApiMessage,
|
|
(PVOID*)&GetCommandHistoryRequest->ExeName.Buffer,
|
|
GetCommandHistoryRequest->ExeName.Length,
|
|
sizeof(BYTE)) )
|
|
{
|
|
return STATUS_INVALID_PARAMETER;
|
|
}
|
|
|
|
Status = ConSrvGetConsole(ProcessData, &Console, TRUE);
|
|
if (NT_SUCCESS(Status))
|
|
{
|
|
Hist = HistoryFindBuffer(Console, &GetCommandHistoryRequest->ExeName);
|
|
if (Hist)
|
|
{
|
|
for (i = 0; i < Hist->NumEntries; i++)
|
|
{
|
|
if (BufferSize < (Hist->Entries[i].Length + sizeof(WCHAR)))
|
|
{
|
|
Status = STATUS_BUFFER_OVERFLOW;
|
|
break;
|
|
}
|
|
memcpy(Buffer, Hist->Entries[i].Buffer, Hist->Entries[i].Length);
|
|
Buffer += Hist->Entries[i].Length;
|
|
*(PWCHAR)Buffer = L'\0';
|
|
Buffer += sizeof(WCHAR);
|
|
}
|
|
}
|
|
GetCommandHistoryRequest->Length = Buffer - (PBYTE)GetCommandHistoryRequest->History;
|
|
ConSrvReleaseConsole(Console, TRUE);
|
|
}
|
|
return Status;
|
|
}
|
|
|
|
CSR_API(SrvGetConsoleCommandHistoryLength)
|
|
{
|
|
PCONSOLE_GETCOMMANDHISTORYLENGTH GetCommandHistoryLengthRequest = &((PCONSOLE_API_MESSAGE)ApiMessage)->Data.GetCommandHistoryLengthRequest;
|
|
PCONSOLE_PROCESS_DATA ProcessData = ConsoleGetPerProcessData(CsrGetClientThread()->Process);
|
|
PCONSOLE Console;
|
|
NTSTATUS Status;
|
|
PHISTORY_BUFFER Hist;
|
|
ULONG Length = 0;
|
|
UINT i;
|
|
|
|
if (!CsrValidateMessageBuffer(ApiMessage,
|
|
(PVOID*)&GetCommandHistoryLengthRequest->ExeName.Buffer,
|
|
GetCommandHistoryLengthRequest->ExeName.Length,
|
|
sizeof(BYTE)))
|
|
{
|
|
return STATUS_INVALID_PARAMETER;
|
|
}
|
|
|
|
Status = ConSrvGetConsole(ProcessData, &Console, TRUE);
|
|
if (NT_SUCCESS(Status))
|
|
{
|
|
Hist = HistoryFindBuffer(Console, &GetCommandHistoryLengthRequest->ExeName);
|
|
if (Hist)
|
|
{
|
|
for (i = 0; i < Hist->NumEntries; i++)
|
|
Length += Hist->Entries[i].Length + sizeof(WCHAR);
|
|
}
|
|
GetCommandHistoryLengthRequest->Length = Length;
|
|
ConSrvReleaseConsole(Console, TRUE);
|
|
}
|
|
return Status;
|
|
}
|
|
|
|
CSR_API(SrvExpungeConsoleCommandHistory)
|
|
{
|
|
PCONSOLE_EXPUNGECOMMANDHISTORY ExpungeCommandHistoryRequest = &((PCONSOLE_API_MESSAGE)ApiMessage)->Data.ExpungeCommandHistoryRequest;
|
|
PCONSOLE_PROCESS_DATA ProcessData = ConsoleGetPerProcessData(CsrGetClientThread()->Process);
|
|
PCONSOLE Console;
|
|
PHISTORY_BUFFER Hist;
|
|
NTSTATUS Status;
|
|
|
|
if (!CsrValidateMessageBuffer(ApiMessage,
|
|
(PVOID*)&ExpungeCommandHistoryRequest->ExeName.Buffer,
|
|
ExpungeCommandHistoryRequest->ExeName.Length,
|
|
sizeof(BYTE)))
|
|
{
|
|
return STATUS_INVALID_PARAMETER;
|
|
}
|
|
|
|
Status = ConSrvGetConsole(ProcessData, &Console, TRUE);
|
|
if (NT_SUCCESS(Status))
|
|
{
|
|
Hist = HistoryFindBuffer(Console, &ExpungeCommandHistoryRequest->ExeName);
|
|
HistoryDeleteBuffer(Hist);
|
|
ConSrvReleaseConsole(Console, TRUE);
|
|
}
|
|
return Status;
|
|
}
|
|
|
|
CSR_API(SrvSetConsoleNumberOfCommands)
|
|
{
|
|
PCONSOLE_SETHISTORYNUMBERCOMMANDS SetHistoryNumberCommandsRequest = &((PCONSOLE_API_MESSAGE)ApiMessage)->Data.SetHistoryNumberCommandsRequest;
|
|
PCONSOLE_PROCESS_DATA ProcessData = ConsoleGetPerProcessData(CsrGetClientThread()->Process);
|
|
PCONSOLE Console;
|
|
PHISTORY_BUFFER Hist;
|
|
NTSTATUS Status;
|
|
UINT MaxEntries = SetHistoryNumberCommandsRequest->NumCommands;
|
|
PUNICODE_STRING OldEntryList, NewEntryList;
|
|
|
|
if (!CsrValidateMessageBuffer(ApiMessage,
|
|
(PVOID*)&SetHistoryNumberCommandsRequest->ExeName.Buffer,
|
|
SetHistoryNumberCommandsRequest->ExeName.Length,
|
|
sizeof(BYTE)))
|
|
{
|
|
return STATUS_INVALID_PARAMETER;
|
|
}
|
|
|
|
Status = ConSrvGetConsole(ProcessData, &Console, TRUE);
|
|
if (NT_SUCCESS(Status))
|
|
{
|
|
Hist = HistoryFindBuffer(Console, &SetHistoryNumberCommandsRequest->ExeName);
|
|
if (Hist)
|
|
{
|
|
OldEntryList = Hist->Entries;
|
|
NewEntryList = ConsoleAllocHeap(0, MaxEntries * sizeof(UNICODE_STRING));
|
|
if (!NewEntryList)
|
|
{
|
|
Status = STATUS_NO_MEMORY;
|
|
}
|
|
else
|
|
{
|
|
/* If necessary, shrink by removing oldest entries */
|
|
for (; Hist->NumEntries > MaxEntries; Hist->NumEntries--)
|
|
{
|
|
RtlFreeUnicodeString(Hist->Entries++);
|
|
Hist->Position += (Hist->Position == 0);
|
|
}
|
|
|
|
Hist->MaxEntries = MaxEntries;
|
|
Hist->Entries = memcpy(NewEntryList, Hist->Entries,
|
|
Hist->NumEntries * sizeof(UNICODE_STRING));
|
|
ConsoleFreeHeap(OldEntryList);
|
|
}
|
|
}
|
|
ConSrvReleaseConsole(Console, TRUE);
|
|
}
|
|
return Status;
|
|
}
|
|
|
|
CSR_API(SrvGetConsoleHistory)
|
|
{
|
|
PCONSOLE_GETSETHISTORYINFO HistoryInfoRequest = &((PCONSOLE_API_MESSAGE)ApiMessage)->Data.HistoryInfoRequest;
|
|
PCONSOLE Console;
|
|
NTSTATUS Status = ConSrvGetConsole(ConsoleGetPerProcessData(CsrGetClientThread()->Process), &Console, TRUE);
|
|
if (NT_SUCCESS(Status))
|
|
{
|
|
HistoryInfoRequest->HistoryBufferSize = Console->HistoryBufferSize;
|
|
HistoryInfoRequest->NumberOfHistoryBuffers = Console->NumberOfHistoryBuffers;
|
|
HistoryInfoRequest->dwFlags = Console->HistoryNoDup;
|
|
ConSrvReleaseConsole(Console, TRUE);
|
|
}
|
|
return Status;
|
|
}
|
|
|
|
CSR_API(SrvSetConsoleHistory)
|
|
{
|
|
PCONSOLE_GETSETHISTORYINFO HistoryInfoRequest = &((PCONSOLE_API_MESSAGE)ApiMessage)->Data.HistoryInfoRequest;
|
|
PCONSOLE Console;
|
|
NTSTATUS Status = ConSrvGetConsole(ConsoleGetPerProcessData(CsrGetClientThread()->Process), &Console, TRUE);
|
|
if (NT_SUCCESS(Status))
|
|
{
|
|
Console->HistoryBufferSize = HistoryInfoRequest->HistoryBufferSize;
|
|
Console->NumberOfHistoryBuffers = HistoryInfoRequest->NumberOfHistoryBuffers;
|
|
Console->HistoryNoDup = HistoryInfoRequest->dwFlags & HISTORY_NO_DUP_FLAG;
|
|
ConSrvReleaseConsole(Console, TRUE);
|
|
}
|
|
return Status;
|
|
}
|
|
|
|
/* EOF */
|