X-Git-Url: https://git.proxmox.com/?p=mirror_edk2.git;a=blobdiff_plain;f=ShellPkg%2FLibrary%2FUefiShellDebug1CommandsLib%2FHexEdit%2FMainHexEditor.c;h=065f8e95a706673e0693e6fb0ae67ccf861bf9b9;hp=961e0cca68f6f08cef73b1212f6c73c476c5430c;hb=5563281fa2b31093a1cbd415553b9264c5136e89;hpb=980d554e3ab6457a4f7a455fabcc297d6e93547f diff --git a/ShellPkg/Library/UefiShellDebug1CommandsLib/HexEdit/MainHexEditor.c b/ShellPkg/Library/UefiShellDebug1CommandsLib/HexEdit/MainHexEditor.c index 961e0cca68..065f8e95a7 100644 --- a/ShellPkg/Library/UefiShellDebug1CommandsLib/HexEdit/MainHexEditor.c +++ b/ShellPkg/Library/UefiShellDebug1CommandsLib/HexEdit/MainHexEditor.c @@ -4,7 +4,7 @@ - Instances of the other objects of the editor - Main Interfaces - Copyright (c) 2005 - 2011, Intel Corporation. All rights reserved.
+ Copyright (c) 2005 - 2018, 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 @@ -37,8 +37,6 @@ BOOLEAN HEditorMouseAction; extern HEFI_EDITOR_BUFFER_IMAGE HBufferImage; extern HEFI_EDITOR_BUFFER_IMAGE HBufferImageBackupVar; -extern HEFI_EDITOR_CLIPBOARD HClipBoard; - extern BOOLEAN HBufferImageMouseNeedRefresh; extern BOOLEAN HBufferImageNeedRefresh; extern BOOLEAN HBufferImageOnlyLineNeedRefresh; @@ -51,15 +49,14 @@ HEFI_EDITOR_GLOBAL_EDITOR HMainEditorBackupVar; // HEFI_EDITOR_GLOBAL_EDITOR HMainEditorConst = { &HBufferImage, - &HClipBoard, { - 0, - 0 + {0, 0} }, { 0, 0 }, + NULL, FALSE, NULL, 0, @@ -68,25 +65,112 @@ HEFI_EDITOR_GLOBAL_EDITOR HMainEditorConst = { 1 }; +/** + Help info that will be displayed. +**/ +EFI_STRING_ID HexMainMenuHelpInfo[] = { + STRING_TOKEN(STR_HEXEDIT_HELP_TITLE), + STRING_TOKEN(STR_HEXEDIT_HELP_BLANK), + STRING_TOKEN(STR_HEXEDIT_HELP_LIST_TITLE), + STRING_TOKEN(STR_HEXEDIT_HELP_DIV), + STRING_TOKEN(STR_HEXEDIT_HELP_GO_TO_OFFSET), + STRING_TOKEN(STR_HEXEDIT_HELP_SAVE_BUFFER), + STRING_TOKEN(STR_HEXEDIT_HELP_EXIT), + STRING_TOKEN(STR_HEXEDIT_HELP_SELECT_START), + STRING_TOKEN(STR_HEXEDIT_HELP_SELECT_END), + STRING_TOKEN(STR_HEXEDIT_HELP_CUT), + STRING_TOKEN(STR_HEXEDIT_HELP_PASTE), + STRING_TOKEN(STR_HEXEDIT_HELP_OPEN_FILE), + STRING_TOKEN(STR_HEXEDIT_HELP_OPEN_DISK), + STRING_TOKEN(STR_HEXEDIT_HELP_OPEN_MEMORY), + STRING_TOKEN(STR_HEXEDIT_HELP_BLANK), + STRING_TOKEN(STR_HEXEDIT_HELP_EXIT_HELP), + STRING_TOKEN(STR_HEXEDIT_HELP_BLANK), + STRING_TOKEN(STR_HEXEDIT_HELP_BLANK), + STRING_TOKEN(STR_HEXEDIT_HELP_BLANK), + STRING_TOKEN(STR_HEXEDIT_HELP_BLANK), + STRING_TOKEN(STR_HEXEDIT_HELP_BLANK), + STRING_TOKEN(STR_HEXEDIT_HELP_BLANK), + STRING_TOKEN(STR_HEXEDIT_HELP_DIV), + 0 +}; + + +/** + show help menu. + + @retval EFI_SUCCESS The operation was successful. +**/ EFI_STATUS -HMainCommandGoToOffset ( +HMainCommandDisplayHelp ( VOID ) -/*++ - -Routine Description: +{ + INT32 CurrentLine; + CHAR16 *InfoString; + EFI_KEY_DATA KeyData; + EFI_STATUS Status; + UINTN EventIndex; + + // + // print helpInfo + // + for (CurrentLine = 0; 0 != HexMainMenuHelpInfo[CurrentLine]; CurrentLine++) { + InfoString = HiiGetString(gShellDebug1HiiHandle, HexMainMenuHelpInfo[CurrentLine] +, NULL); + ShellPrintEx (0,CurrentLine+1,L"%E%s%N",InfoString); + } - move cursor to specified lines + // + // scan for ctrl+w + // + while (TRUE) { + Status = gBS->WaitForEvent (1, &HMainEditor.TextInputEx->WaitForKeyEx, &EventIndex); + if (EFI_ERROR (Status) || (EventIndex != 0)) { + continue; + } + Status = HMainEditor.TextInputEx->ReadKeyStrokeEx (HMainEditor.TextInputEx, &KeyData); + if (EFI_ERROR (Status)) { + continue; + } -Arguments: + if ((KeyData.KeyState.KeyShiftState & EFI_SHIFT_STATE_VALID) == 0) { + // + // For consoles that don't support shift state reporting, + // CTRL+W is translated to L'W' - L'A' + 1. + // + if (KeyData.Key.UnicodeChar == L'W' - L'A' + 1) { + break; + } + } else if (((KeyData.KeyState.KeyShiftState & (EFI_LEFT_CONTROL_PRESSED | EFI_RIGHT_CONTROL_PRESSED)) != 0) && + ((KeyData.KeyState.KeyShiftState & ~(EFI_SHIFT_STATE_VALID | EFI_LEFT_CONTROL_PRESSED | EFI_RIGHT_CONTROL_PRESSED)) == 0)) { + // + // For consoles that supports shift state reporting, + // make sure that only CONTROL shift key is pressed. + // + if ((KeyData.Key.UnicodeChar == 'w') || (KeyData.Key.UnicodeChar == 'W')) { + break; + } + } + } - None + // update screen with buffer's info + HBufferImageNeedRefresh = TRUE; + HBufferImageOnlyLineNeedRefresh = FALSE; + HBufferImageRefresh (); -Returns: + return EFI_SUCCESS; +} - EFI_SUCCESS +/** + Move cursor to specified lines. ---*/ + @retval EFI_SUCCESS The operation was successful. +**/ +EFI_STATUS +HMainCommandGoToOffset ( + VOID + ) { UINTN Size; UINT64 Offset; @@ -157,29 +241,19 @@ Returns: return EFI_SUCCESS; } +/** + Save current opened buffer. + If is file buffer, you can save to current file name or + save to another file name. + + @retval EFI_SUCCESS The operation was successful. + @retval EFI_OUT_OF_RESOURCES A memory allocation occured. + @retval EFI_LOAD_ERROR A load error occured. +**/ EFI_STATUS HMainCommandSaveBuffer ( VOID ) -/*++ - -Routine Description: - - save current opened buffer . - if is file buffer, you can save to current file name or - save to another file name - -Arguments: - - None - -Returns: - - EFI_SUCCESS - EFI_OUT_OF_RESOURCES - EFI_LOAD_ERROR - ---*/ { EFI_STATUS Status; BOOLEAN Done; @@ -190,7 +264,7 @@ Returns: SHELL_FILE_HANDLE ShellFileHandle; if (HMainEditor.BufferImage->BufferType != FileTypeFileBuffer) { - if (HMainEditor.BufferImage->Modified == FALSE) { + if (!HMainEditor.BufferImage->Modified) { return EFI_SUCCESS; } @@ -248,7 +322,6 @@ Returns: // the file won't be saved // return EFI_SUCCESS; - break; case L'c': case L'C': @@ -381,7 +454,7 @@ Returns: // // if the file is read only, so can not write back to it. // - if (HMainEditor.BufferImage->FileImage->ReadOnly == TRUE) { + if (HMainEditor.BufferImage->FileImage->ReadOnly) { StatusBarSetStatusString (L"Access Denied"); SHELL_FREE_NON_NULL (FileName); return EFI_SUCCESS; @@ -467,27 +540,17 @@ Returns: return EFI_SUCCESS; } +/** + Load a disk buffer editor. + + @retval EFI_SUCCESS The operation was successful. + @retval EFI_OUT_OF_RESOURCES A memory allocation occured. + @retval EFI_LOAD_ERROR A load error occured. +**/ EFI_STATUS HMainCommandSelectStart ( VOID ) -/*++ - -Routine Description: - - Load a disk buffer editor - -Arguments: - - None - -Returns: - - EFI_SUCCESS - EFI_LOAD_ERROR - EFI_OUT_OF_RESOURCES - ---*/ { UINTN Start; @@ -515,27 +578,17 @@ Returns: return EFI_SUCCESS; } +/** + Load a disk buffer editor. + + @retval EFI_SUCCESS The operation was successful. + @retval EFI_OUT_OF_RESOURCES A memory allocation occured. + @retval EFI_LOAD_ERROR A load error occured. +**/ EFI_STATUS HMainCommandSelectEnd ( VOID ) -/*++ - -Routine Description: - - Load a disk buffer editor - -Arguments: - - None - -Returns: - - EFI_SUCCESS - EFI_LOAD_ERROR - EFI_OUT_OF_RESOURCES - ---*/ { UINTN End; @@ -563,30 +616,19 @@ Returns: return EFI_SUCCESS; } +/** + Cut current line to clipboard. + + @retval EFI_SUCCESS The operation was successful. + @retval EFI_OUT_OF_RESOURCES A memory allocation occured. + @retval EFI_LOAD_ERROR A load error occured. +**/ EFI_STATUS HMainCommandCut ( VOID ) -/*++ - -Routine Description: - - cut current line to clipboard - -Arguments: - - None - -Returns: - - EFI_SUCCESS - EFI_OUT_OF_RESOURCES - EFI_LOAD_ERROR - ---*/ { UINTN Index; - HEFI_EDITOR_LINE *Line; LIST_ENTRY *Link; UINT8 *Buffer; UINTN Count; @@ -611,8 +653,6 @@ Returns: Link = Link->ForwardLink; } - Line = CR (Link, HEFI_EDITOR_LINE, Link, EFI_EDITOR_LINE_LIST); - Count = HMainEditor.SelectEnd - HMainEditor.SelectStart + 1; Buffer = AllocateZeroPool (Count); if (Buffer == NULL) { @@ -648,28 +688,17 @@ Returns: return EFI_SUCCESS; } +/** + Paste line to file buffer. + + @retval EFI_SUCCESS The operation was successful. + @retval EFI_OUT_OF_RESOURCES A memory allocation occured. + @retval EFI_LOAD_ERROR A load error occured. +**/ EFI_STATUS HMainCommandPaste ( VOID ) -/*++ - -Routine Description: - - paste line to file buffer - -Arguments: - - None - -Returns: - - EFI_SUCCESS - EFI_OUT_OF_RESOURCES - EFI_LOAD_ERROR - - ---*/ { BOOLEAN OnlyLineRefresh; @@ -719,27 +748,17 @@ Returns: } +/** + Exit editor. + + @retval EFI_SUCCESS The operation was successful. + @retval EFI_OUT_OF_RESOURCES A memory allocation occured. + @retval EFI_LOAD_ERROR A load error occured. +**/ EFI_STATUS HMainCommandExit ( VOID ) -/*++ - -Routine Description: - - exit editor - -Arguments: - - None - -Returns: - - EFI_SUCCESS - EFI_OUT_OF_RESOURCES - EFI_LOAD_ERROR - ---*/ { EFI_STATUS Status; @@ -822,27 +841,17 @@ Returns: } +/** + Load a file from disk to editor. + + @retval EFI_SUCCESS The operation was successful. + @retval EFI_OUT_OF_RESOURCES A memory allocation occured. + @retval EFI_LOAD_ERROR A load error occured. +**/ EFI_STATUS HMainCommandOpenFile ( VOID ) -/*++ - -Routine Description: - - Load a file from disk to editor - -Arguments: - - None - -Returns: - - EFI_SUCCESS - EFI_LOAD_ERROR - EFI_OUT_OF_RESOURCES - ---*/ { BOOLEAN Done; EFI_STATUS Status; @@ -1065,28 +1074,18 @@ Returns: return EFI_SUCCESS; } +/** + Load a disk buffer editor. + + @retval EFI_SUCCESS The operation was successful. + @retval EFI_OUT_OF_RESOURCES A memory allocation occured. + @retval EFI_LOAD_ERROR A load error occured. + @retval EFI_NOT_FOUND The disk was not found. +**/ EFI_STATUS HMainCommandOpenDisk ( VOID ) -/*++ - -Routine Description: - - Load a disk buffer editor - -Arguments: - - None - -Returns: - - EFI_SUCCESS - EFI_LOAD_ERROR - EFI_OUT_OF_RESOURCES - EFI_NOT_FOUND - ---*/ { UINT64 Size; UINT64 Offset; @@ -1320,28 +1319,18 @@ Returns: return EFI_SUCCESS; } +/** + Load memory content to editor + + @retval EFI_SUCCESS The operation was successful. + @retval EFI_OUT_OF_RESOURCES A memory allocation occured. + @retval EFI_LOAD_ERROR A load error occured. + @retval EFI_NOT_FOUND The disk was not found. +**/ EFI_STATUS HMainCommandOpenMemory ( VOID ) -/*++ - -Routine Description: - - Load memory content to editor - -Arguments: - - None - -Returns: - - EFI_SUCCESS - EFI_LOAD_ERROR - EFI_OUT_OF_RESOURCES - EFI_NOT_FOUND - ---*/ { UINT64 Size; UINT64 Offset; @@ -1543,6 +1532,36 @@ Returns: } +MENU_ITEM_FUNCTION HexMainControlBasedMenuFunctions[] = { + NULL, + NULL, /* Ctrl - A */ + NULL, /* Ctrl - B */ + NULL, /* Ctrl - C */ + HMainCommandSelectEnd, /* Ctrl - D */ + HMainCommandDisplayHelp, /* Ctrl - E */ + NULL, /* Ctrl - F */ + HMainCommandGoToOffset, /* Ctrl - G */ + NULL, /* Ctrl - H */ + HMainCommandOpenDisk, /* Ctrl - I */ + NULL, /* Ctrl - J */ + NULL, /* Ctrl - K */ + NULL, /* Ctrl - L */ + HMainCommandOpenMemory, /* Ctrl - M */ + NULL, /* Ctrl - N */ + HMainCommandOpenFile, /* Ctrl - O */ + NULL, /* Ctrl - P */ + HMainCommandExit, /* Ctrl - Q */ + NULL, /* Ctrl - R */ + HMainCommandSaveBuffer, /* Ctrl - S */ + HMainCommandSelectStart, /* Ctrl - T */ + NULL, /* Ctrl - U */ + HMainCommandPaste, /* Ctrl - V */ + NULL, /* Ctrl - W */ + HMainCommandCut, /* Ctrl - X */ + NULL, /* Ctrl - Y */ + NULL, /* Ctrl - Z */ +}; + CONST EDITOR_MENU_ITEM HexEditorMenuItems[] = { { STRING_TOKEN(STR_HEXEDIT_LIBMENUBAR_GO_TO_OFFSET), @@ -1604,27 +1623,16 @@ CONST EDITOR_MENU_ITEM HexEditorMenuItems[] = { } }; +/** + Init function for MainEditor + @retval EFI_SUCCESS The operation was successful. + @retval EFI_LOAD_ERROR A load error occured. +**/ EFI_STATUS HMainEditorInit ( VOID ) -/*++ - -Routine Description: - - Init function for MainEditor - -Arguments: - - None - -Returns: - - EFI_SUCCESS - EFI_LOAD_ERROR - ---*/ { EFI_STATUS Status; EFI_HANDLE *HandleBuffer; @@ -1657,6 +1665,19 @@ Returns: &(HMainEditor.ScreenSize.Row) ); + // + // Find TextInEx in System Table ConsoleInHandle + // Per UEFI Spec, TextInEx is required for a console capable platform. + // + Status = gBS->HandleProtocol ( + gST->ConsoleInHandle, + &gEfiSimpleTextInputExProtocolGuid, + (VOID**)&HMainEditor.TextInputEx + ); + if (EFI_ERROR (Status)) { + return Status; + } + // // Find mouse in System Table ConsoleInHandle // @@ -1707,12 +1728,17 @@ Returns: // // below will call the five components' init function // - Status = MainTitleBarInit (NULL); + Status = MainTitleBarInit (L"UEFI HEXEDIT"); if (EFI_ERROR (Status)) { ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HEXEDIT_LIBEDITOR_MAINEDITOR_TITLE), gShellDebug1HiiHandle); return EFI_LOAD_ERROR; } + Status = ControlHotKeyInit (HexMainControlBasedMenuFunctions); + if (EFI_ERROR (Status)) { + ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HEXEDIT_LIBEDITOR_MAINEDITOR_MAINMENU), gShellDebug1HiiHandle); + return EFI_LOAD_ERROR; + } Status = MenuBarInit (HexEditorMenuItems); if (EFI_ERROR (Status)) { ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HEXEDIT_LIBEDITOR_MAINEDITOR_MAINMENU), gShellDebug1HiiHandle); @@ -1725,7 +1751,7 @@ Returns: return EFI_LOAD_ERROR; } - InputBarInit (); + InputBarInit (HMainEditor.TextInputEx); Status = HBufferImageInit (); if (EFI_ERROR (Status)) { @@ -1754,26 +1780,16 @@ Returns: return EFI_SUCCESS; } +/** + Cleanup function for MainEditor. + + @retval EFI_SUCCESS The operation was successful. + @retval EFI_LOAD_ERROR A load error occured. +**/ EFI_STATUS HMainEditorCleanup ( VOID ) -/*++ - -Routine Description: - - cleanup function for MainEditor - -Arguments: - - None - -Returns: - - EFI_SUCCESS - EFI_LOAD_ERROR - ---*/ { EFI_STATUS Status; @@ -1813,25 +1829,15 @@ Returns: return EFI_SUCCESS; } +/** + Refresh function for MainEditor. + + @retval EFI_SUCCESS The operation was successful. +**/ EFI_STATUS HMainEditorRefresh ( VOID ) -/*++ - -Routine Description: - - Refresh function for MainEditor - -Arguments: - - None - -Returns: - - EFI_SUCCESS - ---*/ { BOOLEAN NameChange; BOOLEAN ReadChange; @@ -1839,12 +1845,28 @@ Returns: NameChange = FALSE; ReadChange = FALSE; - if ( HMainEditor.BufferImage->FileImage != NULL && - HMainEditor.BufferImage->FileImage->FileName != NULL && - HBufferImageBackupVar.FileImage != NULL && - HBufferImageBackupVar.FileImage->FileName != NULL && - StrCmp (HMainEditor.BufferImage->FileImage->FileName, HBufferImageBackupVar.FileImage->FileName) != 0 ) { - NameChange = TRUE; + if (HMainEditor.BufferImage->BufferType == FileTypeDiskBuffer) { + if (HMainEditor.BufferImage->DiskImage != NULL && + HBufferImageBackupVar.DiskImage != NULL && + (HMainEditor.BufferImage->DiskImage->Offset != HBufferImageBackupVar.DiskImage->Offset || + HMainEditor.BufferImage->DiskImage->Size != HBufferImageBackupVar.DiskImage->Size) ){ + NameChange = TRUE; + } + } else if (HMainEditor.BufferImage->BufferType == FileTypeMemBuffer) { + if (HMainEditor.BufferImage->MemImage != NULL && + HBufferImageBackupVar.MemImage != NULL && + (HMainEditor.BufferImage->MemImage->Offset != HBufferImageBackupVar.MemImage->Offset || + HMainEditor.BufferImage->MemImage->Size != HBufferImageBackupVar.MemImage->Size) ){ + NameChange = TRUE; + } + } else if (HMainEditor.BufferImage->BufferType == FileTypeFileBuffer) { + if ( HMainEditor.BufferImage->FileImage != NULL && + HMainEditor.BufferImage->FileImage->FileName != NULL && + HBufferImageBackupVar.FileImage != NULL && + HBufferImageBackupVar.FileImage->FileName != NULL && + StrCmp (HMainEditor.BufferImage->FileImage->FileName, HBufferImageBackupVar.FileImage->FileName) != 0 ) { + NameChange = TRUE; + } } if ( HMainEditor.BufferImage->FileImage != NULL && HBufferImageBackupVar.FileImage != NULL && @@ -1868,14 +1890,14 @@ Returns: || ReadChange ) { MainTitleBarRefresh ( - HMainEditor.BufferImage->BufferType == FileTypeFileBuffer?HMainEditor.BufferImage->FileImage->FileName:HMainEditor.BufferImage->BufferType == FileTypeDiskBuffer?HMainEditor.BufferImage->DiskImage->Name:NULL, + HMainEditor.BufferImage->BufferType == FileTypeFileBuffer&&HMainEditor.BufferImage->FileImage!=NULL?HMainEditor.BufferImage->FileImage->FileName:HMainEditor.BufferImage->BufferType == FileTypeDiskBuffer&&HMainEditor.BufferImage->DiskImage!=NULL?HMainEditor.BufferImage->DiskImage->Name:NULL, HMainEditor.BufferImage->BufferType, - HMainEditor.BufferImage->FileImage->ReadOnly, + (BOOLEAN)(HMainEditor.BufferImage->FileImage!=NULL?HMainEditor.BufferImage->FileImage->ReadOnly:FALSE), HMainEditor.BufferImage->Modified, HMainEditor.ScreenSize.Column, HMainEditor.ScreenSize.Row, - HMainEditor.BufferImage->BufferType == FileTypeDiskBuffer?HMainEditor.BufferImage->DiskImage->Offset:HMainEditor.BufferImage->BufferType == FileTypeMemBuffer?HMainEditor.BufferImage->MemImage->Offset:0, - HMainEditor.BufferImage->BufferType == FileTypeDiskBuffer?HMainEditor.BufferImage->DiskImage->Size :HMainEditor.BufferImage->BufferType == FileTypeMemBuffer?HMainEditor.BufferImage->MemImage->Size :0 + HMainEditor.BufferImage->BufferType == FileTypeDiskBuffer&&HMainEditor.BufferImage->DiskImage!=NULL?HMainEditor.BufferImage->DiskImage->Offset:HMainEditor.BufferImage->BufferType == FileTypeMemBuffer&&HMainEditor.BufferImage->MemImage!=NULL?HMainEditor.BufferImage->MemImage->Offset:0, + HMainEditor.BufferImage->BufferType == FileTypeDiskBuffer&&HMainEditor.BufferImage->DiskImage!=NULL?HMainEditor.BufferImage->DiskImage->Size :HMainEditor.BufferImage->BufferType == FileTypeMemBuffer&&HMainEditor.BufferImage->MemImage!=NULL?HMainEditor.BufferImage->MemImage->Size :0 ); HBufferImageRefresh (); } @@ -1888,17 +1910,14 @@ Returns: HEditorFirst, HMainEditor.ScreenSize.Row, HMainEditor.ScreenSize.Column, - 0, - 0, - TRUE + (UINTN)(-1), + (UINTN)(-1), + FALSE ); HBufferImageRefresh (); } if (HEditorFirst) { - MenuBarRefresh ( - HMainEditor.ScreenSize.Row, - HMainEditor.ScreenSize.Column); HBufferImageRefresh (); } @@ -1910,7 +1929,17 @@ Returns: return EFI_SUCCESS; } -STATIC +/** + Handle the mouse input. + + @param[in] MouseState The current mouse state. + @param[out] BeforeLeftButtonDown helps with selections. + + @retval EFI_SUCCESS The operation was successful. + @retval EFI_OUT_OF_RESOURCES A memory allocation occured. + @retval EFI_LOAD_ERROR A load error occured. + @retval EFI_NOT_FOUND The disk was not found. +**/ EFI_STATUS HMainEditorHandleMouseInput ( IN EFI_SIMPLE_POINTER_STATE MouseState, @@ -2046,7 +2075,7 @@ HMainEditorHandleMouseInput ( // // release LButton // - if (*BeforeLeftButtonDown == TRUE) { + if (*BeforeLeftButtonDown) { Action = TRUE; } // @@ -2062,31 +2091,23 @@ HMainEditorHandleMouseInput ( return EFI_NOT_FOUND; } +/** + Handle user key input. will route it to other components handle function. + + @retval EFI_SUCCESS The operation was successful. + @retval EFI_OUT_OF_RESOURCES A memory allocation occured. + @retval EFI_LOAD_ERROR A load error occured. +**/ EFI_STATUS HMainEditorKeyInput ( VOID ) -/*++ - -Routine Description: - - Handle user key input. will route it to other components handle function - -Arguments: - - None - -Returns: - - EFI_SUCCESS - EFI_LOAD_ERROR - EFI_OUT_OF_RESOURCES - ---*/ { - EFI_INPUT_KEY Key; + EFI_KEY_DATA KeyData; EFI_STATUS Status; EFI_SIMPLE_POINTER_STATE MouseState; + UINTN EventIndex; + BOOLEAN NoShiftState; BOOLEAN LengthChange; UINTN Size; UINTN OldSize; @@ -2142,11 +2163,11 @@ Returns: Status = HMainEditorHandleMouseInput (MouseState, &MouseIsDown); if (!EFI_ERROR (Status)) { - if (BeforeMouseIsDown == FALSE) { + if (!BeforeMouseIsDown) { // // mouse down // - if (MouseIsDown == TRUE) { + if (MouseIsDown) { FRow = HBufferImage.BufferPosition.Row; FCol = HBufferImage.BufferPosition.Column; SelectStartBackup = HMainEditor.SelectStart; @@ -2162,8 +2183,8 @@ Returns: // // begin to drag // - if (MouseIsDown == TRUE) { - if (FirstDown == TRUE) { + if (MouseIsDown) { + if (FirstDown) { if (MouseState.RelativeMovementX || MouseState.RelativeMovementY) { HMainEditor.SelectStart = 0; HMainEditor.SelectEnd = 0; @@ -2245,79 +2266,94 @@ Returns: } } - Status = gST->ConIn->ReadKeyStroke (gST->ConIn, &Key); - if (!EFI_ERROR (Status)) { - // - // dispatch to different components' key handling function - // so not everywhere has to set this variable - // - HBufferImageMouseNeedRefresh = TRUE; - - // - // clear previous status string - // - StatusBarSetRefresh(); - - if (Key.ScanCode == SCAN_NULL) { - Status = HBufferImageHandleInput (&Key); - } else if (((Key.ScanCode >= SCAN_UP) && (Key.ScanCode <= SCAN_PAGE_DOWN))) { - Status = HBufferImageHandleInput (&Key); - } else if (((Key.ScanCode >= SCAN_F1) && Key.ScanCode <= (SCAN_F12))) { - Status = MenuBarDispatchFunctionKey (&Key); - } else { - StatusBarSetStatusString (L"Unknown Command"); - - HBufferImageMouseNeedRefresh = FALSE; - } + Status = gBS->WaitForEvent (1, &HMainEditor.TextInputEx->WaitForKeyEx, &EventIndex); + if (!EFI_ERROR (Status) && EventIndex == 0) { + Status = HMainEditor.TextInputEx->ReadKeyStrokeEx (HMainEditor.TextInputEx, &KeyData); + if (!EFI_ERROR (Status)) { + // + // dispatch to different components' key handling function + // so not everywhere has to set this variable + // + HBufferImageMouseNeedRefresh = TRUE; - if (Status != EFI_SUCCESS && Status != EFI_OUT_OF_RESOURCES) { // - // not already has some error status + // clear previous status string + // + StatusBarSetRefresh(); // - if (StrCmp (L"", StatusBarGetString()) == 0) { - StatusBarSetStatusString (L"Disk Error. Try Again"); + // NoShiftState: TRUE when no shift key is pressed. + // + NoShiftState = ((KeyData.KeyState.KeyShiftState & EFI_SHIFT_STATE_VALID) == 0) || (KeyData.KeyState.KeyShiftState == EFI_SHIFT_STATE_VALID); + // + // dispatch to different components' key handling function + // + if (EFI_SUCCESS == MenuBarDispatchControlHotKey(&KeyData)) { + Status = EFI_SUCCESS; + } else if (NoShiftState && KeyData.Key.ScanCode == SCAN_NULL) { + Status = HBufferImageHandleInput (&KeyData.Key); + } else if (NoShiftState && ((KeyData.Key.ScanCode >= SCAN_UP) && (KeyData.Key.ScanCode <= SCAN_PAGE_DOWN))) { + Status = HBufferImageHandleInput (&KeyData.Key); + } else if (NoShiftState && ((KeyData.Key.ScanCode >= SCAN_F1) && KeyData.Key.ScanCode <= SCAN_F12)) { + Status = MenuBarDispatchFunctionKey (&KeyData.Key); + } else { + StatusBarSetStatusString (L"Unknown Command"); + + HBufferImageMouseNeedRefresh = FALSE; + } + + if (Status != EFI_SUCCESS && Status != EFI_OUT_OF_RESOURCES) { + // + // not already has some error status + // + if (StrCmp (L"", StatusBarGetString()) == 0) { + StatusBarSetStatusString (L"Disk Error. Try Again"); + } } } - } - // - // decide if has to set length warning - // - if (HBufferImage.BufferType != HBufferImageBackupVar.BufferType) { - LengthChange = FALSE; - } else { // - // still the old buffer + // decide if has to set length warning // - if (HBufferImage.BufferType != FileTypeFileBuffer) { - Size = HBufferImageGetTotalSize (); - - switch (HBufferImage.BufferType) { - case FileTypeDiskBuffer: - OldSize = HBufferImage.DiskImage->Size * HBufferImage.DiskImage->BlockSize; - break; + if (HBufferImage.BufferType != HBufferImageBackupVar.BufferType) { + LengthChange = FALSE; + } else { + // + // still the old buffer + // + if (HBufferImage.BufferType != FileTypeFileBuffer) { + Size = HBufferImageGetTotalSize (); + + switch (HBufferImage.BufferType) { + case FileTypeDiskBuffer: + OldSize = HBufferImage.DiskImage->Size * HBufferImage.DiskImage->BlockSize; + break; + + case FileTypeMemBuffer: + OldSize = HBufferImage.MemImage->Size; + break; + + default: + OldSize = 0; + break; + } - case FileTypeMemBuffer: - OldSize = HBufferImage.MemImage->Size; - break; - } + if (!LengthChange) { + if (OldSize != Size) { + StatusBarSetStatusString (L"Disk/Mem Buffer Length should not be changed"); + } + } - if (LengthChange == FALSE) { if (OldSize != Size) { - StatusBarSetStatusString (L"Disk/Mem Buffer Length should not be changed"); + LengthChange = TRUE; + } else { + LengthChange = FALSE; } } - - if (OldSize != Size) { - LengthChange = TRUE; - } else { - LengthChange = FALSE; - } } + // + // after handling, refresh editor + // + HMainEditorRefresh (); } - // - // after handling, refresh editor - // - HMainEditorRefresh (); } while (Status != EFI_OUT_OF_RESOURCES && !HEditorExit); @@ -2328,7 +2364,6 @@ Returns: Backup function for MainEditor. **/ VOID -EFIAPI HMainEditorBackup ( VOID )