| /** @file | |
| Main entry point of editor | |
| (C) Copyright 2014-2015 Hewlett-Packard Development Company, L.P.<BR> | |
| Copyright (c) 2005 - 2018, Intel Corporation. All rights reserved. <BR> | |
| SPDX-License-Identifier: BSD-2-Clause-Patent | |
| **/ | |
| #include "UefiShellDebug1CommandsLib.h" | |
| #include "HexEditor.h" | |
| // | |
| // Global Variables | |
| // | |
| STATIC CONST SHELL_PARAM_ITEM ParamList[] = { | |
| {L"-f", TypeFlag}, | |
| {L"-d", TypeFlag}, | |
| {L"-m", TypeFlag}, | |
| {NULL, TypeMax} | |
| }; | |
| /** | |
| Function for 'hexedit' command. | |
| @param[in] ImageHandle Handle to the Image (NULL if Internal). | |
| @param[in] SystemTable Pointer to the System Table (NULL if Internal). | |
| **/ | |
| SHELL_STATUS | |
| EFIAPI | |
| ShellCommandRunHexEdit ( | |
| IN EFI_HANDLE ImageHandle, | |
| IN EFI_SYSTEM_TABLE *SystemTable | |
| ) | |
| { | |
| EFI_STATUS Status; | |
| CHAR16 *Buffer; | |
| CHAR16 *ProblemParam; | |
| SHELL_STATUS ShellStatus; | |
| LIST_ENTRY *Package; | |
| CHAR16 *NewName; | |
| CONST CHAR16 *Name; | |
| UINTN Offset; | |
| UINTN Size; | |
| EDIT_FILE_TYPE WhatToDo; | |
| Buffer = NULL; | |
| ShellStatus = SHELL_SUCCESS; | |
| NewName = NULL; | |
| Buffer = NULL; | |
| Name = NULL; | |
| Offset = 0; | |
| Size = 0; | |
| WhatToDo = FileTypeNone; | |
| // | |
| // initialize the shell lib (we must be in non-auto-init...) | |
| // | |
| Status = ShellInitialize(); | |
| ASSERT_EFI_ERROR(Status); | |
| Status = CommandInit(); | |
| ASSERT_EFI_ERROR(Status); | |
| // | |
| // parse the command line | |
| // | |
| Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE); | |
| if (EFI_ERROR(Status)) { | |
| if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) { | |
| ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellDebug1HiiHandle, L"hexedit", ProblemParam); | |
| FreePool(ProblemParam); | |
| ShellStatus = SHELL_INVALID_PARAMETER; | |
| } else { | |
| ASSERT(FALSE); | |
| } | |
| } else { | |
| // | |
| // Check for -d | |
| // | |
| if (ShellCommandLineGetFlag(Package, L"-d")){ | |
| if (ShellCommandLineGetCount(Package) < 4) { | |
| ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle, L"hexedit"); | |
| ShellStatus = SHELL_INVALID_PARAMETER; | |
| } else if (ShellCommandLineGetCount(Package) > 4) { | |
| ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle, L"hexedit"); | |
| ShellStatus = SHELL_INVALID_PARAMETER; | |
| } else { | |
| WhatToDo = FileTypeDiskBuffer; | |
| Name = ShellCommandLineGetRawValue(Package, 1); | |
| Offset = ShellStrToUintn(ShellCommandLineGetRawValue(Package, 2)); | |
| Size = ShellStrToUintn(ShellCommandLineGetRawValue(Package, 3)); | |
| } | |
| if (Offset == (UINTN)-1 || Size == (UINTN)-1) { | |
| ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_VALUE), gShellDebug1HiiHandle, L"hexedit", L"-d"); | |
| ShellStatus = SHELL_INVALID_PARAMETER; | |
| } | |
| } | |
| // | |
| // check for -f | |
| // | |
| if (ShellCommandLineGetFlag(Package, L"-f") && (WhatToDo == FileTypeNone)){ | |
| if (ShellCommandLineGetCount(Package) < 2) { | |
| ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle, L"hexedit"); | |
| ShellStatus = SHELL_INVALID_PARAMETER; | |
| } else if (ShellCommandLineGetCount(Package) > 2) { | |
| ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle, L"hexedit"); | |
| ShellStatus = SHELL_INVALID_PARAMETER; | |
| } else { | |
| Name = ShellCommandLineGetRawValue(Package, 1); | |
| if (Name == NULL || !IsValidFileName(Name)) { | |
| ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"hexedit", Name); | |
| ShellStatus = SHELL_INVALID_PARAMETER; | |
| } else { | |
| WhatToDo = FileTypeFileBuffer; | |
| } | |
| } | |
| } | |
| // | |
| // check for -m | |
| // | |
| if (ShellCommandLineGetFlag(Package, L"-m") && (WhatToDo == FileTypeNone)){ | |
| if (ShellCommandLineGetCount(Package) < 3) { | |
| ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle, L"hexedit"); | |
| ShellStatus = SHELL_INVALID_PARAMETER; | |
| } else if (ShellCommandLineGetCount(Package) > 3) { | |
| ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle, L"hexedit"); | |
| ShellStatus = SHELL_INVALID_PARAMETER; | |
| } else { | |
| WhatToDo = FileTypeMemBuffer; | |
| Offset = ShellStrToUintn(ShellCommandLineGetRawValue(Package, 1)); | |
| Size = ShellStrToUintn(ShellCommandLineGetRawValue(Package, 2)); | |
| } | |
| } | |
| Name = ShellCommandLineGetRawValue(Package, 1); | |
| if (WhatToDo == FileTypeNone && Name != NULL) { | |
| if (ShellCommandLineGetCount(Package) > 2) { | |
| ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellDebug1HiiHandle, L"hexedit"); | |
| ShellStatus = SHELL_INVALID_PARAMETER; | |
| } else if (!IsValidFileName(Name)) { | |
| ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellDebug1HiiHandle, L"hexedit", Name); | |
| ShellStatus = SHELL_INVALID_PARAMETER; | |
| } else { | |
| WhatToDo = FileTypeFileBuffer; | |
| } | |
| } else if (WhatToDo == FileTypeNone) { | |
| if (gEfiShellProtocol->GetCurDir(NULL) == NULL) { | |
| ShellStatus = SHELL_NOT_FOUND; | |
| ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_CWD), gShellDebug1HiiHandle, L"hexedit"); | |
| } else { | |
| NewName = EditGetDefaultFileName(L"bin"); | |
| Name = NewName; | |
| WhatToDo = FileTypeFileBuffer; | |
| } | |
| } | |
| if (ShellStatus == SHELL_SUCCESS && WhatToDo == FileTypeNone) { | |
| ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellDebug1HiiHandle, L"hexedit"); | |
| ShellStatus = SHELL_INVALID_PARAMETER; | |
| } else if (WhatToDo == FileTypeFileBuffer && ShellGetCurrentDir(NULL) == NULL) { | |
| ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_CWD), gShellDebug1HiiHandle, L"hexedit"); | |
| ShellStatus = SHELL_INVALID_PARAMETER; | |
| } | |
| if (ShellStatus == SHELL_SUCCESS) { | |
| // | |
| // Do the editor | |
| // | |
| Status = HMainEditorInit (); | |
| if (EFI_ERROR (Status)) { | |
| gST->ConOut->ClearScreen (gST->ConOut); | |
| gST->ConOut->EnableCursor (gST->ConOut, TRUE); | |
| ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HEXEDIT_INIT_FAILED), gShellDebug1HiiHandle); | |
| } else { | |
| HMainEditorBackup (); | |
| switch (WhatToDo) { | |
| case FileTypeFileBuffer: | |
| Status = HBufferImageRead ( | |
| Name==NULL?L"":Name, | |
| NULL, | |
| 0, | |
| 0, | |
| 0, | |
| 0, | |
| FileTypeFileBuffer, | |
| FALSE | |
| ); | |
| break; | |
| case FileTypeDiskBuffer: | |
| Status = HBufferImageRead ( | |
| NULL, | |
| Name==NULL?L"":Name, | |
| Offset, | |
| Size, | |
| 0, | |
| 0, | |
| FileTypeDiskBuffer, | |
| FALSE | |
| ); | |
| break; | |
| case FileTypeMemBuffer: | |
| Status = HBufferImageRead ( | |
| NULL, | |
| NULL, | |
| 0, | |
| 0, | |
| (UINT32) Offset, | |
| Size, | |
| FileTypeMemBuffer, | |
| FALSE | |
| ); | |
| break; | |
| default: | |
| Status = EFI_NOT_FOUND; | |
| break; | |
| } | |
| if (!EFI_ERROR (Status)) { | |
| HMainEditorRefresh (); | |
| Status = HMainEditorKeyInput (); | |
| } | |
| if (Status != EFI_OUT_OF_RESOURCES) { | |
| // | |
| // back up the status string | |
| // | |
| Buffer = CatSPrint (NULL, L"%s\r\n", StatusBarGetString()); | |
| } | |
| } | |
| // | |
| // cleanup | |
| // | |
| HMainEditorCleanup (); | |
| if (EFI_ERROR (Status)) { | |
| if (ShellStatus == SHELL_SUCCESS) { | |
| ShellStatus = SHELL_UNSUPPORTED; | |
| } | |
| } | |
| // | |
| // print editor exit code on screen | |
| // | |
| if (Status == EFI_OUT_OF_RESOURCES) { | |
| ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_OUT_MEM), gShellDebug1HiiHandle, L"hexedit"); | |
| } else if (EFI_ERROR(Status)){ | |
| if (Buffer != NULL) { | |
| if (StrCmp (Buffer, L"") != 0) { | |
| // | |
| // print out the status string | |
| // | |
| ShellPrintEx(-1, -1, L"%s", Buffer); | |
| } else { | |
| ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HEXEDIT_UNKNOWN_EDITOR), gShellDebug1HiiHandle); | |
| } | |
| } else { | |
| ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_HEXEDIT_UNKNOWN_EDITOR), gShellDebug1HiiHandle); | |
| } | |
| } | |
| } | |
| ShellCommandLineFreeVarList (Package); | |
| } | |
| SHELL_FREE_NON_NULL (Buffer); | |
| SHELL_FREE_NON_NULL (NewName); | |
| return ShellStatus; | |
| } |