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

/** @file\r
  Functions to deal with Disk buffer.\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 "HexEditor.h"\r
#include <Protocol/BlockIo.h>\r
\r
extern EFI_HANDLE                 HImageHandleBackup;\r
extern HEFI_EDITOR_BUFFER_IMAGE   HBufferImage;\r
\r
extern BOOLEAN                    HBufferImageNeedRefresh;\r
extern BOOLEAN                    HBufferImageOnlyLineNeedRefresh;\r
extern BOOLEAN                    HBufferImageMouseNeedRefresh;\r
\r
extern HEFI_EDITOR_GLOBAL_EDITOR  HMainEditor;\r
\r
HEFI_EDITOR_DISK_IMAGE            HDiskImage;\r
HEFI_EDITOR_DISK_IMAGE            HDiskImageBackupVar;\r
\r
//\r
// for basic initialization of HDiskImage\r
//\r
HEFI_EDITOR_DISK_IMAGE            HDiskImageConst = {\r
  NULL,\r
  0,\r
  0,\r
  0\r
};\r
\r
/**\r
  Initialization function for HDiskImage.\r
 \r
  @retval EFI_SUCCESS     The operation was successful.\r
  @retval EFI_LOAD_ERROR  A load error occured.\r
**/\r
EFI_STATUS\r
HDiskImageInit (\r
  VOID\r
  )\r
{\r
  //\r
  // basically initialize the HDiskImage\r
  //\r
  CopyMem (&HDiskImage, &HDiskImageConst, sizeof (HDiskImage));\r
\r
  CopyMem (&HDiskImageBackupVar, &HDiskImageConst, sizeof (HDiskImageBackupVar));\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Backup function for HDiskImage. Only a few fields need to be backup.   \r
  This is for making the Disk buffer refresh as few as possible.\r
\r
  @retval EFI_SUCCESS           The operation was successful.\r
  @retval EFI_OUT_OF_RESOURCES  gST->ConOut of resources.\r
**/\r
EFI_STATUS\r
HDiskImageBackup (\r
  VOID\r
  )\r
{\r
  //\r
  // backup the disk name, offset and size\r
  //\r
  //\r
  SHELL_FREE_NON_NULL (HDiskImageBackupVar.Name);\r
\r
  HDiskImageBackupVar.Name = CatSPrint(NULL, L"%s", HDiskImage.Name);\r
  if (HDiskImageBackupVar.Name == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  HDiskImageBackupVar.Offset  = HDiskImage.Offset;\r
  HDiskImageBackupVar.Size    = HDiskImage.Size;\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Cleanup function for HDiskImage.\r
\r
  @retval EFI_SUCCESS           The operation was successful.\r
**/\r
EFI_STATUS\r
HDiskImageCleanup (\r
  VOID\r
  )\r
{\r
  SHELL_FREE_NON_NULL (HDiskImage.Name);\r
  SHELL_FREE_NON_NULL (HDiskImageBackupVar.Name);\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Set FileName field in HFileImage.\r
\r
  @param[in] Str      File name to set.\r
  @param[in] Offset   The offset.\r
  @param[in] Size     The size.\r
\r
  @retval EFI_SUCCESS           The operation was successful.\r
  @retval EFI_OUT_OF_RESOURCES  A memory allocation failed.\r
**/\r
EFI_STATUS\r
EFIAPI\r
HDiskImageSetDiskNameOffsetSize (\r
  IN CONST CHAR16   *Str,\r
  IN UINTN    Offset,\r
  IN UINTN    Size\r
  )\r
{\r
  UINTN Len;\r
  UINTN Index;\r
\r
  //\r
  // free the old file name\r
  //\r
  SHELL_FREE_NON_NULL (HDiskImage.Name);\r
\r
  Len             = StrLen (Str);\r
\r
  HDiskImage.Name = AllocateZeroPool (2 * (Len + 1));\r
  if (HDiskImage.Name == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  for (Index = 0; Index < Len; Index++) {\r
    HDiskImage.Name[Index] = Str[Index];\r
  }\r
\r
  HDiskImage.Name[Len]  = L'\0';\r
\r
  HDiskImage.Offset     = Offset;\r
  HDiskImage.Size       = Size;\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Read a disk from disk into HBufferImage.\r
\r
  @param[in] DeviceName   filename to read.\r
  @param[in] Offset       The offset.\r
  @param[in] Size         The size.\r
  @param[in] Recover      if is for recover, no information print.\r
\r
  @retval EFI_SUCCESS           The operation was successful.\r
  @retval EFI_OUT_OF_RESOURCES  A memory allocation failed.\r
  @retval EFI_LOAD_ERROR        A load error occured.\r
  @retval EFI_INVALID_PARAMETER A parameter was invalid.  \r
**/\r
EFI_STATUS\r
HDiskImageRead (\r
  IN CONST CHAR16   *DeviceName,\r
  IN UINTN    Offset,\r
  IN UINTN    Size,\r
  IN BOOLEAN  Recover\r
  )\r
{\r
  CONST EFI_DEVICE_PATH_PROTOCOL  *DevicePath;\r
  EFI_DEVICE_PATH_PROTOCOL        *DupDevicePath;\r
  EFI_DEVICE_PATH_PROTOCOL        *DupDevicePathForFree;\r
  EFI_HANDLE                      Handle;\r
  EFI_BLOCK_IO_PROTOCOL           *BlkIo;\r
  EFI_STATUS                      Status;\r
\r
  VOID                            *Buffer;\r
  CHAR16                          *Str;\r
  UINTN                           Bytes;\r
\r
  HEFI_EDITOR_LINE                *Line;\r
\r
  HBufferImage.BufferType = FileTypeDiskBuffer;\r
\r
  DevicePath              = gEfiShellProtocol->GetDevicePathFromMap(DeviceName);\r
  if (DevicePath == NULL) {\r
    StatusBarSetStatusString (L"Cannot Find Device");\r
    return EFI_INVALID_PARAMETER;\r
  }\r
  DupDevicePath = DuplicateDevicePath(DevicePath);\r
  DupDevicePathForFree = DupDevicePath;\r
  //\r
  // get blkio interface\r
  //\r
  Status = gBS->LocateDevicePath(&gEfiBlockIoProtocolGuid,&DupDevicePath,&Handle);\r
  FreePool(DupDevicePathForFree);\r
  if (EFI_ERROR (Status)) {\r
    StatusBarSetStatusString (L"Read Disk Failed");\r
    return Status;\r
  }\r
  Status = gBS->OpenProtocol(Handle, &gEfiBlockIoProtocolGuid, (VOID**)&BlkIo, gImageHandle, NULL, EFI_OPEN_PROTOCOL_GET_PROTOCOL);\r
  if (EFI_ERROR (Status)) {\r
    StatusBarSetStatusString (L"Read Disk Failed");\r
    return Status;\r
  }\r
  //\r
  // if Offset exceeds LastBlock,\r
  //   return error\r
  //\r
  if (Offset > BlkIo->Media->LastBlock || Offset + Size > BlkIo->Media->LastBlock) {\r
    StatusBarSetStatusString (L"Invalid Offset + Size");\r
    return EFI_LOAD_ERROR;\r
  }\r
\r
  Bytes   = BlkIo->Media->BlockSize * Size;\r
  Buffer  = AllocateZeroPool (Bytes);\r
\r
  if (Buffer == NULL) {\r
    StatusBarSetStatusString (L"Read Disk Failed");\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  //\r
  // read from disk\r
  //\r
  Status = BlkIo->ReadBlocks (\r
                    BlkIo,\r
                    BlkIo->Media->MediaId,\r
                    Offset,\r
                    Bytes,\r
                    Buffer\r
                    );\r
\r
  if (EFI_ERROR (Status)) {\r
    FreePool (Buffer);\r
    StatusBarSetStatusString (L"Read Disk Failed");\r
    return EFI_LOAD_ERROR;\r
  }\r
\r
  HBufferImageFree ();\r
\r
  //\r
  // convert buffer to line list\r
  //\r
  Status = HBufferImageBufferToList (Buffer, Bytes);\r
  FreePool (Buffer);\r
\r
  if (EFI_ERROR (Status)) {\r
    StatusBarSetStatusString (L"Read Disk Failed");\r
    return Status;\r
  }\r
\r
  Status = HDiskImageSetDiskNameOffsetSize (DeviceName, Offset, Size);\r
  if (EFI_ERROR (Status)) {\r
    StatusBarSetStatusString (L"Read Disk Failed");\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
  //\r
  // initialize some variables\r
  //\r
  HDiskImage.BlockSize                = BlkIo->Media->BlockSize;\r
\r
  HBufferImage.DisplayPosition.Row    = 2;\r
  HBufferImage.DisplayPosition.Column = 10;\r
\r
  HBufferImage.MousePosition.Row      = 2;\r
  HBufferImage.MousePosition.Column   = 10;\r
\r
  HBufferImage.LowVisibleRow          = 1;\r
  HBufferImage.HighBits               = TRUE;\r
\r
  HBufferImage.BufferPosition.Row     = 1;\r
  HBufferImage.BufferPosition.Column  = 1;\r
\r
  if (!Recover) {\r
    Str = CatSPrint(NULL, L"%d Lines Read", HBufferImage.NumLines);\r
    if (Str == NULL) {\r
      StatusBarSetStatusString (L"Read Disk Failed");\r
      return EFI_OUT_OF_RESOURCES;\r
    }\r
\r
    StatusBarSetStatusString (Str);\r
    SHELL_FREE_NON_NULL (Str);\r
\r
    HMainEditor.SelectStart = 0;\r
    HMainEditor.SelectEnd   = 0;\r
\r
  }\r
\r
  //\r
  // has line\r
  //\r
  if (HBufferImage.Lines != NULL) {\r
    HBufferImage.CurrentLine = CR (\r
                                HBufferImage.ListHead->ForwardLink,\r
                                HEFI_EDITOR_LINE,\r
                                Link,\r
                                EFI_EDITOR_LINE_LIST\r
                                );\r
  } else {\r
    //\r
    // create a dummy line\r
    //\r
    Line = HBufferImageCreateLine ();\r
    if (Line == NULL) {\r
      StatusBarSetStatusString (L"Read Disk Failed");\r
      return EFI_OUT_OF_RESOURCES;\r
    }\r
\r
    HBufferImage.CurrentLine = Line;\r
  }\r
\r
  HBufferImage.Modified           = FALSE;\r
  HBufferImageNeedRefresh         = TRUE;\r
  HBufferImageOnlyLineNeedRefresh = FALSE;\r
  HBufferImageMouseNeedRefresh    = TRUE;\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Save lines in HBufferImage to disk.\r
  NOT ALLOW TO WRITE TO ANOTHER DISK!!!!!!!!!\r
\r
  @param[in] DeviceName   The device name.\r
  @param[in] Offset       The offset.\r
  @param[in] Size         The size.\r
\r
  @retval EFI_SUCCESS           The operation was successful.\r
  @retval EFI_OUT_OF_RESOURCES  A memory allocation failed.\r
  @retval EFI_LOAD_ERROR        A load error occured.\r
  @retval EFI_INVALID_PARAMETER A parameter was invalid.  \r
**/\r
EFI_STATUS\r
HDiskImageSave (\r
  IN CHAR16 *DeviceName,\r
  IN UINTN  Offset,\r
  IN UINTN  Size\r
  )\r
{\r
\r
  CONST EFI_DEVICE_PATH_PROTOCOL  *DevicePath;\r
  EFI_DEVICE_PATH_PROTOCOL        *DupDevicePath;\r
  EFI_BLOCK_IO_PROTOCOL           *BlkIo;\r
  EFI_STATUS                      Status;\r
  EFI_HANDLE                      Handle;\r
  VOID                            *Buffer;\r
  UINTN                           Bytes;\r
\r
  //\r
  // if not modified, directly return\r
  //\r
  if (HBufferImage.Modified == FALSE) {\r
    return EFI_SUCCESS;\r
  }\r
\r
  HBufferImage.BufferType = FileTypeDiskBuffer;\r
\r
  DevicePath              = gEfiShellProtocol->GetDevicePathFromMap(DeviceName);\r
  if (DevicePath == NULL) {\r
//    StatusBarSetStatusString (L"Cannot Find Device");\r
    return EFI_INVALID_PARAMETER;\r
  }\r
  DupDevicePath = DuplicateDevicePath(DevicePath);\r
\r
  //\r
  // get blkio interface\r
  //\r
  Status = gBS->LocateDevicePath(&gEfiBlockIoProtocolGuid,&DupDevicePath,&Handle);\r
  FreePool(DupDevicePath);\r
  if (EFI_ERROR (Status)) {\r
//    StatusBarSetStatusString (L"Read Disk Failed");\r
    return Status;\r
  }\r
  Status = gBS->OpenProtocol(Handle, &gEfiBlockIoProtocolGuid, (VOID**)&BlkIo, gImageHandle, NULL, EFI_OPEN_PROTOCOL_GET_PROTOCOL);\r
  if (EFI_ERROR (Status)) {\r
//    StatusBarSetStatusString (L"Read Disk Failed");\r
    return Status;\r
  }\r
\r
  Bytes   = BlkIo->Media->BlockSize * Size;\r
  Buffer  = AllocateZeroPool (Bytes);\r
\r
  if (Buffer == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
  //\r
  // concatenate the line list to a buffer\r
  //\r
  Status = HBufferImageListToBuffer (Buffer, Bytes);\r
  if (EFI_ERROR (Status)) {\r
    FreePool (Buffer);\r
    return Status;\r
  }\r
\r
  //\r
  // write the buffer to disk\r
  //\r
  Status = BlkIo->WriteBlocks (\r
                    BlkIo,\r
                    BlkIo->Media->MediaId,\r
                    Offset,\r
                    Bytes,\r
                    Buffer\r
                    );\r
\r
  FreePool (Buffer);\r
\r
  if (EFI_ERROR (Status)) {\r
    return EFI_LOAD_ERROR;\r
  }\r
  //\r
  // now not modified\r
  //\r
  HBufferImage.Modified = FALSE;\r
\r
  return EFI_SUCCESS;\r
}\r
Commit	Line	Data
632820d1	1	/** @file\r
	2	Functions to deal with Disk buffer.\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 "HexEditor.h"\r
	16	#include <Protocol/BlockIo.h>\r
	17	\r
	18	extern EFI_HANDLE HImageHandleBackup;\r
	19	extern HEFI_EDITOR_BUFFER_IMAGE HBufferImage;\r
	20	\r
	21	extern BOOLEAN HBufferImageNeedRefresh;\r
	22	extern BOOLEAN HBufferImageOnlyLineNeedRefresh;\r
	23	extern BOOLEAN HBufferImageMouseNeedRefresh;\r
	24	\r
	25	extern HEFI_EDITOR_GLOBAL_EDITOR HMainEditor;\r
	26	\r
	27	HEFI_EDITOR_DISK_IMAGE HDiskImage;\r
	28	HEFI_EDITOR_DISK_IMAGE HDiskImageBackupVar;\r
	29	\r
	30	//\r
	31	// for basic initialization of HDiskImage\r
	32	//\r
	33	HEFI_EDITOR_DISK_IMAGE HDiskImageConst = {\r
	34	NULL,\r
	35	0,\r
	36	0,\r
	37	0\r
	38	};\r
	39	\r
a1d4bfcc	40	/**\r
	41	Initialization function for HDiskImage.\r
	42	\r
	43	@retval EFI_SUCCESS The operation was successful.\r
	44	@retval EFI_LOAD_ERROR A load error occured.\r
	45	**/\r
632820d1	46	EFI_STATUS\r
	47	HDiskImageInit (\r
	48	VOID\r
	49	)\r
632820d1	50	{\r
	51	//\r
	52	// basically initialize the HDiskImage\r
	53	//\r
	54	CopyMem (&HDiskImage, &HDiskImageConst, sizeof (HDiskImage));\r
	55	\r
	56	CopyMem (&HDiskImageBackupVar, &HDiskImageConst, sizeof (HDiskImageBackupVar));\r
	57	\r
	58	return EFI_SUCCESS;\r
	59	}\r
	60	\r
a1d4bfcc	61	/**\r
	62	Backup function for HDiskImage. Only a few fields need to be backup. \r
	63	This is for making the Disk buffer refresh as few as possible.\r
	64	\r
	65	@retval EFI_SUCCESS The operation was successful.\r
	66	@retval EFI_OUT_OF_RESOURCES gST->ConOut of resources.\r
	67	**/\r
632820d1	68	EFI_STATUS\r
	69	HDiskImageBackup (\r
	70	VOID\r
	71	)\r
632820d1	72	{\r
	73	//\r
	74	// backup the disk name, offset and size\r
	75	//\r
	76	//\r
	77	SHELL_FREE_NON_NULL (HDiskImageBackupVar.Name);\r
	78	\r
	79	HDiskImageBackupVar.Name = CatSPrint(NULL, L"%s", HDiskImage.Name);\r
	80	if (HDiskImageBackupVar.Name == NULL) {\r
	81	return EFI_OUT_OF_RESOURCES;\r
	82	}\r
	83	\r
	84	HDiskImageBackupVar.Offset = HDiskImage.Offset;\r
	85	HDiskImageBackupVar.Size = HDiskImage.Size;\r
	86	\r
	87	return EFI_SUCCESS;\r
	88	}\r
	89	\r
a1d4bfcc	90	/**\r
	91	Cleanup function for HDiskImage.\r
	92	\r
	93	@retval EFI_SUCCESS The operation was successful.\r
	94	**/\r
632820d1	95	EFI_STATUS\r
	96	HDiskImageCleanup (\r
	97	VOID\r
	98	)\r
632820d1	99	{\r
	100	SHELL_FREE_NON_NULL (HDiskImage.Name);\r
	101	SHELL_FREE_NON_NULL (HDiskImageBackupVar.Name);\r
	102	\r
	103	return EFI_SUCCESS;\r
	104	}\r
	105	\r
a1d4bfcc	106	/**\r
	107	Set FileName field in HFileImage.\r
	108	\r
	109	@param[in] Str File name to set.\r
	110	@param[in] Offset The offset.\r
	111	@param[in] Size The size.\r
	112	\r
	113	@retval EFI_SUCCESS The operation was successful.\r
	114	@retval EFI_OUT_OF_RESOURCES A memory allocation failed.\r
	115	**/\r
632820d1	116	EFI_STATUS\r
	117	EFIAPI\r
	118	HDiskImageSetDiskNameOffsetSize (\r
	119	IN CONST CHAR16 *Str,\r
	120	IN UINTN Offset,\r
	121	IN UINTN Size\r
	122	)\r
632820d1	123	{\r
	124	UINTN Len;\r
	125	UINTN Index;\r
	126	\r
	127	//\r
	128	// free the old file name\r
	129	//\r
	130	SHELL_FREE_NON_NULL (HDiskImage.Name);\r
	131	\r
	132	Len = StrLen (Str);\r
	133	\r
	134	HDiskImage.Name = AllocateZeroPool (2 * (Len + 1));\r
	135	if (HDiskImage.Name == NULL) {\r
	136	return EFI_OUT_OF_RESOURCES;\r
	137	}\r
	138	\r
	139	for (Index = 0; Index < Len; Index++) {\r
	140	HDiskImage.Name[Index] = Str[Index];\r
	141	}\r
	142	\r
	143	HDiskImage.Name[Len] = L'\0';\r
	144	\r
	145	HDiskImage.Offset = Offset;\r
	146	HDiskImage.Size = Size;\r
	147	\r
	148	return EFI_SUCCESS;\r
	149	}\r
	150	\r
a1d4bfcc	151	/**\r
	152	Read a disk from disk into HBufferImage.\r
	153	\r
	154	@param[in] DeviceName filename to read.\r
	155	@param[in] Offset The offset.\r
	156	@param[in] Size The size.\r
	157	@param[in] Recover if is for recover, no information print.\r
	158	\r
	159	@retval EFI_SUCCESS The operation was successful.\r
	160	@retval EFI_OUT_OF_RESOURCES A memory allocation failed.\r
	161	@retval EFI_LOAD_ERROR A load error occured.\r
	162	@retval EFI_INVALID_PARAMETER A parameter was invalid. \r
	163	**/\r
632820d1	164	EFI_STATUS\r
	165	HDiskImageRead (\r
	166	IN CONST CHAR16 *DeviceName,\r
	167	IN UINTN Offset,\r
	168	IN UINTN Size,\r
	169	IN BOOLEAN Recover\r
	170	)\r
632820d1	171	{\r
	172	CONST EFI_DEVICE_PATH_PROTOCOL *DevicePath;\r
	173	EFI_DEVICE_PATH_PROTOCOL *DupDevicePath;\r
980d554e	174	EFI_DEVICE_PATH_PROTOCOL *DupDevicePathForFree;\r
632820d1	175	EFI_HANDLE Handle;\r
	176	EFI_BLOCK_IO_PROTOCOL *BlkIo;\r
	177	EFI_STATUS Status;\r
	178	\r
	179	VOID *Buffer;\r
	180	CHAR16 *Str;\r
	181	UINTN Bytes;\r
	182	\r
	183	HEFI_EDITOR_LINE *Line;\r
632820d1	184	\r
632820d1	185	HBufferImage.BufferType = FileTypeDiskBuffer;\r
	186	\r
	187	DevicePath = gEfiShellProtocol->GetDevicePathFromMap(DeviceName);\r
	188	if (DevicePath == NULL) {\r
	189	StatusBarSetStatusString (L"Cannot Find Device");\r
	190	return EFI_INVALID_PARAMETER;\r
	191	}\r
	192	DupDevicePath = DuplicateDevicePath(DevicePath);\r
980d554e	193	DupDevicePathForFree = DupDevicePath;\r
632820d1	194	//\r
	195	// get blkio interface\r
	196	//\r
	197	Status = gBS->LocateDevicePath(&gEfiBlockIoProtocolGuid,&DupDevicePath,&Handle);\r
980d554e	198	FreePool(DupDevicePathForFree);\r
632820d1	199	if (EFI_ERROR (Status)) {\r
	200	StatusBarSetStatusString (L"Read Disk Failed");\r
	201	return Status;\r
	202	}\r
	203	Status = gBS->OpenProtocol(Handle, &gEfiBlockIoProtocolGuid, (VOID**)&BlkIo, gImageHandle, NULL, EFI_OPEN_PROTOCOL_GET_PROTOCOL);\r
	204	if (EFI_ERROR (Status)) {\r
	205	StatusBarSetStatusString (L"Read Disk Failed");\r
	206	return Status;\r
	207	}\r
	208	//\r
	209	// if Offset exceeds LastBlock,\r
	210	// return error\r
	211	//\r
	212	if (Offset > BlkIo->Media->LastBlock \|\| Offset + Size > BlkIo->Media->LastBlock) {\r
	213	StatusBarSetStatusString (L"Invalid Offset + Size");\r
	214	return EFI_LOAD_ERROR;\r
	215	}\r
	216	\r
	217	Bytes = BlkIo->Media->BlockSize * Size;\r
	218	Buffer = AllocateZeroPool (Bytes);\r
	219	\r
	220	if (Buffer == NULL) {\r
	221	StatusBarSetStatusString (L"Read Disk Failed");\r
	222	return EFI_OUT_OF_RESOURCES;\r
	223	}\r
	224	\r
632820d1	225	//\r
	226	// read from disk\r
	227	//\r
	228	Status = BlkIo->ReadBlocks (\r
	229	BlkIo,\r
	230	BlkIo->Media->MediaId,\r
	231	Offset,\r
	232	Bytes,\r
	233	Buffer\r
	234	);\r
	235	\r
	236	if (EFI_ERROR (Status)) {\r
	237	FreePool (Buffer);\r
	238	StatusBarSetStatusString (L"Read Disk Failed");\r
	239	return EFI_LOAD_ERROR;\r
	240	}\r
	241	\r
	242	HBufferImageFree ();\r
	243	\r
	244	//\r
	245	// convert buffer to line list\r
	246	//\r
	247	Status = HBufferImageBufferToList (Buffer, Bytes);\r
	248	FreePool (Buffer);\r
	249	\r
	250	if (EFI_ERROR (Status)) {\r
	251	StatusBarSetStatusString (L"Read Disk Failed");\r
	252	return Status;\r
	253	}\r
	254	\r
	255	Status = HDiskImageSetDiskNameOffsetSize (DeviceName, Offset, Size);\r
	256	if (EFI_ERROR (Status)) {\r
	257	StatusBarSetStatusString (L"Read Disk Failed");\r
	258	return EFI_OUT_OF_RESOURCES;\r
	259	}\r
	260	//\r
	261	// initialize some variables\r
	262	//\r
	263	HDiskImage.BlockSize = BlkIo->Media->BlockSize;\r
	264	\r
	265	HBufferImage.DisplayPosition.Row = 2;\r
	266	HBufferImage.DisplayPosition.Column = 10;\r
	267	\r
	268	HBufferImage.MousePosition.Row = 2;\r
	269	HBufferImage.MousePosition.Column = 10;\r
	270	\r
	271	HBufferImage.LowVisibleRow = 1;\r
	272	HBufferImage.HighBits = TRUE;\r
	273	\r
	274	HBufferImage.BufferPosition.Row = 1;\r
	275	HBufferImage.BufferPosition.Column = 1;\r
	276	\r
	277	if (!Recover) {\r
	278	Str = CatSPrint(NULL, L"%d Lines Read", HBufferImage.NumLines);\r
	279	if (Str == NULL) {\r
	280	StatusBarSetStatusString (L"Read Disk Failed");\r
	281	return EFI_OUT_OF_RESOURCES;\r
	282	}\r
	283	\r
	284	StatusBarSetStatusString (Str);\r
	285	SHELL_FREE_NON_NULL (Str);\r
	286	\r
	287	HMainEditor.SelectStart = 0;\r
	288	HMainEditor.SelectEnd = 0;\r
289	\r
290	}\r
291	\r
292	//\r
293	// has line\r
294	//\r
295	if (HBufferImage.Lines != NULL) {\r
296	HBufferImage.CurrentLine = CR (\r
297	HBufferImage.ListHead->ForwardLink,\r
298	HEFI_EDITOR_LINE,\r
299	Link,\r
300	EFI_EDITOR_LINE_LIST\r
301	);\r
302	} else {\r
303	//\r
304	// create a dummy line\r
305	//\r
306	Line = HBufferImageCreateLine ();\r
307	if (Line == NULL) {\r
308	StatusBarSetStatusString (L"Read Disk Failed");\r
309	return EFI_OUT_OF_RESOURCES;\r
310	}\r
311	\r
312	HBufferImage.CurrentLine = Line;\r
313	}\r
314	\r
315	HBufferImage.Modified = FALSE;\r
316	HBufferImageNeedRefresh = TRUE;\r
317	HBufferImageOnlyLineNeedRefresh = FALSE;\r
318	HBufferImageMouseNeedRefresh = TRUE;\r
319	\r
320	return EFI_SUCCESS;\r
321	}\r
322	\r
a1d4bfcc	323	/**\r
	324	Save lines in HBufferImage to disk.\r
	325	NOT ALLOW TO WRITE TO ANOTHER DISK!!!!!!!!!\r
	326	\r
	327	@param[in] DeviceName The device name.\r
	328	@param[in] Offset The offset.\r
	329	@param[in] Size The size.\r
	330	\r
	331	@retval EFI_SUCCESS The operation was successful.\r
	332	@retval EFI_OUT_OF_RESOURCES A memory allocation failed.\r
	333	@retval EFI_LOAD_ERROR A load error occured.\r
	334	@retval EFI_INVALID_PARAMETER A parameter was invalid. \r
	335	**/\r
632820d1	336	EFI_STATUS\r
	337	HDiskImageSave (\r
	338	IN CHAR16 *DeviceName,\r
	339	IN UINTN Offset,\r
	340	IN UINTN Size\r
	341	)\r
632820d1	342	{\r
	343	\r
	344	CONST EFI_DEVICE_PATH_PROTOCOL *DevicePath;\r
	345	EFI_DEVICE_PATH_PROTOCOL *DupDevicePath;\r
	346	EFI_BLOCK_IO_PROTOCOL *BlkIo;\r
	347	EFI_STATUS Status;\r
	348	EFI_HANDLE Handle;\r
	349	VOID *Buffer;\r
	350	UINTN Bytes;\r
	351	\r
632820d1	352	//\r
	353	// if not modified, directly return\r
	354	//\r
	355	if (HBufferImage.Modified == FALSE) {\r
	356	return EFI_SUCCESS;\r
	357	}\r
	358	\r
632820d1	359	HBufferImage.BufferType = FileTypeDiskBuffer;\r
	360	\r
	361	DevicePath = gEfiShellProtocol->GetDevicePathFromMap(DeviceName);\r
	362	if (DevicePath == NULL) {\r
	363	// StatusBarSetStatusString (L"Cannot Find Device");\r
	364	return EFI_INVALID_PARAMETER;\r
	365	}\r
	366	DupDevicePath = DuplicateDevicePath(DevicePath);\r
	367	\r
	368	//\r
	369	// get blkio interface\r
	370	//\r
	371	Status = gBS->LocateDevicePath(&gEfiBlockIoProtocolGuid,&DupDevicePath,&Handle);\r
	372	FreePool(DupDevicePath);\r
	373	if (EFI_ERROR (Status)) {\r
	374	// StatusBarSetStatusString (L"Read Disk Failed");\r
	375	return Status;\r
	376	}\r
	377	Status = gBS->OpenProtocol(Handle, &gEfiBlockIoProtocolGuid, (VOID**)&BlkIo, gImageHandle, NULL, EFI_OPEN_PROTOCOL_GET_PROTOCOL);\r
	378	if (EFI_ERROR (Status)) {\r
	379	// StatusBarSetStatusString (L"Read Disk Failed");\r
	380	return Status;\r
	381	}\r
	382	\r
	383	Bytes = BlkIo->Media->BlockSize * Size;\r
	384	Buffer = AllocateZeroPool (Bytes);\r
	385	\r
	386	if (Buffer == NULL) {\r
	387	return EFI_OUT_OF_RESOURCES;\r
	388	}\r
	389	//\r
	390	// concatenate the line list to a buffer\r
	391	//\r
	392	Status = HBufferImageListToBuffer (Buffer, Bytes);\r
	393	if (EFI_ERROR (Status)) {\r
	394	FreePool (Buffer);\r
	395	return Status;\r
	396	}\r
	397	\r
632820d1	398	//\r
	399	// write the buffer to disk\r
	400	//\r
	401	Status = BlkIo->WriteBlocks (\r
	402	BlkIo,\r
	403	BlkIo->Media->MediaId,\r
	404	Offset,\r
	405	Bytes,\r
	406	Buffer\r
	407	);\r
	408	\r
	409	FreePool (Buffer);\r
	410	\r
	411	if (EFI_ERROR (Status)) {\r
	412	return EFI_LOAD_ERROR;\r
	413	}\r
	414	//\r
	415	// now not modified\r
	416	//\r
	417	HBufferImage.Modified = FALSE;\r
	418	\r
	419	return EFI_SUCCESS;\r
	420	}\r