BaseTools/Upt: Fix several bugs

[mirror_edk2.git] / ArmPlatformPkg / FileSystem / BootMonFs / BootMonFsReadWrite.c

/** @file\r
*\r
*  Copyright (c) 2012-2014, ARM Limited. All rights reserved.\r
*\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 <Protocol/SimpleFileSystem.h>\r
#include <Library/UefiLib.h>\r
#include <Library/BaseMemoryLib.h>\r
#include <Library/MemoryAllocationLib.h>\r
#include <Library/DebugLib.h>\r
\r
#include "BootMonFsInternal.h"\r
\r
EFIAPI\r
EFI_STATUS\r
BootMonFsReadFile (\r
  IN EFI_FILE_PROTOCOL  *This,\r
  IN OUT UINTN          *BufferSize,\r
  OUT VOID              *Buffer\r
  )\r
{\r
  BOOTMON_FS_INSTANCE   *Instance;\r
  BOOTMON_FS_FILE       *File;\r
  EFI_DISK_IO_PROTOCOL  *DiskIo;\r
  EFI_BLOCK_IO_MEDIA    *Media;\r
  UINT64                FileStart;\r
  EFI_STATUS            Status;\r
  UINTN                 RemainingFileSize;\r
\r
  // Ensure the file has been written in Flash before reading it.\r
  // This keeps the code simple and avoids having to manage a non-flushed file.\r
  BootMonFsFlushFile (This);\r
\r
  File = BOOTMON_FS_FILE_FROM_FILE_THIS (This);\r
  if (File == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  Instance = File->Instance;\r
  DiskIo = Instance->DiskIo;\r
  Media = Instance->Media;\r
  FileStart = (Media->LowestAlignedLba + File->HwDescription.BlockStart) * Media->BlockSize;\r
\r
  if (File->Position >= File->HwDescription.Region[0].Size) {\r
    // The entire file has been read\r
    *BufferSize = 0;\r
    return EFI_DEVICE_ERROR;\r
  }\r
\r
  // This driver assumes that the entire file is in region 0.\r
  RemainingFileSize = File->HwDescription.Region[0].Size - File->Position;\r
\r
  // If read would go past end of file, truncate the read\r
  if (*BufferSize > RemainingFileSize) {\r
    *BufferSize = RemainingFileSize;\r
  }\r
\r
  Status = DiskIo->ReadDisk (\r
                    DiskIo,\r
                    Media->MediaId,\r
                    FileStart + File->Position,\r
                    *BufferSize,\r
                    Buffer\r
                    );\r
  if (EFI_ERROR (Status)) {\r
    *BufferSize = 0;\r
  }\r
\r
  File->Position += *BufferSize;\r
\r
  return Status;\r
}\r
\r
// Inserts an entry into the write chain\r
EFIAPI\r
EFI_STATUS\r
BootMonFsWriteFile (\r
  IN EFI_FILE_PROTOCOL  *This,\r
  IN OUT UINTN          *BufferSize,\r
  IN VOID               *Buffer\r
  )\r
{\r
  BOOTMON_FS_FILE         *File;\r
  BOOTMON_FS_FILE_REGION  *Region;\r
\r
  File = BOOTMON_FS_FILE_FROM_FILE_THIS (This);\r
  if (File == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  if (!(File->OpenMode & EFI_FILE_MODE_WRITE)) {\r
    return EFI_ACCESS_DENIED;\r
  }\r
\r
  // Allocate and initialize the memory region\r
  Region = (BOOTMON_FS_FILE_REGION*)AllocateZeroPool (sizeof (BOOTMON_FS_FILE_REGION));\r
  if (Region == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  Region->Buffer  = AllocateCopyPool (*BufferSize, Buffer);\r
  if (Region->Buffer == NULL) {\r
    FreePool (Region);\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  Region->Size    = *BufferSize;\r
\r
  Region->Offset = File->Position;\r
\r
  InsertTailList (&File->RegionToFlushLink, &Region->Link);\r
\r
  File->Position += *BufferSize;\r
\r
  return EFI_SUCCESS;\r
}\r
\r
EFIAPI\r
EFI_STATUS\r
BootMonFsSetPosition (\r
  IN EFI_FILE_PROTOCOL  *This,\r
  IN UINT64             Position\r
  )\r
{\r
  BOOTMON_FS_FILE         *File;\r
\r
  File = BOOTMON_FS_FILE_FROM_FILE_THIS (This);\r
\r
  // UEFI Spec section 12.5:\r
  // "Seeking to position 0xFFFFFFFFFFFFFFFF causes the current position to\r
  //  be set to the end of the file."\r
  if (Position == 0xFFFFFFFFFFFFFFFF) {\r
    File->Position = BootMonFsGetImageLength (File);\r
  } else {\r
    // NB: Seeking past the end of the file is valid.\r
    File->Position = Position;\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
EFIAPI\r
EFI_STATUS\r
BootMonFsGetPosition (\r
  IN  EFI_FILE_PROTOCOL *This,\r
  OUT UINT64            *Position\r
  ) {\r
  BOOTMON_FS_FILE         *File;\r
\r
  File = BOOTMON_FS_FILE_FROM_FILE_THIS (This);\r
\r
  *Position = File->Position;\r
  return EFI_SUCCESS;\r
}\r