tyler.durden
/
audk
mirror of https://github.com/acidanthera/audk.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
audk/ShellPkg/Application/Shell/ConsoleLogger.c

1244 lines
40 KiB
C
Raw Blame History

/** @file
Provides interface to shell console logger.
(C) Copyright 2013 Hewlett-Packard Development Company, L.P.<BR>
Copyright (c) 2009 - 2018, Intel Corporation. All rights reserved.<BR>
(C) Copyright 2016 Hewlett-Packard Development Company, L.P.<BR>
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.
**/
#include "Shell.h"
/**
Install our intermediate ConOut into the system table to
keep a log of all the info that is displayed to the user.
@param[in] ScreensToSave Sets how many screen-worths of data to save.
@param[out] ConsoleInfo The object to pass into later functions.
@retval EFI_SUCCESS The operation was successful.
@return other The operation failed.
@sa ConsoleLoggerResetBuffers
@sa InstallProtocolInterface
**/
EFI_STATUS
ConsoleLoggerInstall(
IN CONST UINTN ScreensToSave,
OUT CONSOLE_LOGGER_PRIVATE_DATA **ConsoleInfo
)
{
EFI_STATUS Status;
ASSERT(ConsoleInfo != NULL);
(*ConsoleInfo) = AllocateZeroPool(sizeof(CONSOLE_LOGGER_PRIVATE_DATA));
if ((*ConsoleInfo) == NULL) {
return (EFI_OUT_OF_RESOURCES);
}
(*ConsoleInfo)->Signature = CONSOLE_LOGGER_PRIVATE_DATA_SIGNATURE;
(*ConsoleInfo)->OldConOut = gST->ConOut;
(*ConsoleInfo)->OldConHandle = gST->ConsoleOutHandle;
(*ConsoleInfo)->Buffer = NULL;
(*ConsoleInfo)->BufferSize = 0;
(*ConsoleInfo)->OriginalStartRow = 0;
(*ConsoleInfo)->CurrentStartRow = 0;
(*ConsoleInfo)->RowsPerScreen = 0;
(*ConsoleInfo)->ColsPerScreen = 0;
(*ConsoleInfo)->Attributes = NULL;
(*ConsoleInfo)->AttribSize = 0;
(*ConsoleInfo)->ScreenCount = ScreensToSave;
(*ConsoleInfo)->HistoryMode.MaxMode = 1;
(*ConsoleInfo)->HistoryMode.Mode = 0;
(*ConsoleInfo)->HistoryMode.Attribute = 0;
(*ConsoleInfo)->HistoryMode.CursorColumn = 0;
(*ConsoleInfo)->HistoryMode.CursorRow = 0;
(*ConsoleInfo)->HistoryMode.CursorVisible = FALSE;
(*ConsoleInfo)->OurConOut.Reset = ConsoleLoggerReset;
(*ConsoleInfo)->OurConOut.OutputString = ConsoleLoggerOutputString;
(*ConsoleInfo)->OurConOut.TestString = ConsoleLoggerTestString;
(*ConsoleInfo)->OurConOut.QueryMode = ConsoleLoggerQueryMode;
(*ConsoleInfo)->OurConOut.SetMode = ConsoleLoggerSetMode;
(*ConsoleInfo)->OurConOut.SetAttribute = ConsoleLoggerSetAttribute;
(*ConsoleInfo)->OurConOut.ClearScreen = ConsoleLoggerClearScreen;
(*ConsoleInfo)->OurConOut.SetCursorPosition = ConsoleLoggerSetCursorPosition;
(*ConsoleInfo)->OurConOut.EnableCursor = ConsoleLoggerEnableCursor;
(*ConsoleInfo)->OurConOut.Mode = gST->ConOut->Mode;
(*ConsoleInfo)->Enabled = TRUE;
Status = ConsoleLoggerResetBuffers(*ConsoleInfo);
if (EFI_ERROR(Status)) {
SHELL_FREE_NON_NULL((*ConsoleInfo));
*ConsoleInfo = NULL;
return (Status);
}
Status = gBS->InstallProtocolInterface(&gImageHandle, &gEfiSimpleTextOutProtocolGuid, EFI_NATIVE_INTERFACE, (VOID*)&((*ConsoleInfo)->OurConOut));
if (EFI_ERROR(Status)) {
SHELL_FREE_NON_NULL((*ConsoleInfo)->Buffer);
SHELL_FREE_NON_NULL((*ConsoleInfo)->Attributes);
SHELL_FREE_NON_NULL((*ConsoleInfo));
*ConsoleInfo = NULL;
return (Status);
}
gST->ConsoleOutHandle = gImageHandle;
gST->ConOut = &(*ConsoleInfo)->OurConOut;
//
// Update the CRC32 in the EFI System Table header
//
gST->Hdr.CRC32 = 0;
gBS->CalculateCrc32 (
(UINT8 *)&gST->Hdr,
gST->Hdr.HeaderSize,
&gST->Hdr.CRC32
);
return (Status);
}
/**
Return the system to the state it was before InstallConsoleLogger
was installed.
@param[in] ConsoleInfo The object from the install function.
@retval EFI_SUCCESS The operation was successful
@return other The operation failed. This was from UninstallProtocolInterface.
**/
EFI_STATUS
ConsoleLoggerUninstall(
IN CONSOLE_LOGGER_PRIVATE_DATA *ConsoleInfo
)
{
ASSERT(ConsoleInfo != NULL);
ASSERT(ConsoleInfo->OldConOut != NULL);
if (ConsoleInfo->Buffer != NULL) {
FreePool(ConsoleInfo->Buffer);
DEBUG_CODE(ConsoleInfo->Buffer = NULL;);
DEBUG_CODE(ConsoleInfo->BufferSize = 0;);
}
if (ConsoleInfo->Attributes != NULL) {
FreePool(ConsoleInfo->Attributes);
DEBUG_CODE(ConsoleInfo->Attributes = NULL;);
DEBUG_CODE(ConsoleInfo->AttribSize = 0;);
}
gST->ConsoleOutHandle = ConsoleInfo->OldConHandle;
gST->ConOut = ConsoleInfo->OldConOut;
//
// Update the CRC32 in the EFI System Table header
//
gST->Hdr.CRC32 = 0;
gBS->CalculateCrc32 (
(UINT8 *)&gST->Hdr,
gST->Hdr.HeaderSize,
&gST->Hdr.CRC32
);
return (gBS->UninstallProtocolInterface(gImageHandle, &gEfiSimpleTextOutProtocolGuid, (VOID*)&ConsoleInfo->OurConOut));
}
/**
Displays previously logged output back to the screen.
This will scroll the screen forwards and backwards through the log of previous
output. If Rows is 0 then the size of 1/2 the screen will be scrolled. If Rows
is (UINTN)(-1) then the size of the screen will be scrolled.
@param[in] Forward If TRUE then the log will be displayed forwards (scroll to newer).
If FALSE then the log will be displayed backwards (scroll to older).
@param[in] Rows Determines how many rows the log should scroll.
@param[in] ConsoleInfo The pointer to the instance of the console logger information.
**/
EFI_STATUS
ConsoleLoggerDisplayHistory(
IN CONST BOOLEAN Forward,
IN CONST UINTN Rows,
IN CONSOLE_LOGGER_PRIVATE_DATA *ConsoleInfo
)
{
UINTN RowChange;
ASSERT(ConsoleInfo != NULL);
//
// Calculate the row number change
//
switch (Rows) {
case ((UINTN)(-1)):
RowChange = ConsoleInfo->RowsPerScreen;
break;
case (0):
RowChange = ConsoleInfo->RowsPerScreen / 2;
break;
default:
RowChange = Rows;
break;
}
//
// Do the math for direction
//
if (Forward) {
if ((ConsoleInfo->OriginalStartRow - ConsoleInfo->CurrentStartRow) < RowChange) {
RowChange = ConsoleInfo->OriginalStartRow - ConsoleInfo->CurrentStartRow;
}
} else {
if (ConsoleInfo->CurrentStartRow < RowChange) {
RowChange = ConsoleInfo->CurrentStartRow;
}
}
//
// If we are already at one end or the other
//
if (RowChange == 0) {
return (EFI_SUCCESS);
}
//
// Clear the screen
//
ConsoleInfo->OldConOut->ClearScreen(ConsoleInfo->OldConOut);
//
// Set the new start row
//
if (Forward) {
ConsoleInfo->CurrentStartRow += RowChange;
} else {
ConsoleInfo->CurrentStartRow -= RowChange;
}
//
// Change the screen
//
return (UpdateDisplayFromHistory(ConsoleInfo));
}
/**
Function to return to normal output whent he scrolling is complete.
@param[in] ConsoleInfo The pointer to the instance of the console logger information.
@retval EFI_SUCCESS The operation was successful.
@return other The operation failed. See UpdateDisplayFromHistory.
@sa UpdateDisplayFromHistory
**/
EFI_STATUS
ConsoleLoggerStopHistory(
IN CONSOLE_LOGGER_PRIVATE_DATA *ConsoleInfo
)
{
ASSERT(ConsoleInfo != NULL);
if (ConsoleInfo->CurrentStartRow == ConsoleInfo->OriginalStartRow) {
return (EFI_SUCCESS);
}
//
// Clear the screen
//
ConsoleInfo->OldConOut->ClearScreen(ConsoleInfo->OldConOut);
ConsoleInfo->CurrentStartRow = ConsoleInfo->OriginalStartRow;
return (UpdateDisplayFromHistory(ConsoleInfo));
}
/**
Updates the hidden ConOut to be displaying the correct stuff.
@param[in] ConsoleInfo The pointer to the instance of the console logger information.
@retval EFI_SUCCESS The operation was successful.
@return other The operation failed.
**/
EFI_STATUS
UpdateDisplayFromHistory(
IN CONSOLE_LOGGER_PRIVATE_DATA *ConsoleInfo
)
{
EFI_STATUS Status;
EFI_STATUS RetVal;
CHAR16 *Screen;
INT32 *Attributes;
UINTN CurrentRow;
CHAR16 TempCharHolder;
UINTN Column;
INT32 CurrentAttrib;
UINTN CurrentColumn;
CHAR16 *StringSegment;
CHAR16 *StringSegmentEnd;
CHAR16 StringSegmentEndChar;
INT32 OrigAttribute;
ASSERT(ConsoleInfo != NULL);
TempCharHolder = CHAR_NULL;
RetVal = EFI_SUCCESS;
OrigAttribute = ConsoleInfo->OldConOut->Mode->Attribute;
//
// Disable cursor visibility and move it to the top left corner
//
ConsoleInfo->OldConOut->EnableCursor (ConsoleInfo->OldConOut, FALSE);
ConsoleInfo->OldConOut->SetCursorPosition (ConsoleInfo->OldConOut, 0, 0);
Screen = &ConsoleInfo->Buffer[(ConsoleInfo->ColsPerScreen + 2) * ConsoleInfo->CurrentStartRow];
Attributes = &ConsoleInfo->Attributes[ConsoleInfo->ColsPerScreen * ConsoleInfo->CurrentStartRow];
for ( CurrentRow = 0
; CurrentRow < ConsoleInfo->RowsPerScreen
; CurrentRow++
, Screen += (ConsoleInfo->ColsPerScreen + 2)
, Attributes += ConsoleInfo->ColsPerScreen
){
//
// dont use the last char - prevents screen scroll
//
if (CurrentRow == (ConsoleInfo->RowsPerScreen-1)){
TempCharHolder = Screen[ConsoleInfo->ColsPerScreen - 1];
Screen[ConsoleInfo->ColsPerScreen - 1] = CHAR_NULL;
}
for ( Column = 0
; Column < ConsoleInfo->ColsPerScreen
; Column++
){
if (Screen[Column] != CHAR_NULL) {
CurrentAttrib = Attributes[Column];
CurrentColumn = Column;
StringSegment = &Screen[Column];
//
// Find the first char with a different arrribute and make that temporarily NULL
// so we can do fewer printout statements. (later) restore that one and we will
// start at that collumn on the next loop.
//
StringSegmentEndChar = CHAR_NULL;
for ( StringSegmentEnd = StringSegment
; *StringSegmentEnd != CHAR_NULL
; StringSegmentEnd++
, Column++
){
if (Attributes[Column] != CurrentAttrib) {
StringSegmentEndChar = *StringSegmentEnd;
*StringSegmentEnd = CHAR_NULL;
break;
}
} // StringSegmentEnd loop
//
// Now write out as much as had the same Attributes
//
ConsoleInfo->OldConOut->SetAttribute(ConsoleInfo->OldConOut, CurrentAttrib);
ConsoleInfo->OldConOut->SetCursorPosition(ConsoleInfo->OldConOut, CurrentColumn, CurrentRow);
Status = ConsoleInfo->OldConOut->OutputString(ConsoleInfo->OldConOut, StringSegment);
if (EFI_ERROR(Status)) {
ASSERT(FALSE);
RetVal = Status;
}
//
// If we found a change in attribute put the character back and decrement the column
// so when it increments it will point at that character and we will start printing
// a segment with that new attribute
//
if (StringSegmentEndChar != CHAR_NULL) {
*StringSegmentEnd = StringSegmentEndChar;
StringSegmentEndChar = CHAR_NULL;
Column--;
}
}
} // column for loop
//
// If we removed the last char and this was the last row put it back
//
if (TempCharHolder != CHAR_NULL) {
Screen[ConsoleInfo->ColsPerScreen - 1] = TempCharHolder;
TempCharHolder = CHAR_NULL;
}
} // row for loop
//
// If we are setting the screen back to original turn on the cursor and make it visible
// and set the attributes back to what they were
//
if (ConsoleInfo->CurrentStartRow == ConsoleInfo->OriginalStartRow) {
ConsoleInfo->OldConOut->SetAttribute (
ConsoleInfo->OldConOut,
ConsoleInfo->HistoryMode.Attribute
);
ConsoleInfo->OldConOut->SetCursorPosition (
ConsoleInfo->OldConOut,
ConsoleInfo->HistoryMode.CursorColumn,
ConsoleInfo->HistoryMode.CursorRow - ConsoleInfo->OriginalStartRow
);
Status = ConsoleInfo->OldConOut->EnableCursor (
ConsoleInfo->OldConOut,
ConsoleInfo->HistoryMode.CursorVisible
);
if (EFI_ERROR (Status)) {
RetVal = Status;
}
} else {
ConsoleInfo->OldConOut->SetAttribute (
ConsoleInfo->OldConOut,
OrigAttribute
);
}
return (RetVal);
}
/**
Reset the text output device hardware and optionaly run diagnostics
@param This pointer to EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL
@param ExtendedVerification Indicates that a more extensive test may be performed
@retval EFI_SUCCESS The text output device was reset.
@retval EFI_DEVICE_ERROR The text output device is not functioning correctly and
could not be reset.
**/
EFI_STATUS
EFIAPI
ConsoleLoggerReset (
IN EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *This,
IN BOOLEAN ExtendedVerification
)
{
EFI_STATUS Status;
CONSOLE_LOGGER_PRIVATE_DATA *ConsoleInfo;
ConsoleInfo = CONSOLE_LOGGER_PRIVATE_DATA_FROM_THIS(This);
//
// Forward the request to the original ConOut
//
Status = ConsoleInfo->OldConOut->Reset (ConsoleInfo->OldConOut, ExtendedVerification);
//
// Check that the buffers are still correct for logging
//
if (!EFI_ERROR (Status)) {
ConsoleLoggerResetBuffers(ConsoleInfo);
if (ExtendedVerification) {
ConsoleInfo->OriginalStartRow = 0;
ConsoleInfo->CurrentStartRow = 0;
}
}
return Status;
}
/**
Appends a string to the history buffer. If the buffer is full then the oldest
information in the buffer will be dropped. Information is added in a line by
line manner such that an empty line takes up just as much space as a full line.
@param[in] String String pointer to add.
@param[in] ConsoleInfo The pointer to the instance of the console logger information.
**/
EFI_STATUS
AppendStringToHistory(
IN CONST CHAR16 *String,
IN CONSOLE_LOGGER_PRIVATE_DATA *ConsoleInfo
)
{
CONST CHAR16 *Walker;
UINTN CopySize;
UINTN PrintIndex;
UINTN Index;
ASSERT(ConsoleInfo != NULL);
for ( Walker = String
; Walker != NULL && *Walker != CHAR_NULL
; Walker++
){
switch (*Walker) {
case (CHAR_BACKSPACE):
if (ConsoleInfo->HistoryMode.CursorColumn > 0) {
ConsoleInfo->HistoryMode.CursorColumn--;
}
break;
case (CHAR_LINEFEED):
if (ConsoleInfo->HistoryMode.CursorRow >= (INT32)((ConsoleInfo->RowsPerScreen * ConsoleInfo->ScreenCount)-1)) {
//
// Should never be bigger
//
ASSERT(ConsoleInfo->HistoryMode.CursorRow == (INT32)((ConsoleInfo->RowsPerScreen * ConsoleInfo->ScreenCount)-1));
//
// scroll history attributes 'up' 1 row and set the last row to default attribute
//
CopySize = ConsoleInfo->ColsPerScreen
* ((ConsoleInfo->RowsPerScreen * ConsoleInfo->ScreenCount) - 1)
* sizeof(ConsoleInfo->Attributes[0]);
ASSERT(CopySize < ConsoleInfo->AttribSize);
CopyMem(
ConsoleInfo->Attributes,
ConsoleInfo->Attributes + ConsoleInfo->ColsPerScreen,
CopySize
);
for ( Index = 0
; Index < ConsoleInfo->ColsPerScreen
; Index++
){
*(ConsoleInfo->Attributes + (CopySize/sizeof(ConsoleInfo->Attributes[0])) + Index) = ConsoleInfo->HistoryMode.Attribute;
}
//
// scroll history buffer 'up' 1 row and set the last row to spaces (L' ')
//
CopySize = (ConsoleInfo->ColsPerScreen + 2)
* ((ConsoleInfo->RowsPerScreen * ConsoleInfo->ScreenCount) - 1)
* sizeof(ConsoleInfo->Buffer[0]);
ASSERT(CopySize < ConsoleInfo->BufferSize);
CopyMem(
ConsoleInfo->Buffer,
ConsoleInfo->Buffer + (ConsoleInfo->ColsPerScreen + 2),
CopySize
);
//
// Set that last row of chars to spaces
//
SetMem16(((UINT8*)ConsoleInfo->Buffer)+CopySize, ConsoleInfo->ColsPerScreen*sizeof(CHAR16), L' ');
} else {
//
// we are not on the last row
//
//
// We should not be scrolling history
//
ASSERT (ConsoleInfo->OriginalStartRow == ConsoleInfo->CurrentStartRow);
//
// are we at the end of a row?
//
if (ConsoleInfo->HistoryMode.CursorRow == (INT32) (ConsoleInfo->OriginalStartRow + ConsoleInfo->RowsPerScreen - 1)) {
ConsoleInfo->OriginalStartRow++;
ConsoleInfo->CurrentStartRow++;
}
ConsoleInfo->HistoryMode.CursorRow++;
}
break;
case (CHAR_CARRIAGE_RETURN):
//
// Move the cursor to the beginning of the current row.
//
ConsoleInfo->HistoryMode.CursorColumn = 0;
break;
default:
//
// Acrtually print characters into the history buffer
//
PrintIndex = ConsoleInfo->HistoryMode.CursorRow * ConsoleInfo->ColsPerScreen + ConsoleInfo->HistoryMode.CursorColumn;
for ( // no initializer needed
; ConsoleInfo->HistoryMode.CursorColumn < (INT32) ConsoleInfo->ColsPerScreen
; ConsoleInfo->HistoryMode.CursorColumn++
, PrintIndex++
, Walker++
){
if (*Walker == CHAR_NULL
||*Walker == CHAR_BACKSPACE
||*Walker == CHAR_LINEFEED
||*Walker == CHAR_CARRIAGE_RETURN
){
Walker--;
break;
}
//
// The buffer is 2*CursorRow more since it has that many \r\n characters at the end of each row.
//
ASSERT(PrintIndex + ConsoleInfo->HistoryMode.CursorRow < ConsoleInfo->BufferSize);
ConsoleInfo->Buffer[PrintIndex + (2*ConsoleInfo->HistoryMode.CursorRow)] = *Walker;
ASSERT(PrintIndex < ConsoleInfo->AttribSize);
ConsoleInfo->Attributes[PrintIndex] = ConsoleInfo->HistoryMode.Attribute;
} // for loop
//
// Add the carriage return and line feed at the end of the lines
//
if (ConsoleInfo->HistoryMode.CursorColumn >= (INT32)ConsoleInfo->ColsPerScreen) {
AppendStringToHistory(L"\r\n", ConsoleInfo);
Walker--;
}
break;
} // switch for character
} // for loop
return (EFI_SUCCESS);
}
/**
Worker function to handle printing the output to the screen
and the history buffer
@param[in] String The string to output
@param[in] ConsoleInfo The pointer to the instance of the console logger information.
@retval EFI_SUCCESS The string was printed
@retval EFI_DEVICE_ERROR The device reported an error while attempting to output
the text.
@retval EFI_UNSUPPORTED The output device's mode is not currently in a
defined text mode.
@retval EFI_WARN_UNKNOWN_GLYPH This warning code indicates that some of the
characters in the Unicode string could not be
rendered and were skipped.
**/
EFI_STATUS
ConsoleLoggerOutputStringSplit(
IN CONST CHAR16 *String,
IN CONSOLE_LOGGER_PRIVATE_DATA *ConsoleInfo
)
{
EFI_STATUS Status;
//
// Forward the request to the original ConOut
//
Status = ConsoleInfo->OldConOut->OutputString (ConsoleInfo->OldConOut, (CHAR16*)String);
if (EFI_ERROR(Status)) {
return (Status);
}
return (AppendStringToHistory(String, ConsoleInfo));
}
/**
Function to handle page break mode.
This function will prompt for continue or break.
@retval EFI_SUCCESS Continue was choosen
@return other Break was choosen
**/
EFI_STATUS
ConsoleLoggerDoPageBreak(
VOID
)
{
SHELL_PROMPT_RESPONSE *Resp;
EFI_STATUS Status;
Resp = NULL;
ASSERT(ShellInfoObject.PageBreakEnabled);
ShellInfoObject.PageBreakEnabled = FALSE;
Status = ShellPromptForResponseHii(ShellPromptResponseTypeQuitContinue, STRING_TOKEN(STR_SHELL_QUIT_CONT), ShellInfoObject.HiiHandle, (VOID**)&Resp);
ShellInfoObject.PageBreakEnabled = TRUE;
ASSERT(Resp != NULL);
if (Resp == NULL) {
return (EFI_NOT_FOUND);
}
if (EFI_ERROR(Status)) {
if (Resp != NULL) {
FreePool(Resp);
}
return (Status);
}
if (*Resp == ShellPromptResponseContinue) {
FreePool(Resp);
ShellInfoObject.ConsoleInfo->RowCounter = 0;
// ShellInfoObject.ConsoleInfo->OurConOut.Mode->CursorRow = 0;
// ShellInfoObject.ConsoleInfo->OurConOut.Mode->CursorColumn = 0;
return (EFI_SUCCESS);
} else if (*Resp == ShellPromptResponseQuit) {
FreePool(Resp);
ShellInfoObject.ConsoleInfo->Enabled = FALSE;
//
// When user wants to quit, the shell should stop running the command.
//
gBS->SignalEvent (ShellInfoObject.NewEfiShellProtocol->ExecutionBreak);
return (EFI_DEVICE_ERROR);
} else {
ASSERT(FALSE);
}
return (EFI_SUCCESS);
}
/**
Worker function to handle printing the output with page breaks.
@param[in] String The string to output
@param[in] ConsoleInfo The pointer to the instance of the console logger information.
@retval EFI_SUCCESS The string was printed
@retval EFI_DEVICE_ERROR The device reported an error while attempting to output
the text.
@retval EFI_UNSUPPORTED The output device's mode is not currently in a
defined text mode.
@retval EFI_WARN_UNKNOWN_GLYPH This warning code indicates that some of the
characters in the Unicode string could not be
rendered and were skipped.
**/
EFI_STATUS
ConsoleLoggerPrintWithPageBreak(
IN CONST CHAR16 *String,
IN CONSOLE_LOGGER_PRIVATE_DATA *ConsoleInfo
)
{
CONST CHAR16 *Walker;
CONST CHAR16 *LineStart;
CHAR16 *StringCopy;
CHAR16 TempChar;
StringCopy = NULL;
StringCopy = StrnCatGrow(&StringCopy, NULL, String, 0);
if (StringCopy == NULL) {
return (EFI_OUT_OF_RESOURCES);
}
for ( Walker = StringCopy
, LineStart = StringCopy
; Walker != NULL && *Walker != CHAR_NULL
; Walker++
){
switch (*Walker) {
case (CHAR_BACKSPACE):
if (ConsoleInfo->OurConOut.Mode->CursorColumn > 0) {
ConsoleInfo->OurConOut.Mode->CursorColumn--;
}
break;
case (CHAR_LINEFEED):
//
// add a temp NULL terminator
//
TempChar = *(Walker + 1);
*((CHAR16*)(Walker+1)) = CHAR_NULL;
//
// output the string
//
ConsoleLoggerOutputStringSplit (LineStart, ConsoleInfo);
//
// restore the temp NULL terminator to it's original character
//
*((CHAR16*)(Walker+1)) = TempChar;
//
// Update LineStart Variable
//
LineStart = Walker + 1;
//
// increment row count
//
ShellInfoObject.ConsoleInfo->RowCounter++;
ConsoleInfo->OurConOut.Mode->CursorRow++;
break;
case (CHAR_CARRIAGE_RETURN):
//
// Move the cursor to the beginning of the current row.
//
ConsoleInfo->OurConOut.Mode->CursorColumn = 0;
break;
default:
//
// increment column count
//
ConsoleInfo->OurConOut.Mode->CursorColumn++;
//
// check if that is the last column
//
if ((INTN)ConsoleInfo->ColsPerScreen == ConsoleInfo->OurConOut.Mode->CursorColumn + 1) {
//
// output a line similar to the linefeed character.
//
//
// add a temp NULL terminator
//
TempChar = *(Walker + 1);
*((CHAR16*)(Walker+1)) = CHAR_NULL;
//
// output the string
//
ConsoleLoggerOutputStringSplit (LineStart, ConsoleInfo);
//
// restore the temp NULL terminator to it's original character
//
*((CHAR16*)(Walker+1)) = TempChar;
//
// Update LineStart Variable
//
LineStart = Walker + 1;
//
// increment row count and zero the column
//
ShellInfoObject.ConsoleInfo->RowCounter++;
ConsoleInfo->OurConOut.Mode->CursorRow++;
ConsoleInfo->OurConOut.Mode->CursorColumn = 0;
} // last column on line
break;
} // switch for character
//
// check if that was the last printable row. If yes handle PageBreak mode
//
if ((ConsoleInfo->RowsPerScreen) -1 == ShellInfoObject.ConsoleInfo->RowCounter) {
if (EFI_ERROR(ConsoleLoggerDoPageBreak())) {
//
// We got an error which means 'break' and halt the printing
//
SHELL_FREE_NON_NULL(StringCopy);
return (EFI_DEVICE_ERROR);
}
}
} // for loop
if (LineStart != NULL && *LineStart != CHAR_NULL) {
ConsoleLoggerOutputStringSplit (LineStart, ConsoleInfo);
}
SHELL_FREE_NON_NULL(StringCopy);
return (EFI_SUCCESS);
}
/**
Write a Unicode string to the output device.
@param[in] This Protocol instance pointer.
@param[in] WString The NULL-terminated Unicode string to be displayed on the output
device(s). All output devices must also support the Unicode
drawing defined in this file.
@retval EFI_SUCCESS The string was output to the device.
@retval EFI_DEVICE_ERROR The device reported an error while attempting to output
the text.
@retval EFI_UNSUPPORTED The output device's mode is not currently in a
defined text mode.
@retval EFI_WARN_UNKNOWN_GLYPH This warning code indicates that some of the
characters in the Unicode string could not be
rendered and were skipped.
**/
EFI_STATUS
EFIAPI
ConsoleLoggerOutputString (
IN EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *This,
IN CHAR16 *WString
)
{
EFI_STATUS Status;
EFI_SIMPLE_TEXT_INPUT_EX_PROTOCOL *TxtInEx;
EFI_KEY_DATA KeyData;
UINTN EventIndex;
CONSOLE_LOGGER_PRIVATE_DATA *ConsoleInfo;
ConsoleInfo = CONSOLE_LOGGER_PRIVATE_DATA_FROM_THIS(This);
if (ShellInfoObject.ShellInitSettings.BitUnion.Bits.NoConsoleOut) {
return (EFI_UNSUPPORTED);
}
ASSERT(ShellInfoObject.ConsoleInfo == ConsoleInfo);
Status = gBS->HandleProtocol (gST->ConsoleInHandle, &gEfiSimpleTextInputExProtocolGuid, (VOID **) &TxtInEx);
if (!EFI_ERROR (Status)) {
while (ShellInfoObject.HaltOutput) {
ShellInfoObject.HaltOutput = FALSE;
//
// just get some key
//
Status = gBS->WaitForEvent (1, &TxtInEx->WaitForKeyEx, &EventIndex);
ASSERT_EFI_ERROR (Status);
Status = TxtInEx->ReadKeyStrokeEx (TxtInEx, &KeyData);
if (EFI_ERROR(Status)) {
break;
}
if ((KeyData.Key.UnicodeChar == L's') && (KeyData.Key.ScanCode == SCAN_NULL) &&
((KeyData.KeyState.KeyShiftState == (EFI_SHIFT_STATE_VALID | EFI_LEFT_CONTROL_PRESSED)) ||
(KeyData.KeyState.KeyShiftState == (EFI_SHIFT_STATE_VALID | EFI_RIGHT_CONTROL_PRESSED))
)
) {
ShellInfoObject.HaltOutput = TRUE;
}
}
}
if (!ShellInfoObject.ConsoleInfo->Enabled) {
return (EFI_DEVICE_ERROR);
} else if (ShellInfoObject.PageBreakEnabled) {
return (ConsoleLoggerPrintWithPageBreak(WString, ConsoleInfo));
} else {
return (ConsoleLoggerOutputStringSplit(WString, ConsoleInfo));
}
}
/**
Verifies that all characters in a Unicode string can be output to the
target device.
@param[in] This Protocol instance pointer.
@param[in] WString The NULL-terminated Unicode string to be examined for the output
device(s).
@retval EFI_SUCCESS The device(s) are capable of rendering the output string.
@retval EFI_UNSUPPORTED Some of the characters in the Unicode string cannot be
rendered by one or more of the output devices mapped
by the EFI handle.
**/
EFI_STATUS
EFIAPI
ConsoleLoggerTestString (
IN EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *This,
IN CHAR16 *WString
)
{
CONSOLE_LOGGER_PRIVATE_DATA *ConsoleInfo;
ConsoleInfo = CONSOLE_LOGGER_PRIVATE_DATA_FROM_THIS(This);
//
// Forward the request to the original ConOut
//
return (ConsoleInfo->OldConOut->TestString (ConsoleInfo->OldConOut, WString));
}
/**
Returns information for an available text mode that the output device(s)
supports.
@param[in] This Protocol instance pointer.
@param[in] ModeNumber The mode number to return information on.
@param[out] Columns Upon return, the number of columns in the selected geometry
@param[out] Rows Upon return, the number of rows in the selected geometry
@retval EFI_SUCCESS The requested mode information was returned.
@retval EFI_DEVICE_ERROR The device had an error and could not
complete the request.
@retval EFI_UNSUPPORTED The mode number was not valid.
**/
EFI_STATUS
EFIAPI
ConsoleLoggerQueryMode (
IN EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *This,
IN UINTN ModeNumber,
OUT UINTN *Columns,
OUT UINTN *Rows
)
{
CONSOLE_LOGGER_PRIVATE_DATA *ConsoleInfo;
ConsoleInfo = CONSOLE_LOGGER_PRIVATE_DATA_FROM_THIS(This);
//
// Forward the request to the original ConOut
//
return (ConsoleInfo->OldConOut->QueryMode (
ConsoleInfo->OldConOut,
ModeNumber,
Columns,
Rows
));
}
/**
Sets the output device(s) to a specified mode.
@param[in] This Protocol instance pointer.
@param[in] ModeNumber The mode number to set.
@retval EFI_SUCCESS The requested text mode was set.
@retval EFI_DEVICE_ERROR The device had an error and
could not complete the request.
@retval EFI_UNSUPPORTED The mode number was not valid.
**/
EFI_STATUS
EFIAPI
ConsoleLoggerSetMode (
IN EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *This,
IN UINTN ModeNumber
)
{
EFI_STATUS Status;
CONSOLE_LOGGER_PRIVATE_DATA *ConsoleInfo;
ConsoleInfo = CONSOLE_LOGGER_PRIVATE_DATA_FROM_THIS(This);
//
// Forward the request to the original ConOut
//
Status = ConsoleInfo->OldConOut->SetMode (ConsoleInfo->OldConOut, ModeNumber);
//
// Check that the buffers are still correct for logging
//
if (!EFI_ERROR (Status)) {
ConsoleInfo->OurConOut.Mode = ConsoleInfo->OldConOut->Mode;
ConsoleLoggerResetBuffers(ConsoleInfo);
ConsoleInfo->OriginalStartRow = 0;
ConsoleInfo->CurrentStartRow = 0;
ConsoleInfo->OurConOut.ClearScreen (&ConsoleInfo->OurConOut);
}
return Status;
}
/**
Sets the background and foreground colors for the OutputString () and
ClearScreen () functions.
@param[in] This Protocol instance pointer.
@param[in] Attribute The attribute to set. Bits 0..3 are the foreground color, and
bits 4..6 are the background color. All other bits are undefined
and must be zero. The valid Attributes are defined in this file.
@retval EFI_SUCCESS The attribute was set.
@retval EFI_DEVICE_ERROR The device had an error and
could not complete the request.
@retval EFI_UNSUPPORTED The attribute requested is not defined.
**/
EFI_STATUS
EFIAPI
ConsoleLoggerSetAttribute (
IN EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *This,
IN UINTN Attribute
)
{
EFI_STATUS Status;
CONSOLE_LOGGER_PRIVATE_DATA *ConsoleInfo;
ConsoleInfo = CONSOLE_LOGGER_PRIVATE_DATA_FROM_THIS(This);
//
// Forward the request to the original ConOut
//
Status = ConsoleInfo->OldConOut->SetAttribute (ConsoleInfo->OldConOut, Attribute);
//
// Record console output history
//
if (!EFI_ERROR (Status)) {
ConsoleInfo->HistoryMode.Attribute = (INT32) Attribute;
}
return Status;
}
/**
Clears the output device(s) display to the currently selected background
color.
@param[in] This Protocol instance pointer.
@retval EFI_SUCCESS The operation completed successfully.
@retval EFI_DEVICE_ERROR The device had an error and
could not complete the request.
@retval EFI_UNSUPPORTED The output device is not in a valid text mode.
**/
EFI_STATUS
EFIAPI
ConsoleLoggerClearScreen (
IN EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *This
)
{
EFI_STATUS Status;
CHAR16 *Screen;
INT32 *Attributes;
UINTN Row;
UINTN Column;
CONSOLE_LOGGER_PRIVATE_DATA *ConsoleInfo;
if (ShellInfoObject.ShellInitSettings.BitUnion.Bits.NoConsoleOut) {
return (EFI_UNSUPPORTED);
}
ConsoleInfo = CONSOLE_LOGGER_PRIVATE_DATA_FROM_THIS(This);
//
// Forward the request to the original ConOut
//
Status = ConsoleInfo->OldConOut->ClearScreen (ConsoleInfo->OldConOut);
//
// Record console output history
//
if (!EFI_ERROR (Status)) {
Screen = &ConsoleInfo->Buffer[(ConsoleInfo->ColsPerScreen + 2) * ConsoleInfo->CurrentStartRow];
Attributes = &ConsoleInfo->Attributes[ConsoleInfo->ColsPerScreen * ConsoleInfo->CurrentStartRow];
for ( Row = ConsoleInfo->OriginalStartRow
; Row < (ConsoleInfo->RowsPerScreen * ConsoleInfo->ScreenCount)
; Row++
){
for ( Column = 0
; Column < ConsoleInfo->ColsPerScreen
; Column++
, Screen++
, Attributes++
){
*Screen = L' ';
*Attributes = ConsoleInfo->OldConOut->Mode->Attribute;
}
//
// Skip the NULL on each column end in text buffer only
//
Screen += 2;
}
ConsoleInfo->HistoryMode.CursorColumn = 0;
ConsoleInfo->HistoryMode.CursorRow = 0;
}
return Status;
}
/**
Sets the current coordinates of the cursor position
@param[in] This Protocol instance pointer.
@param[in] Column Column to put the cursor in. Must be between zero and Column returned from QueryMode
@param[in] Row Row to put the cursor in. Must be between zero and Row returned from QueryMode
@retval EFI_SUCCESS The operation completed successfully.
@retval EFI_DEVICE_ERROR The device had an error and
could not complete the request.
@retval EFI_UNSUPPORTED The output device is not in a valid text mode, or the
cursor position is invalid for the current mode.
**/
EFI_STATUS
EFIAPI
ConsoleLoggerSetCursorPosition (
IN EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *This,
IN UINTN Column,
IN UINTN Row
)
{
EFI_STATUS Status;
CONSOLE_LOGGER_PRIVATE_DATA *ConsoleInfo;
if (ShellInfoObject.ShellInitSettings.BitUnion.Bits.NoConsoleOut) {
return (EFI_UNSUPPORTED);
}
ConsoleInfo = CONSOLE_LOGGER_PRIVATE_DATA_FROM_THIS(This);
//
// Forward the request to the original ConOut
//
Status = ConsoleInfo->OldConOut->SetCursorPosition (
ConsoleInfo->OldConOut,
Column,
Row
);
//
// Record console output history
//
if (!EFI_ERROR (Status)) {
ConsoleInfo->HistoryMode.CursorColumn = (INT32)Column;
ConsoleInfo->HistoryMode.CursorRow = (INT32)(ConsoleInfo->OriginalStartRow + Row);
}
return Status;
}
/**
Makes the cursor visible or invisible
@param[in] This Protocol instance pointer.
@param[in] Visible If TRUE, the cursor is set to be visible. If FALSE, the cursor is
set to be invisible.
@retval EFI_SUCCESS The operation completed successfully.
@retval EFI_DEVICE_ERROR The device had an error and could not complete the
request, or the device does not support changing
the cursor mode.
@retval EFI_UNSUPPORTED The output device is not in a valid text mode.
**/
EFI_STATUS
EFIAPI
ConsoleLoggerEnableCursor (
IN EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *This,
IN BOOLEAN Visible
)
{
EFI_STATUS Status;
CONSOLE_LOGGER_PRIVATE_DATA *ConsoleInfo;
ConsoleInfo = CONSOLE_LOGGER_PRIVATE_DATA_FROM_THIS(This);
//
// Forward the request to the original ConOut
//
Status = ConsoleInfo->OldConOut->EnableCursor (ConsoleInfo->OldConOut, Visible);
//
// Record console output history
//
if (!EFI_ERROR (Status)) {
ConsoleInfo->HistoryMode.CursorVisible = Visible;
}
return Status;
}
/**
Function to update and verify that the current buffers are correct.
@param[in] ConsoleInfo The pointer to the instance of the console logger information.
This will be used when a mode has changed or a reset ocurred to verify all
history buffers.
**/
EFI_STATUS
ConsoleLoggerResetBuffers(
IN CONSOLE_LOGGER_PRIVATE_DATA *ConsoleInfo
)
{
EFI_STATUS Status;
if (ConsoleInfo->Buffer != NULL) {
FreePool(ConsoleInfo->Buffer);
ConsoleInfo->Buffer = NULL;
ConsoleInfo->BufferSize = 0;
}
if (ConsoleInfo->Attributes != NULL) {
FreePool(ConsoleInfo->Attributes);
ConsoleInfo->Attributes = NULL;
ConsoleInfo->AttribSize = 0;
}
Status = gST->ConOut->QueryMode (gST->ConOut, gST->ConOut->Mode->Mode, &ConsoleInfo->ColsPerScreen, &ConsoleInfo->RowsPerScreen);
if (EFI_ERROR(Status)){
return (Status);
}
ConsoleInfo->BufferSize = (ConsoleInfo->ColsPerScreen + 2) * ConsoleInfo->RowsPerScreen * ConsoleInfo->ScreenCount * sizeof(ConsoleInfo->Buffer[0]);
ConsoleInfo->AttribSize = ConsoleInfo->ColsPerScreen * ConsoleInfo->RowsPerScreen * ConsoleInfo->ScreenCount * sizeof(ConsoleInfo->Attributes[0]);
ConsoleInfo->Buffer = (CHAR16*)AllocateZeroPool(ConsoleInfo->BufferSize);
if (ConsoleInfo->Buffer == NULL) {
return (EFI_OUT_OF_RESOURCES);
}
ConsoleInfo->Attributes = (INT32*)AllocateZeroPool(ConsoleInfo->AttribSize);
if (ConsoleInfo->Attributes == NULL) {
FreePool(ConsoleInfo->Buffer);
ConsoleInfo->Buffer = NULL;
return (EFI_OUT_OF_RESOURCES);
}
CopyMem (&ConsoleInfo->HistoryMode, ConsoleInfo->OldConOut->Mode, sizeof (EFI_SIMPLE_TEXT_OUTPUT_MODE));
return (EFI_SUCCESS);
}
Reference in New Issue View Git Blame Copy Permalink