]> git.proxmox.com Git - mirror_edk2.git/blobdiff - ShellPkg/Library/UefiShellDebug1CommandsLib/HexEdit/MainHexEditor.c
ShellPkg/comp: Use proper parameter names
[mirror_edk2.git] / ShellPkg / Library / UefiShellDebug1CommandsLib / HexEdit / MainHexEditor.c
index 73fa4c238d6c6d7c13abc55983e51ccaee9a6c8b..491acb131ea16d28acae2465287780ff9110b9ec 100644 (file)
@@ -4,7 +4,7 @@
      - Instances of the other objects of the editor\r
      - Main Interfaces\r
   \r
-  Copyright (c) 2005 - 2011, Intel Corporation. All rights reserved. <BR>\r
+  Copyright (c) 2005 - 2012, Intel Corporation. All rights reserved. <BR>\r
   This program and the accompanying materials\r
   are licensed and made available under the terms and conditions of the BSD License\r
   which accompanies this distribution.  The full text of the license may be found at\r
@@ -50,8 +50,7 @@ HEFI_EDITOR_GLOBAL_EDITOR       HMainEditorBackupVar;
 HEFI_EDITOR_GLOBAL_EDITOR       HMainEditorConst = {\r
   &HBufferImage,\r
   {\r
-    0,\r
-    0\r
+    {0, 0}\r
   },\r
   {\r
     0,\r
@@ -65,6 +64,72 @@ HEFI_EDITOR_GLOBAL_EDITOR       HMainEditorConst = {
   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_INPUT_KEY  Key;\r
+\r
+  CurrentLine = 0;\r
+  // print helpInfo      \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
+  // scan for ctrl+w\r
+  do {\r
+    gST->ConIn->ReadKeyStroke (gST->ConIn, &Key);\r
+  } while(SCAN_CONTROL_W != Key.UnicodeChar); \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
@@ -225,7 +290,6 @@ HMainCommandSaveBuffer (
         // the file won't be saved\r
         //\r
         return EFI_SUCCESS;\r
-        break;\r
 \r
       case L'c':\r
       case L'C':\r
@@ -533,7 +597,6 @@ HMainCommandCut (
   )\r
 {\r
   UINTN             Index;\r
-  HEFI_EDITOR_LINE  *Line;\r
   LIST_ENTRY    *Link;\r
   UINT8             *Buffer;\r
   UINTN             Count;\r
@@ -558,8 +621,6 @@ HMainCommandCut (
     Link = Link->ForwardLink;\r
   }\r
 \r
-  Line    = CR (Link, HEFI_EDITOR_LINE, Link, EFI_EDITOR_LINE_LIST);\r
-\r
   Count   = HMainEditor.SelectEnd - HMainEditor.SelectStart + 1;\r
   Buffer  = AllocateZeroPool (Count);\r
   if (Buffer == NULL) {\r
@@ -1439,6 +1500,36 @@ HMainCommandOpenMemory (
 \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
@@ -1592,12 +1683,17 @@ HMainEditorInit (
   //\r
   // below will call the five components' init function\r
   //\r
-  Status = MainTitleBarInit (NULL);\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
@@ -1704,12 +1800,28 @@ HMainEditorRefresh (
   NameChange = FALSE;\r
   ReadChange = FALSE;\r
 \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
+  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
@@ -1753,17 +1865,14 @@ HMainEditorRefresh (
       HEditorFirst,\r
       HMainEditor.ScreenSize.Row,\r
       HMainEditor.ScreenSize.Column,\r
-      0,\r
-      0,\r
-      TRUE\r
+      (UINTN)(-1),\r
+      (UINTN)(-1),\r
+      FALSE\r
       );\r
     HBufferImageRefresh ();\r
   }\r
 \r
   if (HEditorFirst) {\r
-    MenuBarRefresh (    \r
-      HMainEditor.ScreenSize.Row,\r
-      HMainEditor.ScreenSize.Column);\r
     HBufferImageRefresh ();\r
   }\r
 \r
@@ -2122,8 +2231,9 @@ HMainEditorKeyInput (
       // clear previous status string\r
       //\r
       StatusBarSetRefresh();\r
-\r
-      if (Key.ScanCode == SCAN_NULL) {\r
+      if (EFI_SUCCESS == MenuBarDispatchControlHotKey(&Key)) {\r
+        Status = EFI_SUCCESS;\r
+      } else if (Key.ScanCode == SCAN_NULL) {\r
         Status = HBufferImageHandleInput (&Key);\r
       } else if (((Key.ScanCode >= SCAN_UP) && (Key.ScanCode <= SCAN_PAGE_DOWN))) {\r
         Status = HBufferImageHandleInput (&Key);\r
@@ -2164,6 +2274,10 @@ HMainEditorKeyInput (
         case FileTypeMemBuffer:\r
           OldSize = HBufferImage.MemImage->Size;\r
           break;\r
+        \r
+        default:\r
+          OldSize = 0;\r
+          break;\r
         }\r
 \r
         if (!LengthChange) {\r
@@ -2193,7 +2307,6 @@ HMainEditorKeyInput (
   Backup function for MainEditor.\r
 **/\r
 VOID\r
-EFIAPI\r
 HMainEditorBackup (\r
   VOID\r
   )\r