Add WinNtConsole driver into Nt32Pkg.
git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@2779 6f19259b-4bc3-4df7-8a09-765794883524
This commit is contained in:
klu2
379
Nt32Pkg/WinNtConsoleDxe/ConsoleIn.c
Normal file
379
Nt32Pkg/WinNtConsoleDxe/ConsoleIn.c
Normal file
@@ -0,0 +1,379 @@
|
||||
/*++
|
||||
|
||||
Copyright (c) 2006 - 2007, Intel Corporation
|
||||
All rights reserved. This program and the accompanying materials
|
||||
are licensed and made available under the terms and conditions of the BSD License
|
||||
which accompanies this distribution. The full text of the license may be found at
|
||||
http://opensource.org/licenses/bsd-license.php
|
||||
|
||||
THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
|
||||
WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
|
||||
|
||||
Module Name:
|
||||
|
||||
ConsoleIn.c
|
||||
|
||||
Abstract:
|
||||
|
||||
Console based on Win32 APIs.
|
||||
|
||||
This file attaches a SimpleTextIn protocol to a previously open window.
|
||||
|
||||
The constructor for this protocol depends on an open window. Currently
|
||||
the SimpleTextOut protocol creates a window when it's constructor is called.
|
||||
Thus this code must run after the constructor for the SimpleTextOut
|
||||
protocol
|
||||
|
||||
--*/
|
||||
//
|
||||
// The package level header files this module uses
|
||||
//
|
||||
#include <Uefi.h>
|
||||
#include <WinNtDxe.h>
|
||||
//
|
||||
// The protocols, PPI and GUID defintions for this module
|
||||
//
|
||||
#include <Protocol/SimpleTextIn.h>
|
||||
#include <Protocol/WinNtIo.h>
|
||||
#include <Protocol/SimpleTextOut.h>
|
||||
#include <Protocol/ComponentName.h>
|
||||
#include <Protocol/DriverBinding.h>
|
||||
//
|
||||
// The Library classes this module consumes
|
||||
//
|
||||
#include <Library/DebugLib.h>
|
||||
#include <Library/BaseLib.h>
|
||||
#include <Library/UefiDriverEntryPoint.h>
|
||||
#include <Library/UefiLib.h>
|
||||
#include <Library/BaseMemoryLib.h>
|
||||
#include <Library/UefiBootServicesTableLib.h>
|
||||
#include <Library/MemoryAllocationLib.h>
|
||||
#include "Console.h"
|
||||
|
||||
//
|
||||
// Private worker functions
|
||||
//
|
||||
STATIC
|
||||
EFI_STATUS
|
||||
WinNtSimpleTextInCheckKey (
|
||||
WIN_NT_SIMPLE_TEXT_PRIVATE_DATA *Private
|
||||
);
|
||||
|
||||
EFI_STATUS
|
||||
EFIAPI
|
||||
WinNtSimpleTextInReset (
|
||||
IN EFI_SIMPLE_TEXT_INPUT_PROTOCOL *This,
|
||||
IN BOOLEAN ExtendedVerification
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
TODO: Add function description
|
||||
|
||||
Arguments:
|
||||
|
||||
This - TODO: add argument description
|
||||
ExtendedVerification - TODO: add argument description
|
||||
|
||||
Returns:
|
||||
|
||||
EFI_SUCCESS - TODO: Add description for return value
|
||||
|
||||
--*/
|
||||
{
|
||||
WIN_NT_SIMPLE_TEXT_PRIVATE_DATA *Private;
|
||||
|
||||
Private = WIN_NT_SIMPLE_TEXT_IN_PRIVATE_DATA_FROM_THIS (This);
|
||||
return EFI_SUCCESS;
|
||||
}
|
||||
|
||||
STATIC
|
||||
EFI_STATUS
|
||||
WinNtConvertInputRecordToEfiKey (
|
||||
IN INPUT_RECORD *InputRecord,
|
||||
OUT EFI_INPUT_KEY *Key
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
TODO: Add function description
|
||||
|
||||
Arguments:
|
||||
|
||||
InputRecord - TODO: add argument description
|
||||
Key - TODO: add argument description
|
||||
|
||||
Returns:
|
||||
|
||||
EFI_NOT_READY - TODO: Add description for return value
|
||||
EFI_NOT_READY - TODO: Add description for return value
|
||||
EFI_NOT_READY - TODO: Add description for return value
|
||||
EFI_SUCCESS - TODO: Add description for return value
|
||||
|
||||
--*/
|
||||
{
|
||||
//
|
||||
// Make sure InputRecord is an event that represents a keypress
|
||||
//
|
||||
if (InputRecord->EventType == KEY_EVENT) {
|
||||
if (!InputRecord->Event.KeyEvent.bKeyDown) {
|
||||
return EFI_NOT_READY;
|
||||
}
|
||||
} else {
|
||||
return EFI_NOT_READY;
|
||||
}
|
||||
|
||||
//
|
||||
// Check to see if we should return a scan code in place of Unicode character.
|
||||
//
|
||||
Key->ScanCode = 0;
|
||||
Key->UnicodeChar = 0;
|
||||
if ((InputRecord->Event.KeyEvent.dwControlKeyState & (NUMLOCK_ON | ENHANCED_KEY)) != NUMLOCK_ON) {
|
||||
//
|
||||
// Only check these scan codes if num lock is off.
|
||||
//
|
||||
switch (InputRecord->Event.KeyEvent.wVirtualScanCode) {
|
||||
case 0x48: Key->ScanCode = SCAN_UP; break;
|
||||
case 0x50: Key->ScanCode = SCAN_DOWN; break;
|
||||
case 0x4d: Key->ScanCode = SCAN_RIGHT; break;
|
||||
case 0x4b: Key->ScanCode = SCAN_LEFT; break;
|
||||
case 0x47: Key->ScanCode = SCAN_HOME; break;
|
||||
case 0x4F: Key->ScanCode = SCAN_END; break;
|
||||
case 0x52: Key->ScanCode = SCAN_INSERT; break;
|
||||
case 0x53: Key->ScanCode = SCAN_DELETE; break;
|
||||
case 0x49: Key->ScanCode = SCAN_PAGE_UP; break;
|
||||
case 0x51: Key->ScanCode = SCAN_PAGE_DOWN; break;
|
||||
}
|
||||
}
|
||||
|
||||
switch (InputRecord->Event.KeyEvent.wVirtualScanCode) {
|
||||
case 0x3b: Key->ScanCode = SCAN_F1; break;
|
||||
case 0x3c: Key->ScanCode = SCAN_F2; break;
|
||||
case 0x3d: Key->ScanCode = SCAN_F3; break;
|
||||
case 0x3e: Key->ScanCode = SCAN_F4; break;
|
||||
case 0x3f: Key->ScanCode = SCAN_F5; break;
|
||||
case 0x40: Key->ScanCode = SCAN_F6; break;
|
||||
case 0x41: Key->ScanCode = SCAN_F7; break;
|
||||
case 0x42: Key->ScanCode = SCAN_F8; break;
|
||||
case 0x43: Key->ScanCode = SCAN_F9; break;
|
||||
case 0x44: Key->ScanCode = SCAN_F10; break;
|
||||
case 0x01: Key->ScanCode = SCAN_ESC; break;
|
||||
}
|
||||
|
||||
//
|
||||
// If there's a scan code pass it, and don't pass the char code
|
||||
//
|
||||
if (Key->ScanCode == 0) {
|
||||
Key->UnicodeChar = InputRecord->Event.KeyEvent.uChar.UnicodeChar;
|
||||
if (Key->UnicodeChar == 0) {
|
||||
return EFI_NOT_READY;
|
||||
}
|
||||
}
|
||||
|
||||
return EFI_SUCCESS;
|
||||
}
|
||||
|
||||
STATIC
|
||||
EFI_STATUS
|
||||
EFIAPI
|
||||
WinNtSimpleTextInReadKeyStroke (
|
||||
IN EFI_SIMPLE_TEXT_INPUT_PROTOCOL *This,
|
||||
OUT EFI_INPUT_KEY *Key
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
TODO: Add function description
|
||||
|
||||
Arguments:
|
||||
|
||||
This - TODO: add argument description
|
||||
Key - TODO: add argument description
|
||||
|
||||
Returns:
|
||||
|
||||
EFI_DEVICE_ERROR - TODO: Add description for return value
|
||||
EFI_NOT_READY - TODO: Add description for return value
|
||||
|
||||
--*/
|
||||
{
|
||||
EFI_STATUS Status;
|
||||
WIN_NT_SIMPLE_TEXT_PRIVATE_DATA *Private;
|
||||
INPUT_RECORD InputRecord;
|
||||
DWORD NtEventCount;
|
||||
|
||||
Private = WIN_NT_SIMPLE_TEXT_IN_PRIVATE_DATA_FROM_THIS (This);
|
||||
|
||||
Status = WinNtSimpleTextInCheckKey (Private);
|
||||
if (EFI_ERROR (Status)) {
|
||||
return Status;
|
||||
}
|
||||
|
||||
do {
|
||||
|
||||
if (!Private->WinNtThunk->ReadConsoleInput (Private->NtInHandle, &InputRecord, 1, &NtEventCount)) {
|
||||
return EFI_DEVICE_ERROR;
|
||||
}
|
||||
|
||||
if (NtEventCount == 0) {
|
||||
return EFI_NOT_READY;
|
||||
}
|
||||
|
||||
//
|
||||
// Convert the Input Record to an EFI Keystroke.
|
||||
//
|
||||
Status = WinNtConvertInputRecordToEfiKey (&InputRecord, Key);
|
||||
} while (EFI_ERROR (Status));
|
||||
|
||||
return Status;
|
||||
}
|
||||
|
||||
STATIC
|
||||
VOID
|
||||
EFIAPI
|
||||
WinNtSimpleTextInWaitForKey (
|
||||
IN EFI_EVENT Event,
|
||||
IN VOID *Context
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
TODO: Add function description
|
||||
|
||||
Arguments:
|
||||
|
||||
Event - TODO: add argument description
|
||||
Context - TODO: add argument description
|
||||
|
||||
Returns:
|
||||
|
||||
TODO: add return values
|
||||
|
||||
--*/
|
||||
{
|
||||
WIN_NT_SIMPLE_TEXT_PRIVATE_DATA *Private;
|
||||
EFI_STATUS Status;
|
||||
|
||||
Private = (WIN_NT_SIMPLE_TEXT_PRIVATE_DATA *) Context;
|
||||
Status = WinNtSimpleTextInCheckKey (Private);
|
||||
if (!EFI_ERROR (Status)) {
|
||||
gBS->SignalEvent (Event);
|
||||
}
|
||||
}
|
||||
|
||||
STATIC
|
||||
EFI_STATUS
|
||||
WinNtSimpleTextInCheckKey (
|
||||
WIN_NT_SIMPLE_TEXT_PRIVATE_DATA *Private
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
TODO: Add function description
|
||||
|
||||
Arguments:
|
||||
|
||||
Private - TODO: add argument description
|
||||
|
||||
Returns:
|
||||
|
||||
TODO: add return values
|
||||
|
||||
--*/
|
||||
{
|
||||
INPUT_RECORD *InputRecord;
|
||||
DWORD NtEventCount;
|
||||
DWORD ActualNtEventCount;
|
||||
EFI_STATUS Status;
|
||||
BOOLEAN Success;
|
||||
UINTN Index;
|
||||
EFI_INPUT_KEY Key;
|
||||
|
||||
InputRecord = NULL;
|
||||
NtEventCount = 0;
|
||||
Private->WinNtThunk->GetNumberOfConsoleInputEvents (Private->NtInHandle, &NtEventCount);
|
||||
if (NtEventCount == 0) {
|
||||
Status = EFI_NOT_READY;
|
||||
goto Done;
|
||||
}
|
||||
|
||||
InputRecord = AllocatePool (sizeof (INPUT_RECORD) * NtEventCount);
|
||||
if (InputRecord == NULL) {
|
||||
Status = EFI_NOT_READY;
|
||||
goto Done;
|
||||
}
|
||||
|
||||
Success = (BOOLEAN) Private->WinNtThunk->PeekConsoleInput (
|
||||
Private->NtInHandle,
|
||||
InputRecord,
|
||||
NtEventCount,
|
||||
&ActualNtEventCount
|
||||
);
|
||||
if (!Success) {
|
||||
Status = EFI_NOT_READY;
|
||||
goto Done;
|
||||
}
|
||||
|
||||
Status = EFI_NOT_READY;
|
||||
for (Index = 0; Index < (UINTN) ActualNtEventCount; Index++) {
|
||||
//
|
||||
// Convert the Input Record to an EFI Keystroke.
|
||||
//
|
||||
Status = WinNtConvertInputRecordToEfiKey (&InputRecord[Index], &Key);
|
||||
if (!EFI_ERROR (Status)) {
|
||||
Status = EFI_SUCCESS;
|
||||
goto Done;
|
||||
}
|
||||
}
|
||||
|
||||
Done:
|
||||
if (InputRecord != NULL) {
|
||||
FreePool (InputRecord);
|
||||
}
|
||||
|
||||
return Status;
|
||||
}
|
||||
|
||||
EFI_STATUS
|
||||
WinNtSimpleTextInAttachToWindow (
|
||||
IN WIN_NT_SIMPLE_TEXT_PRIVATE_DATA *Private
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
TODO: Add function description
|
||||
|
||||
Arguments:
|
||||
|
||||
Private - TODO: add argument description
|
||||
|
||||
Returns:
|
||||
|
||||
TODO: add return values
|
||||
|
||||
--*/
|
||||
{
|
||||
EFI_STATUS Status;
|
||||
|
||||
Private->NtInHandle = Private->WinNtThunk->GetStdHandle (STD_INPUT_HANDLE);
|
||||
|
||||
Private->SimpleTextIn.Reset = WinNtSimpleTextInReset;
|
||||
Private->SimpleTextIn.ReadKeyStroke = WinNtSimpleTextInReadKeyStroke;
|
||||
|
||||
Status = gBS->CreateEvent (
|
||||
EVT_NOTIFY_WAIT,
|
||||
TPL_NOTIFY,
|
||||
WinNtSimpleTextInWaitForKey,
|
||||
Private,
|
||||
&Private->SimpleTextIn.WaitForKey
|
||||
);
|
||||
ASSERT_EFI_ERROR (Status);
|
||||
|
||||
return Status;
|
||||
}
|
||||
Reference in New Issue
Block a user