Implement ReadConsoleInputExA/W, which are a generalization of ReadConsoleInputA/W and PeekConsoleInputA/W. They allow you to read console input events by either waiting (or not) for their presence, and to remove them from the console input list (or not).
Those functions are the read-equivalent of WriteConsoleInputVDMA/W.
Used in Windows' NTVDM (guess why ;) ).
svn path=/trunk/; revision=60920
static
BOOL
IntGetConsoleInput(HANDLE hConsoleInput,
- BOOL bRead,
PINPUT_RECORD lpBuffer,
DWORD nLength,
LPDWORD lpNumberOfEventsRead,
- BOOL bUnicode)
+ WORD wFlags,
+ BOOLEAN bUnicode)
{
NTSTATUS Status;
CONSOLE_API_MESSAGE ApiMessage;
/* Set up the data to send to the Console Server */
GetInputRequest->InputHandle = hConsoleInput;
- GetInputRequest->Unicode = bUnicode;
- GetInputRequest->bRead = bRead;
GetInputRequest->InputsRead = 0;
GetInputRequest->Length = nLength;
+ GetInputRequest->wFlags = wFlags;
+ GetInputRequest->Unicode = bUnicode;
/* Call the server */
Status = CsrClientCallServer((PCSR_API_MESSAGE)&ApiMessage,
LPDWORD lpNumberOfEventsRead)
{
return IntGetConsoleInput(hConsoleInput,
- FALSE,
lpBuffer,
nLength,
lpNumberOfEventsRead,
+ CONSOLE_READ_KEEPEVENT | CONSOLE_READ_CONTINUE,
TRUE);
}
LPDWORD lpNumberOfEventsRead)
{
return IntGetConsoleInput(hConsoleInput,
- FALSE,
lpBuffer,
nLength,
lpNumberOfEventsRead,
+ CONSOLE_READ_KEEPEVENT | CONSOLE_READ_CONTINUE,
FALSE);
}
LPDWORD lpNumberOfEventsRead)
{
return IntGetConsoleInput(hConsoleInput,
- TRUE,
lpBuffer,
nLength,
lpNumberOfEventsRead,
+ 0,
TRUE);
}
LPDWORD lpNumberOfEventsRead)
{
return IntGetConsoleInput(hConsoleInput,
- TRUE,
lpBuffer,
nLength,
lpNumberOfEventsRead,
+ 0,
FALSE);
}
+/*--------------------------------------------------------------
+ * ReadConsoleInputExW
+ *
+ * @implemented
+ */
BOOL
WINAPI
-ReadConsoleInputExW(HANDLE hConsole, LPVOID lpBuffer, DWORD dwLen, LPDWORD Unknown1, DWORD Unknown2)
+ReadConsoleInputExW(HANDLE hConsoleInput,
+ PINPUT_RECORD lpBuffer,
+ DWORD nLength,
+ LPDWORD lpNumberOfEventsRead,
+ WORD wFlags)
{
- STUB;
- return FALSE;
+ return IntGetConsoleInput(hConsoleInput,
+ lpBuffer,
+ nLength,
+ lpNumberOfEventsRead,
+ wFlags,
+ TRUE);
}
+/*--------------------------------------------------------------
+ * ReadConsoleInputExA
+ *
+ * @implemented
+ */
BOOL
WINAPI
-ReadConsoleInputExA(HANDLE hConsole, LPVOID lpBuffer, DWORD dwLen, LPDWORD Unknown1, DWORD Unknown2)
+ReadConsoleInputExA(HANDLE hConsoleInput,
+ PINPUT_RECORD lpBuffer,
+ DWORD nLength,
+ LPDWORD lpNumberOfEventsRead,
+ WORD wFlags)
{
- STUB;
- return FALSE;
+ return IntGetConsoleInput(hConsoleInput,
+ lpBuffer,
+ nLength,
+ lpNumberOfEventsRead,
+ wFlags,
+ FALSE);
}
#define HISTORY_NO_DUP_FLAG 0x0001
#endif
+/*
+ * Read input flags
+ */
+#define CONSOLE_READ_KEEPEVENT 0x0001
+#define CONSOLE_READ_CONTINUE 0x0002
+
/*
* Event types
*/
BOOL WINAPI GetNumberOfConsoleInputEvents(_In_ HANDLE, _Out_ PDWORD);
BOOL WINAPI GetNumberOfConsoleMouseButtons(_Out_ PDWORD);
+_Success_(return != 0)
BOOL
WINAPI PeekConsoleInputA(
_In_ HANDLE hConsoleInput,
_In_ DWORD nLength,
_Out_ LPDWORD lpNumberOfEventsRead);
+_Success_(return != 0)
BOOL
WINAPI
PeekConsoleInputW(
_In_ DWORD nLength,
_Out_ _Deref_out_range_(<= , nLength) LPDWORD lpNumberOfEventsRead);
+_Success_(return != 0)
+BOOL
+WINAPI
+ReadConsoleInputExA(
+ _In_ HANDLE hConsoleInput,
+ _Out_writes_to_(nLength, *lpNumberOfEventsRead) PINPUT_RECORD lpBuffer,
+ _In_ DWORD nLength,
+ _Out_ _Deref_out_range_(<= , nLength) LPDWORD lpNumberOfEventsRead,
+ _In_ WORD wFlags);
+
+_Success_(return != 0)
+BOOL
+WINAPI
+ReadConsoleInputExW(
+ _In_ HANDLE hConsoleInput,
+ _Out_writes_to_(nLength, *lpNumberOfEventsRead) PINPUT_RECORD lpBuffer,
+ _In_ DWORD nLength,
+ _Out_ _Deref_out_range_(<= , nLength) LPDWORD lpNumberOfEventsRead,
+ _In_ WORD wFlags);
+
BOOL
WINAPI
ReadConsoleOutputAttribute(
typedef struct
{
HANDLE InputHandle;
- BOOL Unicode;
- BOOL bRead; // TRUE --> Read ; FALSE --> Peek
-
ULONG InputsRead;
-
- ULONG Length;
PINPUT_RECORD InputRecord;
+ ULONG Length;
+ WORD wFlags;
+ BOOLEAN Unicode;
} CONSOLE_GETINPUT, *PCONSOLE_GETINPUT;
typedef struct
NTSTATUS NTAPI
ConDrvGetConsoleInput(IN PCONSOLE Console,
IN PCONSOLE_INPUT_BUFFER InputBuffer,
+ IN BOOLEAN KeepEvents,
IN BOOLEAN WaitForMoreEvents,
IN BOOLEAN Unicode,
OUT PINPUT_RECORD InputRecord,
++i;
CurrentInput = CurrentInput->Flink;
- if (WaitForMoreEvents) // TRUE --> Read, we remove inputs from the buffer ; FALSE --> Peek, we keep inputs.
+ /* Remove the events from the queue if needed */
+ if (!KeepEvents)
{
RemoveEntryList(&Input->ListEntry);
ConsoleFreeHeap(Input);
WaitBeforeReading(IN PGET_INPUT_INFO InputInfo,
IN PCSR_API_MESSAGE ApiMessage,
IN CSR_WAIT_FUNCTION WaitFunction OPTIONAL,
- IN BOOL CreateWaitBlock OPTIONAL)
+ IN BOOLEAN CreateWaitBlock OPTIONAL)
{
if (CreateWaitBlock)
{
static NTSTATUS
ReadChars(IN PGET_INPUT_INFO InputInfo,
IN PCSR_API_MESSAGE ApiMessage,
- IN BOOL CreateWaitBlock OPTIONAL);
+ IN BOOLEAN CreateWaitBlock OPTIONAL);
// Wait function CSR_WAIT_FUNCTION
static BOOLEAN
* If we go there, that means we are notified for some new input.
* The console is therefore already locked.
*/
- Status = ReadChars(InputInfo,
- WaitApiMessage,
- FALSE);
+ Status = ReadChars(InputInfo, WaitApiMessage, FALSE);
Quit:
if (Status != STATUS_PENDING)
static NTSTATUS
ReadChars(IN PGET_INPUT_INFO InputInfo,
IN PCSR_API_MESSAGE ApiMessage,
- IN BOOL CreateWaitBlock OPTIONAL)
+ IN BOOLEAN CreateWaitBlock OPTIONAL)
{
NTSTATUS Status;
PCONSOLE_READCONSOLE ReadConsoleRequest = &((PCONSOLE_API_MESSAGE)ApiMessage)->Data.ReadConsoleRequest;
static NTSTATUS
ReadInputBuffer(IN PGET_INPUT_INFO InputInfo,
- IN BOOL Wait,
IN PCSR_API_MESSAGE ApiMessage,
- IN BOOL CreateWaitBlock OPTIONAL);
+ IN BOOLEAN CreateWaitBlock OPTIONAL);
// Wait function CSR_WAIT_FUNCTION
static BOOLEAN
IN ULONG WaitFlags)
{
NTSTATUS Status;
- PCONSOLE_GETINPUT GetInputRequest = &((PCONSOLE_API_MESSAGE)WaitApiMessage)->Data.GetInputRequest;
PGET_INPUT_INFO InputInfo = (PGET_INPUT_INFO)WaitContext;
PVOID InputHandle = WaitArgument2;
* If we go there, that means we are notified for some new input.
* The console is therefore already locked.
*/
- Status = ReadInputBuffer(InputInfo,
- GetInputRequest->bRead,
- WaitApiMessage,
- FALSE);
+ Status = ReadInputBuffer(InputInfo, WaitApiMessage, FALSE);
Quit:
if (Status != STATUS_PENDING)
NTSTATUS NTAPI
ConDrvGetConsoleInput(IN PCONSOLE Console,
IN PCONSOLE_INPUT_BUFFER InputBuffer,
+ IN BOOLEAN KeepEvents,
IN BOOLEAN WaitForMoreEvents,
IN BOOLEAN Unicode,
OUT PINPUT_RECORD InputRecord,
IN ULONG NumEventsToRead,
- OUT PULONG NumEventsRead);
+ OUT PULONG NumEventsRead OPTIONAL);
static NTSTATUS
ReadInputBuffer(IN PGET_INPUT_INFO InputInfo,
- IN BOOL Wait, // TRUE --> Read ; FALSE --> Peek
IN PCSR_API_MESSAGE ApiMessage,
- IN BOOL CreateWaitBlock OPTIONAL)
+ IN BOOLEAN CreateWaitBlock OPTIONAL)
{
NTSTATUS Status;
PCONSOLE_GETINPUT GetInputRequest = &((PCONSOLE_API_MESSAGE)ApiMessage)->Data.GetInputRequest;
Status = ConDrvGetConsoleInput(InputBuffer->Header.Console,
InputBuffer,
- Wait,
+ (GetInputRequest->wFlags & CONSOLE_READ_KEEPEVENT) != 0,
+ (GetInputRequest->wFlags & CONSOLE_READ_CONTINUE ) == 0,
GetInputRequest->Unicode,
GetInputRequest->InputRecord,
GetInputRequest->Length,
InputInfo.HandleEntry = HandleEntry;
InputInfo.InputBuffer = InputBuffer;
- Status = ReadChars(&InputInfo,
- ApiMessage,
- TRUE);
+ Status = ReadChars(&InputInfo, ApiMessage, TRUE);
ConSrvReleaseInputBuffer(InputBuffer, TRUE);
DPRINT("SrvGetConsoleInput\n");
+ if (GetInputRequest->wFlags & ~(CONSOLE_READ_KEEPEVENT | CONSOLE_READ_CONTINUE))
+ return STATUS_INVALID_PARAMETER;
+
if (!CsrValidateMessageBuffer(ApiMessage,
(PVOID*)&GetInputRequest->InputRecord,
GetInputRequest->Length,
InputInfo.HandleEntry = HandleEntry;
InputInfo.InputBuffer = InputBuffer;
- Status = ReadInputBuffer(&InputInfo,
- GetInputRequest->bRead,
- ApiMessage,
- TRUE);
+ Status = ReadInputBuffer(&InputInfo, ApiMessage, TRUE);
ConSrvReleaseInputBuffer(InputBuffer, TRUE);
static NTSTATUS
DoWriteConsole(IN PCSR_API_MESSAGE ApiMessage,
IN PCSR_THREAD ClientThread,
- IN BOOL CreateWaitBlock OPTIONAL);
+ IN BOOLEAN CreateWaitBlock OPTIONAL);
// Wait function CSR_WAIT_FUNCTION
static BOOLEAN
goto Quit;
}
- Status = DoWriteConsole(WaitApiMessage,
- WaitThread,
- FALSE);
+ Status = DoWriteConsole(WaitApiMessage, WaitThread, FALSE);
Quit:
if (Status != STATUS_PENDING)
static NTSTATUS
DoWriteConsole(IN PCSR_API_MESSAGE ApiMessage,
IN PCSR_THREAD ClientThread,
- IN BOOL CreateWaitBlock OPTIONAL)
+ IN BOOLEAN CreateWaitBlock OPTIONAL)
{
NTSTATUS Status;
PCONSOLE_WRITECONSOLE WriteConsoleRequest = &((PCONSOLE_API_MESSAGE)ApiMessage)->Data.WriteConsoleRequest;
return STATUS_INVALID_PARAMETER;
}
- Status = DoWriteConsole(ApiMessage,
- CsrGetClientThread(),
- TRUE);
+ Status = DoWriteConsole(ApiMessage, CsrGetClientThread(), TRUE);
if (Status == STATUS_PENDING) *ReplyCode = CsrReplyPending;