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

/** @file\r
  Main entry point of editor\r
  \r
  Copyright (c) 2005 - 2011, 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
  http://opensource.org/licenses/bsd-license.php\r
\r
  THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
  WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
\r
**/\r
\r
#include "UefiShellDebug1CommandsLib.h"\r
#include "HexEditor.h"\r
\r
//\r
// Global Variables\r
//\r
STATIC CONST SHELL_PARAM_ITEM ParamList[] = {\r
  {L"-f", TypeFlag},\r
  {L"-d", TypeFlag},\r
  {L"-m", TypeFlag},\r
  {NULL, TypeMax}\r
  };\r
\r
/**\r
  Function for 'hexedit' command.\r
\r
  @param[in] ImageHandle  Handle to the Image (NULL if Internal).\r
  @param[in] SystemTable  Pointer to the System Table (NULL if Internal).\r
**/\r
SHELL_STATUS\r
EFIAPI\r
ShellCommandRunHexEdit (\r
  IN EFI_HANDLE        ImageHandle,\r
  IN EFI_SYSTEM_TABLE  *SystemTable\r
  )\r
{\r
  EFI_STATUS              Status;\r
  CHAR16                  *Buffer;\r
  CHAR16                  *ProblemParam;\r
  SHELL_STATUS            ShellStatus;\r
  LIST_ENTRY              *Package;\r
  CONST CHAR16            *Cwd;\r
  CHAR16                  *NewName;\r
  CHAR16                  *Spot;\r
  CONST CHAR16            *Name;\r
  UINTN                   Offset;\r
  UINTN                   Size;\r
  UINT64                  LastOffset;\r
  EDIT_FILE_TYPE          WhatToDo;\r
\r
  Buffer      = NULL;\r
  ShellStatus = SHELL_SUCCESS;\r
  NewName         = NULL;\r
  Cwd         = NULL;\r
  Buffer      = NULL;\r
  Name        = NULL;\r
  Spot        = NULL;\r
  Offset      = 0;\r
  Size        = 0;\r
  LastOffset  = 0;\r
  WhatToDo    = FileTypeNone;\r
\r
  //\r
  // initialize the shell lib (we must be in non-auto-init...)\r
  //\r
  Status = ShellInitialize();\r
  ASSERT_EFI_ERROR(Status);\r
\r
  Status = CommandInit();\r
  ASSERT_EFI_ERROR(Status);\r
\r
  //\r
  // parse the command line\r
  //\r
  Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE);\r
  if (EFI_ERROR(Status)) {\r
    if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {\r
      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, ProblemParam);\r
      FreePool(ProblemParam);\r
      ShellStatus = SHELL_INVALID_PARAMETER;\r
    } else {\r
      ASSERT(FALSE);\r
    }\r
  } else {\r
    //\r
    // Check for -d\r
    //\r
    if (ShellCommandLineGetFlag(Package, L"-d")){\r
      if (ShellCommandLineGetCount(Package) < 4) {\r
        ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle);\r
        ShellStatus = SHELL_INVALID_PARAMETER;\r
      } else if (ShellCommandLineGetCount(Package) > 4) {\r
        ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle);\r
        ShellStatus = SHELL_INVALID_PARAMETER;\r
      } else {\r
        WhatToDo = FileTypeDiskBuffer;\r
        Name    = ShellCommandLineGetRawValue(Package, 1);\r
        Offset  = ShellStrToUintn(ShellCommandLineGetRawValue(Package, 2));\r
        Size    = ShellStrToUintn(ShellCommandLineGetRawValue(Package, 3));\r
      }\r
      if (Offset == (UINTN)-1 || Size == (UINTN)-1) {\r
        ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM_VAL), gShellDebug1HiiHandle, L"-d");\r
        ShellStatus = SHELL_INVALID_PARAMETER;\r
      }\r
    }\r
\r
    //\r
    // check for -f\r
    //\r
    if (ShellCommandLineGetFlag(Package, L"-f") && (WhatToDo == FileTypeNone)){\r
      if (ShellCommandLineGetCount(Package) < 2) {\r
        ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle);\r
        ShellStatus = SHELL_INVALID_PARAMETER;\r
      } else if (ShellCommandLineGetCount(Package) > 2) {\r
        ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle);\r
        ShellStatus = SHELL_INVALID_PARAMETER;\r
      } else {\r
        Name      = ShellCommandLineGetRawValue(Package, 1);\r
        if (Name == NULL || !IsValidFileName(Name)) {\r
          ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, Name);\r
          ShellStatus = SHELL_INVALID_PARAMETER;\r
        } else {\r
          WhatToDo  = FileTypeFileBuffer;\r
        }\r
      }\r
    }\r
\r
    //\r
    // check for -m\r
    //\r
    if (ShellCommandLineGetFlag(Package, L"-m") && (WhatToDo == FileTypeNone)){\r
      if (ShellCommandLineGetCount(Package) < 3) {\r
        ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle);\r
        ShellStatus = SHELL_INVALID_PARAMETER;\r
      } else if (ShellCommandLineGetCount(Package) > 3) {\r
        ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle);\r
        ShellStatus = SHELL_INVALID_PARAMETER;\r
      } else {\r
        WhatToDo = FileTypeMemBuffer;\r
        Offset  = ShellStrToUintn(ShellCommandLineGetRawValue(Package, 1));\r
        Size    = ShellStrToUintn(ShellCommandLineGetRawValue(Package, 2));\r
      }\r
    }\r
    Name = ShellCommandLineGetRawValue(Package, 1);\r
    if (WhatToDo == FileTypeNone && Name != NULL) {\r
      if (ShellCommandLineGetCount(Package) > 2) {\r
        ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle);\r
        ShellStatus = SHELL_INVALID_PARAMETER;\r
      } else if (!IsValidFileName(Name)) {\r
        ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, Name);\r
        ShellStatus = SHELL_INVALID_PARAMETER;\r
      } else {\r
        WhatToDo  = FileTypeFileBuffer;\r
      }\r
    } else if (WhatToDo == FileTypeNone) {\r
      if (gEfiShellProtocol->GetCurDir(NULL) == NULL) {\r
        ShellStatus = SHELL_NOT_FOUND;\r
        ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_CWD), gShellDebug1HiiHandle);\r
      } else {\r
        NewName = EditGetDefaultFileName(L"bin");\r
        Name = NewName;\r
        WhatToDo  = FileTypeFileBuffer;\r
      }\r
    }\r
\r
    if (ShellStatus == SHELL_SUCCESS && WhatToDo == FileTypeNone) {\r
      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle);\r
      ShellStatus = SHELL_INVALID_PARAMETER;\r
    } else if (WhatToDo == FileTypeFileBuffer && ShellGetCurrentDir(NULL) == NULL) {\r
      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_CWD), gShellDebug1HiiHandle);\r
      ShellStatus = SHELL_INVALID_PARAMETER;\r
    }\r
\r
    if (ShellStatus == SHELL_SUCCESS) {\r
      //\r
      // Do the editor\r
      //\r
      Status = HMainEditorInit ();\r
      if (EFI_ERROR (Status)) {\r
        gST->ConOut->ClearScreen (gST->ConOut);\r
        gST->ConOut->EnableCursor (gST->ConOut, TRUE);\r
        ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HEXEDIT_INIT_FAILED), gShellDebug1HiiHandle);\r
      } else {\r
        HMainEditorBackup ();\r
        switch (WhatToDo) {\r
        case FileTypeFileBuffer:\r
          Status = HBufferImageRead (\r
                    Name,\r
                    NULL,\r
                    0,\r
                    0,\r
                    0,\r
                    0,\r
                    FileTypeFileBuffer,\r
                    FALSE\r
                    );\r
          break;\r
\r
        case FileTypeDiskBuffer:\r
          Status = HBufferImageRead (\r
                    NULL,\r
                    Name,\r
                    Offset,\r
                    Size,\r
                    0,\r
                    0,\r
                    FileTypeDiskBuffer,\r
                    FALSE\r
                    );\r
          break;\r
\r
        case FileTypeMemBuffer:\r
          Status = HBufferImageRead (\r
                    NULL,\r
                    NULL,\r
                    0,\r
                    0,\r
                    (UINT32) Offset,\r
                    Size,\r
                    FileTypeMemBuffer,\r
                    FALSE\r
                    );\r
          break;\r
\r
        default:\r
          Status = EFI_NOT_FOUND;\r
          break;\r
        }\r
        if (!EFI_ERROR (Status)) {\r
          HMainEditorRefresh ();\r
          Status = HMainEditorKeyInput ();\r
        }\r
        if (Status != EFI_OUT_OF_RESOURCES) {\r
          //\r
          // back up the status string\r
          //\r
          Buffer = CatSPrint (NULL, L"%s\r\n", StatusBarGetString());\r
        }\r
      }\r
\r
      //\r
      // cleanup\r
      //\r
      HMainEditorCleanup ();\r
\r
      if (!EFI_ERROR (Status)) {\r
        if (ShellStatus == SHELL_SUCCESS) {\r
          ShellStatus = SHELL_UNSUPPORTED;\r
        }\r
      }\r
\r
      //\r
      // print editor exit code on screen\r
      //\r
      if (Status == EFI_OUT_OF_RESOURCES) {\r
        ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_OUT_MEM), gShellDebug1HiiHandle);\r
      } else if (EFI_ERROR(Status)){\r
        if (Buffer != NULL) {\r
          if (StrCmp (Buffer, L"") != 0) {\r
            //\r
            // print out the status string\r
            //\r
            ShellPrintEx(-1, -1, L"%s", Buffer);\r
          } else {\r
            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HEXEDIT_UNKNOWN_EDITOR), gShellDebug1HiiHandle);\r
          }\r
        } else {\r
          ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HEXEDIT_UNKNOWN_EDITOR), gShellDebug1HiiHandle);\r
        }\r
      }\r
    }\r
    ShellCommandLineFreeVarList (Package);\r
  }\r
\r
  SHELL_FREE_NON_NULL (Buffer);\r
  SHELL_FREE_NON_NULL (NewName);\r
  return ShellStatus;\r
}\r
Commit	Line	Data
632820d1	1	/** @file\r
	2	Main entry point of editor\r
	3	\r
	4	Copyright (c) 2005 - 2011, Intel Corporation. All rights reserved. <BR>\r
	5	This program and the accompanying materials\r
	6	are licensed and made available under the terms and conditions of the BSD License\r
	7	which accompanies this distribution. The full text of the license may be found at\r
	8	http://opensource.org/licenses/bsd-license.php\r
	9	\r
	10	THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
	11	WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
	12	\r
	13	**/\r
	14	\r
	15	#include "UefiShellDebug1CommandsLib.h"\r
	16	#include "HexEditor.h"\r
	17	\r
	18	//\r
	19	// Global Variables\r
	20	//\r
	21	STATIC CONST SHELL_PARAM_ITEM ParamList[] = {\r
	22	{L"-f", TypeFlag},\r
	23	{L"-d", TypeFlag},\r
	24	{L"-m", TypeFlag},\r
	25	{NULL, TypeMax}\r
	26	};\r
	27	\r
	28	/**\r
	29	Function for 'hexedit' command.\r
	30	\r
	31	@param[in] ImageHandle Handle to the Image (NULL if Internal).\r
	32	@param[in] SystemTable Pointer to the System Table (NULL if Internal).\r
	33	**/\r
	34	SHELL_STATUS\r
	35	EFIAPI\r
	36	ShellCommandRunHexEdit (\r
	37	IN EFI_HANDLE ImageHandle,\r
	38	IN EFI_SYSTEM_TABLE *SystemTable\r
	39	)\r
	40	{\r
	41	EFI_STATUS Status;\r
	42	CHAR16 *Buffer;\r
	43	CHAR16 *ProblemParam;\r
	44	SHELL_STATUS ShellStatus;\r
	45	LIST_ENTRY *Package;\r
	46	CONST CHAR16 *Cwd;\r
980d554e	47	CHAR16 *NewName;\r
632820d1	48	CHAR16 *Spot;\r
	49	CONST CHAR16 *Name;\r
	50	UINTN Offset;\r
	51	UINTN Size;\r
	52	UINT64 LastOffset;\r
	53	EDIT_FILE_TYPE WhatToDo;\r
	54	\r
	55	Buffer = NULL;\r
	56	ShellStatus = SHELL_SUCCESS;\r
980d554e	57	NewName = NULL;\r
632820d1	58	Cwd = NULL;\r
	59	Buffer = NULL;\r
	60	Name = NULL;\r
	61	Spot = NULL;\r
	62	Offset = 0;\r
	63	Size = 0;\r
	64	LastOffset = 0;\r
	65	WhatToDo = FileTypeNone;\r
	66	\r
	67	//\r
	68	// initialize the shell lib (we must be in non-auto-init...)\r
	69	//\r
	70	Status = ShellInitialize();\r
	71	ASSERT_EFI_ERROR(Status);\r
	72	\r
	73	Status = CommandInit();\r
	74	ASSERT_EFI_ERROR(Status);\r
	75	\r
	76	//\r
	77	// parse the command line\r
	78	//\r
	79	Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE);\r
	80	if (EFI_ERROR(Status)) {\r
	81	if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {\r
	82	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, ProblemParam);\r
	83	FreePool(ProblemParam);\r
	84	ShellStatus = SHELL_INVALID_PARAMETER;\r
	85	} else {\r
	86	ASSERT(FALSE);\r
	87	}\r
	88	} else {\r
	89	//\r
	90	// Check for -d\r
	91	//\r
	92	if (ShellCommandLineGetFlag(Package, L"-d")){\r
392fa5a6	93	if (ShellCommandLineGetCount(Package) < 4) {\r
	94	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle);\r
	95	ShellStatus = SHELL_INVALID_PARAMETER;\r
	96	} else if (ShellCommandLineGetCount(Package) > 4) {\r
632820d1	97	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle);\r
	98	ShellStatus = SHELL_INVALID_PARAMETER;\r
	99	} else {\r
	100	WhatToDo = FileTypeDiskBuffer;\r
	101	Name = ShellCommandLineGetRawValue(Package, 1);\r
	102	Offset = ShellStrToUintn(ShellCommandLineGetRawValue(Package, 2));\r
	103	Size = ShellStrToUintn(ShellCommandLineGetRawValue(Package, 3));\r
	104	}\r
392fa5a6	105	if (Offset == (UINTN)-1 \|\| Size == (UINTN)-1) {\r
	106	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM_VAL), gShellDebug1HiiHandle, L"-d");\r
	107	ShellStatus = SHELL_INVALID_PARAMETER;\r
	108	}\r
632820d1	109	}\r
	110	\r
	111	//\r
	112	// check for -f\r
	113	//\r
	114	if (ShellCommandLineGetFlag(Package, L"-f") && (WhatToDo == FileTypeNone)){\r
392fa5a6	115	if (ShellCommandLineGetCount(Package) < 2) {\r
	116	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle);\r
	117	ShellStatus = SHELL_INVALID_PARAMETER;\r
	118	} else if (ShellCommandLineGetCount(Package) > 2) {\r
632820d1	119	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle);\r
	120	ShellStatus = SHELL_INVALID_PARAMETER;\r
	121	} else {\r
	122	Name = ShellCommandLineGetRawValue(Package, 1);\r
33c031ee	123	if (Name == NULL \|\| !IsValidFileName(Name)) {\r
632820d1	124	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, Name);\r
	125	ShellStatus = SHELL_INVALID_PARAMETER;\r
	126	} else {\r
	127	WhatToDo = FileTypeFileBuffer;\r
	128	}\r
	129	}\r
	130	}\r
	131	\r
	132	//\r
	133	// check for -m\r
	134	//\r
	135	if (ShellCommandLineGetFlag(Package, L"-m") && (WhatToDo == FileTypeNone)){\r
	136	if (ShellCommandLineGetCount(Package) < 3) {\r
	137	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle);\r
	138	ShellStatus = SHELL_INVALID_PARAMETER;\r
	139	} else if (ShellCommandLineGetCount(Package) > 3) {\r
	140	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle);\r
	141	ShellStatus = SHELL_INVALID_PARAMETER;\r
	142	} else {\r
	143	WhatToDo = FileTypeMemBuffer;\r
	144	Offset = ShellStrToUintn(ShellCommandLineGetRawValue(Package, 1));\r
	145	Size = ShellStrToUintn(ShellCommandLineGetRawValue(Package, 2));\r
	146	}\r
	147	}\r
ae724571	148	Name = ShellCommandLineGetRawValue(Package, 1);\r
ae724571	149	if (WhatToDo == FileTypeNone && Name != NULL) {\r
411a3c39	150	if (ShellCommandLineGetCount(Package) > 2) {\r
	151	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle);\r
	152	ShellStatus = SHELL_INVALID_PARAMETER;\r
	153	} else if (!IsValidFileName(Name)) {\r
	154	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, Name);\r
	155	ShellStatus = SHELL_INVALID_PARAMETER;\r
	156	} else {\r
	157	WhatToDo = FileTypeFileBuffer;\r
	158	}\r
980d554e	159	} else if (WhatToDo == FileTypeNone) {\r
	160	if (gEfiShellProtocol->GetCurDir(NULL) == NULL) {\r
	161	ShellStatus = SHELL_NOT_FOUND;\r
	162	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_CWD), gShellDebug1HiiHandle);\r
	163	} else {\r
	164	NewName = EditGetDefaultFileName(L"bin");\r
	165	Name = NewName;\r
	166	WhatToDo = FileTypeFileBuffer;\r
	167	}\r
	168	}\r
632820d1	169	\r
	170	if (ShellStatus == SHELL_SUCCESS && WhatToDo == FileTypeNone) {\r
	171	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle);\r
	172	ShellStatus = SHELL_INVALID_PARAMETER;\r
392fa5a6	173	} else if (WhatToDo == FileTypeFileBuffer && ShellGetCurrentDir(NULL) == NULL) {\r
	174	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_CWD), gShellDebug1HiiHandle);\r
	175	ShellStatus = SHELL_INVALID_PARAMETER;\r
e0c2cc6f	176	}\r
632820d1	177	\r
	178	if (ShellStatus == SHELL_SUCCESS) {\r
	179	//\r
	180	// Do the editor\r
	181	//\r
	182	Status = HMainEditorInit ();\r
	183	if (EFI_ERROR (Status)) {\r
	184	gST->ConOut->ClearScreen (gST->ConOut);\r
	185	gST->ConOut->EnableCursor (gST->ConOut, TRUE);\r
	186	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HEXEDIT_INIT_FAILED), gShellDebug1HiiHandle);\r
	187	} else {\r
	188	HMainEditorBackup ();\r
	189	switch (WhatToDo) {\r
	190	case FileTypeFileBuffer:\r
	191	Status = HBufferImageRead (\r
	192	Name,\r
	193	NULL,\r
	194	0,\r
	195	0,\r
	196	0,\r
	197	0,\r
	198	FileTypeFileBuffer,\r
	199	FALSE\r
	200	);\r
	201	break;\r
	202	\r
	203	case FileTypeDiskBuffer:\r
	204	Status = HBufferImageRead (\r
	205	NULL,\r
	206	Name,\r
	207	Offset,\r
	208	Size,\r
	209	0,\r
	210	0,\r
	211	FileTypeDiskBuffer,\r
	212	FALSE\r
	213	);\r
	214	break;\r
	215	\r
	216	case FileTypeMemBuffer:\r
	217	Status = HBufferImageRead (\r
	218	NULL,\r
	219	NULL,\r
	220	0,\r
	221	0,\r
	222	(UINT32) Offset,\r
	223	Size,\r
	224	FileTypeMemBuffer,\r
	225	FALSE\r
	226	);\r
	227	break;\r
	228	\r
e0c2cc6f	229	default:\r
	230	Status = EFI_NOT_FOUND;\r
	231	break;\r
632820d1	232	}\r
	233	if (!EFI_ERROR (Status)) {\r
	234	HMainEditorRefresh ();\r
	235	Status = HMainEditorKeyInput ();\r
	236	}\r
	237	if (Status != EFI_OUT_OF_RESOURCES) {\r
	238	//\r
	239	// back up the status string\r
	240	//\r
411a3c39	241	Buffer = CatSPrint (NULL, L"%s\r\n", StatusBarGetString());\r
632820d1	242	}\r
	243	}\r
	244	\r
	245	//\r
	246	// cleanup\r
	247	//\r
	248	HMainEditorCleanup ();\r
	249	\r
	250	if (!EFI_ERROR (Status)) {\r
	251	if (ShellStatus == SHELL_SUCCESS) {\r
	252	ShellStatus = SHELL_UNSUPPORTED;\r
	253	}\r
	254	}\r
	255	\r
	256	//\r
	257	// print editor exit code on screen\r
	258	//\r
	259	if (Status == EFI_OUT_OF_RESOURCES) {\r
	260	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_OUT_MEM), gShellDebug1HiiHandle);\r
	261	} else if (EFI_ERROR(Status)){\r
	262	if (Buffer != NULL) {\r
	263	if (StrCmp (Buffer, L"") != 0) {\r
	264	//\r
	265	// print out the status string\r
	266	//\r
411a3c39	267	ShellPrintEx(-1, -1, L"%s", Buffer);\r
632820d1	268	} else {\r
	269	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HEXEDIT_UNKNOWN_EDITOR), gShellDebug1HiiHandle);\r
	270	}\r
	271	} else {\r
	272	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HEXEDIT_UNKNOWN_EDITOR), gShellDebug1HiiHandle);\r
	273	}\r
	274	}\r
	275	}\r
980d554e	276	ShellCommandLineFreeVarList (Package);\r
632820d1	277	}\r
	278	\r
	279	SHELL_FREE_NON_NULL (Buffer);\r
980d554e	280	SHELL_FREE_NON_NULL (NewName);\r
632820d1	281	return ShellStatus;\r
632820d1	282	}\r