Something went wrong. Try again.
Reactos
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768/* * ReactOS kernel * Copyright (C) 2002 ReactOS Team * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License along * with this program; if not, write to the Free Software Foundation, Inc., * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. *//* * COPYRIGHT: See COPYING in the top level directory * PROJECT: ReactOS text-mode setup * FILE: base/setup/usetup/console.c * PURPOSE: Console support functions * PROGRAMMER: */
/* INCLUDES ******************************************************************/
#include <usetup.h>/* Blue Driver Header */#include <blue/ntddblue.h>#include "keytrans.h"
#define NDEBUG#include <debug.h>
/* DATA **********************************************************************/
static BOOLEAN InputQueueEmpty;static BOOLEAN WaitForInput;static KEYBOARD_INPUT_DATA InputDataQueue; // Only one element!static IO_STATUS_BLOCK InputIosb;static UINT LastLoadedCodepage;
/* FUNCTIONS *****************************************************************/
typedef struct _CONSOLE_CABINET_CONTEXT{ CABINET_CONTEXT CabinetContext; PVOID Data; ULONG Size;} CONSOLE_CABINET_CONTEXT, *PCONSOLE_CABINET_CONTEXT;
static PVOIDConsoleCreateFileHandler( IN PCABINET_CONTEXT CabinetContext, IN ULONG FileSize){ PCONSOLE_CABINET_CONTEXT ConsoleCabinetContext;
ConsoleCabinetContext = (PCONSOLE_CABINET_CONTEXT)CabinetContext; ConsoleCabinetContext->Data = RtlAllocateHeap(ProcessHeap, 0, FileSize); if (!ConsoleCabinetContext->Data) { DPRINT("Failed to allocate %d bytes\n", FileSize); return NULL; } ConsoleCabinetContext->Size = FileSize; return ConsoleCabinetContext->Data;}
BOOLWINAPIAllocConsole(VOID){ NTSTATUS Status; UNICODE_STRING ScreenName = RTL_CONSTANT_STRING(L"\\??\\BlueScreen"); UNICODE_STRING KeyboardName = RTL_CONSTANT_STRING(L"\\Device\\KeyboardClass0"); OBJECT_ATTRIBUTES ObjectAttributes; IO_STATUS_BLOCK IoStatusBlock; ULONG Enable;
/* Open the screen */ InitializeObjectAttributes(&ObjectAttributes, &ScreenName, 0, NULL, NULL); Status = NtOpenFile(&StdOutput, FILE_ALL_ACCESS, &ObjectAttributes, &IoStatusBlock, FILE_OPEN, FILE_SYNCHRONOUS_IO_ALERT); if (!NT_SUCCESS(Status)) return FALSE;
/* Enable it */ Enable = TRUE; Status = NtDeviceIoControlFile(StdOutput, NULL, NULL, NULL, &IoStatusBlock, IOCTL_CONSOLE_RESET_SCREEN, &Enable, sizeof(Enable), NULL, 0); if (!NT_SUCCESS(Status)) { NtClose(StdOutput); return FALSE; }
/* Default to en-US output codepage */ SetConsoleOutputCP(437);
/* Open the keyboard */ InitializeObjectAttributes(&ObjectAttributes, &KeyboardName, 0, NULL, NULL); Status = NtOpenFile(&StdInput, FILE_ALL_ACCESS, &ObjectAttributes, &IoStatusBlock, FILE_OPEN, 0); if (!NT_SUCCESS(Status)) { NtClose(StdOutput); return FALSE; }
/* Reset the queue state */ InputQueueEmpty = TRUE; WaitForInput = FALSE;
return TRUE;}
BOOLWINAPIAttachConsole( IN DWORD dwProcessId){ return FALSE;}
BOOLWINAPIFreeConsole(VOID){ /* Reset the queue state */ InputQueueEmpty = TRUE; WaitForInput = FALSE;
if (StdInput != INVALID_HANDLE_VALUE) NtClose(StdInput);
if (StdOutput != INVALID_HANDLE_VALUE) NtClose(StdOutput);
return TRUE;}
BOOLWINAPIWriteConsole( IN HANDLE hConsoleOutput, IN const VOID *lpBuffer, IN DWORD nNumberOfCharsToWrite, OUT LPDWORD lpNumberOfCharsWritten, IN LPVOID lpReserved){ IO_STATUS_BLOCK IoStatusBlock; NTSTATUS Status;
Status = NtWriteFile(hConsoleOutput, NULL, NULL, NULL, &IoStatusBlock, (PVOID)lpBuffer, nNumberOfCharsToWrite, NULL, NULL); if (!NT_SUCCESS(Status)) return FALSE;
*lpNumberOfCharsWritten = IoStatusBlock.Information; return TRUE;}
HANDLEWINAPIGetStdHandle( IN DWORD nStdHandle){ switch (nStdHandle) { case STD_INPUT_HANDLE: return StdInput; case STD_OUTPUT_HANDLE: return StdOutput; default: return INVALID_HANDLE_VALUE; }}
BOOLWINAPIFlushConsoleInputBuffer( IN HANDLE hConsoleInput){ NTSTATUS Status; LARGE_INTEGER Offset, Timeout; IO_STATUS_BLOCK IoStatusBlock; KEYBOARD_INPUT_DATA InputData;
/* Cancel any pending read */ if (WaitForInput) NtCancelIoFile(hConsoleInput, &IoStatusBlock);
/* Reset the queue state */ InputQueueEmpty = TRUE; WaitForInput = FALSE;
/* Flush the keyboard buffer */ do { Offset.QuadPart = 0; Status = NtReadFile(hConsoleInput, NULL, NULL, NULL, &IoStatusBlock, &InputData, sizeof(InputData), &Offset, NULL); if (Status == STATUS_PENDING) { Timeout.QuadPart = -100; Status = NtWaitForSingleObject(hConsoleInput, FALSE, &Timeout); if (Status == STATUS_TIMEOUT) { NtCancelIoFile(hConsoleInput, &IoStatusBlock); return TRUE; } } } while (NT_SUCCESS(Status)); return FALSE;}
BOOLWINAPIPeekConsoleInput( IN HANDLE hConsoleInput, OUT PINPUT_RECORD lpBuffer, IN DWORD nLength, OUT LPDWORD lpNumberOfEventsRead){ NTSTATUS Status; LARGE_INTEGER Offset, Timeout; KEYBOARD_INPUT_DATA InputData;
if (InputQueueEmpty) { /* Read the keyboard for an event, without waiting */ if (!WaitForInput) { Offset.QuadPart = 0; Status = NtReadFile(hConsoleInput, NULL, NULL, NULL, &InputIosb, &InputDataQueue, sizeof(InputDataQueue), &Offset, NULL); if (!NT_SUCCESS(Status)) return FALSE; if (Status == STATUS_PENDING) { /* No input yet, we will have to wait next time */ *lpNumberOfEventsRead = 0; WaitForInput = TRUE; return TRUE; } } else { /* * We already tried to read from the keyboard and are * waiting for data, check whether something showed up. */ Timeout.QuadPart = -100; // Wait just a little bit. Status = NtWaitForSingleObject(hConsoleInput, FALSE, &Timeout); if (Status == STATUS_TIMEOUT) { /* Nothing yet, continue waiting next time */ *lpNumberOfEventsRead = 0; WaitForInput = TRUE; return TRUE; } WaitForInput = FALSE; if (!NT_SUCCESS(Status)) return FALSE; }
/* We got something in the queue */ InputQueueEmpty = FALSE; WaitForInput = FALSE; }
/* Fetch from the queue but keep it inside */ InputData = InputDataQueue;
lpBuffer->EventType = KEY_EVENT; Status = IntTranslateKey(hConsoleInput, &InputData, &lpBuffer->Event.KeyEvent); if (!NT_SUCCESS(Status)) return FALSE;
*lpNumberOfEventsRead = 1; return TRUE;}
BOOLWINAPIReadConsoleInput( IN HANDLE hConsoleInput, OUT PINPUT_RECORD lpBuffer, IN DWORD nLength, OUT LPDWORD lpNumberOfEventsRead){ NTSTATUS Status; LARGE_INTEGER Offset; KEYBOARD_INPUT_DATA InputData;
if (InputQueueEmpty) { /* Read the keyboard and wait for an event, skipping the queue */ if (!WaitForInput) { Offset.QuadPart = 0; Status = NtReadFile(hConsoleInput, NULL, NULL, NULL, &InputIosb, &InputDataQueue, sizeof(InputDataQueue), &Offset, NULL); if (Status == STATUS_PENDING) { /* Block and wait for input */ WaitForInput = TRUE; Status = NtWaitForSingleObject(hConsoleInput, FALSE, NULL); WaitForInput = FALSE; Status = InputIosb.Status; } if (!NT_SUCCESS(Status)) return FALSE; } else { /* * We already tried to read from the keyboard and are * waiting for data, block and wait for input. */ Status = NtWaitForSingleObject(hConsoleInput, FALSE, NULL); WaitForInput = FALSE; Status = InputIosb.Status; if (!NT_SUCCESS(Status)) return FALSE; } }
/* Fetch from the queue and empty it */ InputData = InputDataQueue; InputQueueEmpty = TRUE;
lpBuffer->EventType = KEY_EVENT; Status = IntTranslateKey(hConsoleInput, &InputData, &lpBuffer->Event.KeyEvent); if (!NT_SUCCESS(Status)) return FALSE;
*lpNumberOfEventsRead = 1; return TRUE;}
BOOLWINAPIWriteConsoleOutputCharacterA( HANDLE hConsoleOutput, IN LPCSTR lpCharacter, IN DWORD nLength, IN COORD dwWriteCoord, OUT LPDWORD lpNumberOfCharsWritten){ IO_STATUS_BLOCK IoStatusBlock; PCHAR Buffer; COORD *pCoord; PCHAR pText; NTSTATUS Status;
Buffer = (CHAR*)RtlAllocateHeap(ProcessHeap, 0, nLength + sizeof(COORD)); pCoord = (COORD *)Buffer; pText = (PCHAR)(pCoord + 1);
*pCoord = dwWriteCoord; memcpy(pText, lpCharacter, nLength);
Status = NtDeviceIoControlFile(hConsoleOutput, NULL, NULL, NULL, &IoStatusBlock, IOCTL_CONSOLE_WRITE_OUTPUT_CHARACTER, NULL, 0, Buffer, nLength + sizeof(COORD));
RtlFreeHeap(ProcessHeap, 0, Buffer); if (!NT_SUCCESS(Status)) return FALSE;
*lpNumberOfCharsWritten = IoStatusBlock.Information; return TRUE;}
BOOLWINAPIWriteConsoleOutputCharacterW( HANDLE hConsoleOutput, IN LPCWSTR lpCharacter, IN DWORD nLength, IN COORD dwWriteCoord, OUT LPDWORD lpNumberOfCharsWritten){ IO_STATUS_BLOCK IoStatusBlock; PCHAR Buffer; COORD *pCoord; PCHAR pText; NTSTATUS Status;// ULONG i;
UNICODE_STRING UnicodeString; OEM_STRING OemString; ULONG OemLength;
UnicodeString.Length = nLength * sizeof(WCHAR); UnicodeString.MaximumLength = nLength * sizeof(WCHAR); UnicodeString.Buffer = (PWSTR)lpCharacter;
OemLength = RtlUnicodeStringToOemSize(&UnicodeString);
Buffer = (CHAR*)RtlAllocateHeap(ProcessHeap, 0, OemLength + sizeof(COORD));// nLength + sizeof(COORD)); if (Buffer== NULL) return FALSE;
pCoord = (COORD *)Buffer; pText = (PCHAR)(pCoord + 1);
*pCoord = dwWriteCoord;
OemString.Length = 0; OemString.MaximumLength = OemLength; OemString.Buffer = pText;
Status = RtlUnicodeStringToOemString(&OemString, &UnicodeString, FALSE); if (!NT_SUCCESS(Status)) goto done;
/* FIXME: use real unicode->oem conversion */// for (i = 0; i < nLength; i++)// pText[i] = (CHAR)lpCharacter[i];
Status = NtDeviceIoControlFile(hConsoleOutput, NULL, NULL, NULL, &IoStatusBlock, IOCTL_CONSOLE_WRITE_OUTPUT_CHARACTER, NULL, 0, Buffer, nLength + sizeof(COORD));
done: RtlFreeHeap(ProcessHeap, 0, Buffer); if (!NT_SUCCESS(Status)) return FALSE;
*lpNumberOfCharsWritten = IoStatusBlock.Information; return TRUE;}
BOOLWINAPIFillConsoleOutputAttribute( IN HANDLE hConsoleOutput, IN WORD wAttribute, IN DWORD nLength, IN COORD dwWriteCoord, OUT LPDWORD lpNumberOfAttrsWritten){ IO_STATUS_BLOCK IoStatusBlock; OUTPUT_ATTRIBUTE Buffer; NTSTATUS Status;
Buffer.wAttribute = wAttribute; Buffer.nLength = nLength; Buffer.dwCoord = dwWriteCoord;
Status = NtDeviceIoControlFile(hConsoleOutput, NULL, NULL, NULL, &IoStatusBlock, IOCTL_CONSOLE_FILL_OUTPUT_ATTRIBUTE, &Buffer, sizeof(OUTPUT_ATTRIBUTE), &Buffer, sizeof(OUTPUT_ATTRIBUTE)); if (!NT_SUCCESS(Status)) return FALSE;
*lpNumberOfAttrsWritten = Buffer.dwTransfered; return TRUE;}
BOOLWINAPIFillConsoleOutputCharacterA( IN HANDLE hConsoleOutput, IN CHAR cCharacter, IN DWORD nLength, IN COORD dwWriteCoord, OUT LPDWORD lpNumberOfCharsWritten){ IO_STATUS_BLOCK IoStatusBlock; OUTPUT_CHARACTER Buffer; NTSTATUS Status;
Buffer.cCharacter = cCharacter; Buffer.nLength = nLength; Buffer.dwCoord = dwWriteCoord;
Status = NtDeviceIoControlFile(hConsoleOutput, NULL, NULL, NULL, &IoStatusBlock, IOCTL_CONSOLE_FILL_OUTPUT_CHARACTER, &Buffer, sizeof(OUTPUT_CHARACTER), &Buffer, sizeof(OUTPUT_CHARACTER)); if (!NT_SUCCESS(Status)) return FALSE;
*lpNumberOfCharsWritten = Buffer.dwTransfered; return TRUE;}
BOOLWINAPIGetConsoleScreenBufferInfo( IN HANDLE hConsoleOutput, OUT PCONSOLE_SCREEN_BUFFER_INFO lpConsoleScreenBufferInfo){ IO_STATUS_BLOCK IoStatusBlock; NTSTATUS Status;
Status = NtDeviceIoControlFile(hConsoleOutput, NULL, NULL, NULL, &IoStatusBlock, IOCTL_CONSOLE_GET_SCREEN_BUFFER_INFO, NULL, 0, lpConsoleScreenBufferInfo, sizeof(CONSOLE_SCREEN_BUFFER_INFO)); return NT_SUCCESS(Status);}
BOOLWINAPISetConsoleCursorInfo( IN HANDLE hConsoleOutput, IN const CONSOLE_CURSOR_INFO *lpConsoleCursorInfo){ IO_STATUS_BLOCK IoStatusBlock; NTSTATUS Status;
Status = NtDeviceIoControlFile(hConsoleOutput, NULL, NULL, NULL, &IoStatusBlock, IOCTL_CONSOLE_SET_CURSOR_INFO, (PCONSOLE_CURSOR_INFO)lpConsoleCursorInfo, sizeof(CONSOLE_CURSOR_INFO), NULL, 0); return NT_SUCCESS(Status);}
BOOLWINAPISetConsoleCursorPosition( IN HANDLE hConsoleOutput, IN COORD dwCursorPosition){ CONSOLE_SCREEN_BUFFER_INFO ConsoleScreenBufferInfo; IO_STATUS_BLOCK IoStatusBlock; NTSTATUS Status;
Status = GetConsoleScreenBufferInfo(hConsoleOutput, &ConsoleScreenBufferInfo); if (!NT_SUCCESS(Status)) return FALSE;
ConsoleScreenBufferInfo.dwCursorPosition.X = dwCursorPosition.X; ConsoleScreenBufferInfo.dwCursorPosition.Y = dwCursorPosition.Y;
Status = NtDeviceIoControlFile(hConsoleOutput, NULL, NULL, NULL, &IoStatusBlock, IOCTL_CONSOLE_SET_SCREEN_BUFFER_INFO, &ConsoleScreenBufferInfo, sizeof(CONSOLE_SCREEN_BUFFER_INFO), NULL, 0); return NT_SUCCESS(Status);}
BOOLWINAPISetConsoleTextAttribute( IN HANDLE hConsoleOutput, IN WORD wAttributes){ IO_STATUS_BLOCK IoStatusBlock; NTSTATUS Status;
Status = NtDeviceIoControlFile(hConsoleOutput, NULL, NULL, NULL, &IoStatusBlock, IOCTL_CONSOLE_SET_TEXT_ATTRIBUTE, &wAttributes, sizeof(USHORT), NULL, 0); return NT_SUCCESS(Status);}
BOOLWINAPISetConsoleOutputCP( IN UINT wCodepage){ static PCWSTR FontFile = L"\\SystemRoot\\vgafonts.cab"; WCHAR FontName[20]; CONSOLE_CABINET_CONTEXT ConsoleCabinetContext; PCABINET_CONTEXT CabinetContext = &ConsoleCabinetContext.CabinetContext; CAB_SEARCH Search; ULONG CabStatus; HANDLE hConsoleOutput; IO_STATUS_BLOCK IoStatusBlock; NTSTATUS Status;
if (wCodepage == LastLoadedCodepage) return TRUE;
hConsoleOutput = GetStdHandle(STD_OUTPUT_HANDLE);
CabinetInitialize(CabinetContext); CabinetSetEventHandlers(CabinetContext, NULL, NULL, NULL, ConsoleCreateFileHandler); CabinetSetCabinetName(CabinetContext, FontFile);
CabStatus = CabinetOpen(CabinetContext); if (CabStatus != CAB_STATUS_SUCCESS) { DPRINT("CabinetOpen('%S') returned 0x%08x\n", FontFile, CabStatus); return FALSE; }
RtlStringCbPrintfW(FontName, sizeof(FontName), L"%u-8x8.bin", wCodepage); CabStatus = CabinetFindFirst(CabinetContext, FontName, &Search); if (CabStatus != CAB_STATUS_SUCCESS) { DPRINT("CabinetFindFirst('%S', '%S') returned 0x%08x\n", FontFile, FontName, CabStatus); CabinetClose(CabinetContext); return FALSE; }
CabStatus = CabinetExtractFile(CabinetContext, &Search); CabinetClose(CabinetContext); if (CabStatus != CAB_STATUS_SUCCESS) { DPRINT("CabinetExtractFile('%S', '%S') returned 0x%08x\n", FontFile, FontName, CabStatus); if (ConsoleCabinetContext.Data) RtlFreeHeap(ProcessHeap, 0, ConsoleCabinetContext.Data); return FALSE; } ASSERT(ConsoleCabinetContext.Data);
Status = NtDeviceIoControlFile(hConsoleOutput, NULL, NULL, NULL, &IoStatusBlock, IOCTL_CONSOLE_LOADFONT, ConsoleCabinetContext.Data, ConsoleCabinetContext.Size, NULL, 0);
RtlFreeHeap(ProcessHeap, 0, ConsoleCabinetContext.Data);
if (!NT_SUCCESS(Status)) return FALSE;
LastLoadedCodepage = wCodepage; return TRUE;}
/* EOF */