mirror of https://github.com/acidanthera/audk.git
358 lines
8.5 KiB
C
358 lines
8.5 KiB
C
/*++
|
|
|
|
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
|
|
|
|
--*/
|
|
|
|
#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_IN_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_IN_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;
|
|
}
|