VOID TerminateScreen(VOID)
{
// clear text screen
refit_call2_wrapper(ST->ConOut->SetAttribute, ST->ConOut, ATTR_BASIC);
refit_call1_wrapper(ST->ConOut->ClearScreen, ST->ConOut);
// enable cursor
refit_call2_wrapper(ST->ConOut->EnableCursor, ST->ConOut, TRUE);
}
BOOLEAN CheckError(IN EFI_STATUS Status, IN CHAR16 *where)
{
CHAR16 ErrorName[64];
if (!EFI_ERROR(Status))
return FALSE;
StatusToString(ErrorName, Status);
refit_call2_wrapper(ST->ConOut->SetAttribute, ST->ConOut, ATTR_ERROR);
Print(L"Error: %s %s\n", ErrorName, where);
refit_call2_wrapper(ST->ConOut->SetAttribute, ST->ConOut, ATTR_BASIC);
haveError = TRUE;
return TRUE;
}
VOID InitScreen(VOID)
{
// initialize libeg
egInitScreen();
if (egHasGraphicsMode()) {
egGetScreenSize(&UGAWidth, &UGAHeight);
AllowGraphicsMode = TRUE;
} else {
AllowGraphicsMode = FALSE;
egSetTextMode(GlobalConfig.RequestedTextMode);
egSetGraphicsModeEnabled(FALSE); // just to be sure we are in text mode
}
GraphicsScreenDirty = TRUE;
// disable cursor
refit_call2_wrapper(ST->ConOut->EnableCursor, ST->ConOut, FALSE);
// get size of text console
if (refit_call4_wrapper(ST->ConOut->QueryMode, ST->ConOut, ST->ConOut->Mode->Mode, &ConWidth, &ConHeight) != EFI_SUCCESS) {
// use default values on error
ConWidth = 80;
ConHeight = 25;
}
PrepareBlankLine();
// show the banner if in text mode
if (GlobalConfig.TextOnly && (GlobalConfig.ScreensaverTime != -1))
DrawScreenHeader(L"Initializing...");
}
BOOLEAN CheckFatalError(IN EFI_STATUS Status, IN CHAR16 *where)
{
CHAR16 ErrorName[64];
if (!EFI_ERROR(Status))
return FALSE;
StatusToString(ErrorName, Status);
refit_call2_wrapper(ST->ConOut->SetAttribute, ST->ConOut, ATTR_ERROR);
Print(L"Fatal Error: %s %s\n", ErrorName, where);
refit_call2_wrapper(ST->ConOut->SetAttribute, ST->ConOut, ATTR_BASIC);
haveError = TRUE;
//BS->Exit(ImageHandle, ExitStatus, ExitDataSize, ExitData);
return TRUE;
}
// Sets the screen resolution to the specified value, if possible.
// If the specified value is not valid, displays a warning with the valid
// modes on UEFI systems, or silently fails on EFI 1.x systems. Note that
// this function attempts to set ANY screen resolution, even 0x0 or
// ridiculously large values.
// Returns TRUE if successful, FALSE if not.
BOOLEAN egSetScreenSize(IN UINTN ScreenWidth, IN UINTN ScreenHeight) {
EFI_STATUS Status = EFI_SUCCESS;
EFI_GRAPHICS_OUTPUT_MODE_INFORMATION *Info;
UINT32 ModeNum = 0;
UINTN Size;
BOOLEAN ModeSet = FALSE;
UINT32 UGAWidth, UGAHeight, UGADepth, UGARefreshRate;
if (GraphicsOutput != NULL) { // GOP mode (UEFI)
// Do a loop through the modes to see if the specified one is available;
// and if so, switch to it....
while ((Status == EFI_SUCCESS) && (!ModeSet)) {
Status = refit_call4_wrapper(GraphicsOutput->QueryMode, GraphicsOutput, ModeNum, &Size, &Info);
if ((Status == EFI_SUCCESS) && (Size >= sizeof(*Info)) &&
(Info->HorizontalResolution == ScreenWidth) && (Info->VerticalResolution == ScreenHeight)) {
Status = refit_call2_wrapper(GraphicsOutput->SetMode, GraphicsOutput, ModeNum);
ModeSet = (Status == EFI_SUCCESS);
} // if
ModeNum++;
} // while()
if (ModeSet) {
egScreenWidth = ScreenWidth;
egScreenHeight = ScreenHeight;
} else {// If unsuccessful, display an error message for the user....
Print(L"Error setting mode %d x %d; using default mode!\nAvailable modes are:\n", ScreenWidth, ScreenHeight);
ModeNum = 0;
Status = EFI_SUCCESS;
while (Status == EFI_SUCCESS) {
Status = refit_call4_wrapper(GraphicsOutput->QueryMode, GraphicsOutput, ModeNum, &Size, &Info);
if ((Status == EFI_SUCCESS) && (Size >= sizeof(*Info))) {
Print(L"Mode %d: %d x %d\n", ModeNum, Info->HorizontalResolution, Info->VerticalResolution);
} // else
ModeNum++;
} // while()
PauseForKey();
} // if()
} else if (UgaDraw != NULL) { // UGA mode (EFI 1.x)
// Try to use current color depth & refresh rate for new mode. Maybe not the best choice
// in all cases, but I don't know how to probe for alternatives....
Status = refit_call5_wrapper(UgaDraw->GetMode, UgaDraw, &UGAWidth, &UGAHeight, &UGADepth, &UGARefreshRate);
Status = refit_call5_wrapper(UgaDraw->SetMode, UgaDraw, ScreenWidth, ScreenHeight, UGADepth, UGARefreshRate);
if (Status == EFI_SUCCESS) {
egScreenWidth = ScreenWidth;
egScreenHeight = ScreenHeight;
ModeSet = TRUE;
} else {
// TODO: Find a list of supported modes and display it.
// NOTE: Below doesn't actually appear unless we explicitly switch to text mode.
// This is just a placeholder until something better can be done....
Print(L"Error setting mode %d x %d; unsupported mode!\n");
} // if/else
} // if/else if
return (ModeSet);
} // BOOLEAN egSetScreenSize()
VOID SwitchToText(IN BOOLEAN CursorEnabled)
{
egSetGraphicsModeEnabled(FALSE);
refit_call2_wrapper(ST->ConOut->EnableCursor, ST->ConOut, CursorEnabled);
// get size of text console
if (refit_call4_wrapper(ST->ConOut->QueryMode, ST->ConOut, ST->ConOut->Mode->Mode, &ConWidth, &ConHeight) != EFI_SUCCESS) {
// use default values on error
ConWidth = 80;
ConHeight = 25;
}
PrepareBlankLine();
}
VOID egSetGraphicsModeEnabled(IN BOOLEAN Enable)
{
EFI_CONSOLE_CONTROL_SCREEN_MODE CurrentMode;
EFI_CONSOLE_CONTROL_SCREEN_MODE NewMode;
if (ConsoleControl != NULL) {
refit_call4_wrapper(ConsoleControl->GetMode, ConsoleControl, &CurrentMode, NULL, NULL);
NewMode = Enable ? EfiConsoleControlScreenGraphics
: EfiConsoleControlScreenText;
if (CurrentMode != NewMode)
refit_call2_wrapper(ConsoleControl->SetMode, ConsoleControl, NewMode);
}
}
static VOID DrawScreenHeader(IN CHAR16 *Title)
{
UINTN y;
// clear to black background
egClearScreen(&DarkBackgroundPixel); // first clear in graphics mode
refit_call2_wrapper(ST->ConOut->SetAttribute, ST->ConOut, ATTR_BASIC);
refit_call1_wrapper(ST->ConOut->ClearScreen, ST->ConOut); // then clear in text mode
// paint header background
refit_call2_wrapper(ST->ConOut->SetAttribute, ST->ConOut, ATTR_BANNER);
for (y = 0; y < 3; y++) {
refit_call3_wrapper(ST->ConOut->SetCursorPosition, ST->ConOut, 0, y);
Print(BlankLine);
}
// print header text
refit_call3_wrapper(ST->ConOut->SetCursorPosition, ST->ConOut, 3, 1);
Print(L"rEFInd - %s", Title);
// reposition cursor
refit_call2_wrapper(ST->ConOut->SetAttribute, ST->ConOut, ATTR_BASIC);
refit_call3_wrapper(ST->ConOut->SetCursorPosition, ST->ConOut, 0, 4);
}
BOOLEAN ReadAllKeyStrokes(VOID)
{
BOOLEAN GotKeyStrokes;
EFI_STATUS Status;
EFI_INPUT_KEY key;
GotKeyStrokes = FALSE;
for (;;) {
Status = refit_call2_wrapper(ST->ConIn->ReadKeyStroke, ST->ConIn, &key);
if (Status == EFI_SUCCESS) {
GotKeyStrokes = TRUE;
continue;
}
break;
}
return GotKeyStrokes;
}