[mirror_edk2.git] / ShellPkg / Library / UefiShellDebug1CommandsLib / HexEdit / MainHexEditor.c

/** @file\r
    Defines the Main Editor data type -\r
     - Global variables\r
     - Instances of the other objects of the editor\r
     - Main Interfaces\r
\r
  Copyright (c) 2005 - 2018, Intel Corporation. All rights reserved. <BR>\r
  SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include "HexEditor.h"\r
#include "EditStatusBar.h"\r
#include "EditInputBar.h"\r
\r
HEFI_EDITOR_COLOR_ATTRIBUTES    HOriginalColors;\r
INTN                            HOriginalMode;\r
\r
//\r
// the first time editor launch\r
//\r
BOOLEAN                         HEditorFirst;\r
\r
//\r
// it's time editor should exit\r
//\r
BOOLEAN                         HEditorExit;\r
\r
BOOLEAN                         HEditorMouseAction;\r
\r
extern HEFI_EDITOR_BUFFER_IMAGE HBufferImage;\r
extern HEFI_EDITOR_BUFFER_IMAGE HBufferImageBackupVar;\r
\r
extern BOOLEAN                  HBufferImageMouseNeedRefresh;\r
extern BOOLEAN                  HBufferImageNeedRefresh;\r
extern BOOLEAN                  HBufferImageOnlyLineNeedRefresh;\r
\r
HEFI_EDITOR_GLOBAL_EDITOR       HMainEditor;\r
HEFI_EDITOR_GLOBAL_EDITOR       HMainEditorBackupVar;\r
\r
//\r
// basic initialization for MainEditor\r
//\r
HEFI_EDITOR_GLOBAL_EDITOR       HMainEditorConst = {\r
  &HBufferImage,\r
  {\r
    {0, 0}\r
  },\r
  {\r
    0,\r
    0\r
  },\r
  NULL,\r
  FALSE,\r
  NULL,\r
  0,\r
  0,\r
  1,\r
  1\r
};\r
\r
/**\r
  Help info that will be displayed.\r
**/\r
EFI_STRING_ID  HexMainMenuHelpInfo[] = {\r
  STRING_TOKEN(STR_HEXEDIT_HELP_TITLE),\r
  STRING_TOKEN(STR_HEXEDIT_HELP_BLANK),\r
  STRING_TOKEN(STR_HEXEDIT_HELP_LIST_TITLE),\r
  STRING_TOKEN(STR_HEXEDIT_HELP_DIV),\r
  STRING_TOKEN(STR_HEXEDIT_HELP_GO_TO_OFFSET),\r
  STRING_TOKEN(STR_HEXEDIT_HELP_SAVE_BUFFER),\r
  STRING_TOKEN(STR_HEXEDIT_HELP_EXIT),\r
  STRING_TOKEN(STR_HEXEDIT_HELP_SELECT_START),\r
  STRING_TOKEN(STR_HEXEDIT_HELP_SELECT_END),\r
  STRING_TOKEN(STR_HEXEDIT_HELP_CUT),\r
  STRING_TOKEN(STR_HEXEDIT_HELP_PASTE),\r
  STRING_TOKEN(STR_HEXEDIT_HELP_OPEN_FILE),\r
  STRING_TOKEN(STR_HEXEDIT_HELP_OPEN_DISK),\r
  STRING_TOKEN(STR_HEXEDIT_HELP_OPEN_MEMORY),\r
  STRING_TOKEN(STR_HEXEDIT_HELP_BLANK),\r
  STRING_TOKEN(STR_HEXEDIT_HELP_EXIT_HELP),\r
  STRING_TOKEN(STR_HEXEDIT_HELP_BLANK),\r
  STRING_TOKEN(STR_HEXEDIT_HELP_BLANK),\r
  STRING_TOKEN(STR_HEXEDIT_HELP_BLANK),\r
  STRING_TOKEN(STR_HEXEDIT_HELP_BLANK),\r
  STRING_TOKEN(STR_HEXEDIT_HELP_BLANK),\r
  STRING_TOKEN(STR_HEXEDIT_HELP_BLANK),\r
  STRING_TOKEN(STR_HEXEDIT_HELP_DIV),\r
  0\r
};\r
\r
\r
/**\r
  show help menu.\r
\r
  @retval EFI_SUCCESS             The operation was successful.\r
**/\r
EFI_STATUS\r
HMainCommandDisplayHelp (\r
  VOID\r
  )\r
{\r
  INT32           CurrentLine;\r
  CHAR16          *InfoString;\r
  EFI_KEY_DATA    KeyData;\r
  EFI_STATUS      Status;\r
  UINTN           EventIndex;\r
\r
  //\r
  // print helpInfo\r
  //\r
  for (CurrentLine = 0; 0 != HexMainMenuHelpInfo[CurrentLine]; CurrentLine++) {\r
    InfoString = HiiGetString(gShellDebug1HiiHandle, HexMainMenuHelpInfo[CurrentLine]\r
, NULL);\r
    ShellPrintEx (0,CurrentLine+1,L"%E%s%N",InfoString);\r
  }\r
\r
  //\r
  // scan for ctrl+w\r
  //\r
  while (TRUE) {\r
    Status = gBS->WaitForEvent (1, &HMainEditor.TextInputEx->WaitForKeyEx, &EventIndex);\r
    if (EFI_ERROR (Status) || (EventIndex != 0)) {\r
      continue;\r
    }\r
    Status = HMainEditor.TextInputEx->ReadKeyStrokeEx (HMainEditor.TextInputEx, &KeyData);\r
    if (EFI_ERROR (Status)) {\r
      continue;\r
    }\r
\r
    if (((KeyData.KeyState.KeyShiftState & EFI_SHIFT_STATE_VALID) == 0) ||\r
        (KeyData.KeyState.KeyShiftState == EFI_SHIFT_STATE_VALID)) {\r
      //\r
      // For consoles that don't support/report shift state,\r
      // CTRL+W is translated to L'W' - L'A' + 1.\r
      //\r
      if (KeyData.Key.UnicodeChar == L'W' - L'A' + 1) {\r
        break;\r
      }\r
    } else if (((KeyData.KeyState.KeyShiftState & EFI_SHIFT_STATE_VALID) != 0) &&\r
               ((KeyData.KeyState.KeyShiftState & (EFI_LEFT_CONTROL_PRESSED | EFI_RIGHT_CONTROL_PRESSED)) != 0) &&\r
               ((KeyData.KeyState.KeyShiftState & ~(EFI_SHIFT_STATE_VALID | EFI_LEFT_CONTROL_PRESSED | EFI_RIGHT_CONTROL_PRESSED)) == 0)) {\r
      //\r
      // For consoles that supports/reports shift state,\r
      // make sure that only CONTROL shift key is pressed.\r
      //\r
      if ((KeyData.Key.UnicodeChar == 'w') || (KeyData.Key.UnicodeChar == 'W')) {\r
        break;\r
      }\r
    }\r
  }\r
\r
  // update screen with buffer's info\r
  HBufferImageNeedRefresh = TRUE;\r
  HBufferImageOnlyLineNeedRefresh = FALSE;\r
  HBufferImageRefresh ();\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Move cursor to specified lines.\r
\r
  @retval EFI_SUCCESS   The operation was successful.\r
**/\r
EFI_STATUS\r
HMainCommandGoToOffset (\r
  VOID\r
  )\r
{\r
  UINTN       Size;\r
  UINT64      Offset;\r
  EFI_STATUS  Status;\r
  UINTN       FRow;\r
  UINTN       FCol;\r
\r
  //\r
  // variable initialization\r
  //\r
  Size    = 0;\r
  Offset  = 0;\r
  FRow    = 0;\r
  FCol    = 0;\r
\r
  //\r
  // get offset\r
  //\r
  Status = InputBarSetPrompt (L"Go To Offset: ");\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  Status = InputBarSetStringSize (8);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  while (1) {\r
    Status = InputBarRefresh (HMainEditor.ScreenSize.Row, HMainEditor.ScreenSize.Column);\r
\r
    //\r
    // ESC pressed\r
    //\r
    if (Status == EFI_NOT_READY) {\r
\r
      return EFI_SUCCESS;\r
    }\r
    //\r
    // THE input string length should > 0\r
    //\r
    if (StrLen (InputBarGetString()) > 0) {\r
      Status = ShellConvertStringToUint64 (InputBarGetString(), &Offset, TRUE, FALSE);\r
      if (EFI_ERROR (Status)) {\r
        StatusBarSetStatusString (L"Invalid Offset");\r
        return EFI_SUCCESS;\r
      }\r
\r
      break;\r
    }\r
  }\r
\r
  Size = HBufferImageGetTotalSize ();\r
  if (Offset >= Size) {\r
    StatusBarSetStatusString (L"Invalid Offset");\r
    return EFI_SUCCESS;\r
  }\r
\r
  FRow  = (UINTN)DivU64x32(Offset , 0x10) + 1;\r
  FCol  = (UINTN)ModU64x32(Offset , 0x10) + 1;\r
\r
  HBufferImageMovePosition (FRow, FCol, TRUE);\r
\r
  HBufferImageNeedRefresh         = TRUE;\r
  HBufferImageOnlyLineNeedRefresh = FALSE;\r
  HBufferImageMouseNeedRefresh    = TRUE;\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Save current opened buffer.\r
  If is file buffer, you can save to current file name or\r
  save to another file name.\r
\r
  @retval EFI_SUCCESS             The operation was successful.\r
  @retval EFI_OUT_OF_RESOURCES    A memory allocation occured.\r
  @retval EFI_LOAD_ERROR          A load error occured.\r
**/\r
EFI_STATUS\r
HMainCommandSaveBuffer (\r
  VOID\r
  )\r
{\r
  EFI_STATUS          Status;\r
  BOOLEAN             Done;\r
  CHAR16              *FileName;\r
  BOOLEAN             OldFile;\r
  CHAR16              *Str;\r
  EFI_FILE_INFO       *Info;\r
  SHELL_FILE_HANDLE   ShellFileHandle;\r
\r
  if (HMainEditor.BufferImage->BufferType != FileTypeFileBuffer) {\r
    if (!HMainEditor.BufferImage->Modified) {\r
      return EFI_SUCCESS;\r
    }\r
\r
    Status = InputBarSetPrompt (L"Dangerous to save disk/mem buffer. Save (Yes/No/Cancel) ? ");\r
    if (EFI_ERROR (Status)) {\r
      return Status;\r
    }\r
    //\r
    // the answer is just one character\r
    //\r
    Status = InputBarSetStringSize (1);\r
    if (EFI_ERROR (Status)) {\r
      return Status;\r
    }\r
    //\r
    // loop for user's answer\r
    // valid answer is just 'y' 'Y', 'n' 'N', 'c' 'C'\r
    //\r
    while (1) {\r
      Status = InputBarRefresh (HMainEditor.ScreenSize.Row, HMainEditor.ScreenSize.Column);\r
\r
      //\r
      // ESC pressed\r
      //\r
      if (Status == EFI_NOT_READY) {\r
        return EFI_SUCCESS;\r
      }\r
\r
      switch (InputBarGetString()[0]) {\r
      case L'y':\r
      case L'Y':\r
        //\r
        // want to save this buffer first\r
        //\r
        Status = HBufferImageSave (\r
                  NULL,\r
                  HMainEditor.BufferImage->DiskImage->Name,\r
                  HMainEditor.BufferImage->DiskImage->Offset,\r
                  HMainEditor.BufferImage->DiskImage->Size,\r
                  HMainEditor.BufferImage->MemImage->Offset,\r
                  HMainEditor.BufferImage->MemImage->Size,\r
                  HMainEditor.BufferImage->BufferType\r
                  );\r
\r
        if (EFI_ERROR (Status)) {\r
          StatusBarSetStatusString (L"BufferSave: Problems Writing");\r
          return Status;\r
        }\r
\r
        return EFI_SUCCESS;\r
\r
      case L'n':\r
      case L'N':\r
        //\r
        // the file won't be saved\r
        //\r
        return EFI_SUCCESS;\r
\r
      case L'c':\r
      case L'C':\r
        return EFI_SUCCESS;\r
      }\r
      //\r
      // end of switch\r
      //\r
    }\r
    //\r
    // ENDOF WHILE\r
    //\r
  }\r
  //\r
  // ENDOF != FILEBUFFER\r
  //\r
  // This command will save currently opened file to disk.\r
  // You can choose save to another file name or just save to\r
  // current file name.\r
  // Below is the scenario of Save File command: (\r
  //    Suppose the old file name is A )\r
  // 1. An Input Bar will be prompted:    "File To Save: [ old file name]"\r
  //    IF user press ESC, Save File command ends .\r
  //    IF user press Enter, input file name will be A.\r
  //    IF user inputs a new file name B,  input file name will be B.\r
  //\r
  // 2. IF input file name is A, go to do Step 3.\r
  //    IF input file name is B, go to do Step 4.\r
  //\r
  // 3. IF A is read only, Status Bar will show "Access Denied"\r
  //       and Save File commands ends.\r
  //    IF A is not read only, save file buffer to disk\r
  //       and remove Modified flag in Title Bar , then Save File command ends.\r
  //\r
  // 4. IF B does not exist, create this file and save file buffer to it.\r
  //       Go to do Step 7.\r
  //    IF B exits, do Step 5.\r
  //\r
  // 5. An Input Bar will be prompted:\r
  //       "File Exists. Overwrite ( Yes/No/Cancel ) ?"\r
  //      IF user press 'y' or 'Y', do Step 6.\r
  //      IF user press 'n' or 'N', Save File commands ends.\r
  //      IF user press 'c' or 'C' or ESC, Save File commands ends.\r
  //\r
  // 6. IF B is a read-only file, Status Bar will show "Access Denied"\r
  //       and Save File commands ends.\r
  //    IF B can be read and write, save file buffer to B.\r
  //\r
  // 7. Update File Name field in Title Bar to B\r
  //       and remove the Modified flag in Title Bar.\r
  //\r
  Str = CatSPrint(NULL,\r
          L"File to Save: [%s]",\r
          HMainEditor.BufferImage->FileImage->FileName\r
          );\r
  if (Str == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  if (StrLen (Str) >= 50) {\r
    //\r
    // replace the long file name with "..."\r
    //\r
    Str[46] = L'.';\r
    Str[47] = L'.';\r
    Str[48] = L'.';\r
    Str[49] = L']';\r
    Str[50] = L'\0';\r
  }\r
\r
  Status = InputBarSetPrompt (Str);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  Status = InputBarSetStringSize (100);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
  //\r
  // get new file name\r
  //\r
  Status = InputBarRefresh (HMainEditor.ScreenSize.Row, HMainEditor.ScreenSize.Column);\r
\r
  //\r
  // if user pressed ESC\r
  //\r
  if (Status == EFI_NOT_READY) {\r
    SHELL_FREE_NON_NULL (Str);\r
    return EFI_SUCCESS;\r
  }\r
\r
  SHELL_FREE_NON_NULL (Str);\r
\r
  //\r
  // if just enter pressed, so think save to current file name\r
  //\r
  if (StrLen (InputBarGetString()) == 0) {\r
    FileName = CatSPrint(NULL,\r
                L"%s",\r
                HMainEditor.BufferImage->FileImage->FileName\r
                );\r
  } else {\r
    FileName = CatSPrint(NULL, L"%s", InputBarGetString());\r
  }\r
\r
  if (FileName == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  if (!IsValidFileName (FileName)) {\r
    StatusBarSetStatusString (L"Invalid File Name");\r
    SHELL_FREE_NON_NULL (FileName);\r
    return EFI_SUCCESS;\r
  }\r
\r
  OldFile = FALSE;\r
\r
  //\r
  // save to the old file\r
  //\r
  if (StringNoCaseCompare (\r
        &FileName,\r
        &HMainEditor.BufferImage->FileImage->FileName\r
        ) == 0) {\r
    OldFile = TRUE;\r
  }\r
\r
  if (OldFile) {\r
    //\r
    // if the file is read only, so can not write back to it.\r
    //\r
    if (HMainEditor.BufferImage->FileImage->ReadOnly) {\r
      StatusBarSetStatusString (L"Access Denied");\r
      SHELL_FREE_NON_NULL (FileName);\r
      return EFI_SUCCESS;\r
    }\r
  } else {\r
    Status = ShellOpenFileByName (FileName, &ShellFileHandle, EFI_FILE_MODE_READ, 0);\r
\r
    if (!EFI_ERROR (Status)) {\r
\r
      Info = ShellGetFileInfo(ShellFileHandle);\r
\r
      ShellCloseFile(&ShellFileHandle);\r
      //\r
      // check if read only\r
      //\r
      if (Info->Attribute & EFI_FILE_READ_ONLY) {\r
        StatusBarSetStatusString (L"Access Denied");\r
        SHELL_FREE_NON_NULL (FileName);\r
        return EFI_SUCCESS;\r
      }\r
\r
      SHELL_FREE_NON_NULL(Info);\r
      //\r
      // ask user whether to overwrite this file\r
      //\r
      Status = InputBarSetPrompt (L"File exists. Overwrite (Yes/No/Cancel) ? ");\r
      if (EFI_ERROR (Status)) {\r
        SHELL_FREE_NON_NULL (FileName);\r
        return Status;\r
      }\r
\r
      Status = InputBarSetStringSize (1);\r
      if (EFI_ERROR (Status)) {\r
        SHELL_FREE_NON_NULL (FileName);\r
        return Status;\r
      }\r
\r
      Done = FALSE;\r
      while (!Done) {\r
        Status = InputBarRefresh (HMainEditor.ScreenSize.Row, HMainEditor.ScreenSize.Column);\r
\r
        if (Status == EFI_NOT_READY) {\r
          SHELL_FREE_NON_NULL (FileName);\r
          return EFI_SUCCESS;\r
        }\r
\r
        switch (InputBarGetString()[0]) {\r
        case L'y':\r
        case L'Y':\r
          Done = TRUE;\r
          break;\r
        case L'n':\r
        case L'N':\r
          SHELL_FREE_NON_NULL (FileName);\r
          return EFI_SUCCESS;\r
        case L'c':\r
        case L'C':\r
          SHELL_FREE_NON_NULL (FileName);\r
          return EFI_SUCCESS;\r
        } // switch\r
      } // while\r
    } // if opened existing file\r
  } // if OldFile\r
\r
  //\r
  // save file back to disk\r
  //\r
  Status = HBufferImageSave (\r
            FileName,\r
            HMainEditor.BufferImage->DiskImage->Name,\r
            HMainEditor.BufferImage->DiskImage->Offset,\r
            HMainEditor.BufferImage->DiskImage->Size,\r
            HMainEditor.BufferImage->MemImage->Offset,\r
            HMainEditor.BufferImage->MemImage->Size,\r
            FileTypeFileBuffer\r
            );\r
  SHELL_FREE_NON_NULL (FileName);\r
\r
  if (EFI_ERROR (Status)) {\r
    return EFI_LOAD_ERROR;\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Load a disk buffer editor.\r
\r
  @retval EFI_SUCCESS             The operation was successful.\r
  @retval EFI_OUT_OF_RESOURCES    A memory allocation occured.\r
  @retval EFI_LOAD_ERROR          A load error occured.\r
**/\r
EFI_STATUS\r
HMainCommandSelectStart (\r
  VOID\r
  )\r
{\r
  UINTN Start;\r
\r
  Start = (HMainEditor.BufferImage->BufferPosition.Row - 1) * 0x10 + HMainEditor.BufferImage->BufferPosition.Column;\r
\r
  //\r
  // last line\r
  //\r
  if (HMainEditor.BufferImage->CurrentLine->Link.ForwardLink == HMainEditor.BufferImage->ListHead) {\r
    if (HMainEditor.BufferImage->BufferPosition.Column > HMainEditor.BufferImage->CurrentLine->Size) {\r
      StatusBarSetStatusString (L"Invalid Block Start");\r
      return EFI_LOAD_ERROR;\r
    }\r
  }\r
\r
  if (HMainEditor.SelectEnd != 0 && Start > HMainEditor.SelectEnd) {\r
    StatusBarSetStatusString (L"Invalid Block Start");\r
    return EFI_LOAD_ERROR;\r
  }\r
\r
  HMainEditor.SelectStart = Start;\r
\r
  HBufferImageNeedRefresh = TRUE;\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Load a disk buffer editor.\r
\r
  @retval EFI_SUCCESS             The operation was successful.\r
  @retval EFI_OUT_OF_RESOURCES    A memory allocation occured.\r
  @retval EFI_LOAD_ERROR          A load error occured.\r
**/\r
EFI_STATUS\r
HMainCommandSelectEnd (\r
  VOID\r
  )\r
{\r
  UINTN End;\r
\r
  End = (HMainEditor.BufferImage->BufferPosition.Row - 1) * 0x10 + HMainEditor.BufferImage->BufferPosition.Column;\r
\r
  //\r
  // last line\r
  //\r
  if (HMainEditor.BufferImage->CurrentLine->Link.ForwardLink == HMainEditor.BufferImage->ListHead) {\r
    if (HMainEditor.BufferImage->BufferPosition.Column > HMainEditor.BufferImage->CurrentLine->Size) {\r
      StatusBarSetStatusString (L"Invalid Block End");\r
      return EFI_LOAD_ERROR;\r
    }\r
  }\r
\r
  if (HMainEditor.SelectStart != 0 && End < HMainEditor.SelectStart) {\r
    StatusBarSetStatusString (L"Invalid Block End");\r
    return EFI_SUCCESS;\r
  }\r
\r
  HMainEditor.SelectEnd   = End;\r
\r
  HBufferImageNeedRefresh = TRUE;\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Cut current line to clipboard.\r
\r
  @retval EFI_SUCCESS             The operation was successful.\r
  @retval EFI_OUT_OF_RESOURCES    A memory allocation occured.\r
  @retval EFI_LOAD_ERROR          A load error occured.\r
**/\r
EFI_STATUS\r
HMainCommandCut (\r
  VOID\r
  )\r
{\r
  UINTN             Index;\r
  LIST_ENTRY    *Link;\r
  UINT8             *Buffer;\r
  UINTN             Count;\r
\r
  //\r
  // not select, so not allowed to cut\r
  //\r
  if (HMainEditor.SelectStart == 0) {\r
    StatusBarSetStatusString (L"No Block is Selected");\r
    return EFI_SUCCESS;\r
  }\r
  //\r
  // not select, so not allowed to cut\r
  //\r
  if (HMainEditor.SelectEnd == 0) {\r
    StatusBarSetStatusString (L"No Block is Selected");\r
    return EFI_SUCCESS;\r
  }\r
\r
  Link = HMainEditor.BufferImage->ListHead->ForwardLink;\r
  for (Index = 0; Index < (HMainEditor.SelectEnd - 1) / 0x10; Index++) {\r
    Link = Link->ForwardLink;\r
  }\r
\r
  Count   = HMainEditor.SelectEnd - HMainEditor.SelectStart + 1;\r
  Buffer  = AllocateZeroPool (Count);\r
  if (Buffer == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
  //\r
  // cut the selected area\r
  //\r
  HBufferImageDeleteCharacterFromBuffer (\r
    HMainEditor.SelectStart - 1,\r
    Count,\r
    Buffer\r
    );\r
\r
  //\r
  // put to clipboard\r
  //\r
  HClipBoardSet (Buffer, Count);\r
\r
  HBufferImageNeedRefresh         = TRUE;\r
  HBufferImageOnlyLineNeedRefresh = FALSE;\r
\r
  if (!HMainEditor.BufferImage->Modified) {\r
    HMainEditor.BufferImage->Modified = TRUE;\r
  }\r
\r
  //\r
  // now no select area\r
  //\r
  HMainEditor.SelectStart = 0;\r
  HMainEditor.SelectEnd   = 0;\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Paste line to file buffer.\r
\r
  @retval EFI_SUCCESS             The operation was successful.\r
  @retval EFI_OUT_OF_RESOURCES    A memory allocation occured.\r
  @retval EFI_LOAD_ERROR          A load error occured.\r
**/\r
EFI_STATUS\r
HMainCommandPaste (\r
  VOID\r
  )\r
{\r
\r
  BOOLEAN           OnlyLineRefresh;\r
  HEFI_EDITOR_LINE  *Line;\r
  UINT8             *Buffer;\r
  UINTN             Count;\r
  UINTN             FPos;\r
\r
  Count = HClipBoardGet (&Buffer);\r
  if (Count == 0 || Buffer == NULL) {\r
    StatusBarSetStatusString (L"Nothing to Paste");\r
    return EFI_SUCCESS;\r
  }\r
\r
  Line            = HMainEditor.BufferImage->CurrentLine;\r
\r
  OnlyLineRefresh = FALSE;\r
  if (Line->Link.ForwardLink == HMainEditor.BufferImage->ListHead && Line->Size + Count < 0x10) {\r
    //\r
    // is at last line, and after paste will not exceed\r
    // so only this line need to be refreshed\r
    //\r
    // if after add, the line is 0x10, then will append a new line\r
    // so the whole page will need be refreshed\r
    //\r
    OnlyLineRefresh = TRUE;\r
\r
  }\r
\r
  FPos = 0x10 * (HMainEditor.BufferImage->BufferPosition.Row - 1) + HMainEditor.BufferImage->BufferPosition.Column - 1;\r
\r
  HBufferImageAddCharacterToBuffer (FPos, Count, Buffer);\r
\r
  if (OnlyLineRefresh) {\r
    HBufferImageNeedRefresh         = FALSE;\r
    HBufferImageOnlyLineNeedRefresh = TRUE;\r
  } else {\r
    HBufferImageNeedRefresh         = TRUE;\r
    HBufferImageOnlyLineNeedRefresh = FALSE;\r
  }\r
\r
  if (!HMainEditor.BufferImage->Modified) {\r
    HMainEditor.BufferImage->Modified = TRUE;\r
  }\r
\r
  return EFI_SUCCESS;\r
\r
}\r
\r
/**\r
  Exit editor.\r
\r
  @retval EFI_SUCCESS             The operation was successful.\r
  @retval EFI_OUT_OF_RESOURCES    A memory allocation occured.\r
  @retval EFI_LOAD_ERROR          A load error occured.\r
**/\r
EFI_STATUS\r
HMainCommandExit (\r
  VOID\r
  )\r
{\r
  EFI_STATUS  Status;\r
\r
  //\r
  // Below is the scenario of Exit command:\r
  // 1. IF currently opened file is not modified, exit the editor and\r
  //       Exit command ends.\r
  //    IF currently opened file is modified, do Step 2\r
  //\r
  // 2. An Input Bar will be prompted:\r
  //       "File modified. Save ( Yes/No/Cancel )?"\r
  //      IF user press 'y' or 'Y', currently opened file will be saved and\r
  //         Editor exits\r
  //      IF user press 'n' or 'N', currently opened file will not be saved\r
  //         and Editor exits.\r
  //      IF user press 'c' or 'C' or ESC, Exit command ends.\r
  //\r
  //\r
  // if file has been modified, so will prompt user\r
  //       whether to save the changes\r
  //\r
  if (HMainEditor.BufferImage->Modified) {\r
\r
    Status = InputBarSetPrompt (L"Buffer modified. Save (Yes/No/Cancel) ? ");\r
    if (EFI_ERROR (Status)) {\r
      return Status;\r
    }\r
\r
    Status = InputBarSetStringSize (1);\r
    if (EFI_ERROR (Status)) {\r
      return Status;\r
    }\r
\r
    while (1) {\r
      Status = InputBarRefresh (HMainEditor.ScreenSize.Row, HMainEditor.ScreenSize.Column);\r
\r
      //\r
      // ESC pressed\r
      //\r
      if (Status == EFI_NOT_READY) {\r
        return EFI_SUCCESS;\r
      }\r
\r
      switch (InputBarGetString()[0]) {\r
      case L'y':\r
      case L'Y':\r
        //\r
        // write file back to disk\r
        //\r
        Status = HBufferImageSave (\r
                  HMainEditor.BufferImage->FileImage->FileName,\r
                  HMainEditor.BufferImage->DiskImage->Name,\r
                  HMainEditor.BufferImage->DiskImage->Offset,\r
                  HMainEditor.BufferImage->DiskImage->Size,\r
                  HMainEditor.BufferImage->MemImage->Offset,\r
                  HMainEditor.BufferImage->MemImage->Size,\r
                  HMainEditor.BufferImage->BufferType\r
                  );\r
        if (!EFI_ERROR (Status)) {\r
          HEditorExit = TRUE;\r
        }\r
\r
        return Status;\r
\r
      case L'n':\r
      case L'N':\r
        HEditorExit = TRUE;\r
        return EFI_SUCCESS;\r
\r
      case L'c':\r
      case L'C':\r
        return EFI_SUCCESS;\r
\r
      }\r
    }\r
  }\r
\r
  HEditorExit = TRUE;\r
  return EFI_SUCCESS;\r
\r
}\r
\r
/**\r
  Load a file from disk to editor.\r
\r
  @retval EFI_SUCCESS             The operation was successful.\r
  @retval EFI_OUT_OF_RESOURCES    A memory allocation occured.\r
  @retval EFI_LOAD_ERROR          A load error occured.\r
**/\r
EFI_STATUS\r
HMainCommandOpenFile (\r
  VOID\r
  )\r
{\r
  BOOLEAN                         Done;\r
  EFI_STATUS                      Status;\r
  EDIT_FILE_TYPE                  BufferType;\r
\r
  BufferType = HMainEditor.BufferImage->BufferType;\r
\r
  //\r
  //  This command will open a file from current working directory.\r
  //  Read-only file can also be opened. But it can not be modified.\r
  // Below is the scenario of Open File command:\r
  // 1. IF currently opened file has not been modified, directly go to step .\r
  //  IF currently opened file has been modified, an Input Bar will be\r
  //     prompted as :\r
  //      "File Modified. Save ( Yes/No/Cancel) ?"\r
  //          IF user press 'y' or 'Y', currently opened file will be saved.\r
  //          IF user press 'n' or 'N', currently opened file will\r
  //             not be saved.\r
  //          IF user press 'c' or 'C' or ESC, Open File command ends and\r
  //             currently opened file is still opened.\r
  //\r
  // 2. An Input Bar will be prompted as :  "File Name to Open: "\r
  //      IF user press ESC, Open File command ends and\r
  //         currently opened file is still opened.\r
  //      Any other inputs with a Return will cause\r
  //          currently opened file close.\r
  //\r
  // 3. IF user input file name is an existing file ,\r
  //       this file will be read and opened.\r
  //    IF user input file name is a new file, this file will be created\r
  //       and opened. This file's type ( UNICODE or ASCII ) is the same with\r
  //       the old file.\r
  //\r
  //\r
  // if current file is modified, so you need to choose whether to\r
  //    save it first.\r
  //\r
  if (HMainEditor.BufferImage->Modified) {\r
\r
    Status = InputBarSetPrompt (L"Buffer modified. Save (Yes/No/Cancel) ? ");\r
    if (EFI_ERROR (Status)) {\r
      return Status;\r
    }\r
    //\r
    // the answer is just one character\r
    //\r
    Status = InputBarSetStringSize (1);\r
    if (EFI_ERROR (Status)) {\r
      return Status;\r
    }\r
    //\r
    // loop for user's answer\r
    // valid answer is just 'y' 'Y', 'n' 'N', 'c' 'C'\r
    //\r
    Done = FALSE;\r
    while (!Done) {\r
      Status = InputBarRefresh (HMainEditor.ScreenSize.Row, HMainEditor.ScreenSize.Column);\r
\r
      //\r
      // ESC pressed\r
      //\r
      if (Status == EFI_NOT_READY) {\r
        return EFI_SUCCESS;\r
      }\r
\r
      switch (InputBarGetString()[0]) {\r
      case L'y':\r
      case L'Y':\r
        //\r
        // want to save this buffer first\r
        //\r
        Status = HBufferImageSave (\r
                  HMainEditor.BufferImage->FileImage->FileName,\r
                  HMainEditor.BufferImage->DiskImage->Name,\r
                  HMainEditor.BufferImage->DiskImage->Offset,\r
                  HMainEditor.BufferImage->DiskImage->Size,\r
                  HMainEditor.BufferImage->MemImage->Offset,\r
                  HMainEditor.BufferImage->MemImage->Size,\r
                  HMainEditor.BufferImage->BufferType\r
                  );\r
        if (EFI_ERROR (Status)) {\r
          return Status;\r
        }\r
\r
        MainTitleBarRefresh (\r
          HMainEditor.BufferImage->BufferType == FileTypeFileBuffer?HMainEditor.BufferImage->FileImage->FileName:HMainEditor.BufferImage->BufferType == FileTypeDiskBuffer?HMainEditor.BufferImage->DiskImage->Name:NULL,\r
          HMainEditor.BufferImage->BufferType,\r
          HMainEditor.BufferImage->FileImage->ReadOnly,\r
          FALSE,\r
          HMainEditor.ScreenSize.Column,\r
          HMainEditor.ScreenSize.Row,\r
          HMainEditor.BufferImage->BufferType == FileTypeDiskBuffer?HMainEditor.BufferImage->DiskImage->Offset:HMainEditor.BufferImage->BufferType == FileTypeMemBuffer?HMainEditor.BufferImage->MemImage->Offset:0,\r
          HMainEditor.BufferImage->BufferType == FileTypeDiskBuffer?HMainEditor.BufferImage->DiskImage->Size  :HMainEditor.BufferImage->BufferType == FileTypeMemBuffer?HMainEditor.BufferImage->MemImage->Size  :0\r
          );\r
        Done = TRUE;\r
        break;\r
\r
      case L'n':\r
      case L'N':\r
        //\r
        // the file won't be saved\r
        //\r
        Done = TRUE;\r
        break;\r
\r
      case L'c':\r
      case L'C':\r
        return EFI_SUCCESS;\r
      }\r
    }\r
  }\r
  //\r
  // TO get the open file name\r
  //\r
  Status = InputBarSetPrompt (L"File Name to Open: ");\r
  if (EFI_ERROR (Status)) {\r
    HBufferImageRead (\r
      HMainEditor.BufferImage->FileImage->FileName,\r
      HMainEditor.BufferImage->DiskImage->Name,\r
      HMainEditor.BufferImage->DiskImage->Offset,\r
      HMainEditor.BufferImage->DiskImage->Size,\r
      HMainEditor.BufferImage->MemImage->Offset,\r
      HMainEditor.BufferImage->MemImage->Size,\r
      BufferType,\r
      TRUE\r
      );\r
    return Status;\r
  }\r
\r
  Status = InputBarSetStringSize (100);\r
  if (EFI_ERROR (Status)) {\r
    Status = HBufferImageRead (\r
              HMainEditor.BufferImage->FileImage->FileName,\r
              HMainEditor.BufferImage->DiskImage->Name,\r
              HMainEditor.BufferImage->DiskImage->Offset,\r
              HMainEditor.BufferImage->DiskImage->Size,\r
              HMainEditor.BufferImage->MemImage->Offset,\r
              HMainEditor.BufferImage->MemImage->Size,\r
              BufferType,\r
              TRUE\r
              );\r
    return Status;\r
  }\r
\r
  while (1) {\r
    Status = InputBarRefresh (HMainEditor.ScreenSize.Row, HMainEditor.ScreenSize.Column);\r
\r
    //\r
    // ESC pressed\r
    //\r
    if (Status == EFI_NOT_READY) {\r
      Status = HBufferImageRead (\r
                HMainEditor.BufferImage->FileImage->FileName,\r
                HMainEditor.BufferImage->DiskImage->Name,\r
                HMainEditor.BufferImage->DiskImage->Offset,\r
                HMainEditor.BufferImage->DiskImage->Size,\r
                HMainEditor.BufferImage->MemImage->Offset,\r
                HMainEditor.BufferImage->MemImage->Size,\r
                BufferType,\r
                TRUE\r
                );\r
\r
      return Status;\r
    }\r
    //\r
    // THE input string length should > 0\r
    //\r
    if (StrLen (InputBarGetString()) > 0) {\r
      //\r
      // CHECK if filename's valid\r
      //\r
      if (!IsValidFileName (InputBarGetString())) {\r
        HBufferImageRead (\r
          HMainEditor.BufferImage->FileImage->FileName,\r
          HMainEditor.BufferImage->DiskImage->Name,\r
          HMainEditor.BufferImage->DiskImage->Offset,\r
          HMainEditor.BufferImage->DiskImage->Size,\r
          HMainEditor.BufferImage->MemImage->Offset,\r
          HMainEditor.BufferImage->MemImage->Size,\r
          BufferType,\r
          TRUE\r
          );\r
\r
        StatusBarSetStatusString (L"Invalid File Name");\r
        return EFI_SUCCESS;\r
      }\r
\r
      break;\r
    }\r
  }\r
  //\r
  // read from disk\r
  //\r
  Status = HBufferImageRead (\r
            InputBarGetString(),\r
            HMainEditor.BufferImage->DiskImage->Name,\r
            HMainEditor.BufferImage->DiskImage->Offset,\r
            HMainEditor.BufferImage->DiskImage->Size,\r
            HMainEditor.BufferImage->MemImage->Offset,\r
            HMainEditor.BufferImage->MemImage->Size,\r
            FileTypeFileBuffer,\r
            FALSE\r
            );\r
\r
  if (EFI_ERROR (Status)) {\r
    HBufferImageRead (\r
      HMainEditor.BufferImage->FileImage->FileName,\r
      HMainEditor.BufferImage->DiskImage->Name,\r
      HMainEditor.BufferImage->DiskImage->Offset,\r
      HMainEditor.BufferImage->DiskImage->Size,\r
      HMainEditor.BufferImage->MemImage->Offset,\r
      HMainEditor.BufferImage->MemImage->Size,\r
      BufferType,\r
      TRUE\r
      );\r
\r
    return EFI_LOAD_ERROR;\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Load a disk buffer editor.\r
\r
  @retval EFI_SUCCESS             The operation was successful.\r
  @retval EFI_OUT_OF_RESOURCES    A memory allocation occured.\r
  @retval EFI_LOAD_ERROR          A load error occured.\r
  @retval EFI_NOT_FOUND           The disk was not found.\r
**/\r
EFI_STATUS\r
HMainCommandOpenDisk (\r
  VOID\r
  )\r
{\r
  UINT64                          Size;\r
  UINT64                          Offset;\r
  CHAR16                          *DeviceName;\r
  EFI_STATUS                      Status;\r
  BOOLEAN                         Done;\r
\r
  EDIT_FILE_TYPE                  BufferType;\r
\r
  //\r
  // variable initialization\r
  //\r
  Size        = 0;\r
  Offset      = 0;\r
  BufferType  = HMainEditor.BufferImage->BufferType;\r
\r
  //\r
  // if current file is modified, so you need to choose\r
  // whether to save it first.\r
  //\r
  if (HMainEditor.BufferImage->Modified) {\r
\r
    Status = InputBarSetPrompt (L"Buffer modified. Save (Yes/No/Cancel) ? ");\r
    if (EFI_ERROR (Status)) {\r
      return Status;\r
    }\r
    //\r
    // the answer is just one character\r
    //\r
    Status = InputBarSetStringSize (1);\r
    if (EFI_ERROR (Status)) {\r
      return Status;\r
    }\r
    //\r
    // loop for user's answer\r
    // valid answer is just 'y' 'Y', 'n' 'N', 'c' 'C'\r
    //\r
    Done = FALSE;\r
    while (!Done) {\r
      Status = InputBarRefresh (HMainEditor.ScreenSize.Row, HMainEditor.ScreenSize.Column);\r
\r
      //\r
      // ESC pressed\r
      //\r
      if (Status == EFI_NOT_READY) {\r
        return EFI_SUCCESS;\r
      }\r
\r
      switch (InputBarGetString()[0]) {\r
      case L'y':\r
      case L'Y':\r
        //\r
        // want to save this buffer first\r
        //\r
        Status = HBufferImageSave (\r
                  HMainEditor.BufferImage->FileImage->FileName,\r
                  HMainEditor.BufferImage->DiskImage->Name,\r
                  HMainEditor.BufferImage->DiskImage->Offset,\r
                  HMainEditor.BufferImage->DiskImage->Size,\r
                  HMainEditor.BufferImage->MemImage->Offset,\r
                  HMainEditor.BufferImage->MemImage->Size,\r
                  BufferType\r
                  );\r
        if (EFI_ERROR (Status)) {\r
          return Status;\r
        }\r
\r
        MainTitleBarRefresh (\r
          HMainEditor.BufferImage->BufferType == FileTypeFileBuffer?HMainEditor.BufferImage->FileImage->FileName:HMainEditor.BufferImage->BufferType == FileTypeDiskBuffer?HMainEditor.BufferImage->DiskImage->Name:NULL,\r
          HMainEditor.BufferImage->BufferType,\r
          HMainEditor.BufferImage->FileImage->ReadOnly,\r
          FALSE,\r
          HMainEditor.ScreenSize.Column,\r
          HMainEditor.ScreenSize.Row,\r
          HMainEditor.BufferImage->BufferType == FileTypeDiskBuffer?HMainEditor.BufferImage->DiskImage->Offset:HMainEditor.BufferImage->BufferType == FileTypeMemBuffer?HMainEditor.BufferImage->MemImage->Offset:0,\r
          HMainEditor.BufferImage->BufferType == FileTypeDiskBuffer?HMainEditor.BufferImage->DiskImage->Size  :HMainEditor.BufferImage->BufferType == FileTypeMemBuffer?HMainEditor.BufferImage->MemImage->Size  :0\r
          );\r
        Done = TRUE;\r
        break;\r
\r
      case L'n':\r
      case L'N':\r
        //\r
        // the file won't be saved\r
        //\r
        Done = TRUE;\r
        break;\r
\r
      case L'c':\r
      case L'C':\r
        return EFI_SUCCESS;\r
      }\r
    }\r
  }\r
  //\r
  // get disk block device name\r
  //\r
  Status = InputBarSetPrompt (L"Block Device to Open: ");\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  Status = InputBarSetStringSize (100);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  while (1) {\r
    Status = InputBarRefresh (HMainEditor.ScreenSize.Row, HMainEditor.ScreenSize.Column);\r
\r
    //\r
    // ESC pressed\r
    //\r
    if (Status == EFI_NOT_READY) {\r
\r
      return EFI_SUCCESS;\r
    }\r
    //\r
    // THE input string length should > 0\r
    //\r
    if (StrLen (InputBarGetString()) > 0) {\r
      break;\r
    }\r
  }\r
\r
  DeviceName = CatSPrint(NULL, L"%s", InputBarGetString());\r
  if (DeviceName == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
  //\r
  // get starting offset\r
  //\r
  Status = InputBarSetPrompt (L"First Block No.: ");\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  Status = InputBarSetStringSize (16);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  while (1) {\r
    Status = InputBarRefresh (HMainEditor.ScreenSize.Row, HMainEditor.ScreenSize.Column);\r
\r
    //\r
    // ESC pressed\r
    //\r
    if (Status == EFI_NOT_READY) {\r
\r
      return EFI_SUCCESS;\r
    }\r
    //\r
    // THE input string length should > 0\r
    //\r
    if (StrLen (InputBarGetString()) > 0) {\r
      Status = ShellConvertStringToUint64 (InputBarGetString(), &Offset, TRUE, FALSE);\r
      if (EFI_ERROR (Status)) {\r
        continue;\r
      }\r
\r
      break;\r
    }\r
  }\r
  //\r
  // get Number of Blocks:\r
  //\r
  Status = InputBarSetPrompt (L"Number of Blocks: ");\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  Status = InputBarSetStringSize (8);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  while (1) {\r
    Status = InputBarRefresh (HMainEditor.ScreenSize.Row, HMainEditor.ScreenSize.Column);\r
\r
    //\r
    // ESC pressed\r
    //\r
    if (Status == EFI_NOT_READY) {\r
\r
      return EFI_SUCCESS;\r
    }\r
    //\r
    // THE input string length should > 0\r
    //\r
    if (StrLen (InputBarGetString()) > 0) {\r
      Status = ShellConvertStringToUint64 (InputBarGetString(), &Size, TRUE, FALSE);\r
      if (EFI_ERROR (Status)) {\r
        continue;\r
      }\r
\r
      if (Size == 0) {\r
        continue;\r
      }\r
\r
      break;\r
    }\r
  }\r
\r
  Status = HBufferImageRead (\r
            NULL,\r
            DeviceName,\r
            (UINTN)Offset,\r
            (UINTN)Size,\r
            0,\r
            0,\r
            FileTypeDiskBuffer,\r
            FALSE\r
            );\r
\r
  if (EFI_ERROR (Status)) {\r
\r
    HBufferImageRead (\r
      HMainEditor.BufferImage->FileImage->FileName,\r
      HMainEditor.BufferImage->DiskImage->Name,\r
      HMainEditor.BufferImage->DiskImage->Offset,\r
      HMainEditor.BufferImage->DiskImage->Size,\r
      HMainEditor.BufferImage->MemImage->Offset,\r
      HMainEditor.BufferImage->MemImage->Size,\r
      BufferType,\r
      TRUE\r
      );\r
    return EFI_NOT_FOUND;\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Load memory content to editor\r
\r
  @retval EFI_SUCCESS             The operation was successful.\r
  @retval EFI_OUT_OF_RESOURCES    A memory allocation occured.\r
  @retval EFI_LOAD_ERROR          A load error occured.\r
  @retval EFI_NOT_FOUND           The disk was not found.\r
**/\r
EFI_STATUS\r
HMainCommandOpenMemory (\r
  VOID\r
  )\r
{\r
  UINT64                          Size;\r
  UINT64                          Offset;\r
  EFI_STATUS                      Status;\r
  BOOLEAN                         Done;\r
  EDIT_FILE_TYPE                  BufferType;\r
\r
  //\r
  // variable initialization\r
  //\r
  Size        = 0;\r
  Offset      = 0;\r
  BufferType  = HMainEditor.BufferImage->BufferType;\r
\r
  //\r
  // if current buffer is modified, so you need to choose\r
  // whether to save it first.\r
  //\r
  if (HMainEditor.BufferImage->Modified) {\r
\r
    Status = InputBarSetPrompt (L"Buffer modified. Save (Yes/No/Cancel) ? ");\r
    if (EFI_ERROR (Status)) {\r
      return Status;\r
    }\r
    //\r
    // the answer is just one character\r
    //\r
    Status = InputBarSetStringSize (1);\r
    if (EFI_ERROR (Status)) {\r
      return Status;\r
    }\r
    //\r
    // loop for user's answer\r
    // valid answer is just 'y' 'Y', 'n' 'N', 'c' 'C'\r
    //\r
    Done = FALSE;\r
    while (!Done) {\r
      Status = InputBarRefresh (HMainEditor.ScreenSize.Row, HMainEditor.ScreenSize.Column);\r
\r
      //\r
      // ESC pressed\r
      //\r
      if (Status == EFI_NOT_READY) {\r
        return EFI_SUCCESS;\r
      }\r
\r
      switch (InputBarGetString()[0]) {\r
      case L'y':\r
      case L'Y':\r
        //\r
        // want to save this buffer first\r
        //\r
        Status = HBufferImageSave (\r
                  HMainEditor.BufferImage->FileImage->FileName,\r
                  HMainEditor.BufferImage->DiskImage->Name,\r
                  HMainEditor.BufferImage->DiskImage->Offset,\r
                  HMainEditor.BufferImage->DiskImage->Size,\r
                  HMainEditor.BufferImage->MemImage->Offset,\r
                  HMainEditor.BufferImage->MemImage->Size,\r
                  BufferType\r
                  );\r
        if (EFI_ERROR (Status)) {\r
          return Status;\r
        }\r
\r
        MainTitleBarRefresh (\r
          HMainEditor.BufferImage->BufferType == FileTypeFileBuffer?HMainEditor.BufferImage->FileImage->FileName:HMainEditor.BufferImage->BufferType == FileTypeDiskBuffer?HMainEditor.BufferImage->DiskImage->Name:NULL,\r
          HMainEditor.BufferImage->BufferType,\r
          HMainEditor.BufferImage->FileImage->ReadOnly,\r
          FALSE,\r
          HMainEditor.ScreenSize.Column,\r
          HMainEditor.ScreenSize.Row,\r
          HMainEditor.BufferImage->BufferType == FileTypeDiskBuffer?HMainEditor.BufferImage->DiskImage->Offset:HMainEditor.BufferImage->BufferType == FileTypeMemBuffer?HMainEditor.BufferImage->MemImage->Offset:0,\r
          HMainEditor.BufferImage->BufferType == FileTypeDiskBuffer?HMainEditor.BufferImage->DiskImage->Size  :HMainEditor.BufferImage->BufferType == FileTypeMemBuffer?HMainEditor.BufferImage->MemImage->Size  :0\r
          );\r
        Done = TRUE;\r
        break;\r
\r
      case L'n':\r
      case L'N':\r
        //\r
        // the file won't be saved\r
        //\r
        Done = TRUE;\r
        break;\r
\r
      case L'c':\r
      case L'C':\r
        return EFI_SUCCESS;\r
      }\r
    }\r
  }\r
  //\r
  // get starting offset\r
  //\r
  Status = InputBarSetPrompt (L"Starting Offset: ");\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  Status = InputBarSetStringSize (8);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  while (1) {\r
    Status = InputBarRefresh (HMainEditor.ScreenSize.Row, HMainEditor.ScreenSize.Column);\r
\r
    //\r
    // ESC pressed\r
    //\r
    if (Status == EFI_NOT_READY) {\r
\r
      return EFI_SUCCESS;\r
    }\r
    //\r
    // THE input string length should > 0\r
    //\r
    if (StrLen (InputBarGetString()) > 0) {\r
      Status = ShellConvertStringToUint64 (InputBarGetString(), &Offset, TRUE, FALSE);\r
      if (EFI_ERROR (Status)) {\r
        continue;\r
      }\r
\r
      break;\r
    }\r
  }\r
  //\r
  // get Number of Blocks:\r
  //\r
  Status = InputBarSetPrompt (L"Buffer Size: ");\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  Status = InputBarSetStringSize (8);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  while (1) {\r
    Status = InputBarRefresh (HMainEditor.ScreenSize.Row, HMainEditor.ScreenSize.Column);\r
\r
    //\r
    // ESC pressed\r
    //\r
    if (Status == EFI_NOT_READY) {\r
\r
      return EFI_SUCCESS;\r
    }\r
    //\r
    // THE input string length should > 0\r
    //\r
    if (StrLen (InputBarGetString()) > 0) {\r
      Status = ShellConvertStringToUint64 (InputBarGetString(), &Size, TRUE, FALSE);\r
      if (EFI_ERROR (Status)) {\r
        continue;\r
      }\r
\r
      if (Size == 0) {\r
        continue;\r
      }\r
\r
      break;\r
    }\r
  }\r
\r
  if ((Offset + Size - 1)> 0xffffffff) {\r
    StatusBarSetStatusString (L"Invalid parameter");\r
    return EFI_LOAD_ERROR;\r
  }\r
\r
  Status = HBufferImageRead (\r
            NULL,\r
            NULL,\r
            0,\r
            0,\r
            (UINTN)Offset,\r
            (UINTN)Size,\r
            FileTypeMemBuffer,\r
            FALSE\r
            );\r
\r
  if (EFI_ERROR (Status)) {\r
    StatusBarSetStatusString (L"Read Device Error!");\r
    HBufferImageRead (\r
      HMainEditor.BufferImage->FileImage->FileName,\r
      HMainEditor.BufferImage->DiskImage->Name,\r
      HMainEditor.BufferImage->DiskImage->Offset,\r
      HMainEditor.BufferImage->DiskImage->Size,\r
      HMainEditor.BufferImage->MemImage->Offset,\r
      HMainEditor.BufferImage->MemImage->Size,\r
      BufferType,\r
      TRUE\r
      );\r
    return EFI_NOT_FOUND;\r
  }\r
  return EFI_SUCCESS;\r
\r
}\r
\r
MENU_ITEM_FUNCTION HexMainControlBasedMenuFunctions[] = {\r
  NULL,\r
  NULL,                      /* Ctrl - A */\r
  NULL,                      /* Ctrl - B */\r
  NULL,                      /* Ctrl - C */\r
  HMainCommandSelectEnd,     /* Ctrl - D */\r
  HMainCommandDisplayHelp,   /* Ctrl - E */\r
  NULL,                      /* Ctrl - F */\r
  HMainCommandGoToOffset,    /* Ctrl - G */\r
  NULL,                      /* Ctrl - H */\r
  HMainCommandOpenDisk,      /* Ctrl - I */\r
  NULL,                      /* Ctrl - J */\r
  NULL,                      /* Ctrl - K */\r
  NULL,                      /* Ctrl - L */\r
  HMainCommandOpenMemory,    /* Ctrl - M */\r
  NULL,                      /* Ctrl - N */\r
  HMainCommandOpenFile,      /* Ctrl - O */\r
  NULL,                      /* Ctrl - P */\r
  HMainCommandExit,          /* Ctrl - Q */\r
  NULL,                      /* Ctrl - R */\r
  HMainCommandSaveBuffer,    /* Ctrl - S */\r
  HMainCommandSelectStart,   /* Ctrl - T */\r
  NULL,                      /* Ctrl - U */\r
  HMainCommandPaste,         /* Ctrl - V */\r
  NULL,                      /* Ctrl - W */\r
  HMainCommandCut,           /* Ctrl - X */\r
  NULL,                      /* Ctrl - Y */\r
  NULL,                      /* Ctrl - Z */\r
};\r
\r
CONST EDITOR_MENU_ITEM HexEditorMenuItems[] = {\r
  {\r
    STRING_TOKEN(STR_HEXEDIT_LIBMENUBAR_GO_TO_OFFSET),\r
    STRING_TOKEN(STR_EDIT_LIBMENUBAR_F1),\r
    HMainCommandGoToOffset\r
  },\r
  {\r
    STRING_TOKEN(STR_HEXEDIT_LIBMENUBAR_SAVE_BUFFER),\r
    STRING_TOKEN(STR_EDIT_LIBMENUBAR_F2),\r
    HMainCommandSaveBuffer\r
  },\r
  {\r
    STRING_TOKEN(STR_EDIT_LIBMENUBAR_EXIT),\r
    STRING_TOKEN(STR_EDIT_LIBMENUBAR_F3),\r
    HMainCommandExit\r
  },\r
\r
  {\r
    STRING_TOKEN(STR_HEXEDIT_LIBMENUBAR_SELECT_START),\r
    STRING_TOKEN(STR_EDIT_LIBMENUBAR_F4),\r
    HMainCommandSelectStart\r
  },\r
  {\r
    STRING_TOKEN(STR_HEXEDIT_LIBMENUBAR_SELECT_END),\r
    STRING_TOKEN(STR_EDIT_LIBMENUBAR_F5),\r
    HMainCommandSelectEnd\r
  },\r
  {\r
    STRING_TOKEN(STR_HEXEDIT_LIBMENUBAR_CUT),\r
    STRING_TOKEN(STR_EDIT_LIBMENUBAR_F6),\r
    HMainCommandCut\r
  },\r
  {\r
    STRING_TOKEN(STR_HEXEDIT_LIBMENUBAR_PASTE),\r
    STRING_TOKEN(STR_EDIT_LIBMENUBAR_F7),\r
    HMainCommandPaste\r
  },\r
\r
  {\r
    STRING_TOKEN(STR_HEXEDIT_LIBMENUBAR_OPEN_FILE),\r
    STRING_TOKEN(STR_EDIT_LIBMENUBAR_F8),\r
    HMainCommandOpenFile\r
  },\r
  {\r
    STRING_TOKEN(STR_HEXEDIT_LIBMENUBAR_OPEN_DISK),\r
    STRING_TOKEN(STR_EDIT_LIBMENUBAR_F9),\r
    HMainCommandOpenDisk\r
  },\r
  {\r
    STRING_TOKEN(STR_HEXEDIT_LIBMENUBAR_OPEN_MEMORY),\r
    STRING_TOKEN(STR_EDIT_LIBMENUBAR_F10),\r
    HMainCommandOpenMemory\r
  },\r
\r
  {\r
    0,\r
    0,\r
    NULL\r
  }\r
};\r
\r
/**\r
  Init function for MainEditor\r
\r
  @retval EFI_SUCCESS             The operation was successful.\r
  @retval EFI_LOAD_ERROR          A load error occured.\r
**/\r
EFI_STATUS\r
HMainEditorInit (\r
  VOID\r
  )\r
{\r
  EFI_STATUS  Status;\r
  EFI_HANDLE  *HandleBuffer;\r
  UINTN       HandleCount;\r
  UINTN       Index;\r
\r
  //\r
  // basic initialization\r
  //\r
  CopyMem (&HMainEditor, &HMainEditorConst, sizeof (HMainEditor));\r
\r
  //\r
  // set screen attributes\r
  //\r
  HMainEditor.ColorAttributes.Colors.Foreground = gST->ConOut->Mode->Attribute & 0x000000ff;\r
\r
  HMainEditor.ColorAttributes.Colors.Background = (UINT8) (gST->ConOut->Mode->Attribute >> 4);\r
\r
  HOriginalColors = HMainEditor.ColorAttributes.Colors;\r
\r
  HOriginalMode = gST->ConOut->Mode->Mode;\r
\r
  //\r
  // query screen size\r
  //\r
  gST->ConOut->QueryMode (\r
        gST->ConOut,\r
        gST->ConOut->Mode->Mode,\r
        &(HMainEditor.ScreenSize.Column),\r
        &(HMainEditor.ScreenSize.Row)\r
        );\r
\r
  //\r
  // Find TextInEx in System Table ConsoleInHandle\r
  // Per UEFI Spec, TextInEx is required for a console capable platform.\r
  //\r
  Status = gBS->HandleProtocol (\r
              gST->ConsoleInHandle,\r
              &gEfiSimpleTextInputExProtocolGuid,\r
              (VOID**)&HMainEditor.TextInputEx\r
              );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  //\r
  // Find mouse in System Table ConsoleInHandle\r
  //\r
  Status = gBS->HandleProtocol (\r
                gST->ConsoleInHandle,\r
                &gEfiSimplePointerProtocolGuid,\r
                (VOID**)&HMainEditor.MouseInterface\r
                );\r
  if (EFI_ERROR (Status)) {\r
    //\r
    // If there is no Simple Pointer Protocol on System Table\r
    //\r
    HandleBuffer = NULL;\r
    HMainEditor.MouseInterface = NULL;\r
    Status = gBS->LocateHandleBuffer (\r
                  ByProtocol,\r
                  &gEfiSimplePointerProtocolGuid,\r
                  NULL,\r
                  &HandleCount,\r
                  &HandleBuffer\r
                  );\r
    if (!EFI_ERROR (Status) && HandleCount > 0) {\r
      //\r
      // Try to find the first available mouse device\r
      //\r
      for (Index = 0; Index < HandleCount; Index++) {\r
        Status = gBS->HandleProtocol (\r
                      HandleBuffer[Index],\r
                      &gEfiSimplePointerProtocolGuid,\r
                      (VOID**)&HMainEditor.MouseInterface\r
                      );\r
        if (!EFI_ERROR (Status)) {\r
          break;\r
        }\r
      }\r
    }\r
    if (HandleBuffer != NULL) {\r
      FreePool (HandleBuffer);\r
    }\r
  }\r
\r
  if (!EFI_ERROR (Status) && HMainEditor.MouseInterface != NULL) {\r
    HMainEditor.MouseAccumulatorX  = 0;\r
    HMainEditor.MouseAccumulatorY  = 0;\r
    HMainEditor.MouseSupported     = TRUE;\r
  }\r
\r
  //\r
  // below will call the five components' init function\r
  //\r
  Status = MainTitleBarInit (L"UEFI HEXEDIT");\r
  if (EFI_ERROR (Status)) {\r
    ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HEXEDIT_LIBEDITOR_MAINEDITOR_TITLE), gShellDebug1HiiHandle);\r
    return EFI_LOAD_ERROR;\r
  }\r
\r
  Status = ControlHotKeyInit (HexMainControlBasedMenuFunctions);\r
  if (EFI_ERROR (Status)) {\r
    ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HEXEDIT_LIBEDITOR_MAINEDITOR_MAINMENU), gShellDebug1HiiHandle);\r
    return EFI_LOAD_ERROR;\r
  }\r
  Status = MenuBarInit (HexEditorMenuItems);\r
  if (EFI_ERROR (Status)) {\r
    ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HEXEDIT_LIBEDITOR_MAINEDITOR_MAINMENU), gShellDebug1HiiHandle);\r
    return EFI_LOAD_ERROR;\r
  }\r
\r
  Status = StatusBarInit ();\r
  if (EFI_ERROR (Status)) {\r
    ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HEXEDIT_LIBEDITOR_MAINEDITOR_STATUS), gShellDebug1HiiHandle);\r
    return EFI_LOAD_ERROR;\r
  }\r
\r
  InputBarInit (HMainEditor.TextInputEx);\r
\r
  Status = HBufferImageInit ();\r
  if (EFI_ERROR (Status)) {\r
    ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HEXEDIT_LIBEDITOR_MAINEDITOR_BUFFERIMAGE), gShellDebug1HiiHandle);\r
    return EFI_LOAD_ERROR;\r
  }\r
\r
  Status = HClipBoardInit ();\r
  if (EFI_ERROR (Status)) {\r
    ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HEXEDIT_LIBEDITOR_MAINEDITOR_CLIPBOARD), gShellDebug1HiiHandle);\r
    return EFI_LOAD_ERROR;\r
  }\r
  //\r
  // clear whole screen and enable cursor\r
  //\r
  gST->ConOut->ClearScreen (gST->ConOut);\r
  gST->ConOut->EnableCursor (gST->ConOut, TRUE);\r
\r
  //\r
  // initialize EditorFirst and EditorExit\r
  //\r
  HEditorFirst        = TRUE;\r
  HEditorExit         = FALSE;\r
  HEditorMouseAction  = FALSE;\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Cleanup function for MainEditor.\r
\r
  @retval EFI_SUCCESS             The operation was successful.\r
  @retval EFI_LOAD_ERROR          A load error occured.\r
**/\r
EFI_STATUS\r
HMainEditorCleanup (\r
  VOID\r
  )\r
{\r
  EFI_STATUS  Status;\r
\r
  //\r
  // call the five components' cleanup function\r
  //\r
  MainTitleBarCleanup ();\r
\r
  MenuBarCleanup ();\r
\r
  StatusBarCleanup ();\r
\r
  InputBarCleanup ();\r
\r
  Status = HBufferImageCleanup ();\r
  if (EFI_ERROR (Status)) {\r
    ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HEXEDIT_LIBEDITOR_BUFFERIMAGE_CLEAN), gShellDebug1HiiHandle);\r
  }\r
\r
  Status = HClipBoardCleanup ();\r
  if (EFI_ERROR (Status)) {\r
    ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HEXEDIT_LIBEDITOR_CLIPBOARD_CLEAN), gShellDebug1HiiHandle);\r
  }\r
  //\r
  // restore old mode\r
  //\r
  if (HOriginalMode != gST->ConOut->Mode->Mode) {\r
    gST->ConOut->SetMode (gST->ConOut, HOriginalMode);\r
  }\r
\r
  gST->ConOut->SetAttribute (\r
        gST->ConOut,\r
        EFI_TEXT_ATTR (HOriginalColors.Foreground, HOriginalColors.Background)\r
        );\r
  gST->ConOut->ClearScreen (gST->ConOut);\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Refresh function for MainEditor.\r
\r
  @retval EFI_SUCCESS             The operation was successful.\r
**/\r
EFI_STATUS\r
HMainEditorRefresh (\r
  VOID\r
  )\r
{\r
  BOOLEAN NameChange;\r
  BOOLEAN ReadChange;\r
\r
  NameChange = FALSE;\r
  ReadChange = FALSE;\r
\r
  if (HMainEditor.BufferImage->BufferType == FileTypeDiskBuffer) {\r
    if (HMainEditor.BufferImage->DiskImage != NULL &&\r
        HBufferImageBackupVar.DiskImage != NULL &&\r
        (HMainEditor.BufferImage->DiskImage->Offset != HBufferImageBackupVar.DiskImage->Offset ||\r
           HMainEditor.BufferImage->DiskImage->Size != HBufferImageBackupVar.DiskImage->Size) ){\r
      NameChange = TRUE;\r
    }\r
  } else if (HMainEditor.BufferImage->BufferType == FileTypeMemBuffer) {\r
    if (HMainEditor.BufferImage->MemImage != NULL &&\r
        HBufferImageBackupVar.MemImage != NULL &&\r
        (HMainEditor.BufferImage->MemImage->Offset != HBufferImageBackupVar.MemImage->Offset ||\r
           HMainEditor.BufferImage->MemImage->Size != HBufferImageBackupVar.MemImage->Size) ){\r
      NameChange = TRUE;\r
    }\r
  } else if (HMainEditor.BufferImage->BufferType == FileTypeFileBuffer) {\r
    if ( HMainEditor.BufferImage->FileImage != NULL &&\r
         HMainEditor.BufferImage->FileImage->FileName != NULL &&\r
         HBufferImageBackupVar.FileImage != NULL &&\r
         HBufferImageBackupVar.FileImage->FileName != NULL &&\r
         StrCmp (HMainEditor.BufferImage->FileImage->FileName, HBufferImageBackupVar.FileImage->FileName) != 0 ) {\r
      NameChange = TRUE;\r
    }\r
  }\r
  if ( HMainEditor.BufferImage->FileImage != NULL &&\r
       HBufferImageBackupVar.FileImage != NULL &&\r
       HMainEditor.BufferImage->FileImage->ReadOnly != HBufferImageBackupVar.FileImage->ReadOnly ) {\r
    ReadChange = TRUE;\r
  }\r
\r
  //\r
  // to aVOID screen flicker\r
  // the stall value is from experience\r
  //\r
  gBS->Stall (50);\r
\r
  //\r
  // call the components refresh function\r
  //\r
  if (HEditorFirst\r
    || NameChange\r
    || HMainEditor.BufferImage->BufferType != HBufferImageBackupVar.BufferType\r
    || HBufferImageBackupVar.Modified != HMainEditor.BufferImage->Modified\r
    || ReadChange ) {\r
\r
    MainTitleBarRefresh (\r
      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,\r
      HMainEditor.BufferImage->BufferType,\r
      (BOOLEAN)(HMainEditor.BufferImage->FileImage!=NULL?HMainEditor.BufferImage->FileImage->ReadOnly:FALSE),\r
      HMainEditor.BufferImage->Modified,\r
      HMainEditor.ScreenSize.Column,\r
      HMainEditor.ScreenSize.Row,\r
      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,\r
      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\r
      );\r
    HBufferImageRefresh ();\r
  }\r
  if (HEditorFirst\r
    || HBufferImageBackupVar.DisplayPosition.Row != HMainEditor.BufferImage->DisplayPosition.Row\r
    || HBufferImageBackupVar.DisplayPosition.Column != HMainEditor.BufferImage->DisplayPosition.Column\r
    || StatusBarGetRefresh()) {\r
\r
    StatusBarRefresh (\r
      HEditorFirst,\r
      HMainEditor.ScreenSize.Row,\r
      HMainEditor.ScreenSize.Column,\r
      (UINTN)(-1),\r
      (UINTN)(-1),\r
      FALSE\r
      );\r
    HBufferImageRefresh ();\r
  }\r
\r
  if (HEditorFirst) {\r
    HBufferImageRefresh ();\r
  }\r
\r
  //\r
  // EditorFirst is now set to FALSE\r
  //\r
  HEditorFirst = FALSE;\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Handle the mouse input.\r
\r
  @param[in] MouseState             The current mouse state.\r
  @param[out] BeforeLeftButtonDown  helps with selections.\r
\r
  @retval EFI_SUCCESS             The operation was successful.\r
  @retval EFI_OUT_OF_RESOURCES    A memory allocation occured.\r
  @retval EFI_LOAD_ERROR          A load error occured.\r
  @retval EFI_NOT_FOUND           The disk was not found.\r
**/\r
EFI_STATUS\r
HMainEditorHandleMouseInput (\r
  IN  EFI_SIMPLE_POINTER_STATE       MouseState,\r
  OUT BOOLEAN                        *BeforeLeftButtonDown\r
  )\r
{\r
\r
  INT32             TextX;\r
  INT32             TextY;\r
  UINTN             FRow;\r
  UINTN             FCol;\r
  BOOLEAN           HighBits;\r
  LIST_ENTRY    *Link;\r
  HEFI_EDITOR_LINE  *Line;\r
  UINTN             Index;\r
  BOOLEAN           Action;\r
\r
  Action = FALSE;\r
\r
  //\r
  // have mouse movement\r
  //\r
  if (MouseState.RelativeMovementX || MouseState.RelativeMovementY) {\r
    //\r
    // handle\r
    //\r
    TextX = HGetTextX (MouseState.RelativeMovementX);\r
    TextY = HGetTextY (MouseState.RelativeMovementY);\r
\r
    HBufferImageAdjustMousePosition (TextX, TextY);\r
\r
    Action = TRUE;\r
\r
  }\r
\r
  if (MouseState.LeftButton) {\r
    HighBits = HBufferImageIsAtHighBits (\r
                HMainEditor.BufferImage->MousePosition.Column,\r
                &FCol\r
                );\r
\r
    //\r
    // not at an movable place\r
    //\r
    if (FCol == 0) {\r
      //\r
      // now just move mouse pointer to legal position\r
      //\r
      //\r
      // move mouse position to legal position\r
      //\r
      HMainEditor.BufferImage->MousePosition.Column -= 10;\r
      if (HMainEditor.BufferImage->MousePosition.Column > 24) {\r
        HMainEditor.BufferImage->MousePosition.Column--;\r
        FCol = HMainEditor.BufferImage->MousePosition.Column / 3 + 1 + 1;\r
      } else {\r
        if (HMainEditor.BufferImage->MousePosition.Column < 24) {\r
          FCol = HMainEditor.BufferImage->MousePosition.Column / 3 + 1 + 1;\r
        } else {\r
          //\r
          // == 24\r
          //\r
          FCol = 9;\r
        }\r
      }\r
\r
      HighBits = TRUE;\r
\r
    }\r
\r
    FRow = HMainEditor.BufferImage->BufferPosition.Row +\r
      HMainEditor.BufferImage->MousePosition.Row -\r
      HMainEditor.BufferImage->DisplayPosition.Row;\r
\r
    if (HMainEditor.BufferImage->NumLines < FRow) {\r
      //\r
      // dragging\r
      //\r
      //\r
      // now just move mouse pointer to legal position\r
      //\r
      FRow      = HMainEditor.BufferImage->NumLines;\r
      HighBits  = TRUE;\r
    }\r
\r
    Link = HMainEditor.BufferImage->ListHead->ForwardLink;\r
    for (Index = 0; Index < FRow - 1; Index++) {\r
      Link = Link->ForwardLink;\r
    }\r
\r
    Line = CR (Link, HEFI_EDITOR_LINE, Link, EFI_EDITOR_LINE_LIST);\r
\r
    //\r
    // dragging\r
    //\r
    //\r
    // now just move mouse pointer to legal position\r
    //\r
    if (FCol > Line->Size) {\r
      if (*BeforeLeftButtonDown) {\r
        HighBits = FALSE;\r
\r
        if (Line->Size == 0) {\r
          if (FRow > 1) {\r
            FRow--;\r
            FCol = 16;\r
          } else {\r
            FRow  = 1;\r
            FCol  = 1;\r
          }\r
\r
        } else {\r
          FCol = Line->Size;\r
        }\r
      } else {\r
        FCol      = Line->Size + 1;\r
        HighBits  = TRUE;\r
      }\r
    }\r
\r
    HBufferImageMovePosition (FRow, FCol, HighBits);\r
\r
    HMainEditor.BufferImage->MousePosition.Row    = HMainEditor.BufferImage->DisplayPosition.Row;\r
\r
    HMainEditor.BufferImage->MousePosition.Column = HMainEditor.BufferImage->DisplayPosition.Column;\r
\r
    *BeforeLeftButtonDown                         = TRUE;\r
\r
    Action = TRUE;\r
  } else {\r
    //\r
    // else of if LButton\r
    //\r
    // release LButton\r
    //\r
    if (*BeforeLeftButtonDown) {\r
      Action = TRUE;\r
    }\r
    //\r
    // mouse up\r
    //\r
    *BeforeLeftButtonDown = FALSE;\r
  }\r
\r
  if (Action) {\r
    return EFI_SUCCESS;\r
  }\r
\r
  return EFI_NOT_FOUND;\r
}\r
\r
/**\r
  Handle user key input. will route it to other components handle function.\r
\r
  @retval EFI_SUCCESS             The operation was successful.\r
  @retval EFI_OUT_OF_RESOURCES    A memory allocation occured.\r
  @retval EFI_LOAD_ERROR          A load error occured.\r
**/\r
EFI_STATUS\r
HMainEditorKeyInput (\r
  VOID\r
  )\r
{\r
  EFI_KEY_DATA              KeyData;\r
  EFI_STATUS                Status;\r
  EFI_SIMPLE_POINTER_STATE  MouseState;\r
  BOOLEAN                   NoShiftState;\r
  BOOLEAN                   LengthChange;\r
  UINTN                     Size;\r
  UINTN                     OldSize;\r
  BOOLEAN                   BeforeMouseIsDown;\r
  BOOLEAN                   MouseIsDown;\r
  BOOLEAN                   FirstDown;\r
  BOOLEAN                   MouseDrag;\r
  UINTN                     FRow;\r
  UINTN                     FCol;\r
  UINTN                     SelectStartBackup;\r
  UINTN                     SelectEndBackup;\r
\r
  //\r
  // variable initialization\r
  //\r
  OldSize       = 0;\r
  FRow          = 0;\r
  FCol          = 0;\r
  LengthChange  = FALSE;\r
\r
  MouseIsDown   = FALSE;\r
  FirstDown     = FALSE;\r
  MouseDrag     = FALSE;\r
\r
  do {\r
\r
    Status              = EFI_SUCCESS;\r
\r
    HEditorMouseAction  = FALSE;\r
\r
    //\r
    // backup some key elements, so that can aVOID some refresh work\r
    //\r
    HMainEditorBackup ();\r
\r
    //\r
    // wait for user key input\r
    //\r
    //\r
    // change priority of checking mouse/keyboard activity dynamically\r
    // so prevent starvation of keyboard.\r
    // if last time, mouse moves then this time check keyboard\r
    //\r
    if (HMainEditor.MouseSupported) {\r
      Status = HMainEditor.MouseInterface->GetState (\r
                                            HMainEditor.MouseInterface,\r
                                            &MouseState\r
                                            );\r
      if (!EFI_ERROR (Status)) {\r
\r
        BeforeMouseIsDown = MouseIsDown;\r
\r
        Status            = HMainEditorHandleMouseInput (MouseState, &MouseIsDown);\r
\r
        if (!EFI_ERROR (Status)) {\r
          if (!BeforeMouseIsDown) {\r
            //\r
            // mouse down\r
            //\r
            if (MouseIsDown) {\r
              FRow              = HBufferImage.BufferPosition.Row;\r
              FCol              = HBufferImage.BufferPosition.Column;\r
              SelectStartBackup = HMainEditor.SelectStart;\r
              SelectEndBackup   = HMainEditor.SelectEnd;\r
\r
              FirstDown         = TRUE;\r
            }\r
          } else {\r
\r
            SelectStartBackup = HMainEditor.SelectStart;\r
            SelectEndBackup   = HMainEditor.SelectEnd;\r
\r
            //\r
            // begin to drag\r
            //\r
            if (MouseIsDown) {\r
              if (FirstDown) {\r
                if (MouseState.RelativeMovementX || MouseState.RelativeMovementY) {\r
                  HMainEditor.SelectStart = 0;\r
                  HMainEditor.SelectEnd   = 0;\r
                  HMainEditor.SelectStart = (FRow - 1) * 0x10 + FCol;\r
\r
                  MouseDrag               = TRUE;\r
                  FirstDown               = FALSE;\r
                }\r
              } else {\r
                if ((\r
                      (HBufferImage.BufferPosition.Row - 1) *\r
                    0x10 +\r
                    HBufferImage.BufferPosition.Column\r
                      ) >= HMainEditor.SelectStart\r
                        ) {\r
                  HMainEditor.SelectEnd = (HBufferImage.BufferPosition.Row - 1) *\r
                    0x10 +\r
                    HBufferImage.BufferPosition.Column;\r
                } else {\r
                  HMainEditor.SelectEnd = 0;\r
                }\r
              }\r
              //\r
              // end of if RelativeX/Y\r
              //\r
            } else {\r
              //\r
              // mouse is up\r
              //\r
              if (MouseDrag) {\r
                if (HBufferImageGetTotalSize () == 0) {\r
                  HMainEditor.SelectStart = 0;\r
                  HMainEditor.SelectEnd   = 0;\r
                  FirstDown               = FALSE;\r
                  MouseDrag               = FALSE;\r
                }\r
\r
                if ((\r
                      (HBufferImage.BufferPosition.Row - 1) *\r
                    0x10 +\r
                    HBufferImage.BufferPosition.Column\r
                      ) >= HMainEditor.SelectStart\r
                        ) {\r
                  HMainEditor.SelectEnd = (HBufferImage.BufferPosition.Row - 1) *\r
                    0x10 +\r
                    HBufferImage.BufferPosition.Column;\r
                } else {\r
                  HMainEditor.SelectEnd = 0;\r
                }\r
\r
                if (HMainEditor.SelectEnd == 0) {\r
                  HMainEditor.SelectStart = 0;\r
                }\r
              }\r
\r
              FirstDown = FALSE;\r
              MouseDrag = FALSE;\r
            }\r
\r
            if (SelectStartBackup != HMainEditor.SelectStart || SelectEndBackup != HMainEditor.SelectEnd) {\r
              if ((SelectStartBackup - 1) / 0x10 != (HMainEditor.SelectStart - 1) / 0x10) {\r
                HBufferImageNeedRefresh = TRUE;\r
              } else {\r
                if ((SelectEndBackup - 1) / 0x10 != (HMainEditor.SelectEnd - 1) / 0x10) {\r
                  HBufferImageNeedRefresh = TRUE;\r
                } else {\r
                  HBufferImageOnlyLineNeedRefresh = TRUE;\r
                }\r
              }\r
            }\r
          }\r
\r
          HEditorMouseAction            = TRUE;\r
          HBufferImageMouseNeedRefresh  = TRUE;\r
\r
        } else if (Status == EFI_LOAD_ERROR) {\r
          StatusBarSetStatusString (L"Invalid Mouse Movement ");\r
        }\r
      }\r
    }\r
\r
    //\r
    // CheckEvent() returns Success when non-partial key is pressed.\r
    //\r
    Status = gBS->CheckEvent (HMainEditor.TextInputEx->WaitForKeyEx);\r
    if (!EFI_ERROR (Status)) {\r
      Status = HMainEditor.TextInputEx->ReadKeyStrokeEx (HMainEditor.TextInputEx, &KeyData);\r
      if (!EFI_ERROR (Status)) {\r
        //\r
        // dispatch to different components' key handling function\r
        // so not everywhere has to set this variable\r
        //\r
        HBufferImageMouseNeedRefresh = TRUE;\r
\r
        //\r
        // clear previous status string\r
        //\r
        StatusBarSetRefresh();\r
        //\r
        // NoShiftState: TRUE when no shift key is pressed.\r
        //\r
        NoShiftState = ((KeyData.KeyState.KeyShiftState & EFI_SHIFT_STATE_VALID) == 0) || (KeyData.KeyState.KeyShiftState == EFI_SHIFT_STATE_VALID);\r
        //\r
        // dispatch to different components' key handling function\r
        //\r
        if (EFI_SUCCESS == MenuBarDispatchControlHotKey(&KeyData)) {\r
          Status = EFI_SUCCESS;\r
        } else if (NoShiftState && KeyData.Key.ScanCode == SCAN_NULL) {\r
          Status = HBufferImageHandleInput (&KeyData.Key);\r
        } else if (NoShiftState && ((KeyData.Key.ScanCode >= SCAN_UP) && (KeyData.Key.ScanCode <= SCAN_PAGE_DOWN))) {\r
          Status = HBufferImageHandleInput (&KeyData.Key);\r
        } else if (NoShiftState && ((KeyData.Key.ScanCode >= SCAN_F1) && KeyData.Key.ScanCode <= SCAN_F12)) {\r
          Status = MenuBarDispatchFunctionKey (&KeyData.Key);\r
        } else {\r
          StatusBarSetStatusString (L"Unknown Command");\r
\r
          HBufferImageMouseNeedRefresh = FALSE;\r
        }\r
\r
        if (Status != EFI_SUCCESS && Status != EFI_OUT_OF_RESOURCES) {\r
          //\r
          // not already has some error status\r
          //\r
          if (StrCmp (L"", StatusBarGetString()) == 0) {\r
            StatusBarSetStatusString (L"Disk Error. Try Again");\r
          }\r
        }\r
      }\r
      //\r
      // decide if has to set length warning\r
      //\r
      if (HBufferImage.BufferType != HBufferImageBackupVar.BufferType) {\r
        LengthChange = FALSE;\r
      } else {\r
        //\r
        // still the old buffer\r
        //\r
        if (HBufferImage.BufferType != FileTypeFileBuffer) {\r
          Size = HBufferImageGetTotalSize ();\r
\r
          switch (HBufferImage.BufferType) {\r
          case FileTypeDiskBuffer:\r
            OldSize = HBufferImage.DiskImage->Size * HBufferImage.DiskImage->BlockSize;\r
            break;\r
\r
          case FileTypeMemBuffer:\r
            OldSize = HBufferImage.MemImage->Size;\r
            break;\r
\r
          default:\r
            OldSize = 0;\r
            break;\r
          }\r
\r
          if (!LengthChange) {\r
            if (OldSize != Size) {\r
              StatusBarSetStatusString (L"Disk/Mem Buffer Length should not be changed");\r
            }\r
          }\r
\r
          if (OldSize != Size) {\r
            LengthChange = TRUE;\r
          } else {\r
            LengthChange = FALSE;\r
          }\r
        }\r
      }\r
    }\r
    //\r
    // after handling, refresh editor\r
    //\r
    HMainEditorRefresh ();\r
\r
  } while (Status != EFI_OUT_OF_RESOURCES && !HEditorExit);\r
\r
  return Status;\r
}\r
\r
/**\r
  Backup function for MainEditor.\r
**/\r
VOID\r
HMainEditorBackup (\r
  VOID\r
  )\r
{\r
  HMainEditorBackupVar.SelectStart  = HMainEditor.SelectStart;\r
  HMainEditorBackupVar.SelectEnd    = HMainEditor.SelectEnd;\r
  HBufferImageBackup ();\r
}\r