147 lines
3.5 KiB
C
147 lines
3.5 KiB
C
|
/*++
|
||
|
|
|
||
|
|
Copyright (c) 1999 Intel Corporation
|
||
|
|
|
||
|
|
Module Name:
|
||
|
|
main.c
|
||
|
|
|
||
|
|
Abstract:
|
||
|
|
|
||
|
|
|
||
|
|
--*/
|
||
|
|
|
||
|
|
#include "hexedit.h"
|
||
|
|
|
||
|
|
typedef enum {
|
||
|
|
NEW_FILE,
|
||
|
|
OPEN_FILE,
|
||
|
|
OPEN_DISK,
|
||
|
|
OPEN_MEMORY
|
||
|
|
} IMAGE_TYPE;
|
||
|
|
|
||
|
|
|
||
|
|
STATIC
|
||
|
|
VOID
|
||
|
|
PrintUsage (
|
||
|
|
VOID
|
||
|
|
)
|
||
|
|
{
|
||
|
|
Print(L"\n\n %Ehex Usage%N\n");
|
||
|
|
Print(L" %H[-f] Filename%N Open File For Editing\n");
|
||
|
|
Print(L" %H-d Offset Size%N Open Disk Area For Editing\n");
|
||
|
|
Print(L" %H-m Offset Size%N Open Memory Area For Editing\n");
|
||
|
|
Print(L" %H-h%N Print This Screen\n");
|
||
|
|
Print(L"\n\n");
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
EFI_STATUS
|
||
|
|
InitializeEFIHexEditor (
|
||
|
|
IN EFI_HANDLE ImageHandle,
|
||
|
|
IN EFI_SYSTEM_TABLE *SystemTable
|
||
|
|
)
|
||
|
|
{
|
||
|
|
EFI_STATUS Status;
|
||
|
|
CHAR16 *Name;
|
||
|
|
UINTN Offset = 0;
|
||
|
|
UINTN Size = 0;
|
||
|
|
IMAGE_TYPE WhatToDo = NEW_FILE;
|
||
|
|
|
||
|
|
InitializeLib (ImageHandle, SystemTable);
|
||
|
|
InitializeShellApplication (ImageHandle, SystemTable);
|
||
|
|
|
||
|
|
Name = PoolPrint(L"New File");
|
||
|
|
|
||
|
|
if ( SI->Argc > 1 ) {
|
||
|
|
if (StrCmp(SI->Argv[1],L"-h") == 0) {
|
||
|
|
PrintUsage();
|
||
|
|
return EFI_SUCCESS;
|
||
|
|
} else if (StrCmp(SI->Argv[1],L"-d") == 0) {
|
||
|
|
if (SI->Argc < 5) {
|
||
|
|
PrintUsage();
|
||
|
|
return EFI_SUCCESS;
|
||
|
|
}
|
||
|
|
Name = SI->Argv[2];
|
||
|
|
Offset = xtoi(SI->Argv[3]);
|
||
|
|
Size = xtoi(SI->Argv[4]);
|
||
|
|
WhatToDo = OPEN_DISK;
|
||
|
|
} else if (StrCmp(SI->Argv[1],L"-m") == 0) {
|
||
|
|
if (SI->Argc < 4) {
|
||
|
|
PrintUsage();
|
||
|
|
return EFI_SUCCESS;
|
||
|
|
}
|
||
|
|
|
||
|
|
Offset = xtoi(SI->Argv[2]);
|
||
|
|
Size = xtoi(SI->Argv[3]);
|
||
|
|
WhatToDo = OPEN_MEMORY;
|
||
|
|
} else {
|
||
|
|
if (StrCmp(SI->Argv[1],L"-f") == 0) {
|
||
|
|
if (SI->Argc > 2) {
|
||
|
|
Name = SI->Argv[2];
|
||
|
|
} else {
|
||
|
|
PrintUsage();
|
||
|
|
return EFI_SUCCESS;
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
Name = SI->Argv[1];
|
||
|
|
}
|
||
|
|
WhatToDo = OPEN_FILE;
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
Status = MainEditor.Init(&ImageHandle);
|
||
|
|
if (EFI_ERROR(Status)) {
|
||
|
|
EditorError(Status,L"Initialization Failed");
|
||
|
|
return EFI_SUCCESS;
|
||
|
|
}
|
||
|
|
Status = MainEditor.BufferImage->FileImage->Init ();
|
||
|
|
if (EFI_ERROR(Status)) {
|
||
|
|
EditorError(Status,L"File Handle Initialization Failed");
|
||
|
|
return EFI_SUCCESS;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
switch (WhatToDo) {
|
||
|
|
case OPEN_FILE:
|
||
|
|
MainEditor.BufferImage->FileImage->Init();
|
||
|
|
MainEditor.BufferImage->FileImage->SetFilename(Name);
|
||
|
|
break;
|
||
|
|
case OPEN_DISK:
|
||
|
|
MainEditor.BufferImage->DiskImage->Init();
|
||
|
|
MainEditor.BufferImage->DiskImage->SetDevice(Name);
|
||
|
|
MainEditor.BufferImage->DiskImage->SetOffset(Offset);
|
||
|
|
MainEditor.BufferImage->DiskImage->SetSize(Size);
|
||
|
|
break;
|
||
|
|
case OPEN_MEMORY:
|
||
|
|
MainEditor.BufferImage->MemImage->Init();
|
||
|
|
MainEditor.BufferImage->DiskImage->SetOffset(Offset);
|
||
|
|
MainEditor.BufferImage->DiskImage->SetSize(Size);
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (WhatToDo != NEW_FILE) {
|
||
|
|
Status = MainEditor.BufferImage->Open();
|
||
|
|
if ( EFI_ERROR(Status) ) {
|
||
|
|
EditorError(Status,L"Could Not Open File");
|
||
|
|
return EFI_SUCCESS;
|
||
|
|
}
|
||
|
|
MainEditor.TitleBar->SetTitleString (Name);
|
||
|
|
MainEditor.BufferImage->Read();
|
||
|
|
}
|
||
|
|
|
||
|
|
MainEditor.Refresh ();
|
||
|
|
MainEditor.MenuBar->Refresh();
|
||
|
|
|
||
|
|
MainEditor.KeyInput ();
|
||
|
|
|
||
|
|
MainEditor.Cleanup();
|
||
|
|
|
||
|
|
return EFI_SUCCESS;
|
||
|
|
}
|
||
|
|
|