[mirror_edk2.git] / MdeModulePkg / Universal / FaultTolerantWriteDxe / FaultTolerantWrite.h

/** @file\r
\r
  The internal header file includes the common header files, defines\r
  internal structure and functions used by Ftw module.\r
\r
Copyright (c) 2006 - 2014, 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
#ifndef _EFI_FAULT_TOLERANT_WRITE_H_\r
#define _EFI_FAULT_TOLERANT_WRITE_H_\r
\r
#include <PiDxe.h>\r
\r
#include <Guid/SystemNvDataGuid.h>\r
#include <Guid/ZeroGuid.h>\r
#include <Protocol/FaultTolerantWrite.h>\r
#include <Protocol/FirmwareVolumeBlock.h>\r
#include <Protocol/SwapAddressRange.h>\r
\r
#include <Library/PcdLib.h>\r
#include <Library/DebugLib.h>\r
#include <Library/UefiLib.h>\r
#include <Library/UefiDriverEntryPoint.h>\r
#include <Library/BaseMemoryLib.h>\r
#include <Library/MemoryAllocationLib.h>\r
#include <Library/UefiBootServicesTableLib.h>\r
#include <Library/ReportStatusCodeLib.h>\r
\r
//\r
// Flash erase polarity is 1\r
//\r
#define FTW_ERASE_POLARITY  1\r
\r
#define FTW_ERASED_BYTE     ((UINT8) (255))\r
#define FTW_POLARITY_REVERT ((UINT8) (255))\r
\r
#define HEADER_ALLOCATED  0x1\r
#define WRITES_ALLOCATED  0x2\r
#define WRITES_COMPLETED  0x4\r
\r
#define BOOT_BLOCK_UPDATE 0x1\r
#define SPARE_COMPLETED   0x2\r
#define DEST_COMPLETED    0x4\r
\r
#define FTW_BLOCKS(Length, BlockSize) ((UINTN) ((Length) / (BlockSize) + (((Length) & ((BlockSize) - 1)) ? 1 : 0)))\r
\r
#define FTW_DEVICE_SIGNATURE  SIGNATURE_32 ('F', 'T', 'W', 'D')\r
\r
//\r
// EFI Fault tolerant protocol private data structure\r
//\r
typedef struct {\r
  UINTN                                   Signature;\r
  EFI_HANDLE                              Handle;\r
  EFI_FAULT_TOLERANT_WRITE_PROTOCOL       FtwInstance;\r
  EFI_PHYSICAL_ADDRESS                    WorkSpaceAddress;   // Base address of working space range in flash.\r
  EFI_PHYSICAL_ADDRESS                    SpareAreaAddress;   // Base address of spare range in flash.\r
  UINTN                                   WorkSpaceLength;    // Size of working space range in flash.\r
  UINTN                                   NumberOfWorkSpaceBlock; // Number of the blocks in work block for work space.\r
  UINTN                                   WorkBlockSize;      // Block size in bytes of the work blocks in flash\r
  UINTN                                   SpareAreaLength;    // Size of spare range in flash.\r
  UINTN                                   NumberOfSpareBlock; // Number of the blocks in spare block.\r
  UINTN                                   SpareBlockSize;     // Block size in bytes of the spare blocks in flash\r
  EFI_FAULT_TOLERANT_WORKING_BLOCK_HEADER *FtwWorkSpaceHeader;// Pointer to Working Space Header in memory buffer\r
  EFI_FAULT_TOLERANT_WRITE_HEADER         *FtwLastWriteHeader;// Pointer to last record header in memory buffer\r
  EFI_FAULT_TOLERANT_WRITE_RECORD         *FtwLastWriteRecord;// Pointer to last record in memory buffer\r
  EFI_FIRMWARE_VOLUME_BLOCK_PROTOCOL      *FtwFvBlock;        // FVB of working block\r
  EFI_FIRMWARE_VOLUME_BLOCK_PROTOCOL      *FtwBackupFvb;      // FVB of spare block\r
  EFI_LBA                                 FtwSpareLba;        // Start LBA of spare block\r
  EFI_LBA                                 FtwWorkBlockLba;    // Start LBA of working block that contains working space in its last block.\r
  UINTN                                   NumberOfWorkBlock;  // Number of the blocks in work block.\r
  EFI_LBA                                 FtwWorkSpaceLba;    // Start LBA of working space\r
  UINTN                                   FtwWorkSpaceBase;   // Offset into the FtwWorkSpaceLba block.\r
  UINTN                                   FtwWorkSpaceSize;   // Size of working space range that stores write record.\r
  EFI_LBA                                 FtwWorkSpaceLbaInSpare; // Start LBA of working space in spare block.\r
  UINTN                                   FtwWorkSpaceBaseInSpare;// Offset into the FtwWorkSpaceLbaInSpare block.\r
  UINT8                                   *FtwWorkSpace;      // Point to Work Space in memory buffer \r
  //\r
  // Following a buffer of FtwWorkSpace[FTW_WORK_SPACE_SIZE],\r
  // Allocated with EFI_FTW_DEVICE.\r
  //\r
} EFI_FTW_DEVICE;\r
\r
#define FTW_CONTEXT_FROM_THIS(a)  CR (a, EFI_FTW_DEVICE, FtwInstance, FTW_DEVICE_SIGNATURE)\r
\r
//\r
// Driver entry point\r
//\r
/**\r
  This function is the entry point of the Fault Tolerant Write driver.\r
\r
  @param ImageHandle     A handle for the image that is initializing this driver\r
  @param SystemTable     A pointer to the EFI system table\r
\r
  @return EFI_SUCCESS           FTW has finished the initialization\r
  @retval EFI_NOT_FOUND         Locate FVB protocol error\r
  @retval EFI_OUT_OF_RESOURCES  Allocate memory error\r
  @retval EFI_VOLUME_CORRUPTED  Firmware volume is error\r
  @retval EFI_ABORTED           FTW initialization error\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
InitializeFaultTolerantWrite (\r
  IN EFI_HANDLE         ImageHandle,\r
  IN EFI_SYSTEM_TABLE   *SystemTable\r
  );\r
\r
//\r
// Fault Tolerant Write Protocol API\r
//\r
\r
/**\r
  Query the largest block that may be updated in a fault tolerant manner.\r
\r
\r
  @param This            Indicates a pointer to the calling context.\r
  @param BlockSize       A pointer to a caller allocated UINTN that is updated to\r
                         indicate the size of the largest block that can be updated.\r
\r
  @return EFI_SUCCESS   The function completed successfully\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
FtwGetMaxBlockSize (\r
  IN EFI_FAULT_TOLERANT_WRITE_PROTOCOL    *This,\r
  OUT UINTN                               *BlockSize\r
  );\r
\r
/**\r
  Allocates space for the protocol to maintain information about writes.\r
  Since writes must be completed in a fault tolerant manner and multiple\r
  updates will require more resources to be successful, this function\r
  enables the protocol to ensure that enough space exists to track\r
  information about the upcoming writes.\r
\r
  All writes must be completed or aborted before another fault tolerant write can occur.\r
\r
  @param This            Indicates a pointer to the calling context.\r
  @param CallerId        The GUID identifying the write.\r
  @param PrivateDataSize The size of the caller's private data\r
                         that must be recorded for each write.\r
  @param NumberOfWrites  The number of fault tolerant block writes\r
                         that will need to occur.\r
\r
  @return EFI_SUCCESS        The function completed successfully\r
  @retval EFI_ABORTED        The function could not complete successfully.\r
  @retval EFI_ACCESS_DENIED  All allocated writes have not been completed.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
FtwAllocate (\r
  IN EFI_FAULT_TOLERANT_WRITE_PROTOCOL    *This,\r
  IN EFI_GUID                             *CallerId,\r
  IN UINTN                                PrivateDataSize,\r
  IN UINTN                                NumberOfWrites\r
  );\r
\r
/**\r
  Starts a target block update. This function will record data about write\r
  in fault tolerant storage and will complete the write in a recoverable\r
  manner, ensuring at all times that either the original contents or\r
  the modified contents are available.\r
\r
\r
  @param This            Calling context\r
  @param Lba             The logical block address of the target block.\r
  @param Offset          The offset within the target block to place the data.\r
  @param Length          The number of bytes to write to the target block.\r
  @param PrivateData     A pointer to private data that the caller requires to\r
                         complete any pending writes in the event of a fault.\r
  @param FvBlockHandle   The handle of FVB protocol that provides services for\r
                         reading, writing, and erasing the target block.\r
  @param Buffer          The data to write.\r
\r
  @retval EFI_SUCCESS          The function completed successfully \r
  @retval EFI_ABORTED          The function could not complete successfully. \r
  @retval EFI_BAD_BUFFER_SIZE  The input data can't fit within the spare block. \r
                               Offset + *NumBytes > SpareAreaLength.\r
  @retval EFI_ACCESS_DENIED    No writes have been allocated. \r
  @retval EFI_OUT_OF_RESOURCES Cannot allocate enough memory resource.\r
  @retval EFI_NOT_FOUND        Cannot find FVB protocol by handle.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
FtwWrite (\r
  IN EFI_FAULT_TOLERANT_WRITE_PROTOCOL     *This,\r
  IN EFI_LBA                               Lba,\r
  IN UINTN                                 Offset,\r
  IN UINTN                                 Length,\r
  IN VOID                                  *PrivateData,\r
  IN EFI_HANDLE                            FvBlockHandle,\r
  IN VOID                                  *Buffer\r
  );\r
\r
/**\r
  Restarts a previously interrupted write. The caller must provide the\r
  block protocol needed to complete the interrupted write.\r
\r
  @param This            Calling context.\r
  @param FvBlockHandle   The handle of FVB protocol that provides services for\r
                         reading, writing, and erasing the target block.\r
\r
  @retval  EFI_SUCCESS          The function completed successfully\r
  @retval  EFI_ACCESS_DENIED    No pending writes exist\r
  @retval  EFI_NOT_FOUND        FVB protocol not found by the handle\r
  @retval  EFI_ABORTED          The function could not complete successfully\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
FtwRestart (\r
  IN EFI_FAULT_TOLERANT_WRITE_PROTOCOL     *This,\r
  IN EFI_HANDLE                            FvBlockHandle\r
  );\r
\r
/**\r
  Aborts all previous allocated writes.\r
\r
  @param  This                 Calling context\r
\r
  @retval EFI_SUCCESS          The function completed successfully\r
  @retval EFI_ABORTED          The function could not complete successfully.\r
  @retval EFI_NOT_FOUND        No allocated writes exist.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
FtwAbort (\r
  IN EFI_FAULT_TOLERANT_WRITE_PROTOCOL     *This\r
  );\r
\r
/**\r
  Starts a target block update. This records information about the write\r
  in fault tolerant storage and will complete the write in a recoverable\r
  manner, ensuring at all times that either the original contents or\r
  the modified contents are available.\r
\r
  @param This            Indicates a pointer to the calling context.\r
  @param CallerId        The GUID identifying the last write.\r
  @param Lba             The logical block address of the last write.\r
  @param Offset          The offset within the block of the last write.\r
  @param Length          The length of the last write.\r
  @param PrivateDataSize bytes from the private data\r
                         stored for this write.\r
  @param PrivateData     A pointer to a buffer. The function will copy\r
  @param Complete        A Boolean value with TRUE indicating\r
                         that the write was completed.\r
\r
  @retval EFI_SUCCESS           The function completed successfully\r
  @retval EFI_ABORTED           The function could not complete successfully\r
  @retval EFI_NOT_FOUND         No allocated writes exist\r
  @retval EFI_BUFFER_TOO_SMALL  Input buffer is not larget enough\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
FtwGetLastWrite (\r
  IN EFI_FAULT_TOLERANT_WRITE_PROTOCOL     *This,\r
  OUT EFI_GUID                             *CallerId,\r
  OUT EFI_LBA                              *Lba,\r
  OUT UINTN                                *Offset,\r
  OUT UINTN                                *Length,\r
  IN OUT UINTN                             *PrivateDataSize,\r
  OUT VOID                                 *PrivateData,\r
  OUT BOOLEAN                              *Complete\r
  );\r
\r
/**\r
  Erase spare block.\r
\r
  @param FtwDevice        The private data of FTW driver\r
\r
  @retval EFI_SUCCESS           The erase request was successfully completed.\r
  @retval EFI_ACCESS_DENIED     The firmware volume is in the WriteDisabled state.\r
  @retval EFI_DEVICE_ERROR      The block device is not functioning\r
                                correctly and could not be written.\r
                                The firmware device may have been\r
                                partially erased.\r
  @retval EFI_INVALID_PARAMETER One or more of the LBAs listed\r
                                in the variable argument list do\r
                                not exist in the firmware volume.  \r
\r
\r
**/\r
EFI_STATUS\r
FtwEraseSpareBlock (\r
  IN EFI_FTW_DEVICE   *FtwDevice\r
  );\r
\r
/**\r
  Retrieve the proper FVB protocol interface by HANDLE.\r
\r
\r
  @param FvBlockHandle   The handle of FVB protocol that provides services for\r
                         reading, writing, and erasing the target block.\r
  @param FvBlock         The interface of FVB protocol\r
\r
  @retval  EFI_SUCCESS   The function completed successfully\r
  @retval  EFI_ABORTED   The function could not complete successfully\r
\r
**/\r
EFI_STATUS\r
FtwGetFvbByHandle (\r
  IN EFI_HANDLE                           FvBlockHandle,\r
  OUT EFI_FIRMWARE_VOLUME_BLOCK_PROTOCOL  **FvBlock\r
  );\r
\r
/**\r
\r
  Is it in working block?\r
\r
  @param FtwDevice       The private data of FTW driver\r
  @param FvBlock         Fvb protocol instance\r
  @param Lba             The block specified\r
\r
  @return A BOOLEAN value indicating in working block or not.\r
\r
**/\r
BOOLEAN\r
IsWorkingBlock (\r
  EFI_FTW_DEVICE                      *FtwDevice,\r
  EFI_FIRMWARE_VOLUME_BLOCK_PROTOCOL  *FvBlock,\r
  EFI_LBA                             Lba\r
  );\r
\r
/**\r
\r
  Is it in boot block?\r
\r
  @param FtwDevice       The private data of FTW driver\r
  @param FvBlock         Fvb protocol instance\r
\r
  @return A BOOLEAN value indicating in boot block or not.\r
\r
**/\r
BOOLEAN\r
IsBootBlock (\r
  EFI_FTW_DEVICE                      *FtwDevice,\r
  EFI_FIRMWARE_VOLUME_BLOCK_PROTOCOL  *FvBlock\r
  );\r
\r
/**\r
  Copy the content of spare block to a target block. Size is FTW_BLOCK_SIZE.\r
  Spare block is accessed by FTW backup FVB protocol interface.\r
  Target block is accessed by FvBlock protocol interface.\r
\r
\r
  @param FtwDevice       The private data of FTW driver\r
  @param FvBlock         FVB Protocol interface to access target block\r
  @param Lba             Lba of the target block\r
  @param BlockSize       The size of the block\r
  @param NumberOfBlocks  The number of consecutive blocks starting with Lba\r
\r
  @retval  EFI_SUCCESS               Spare block content is copied to target block\r
  @retval  EFI_INVALID_PARAMETER     Input parameter error\r
  @retval  EFI_OUT_OF_RESOURCES      Allocate memory error\r
  @retval  EFI_ABORTED               The function could not complete successfully\r
\r
**/\r
EFI_STATUS\r
FlushSpareBlockToTargetBlock (\r
  EFI_FTW_DEVICE                      *FtwDevice,\r
  EFI_FIRMWARE_VOLUME_BLOCK_PROTOCOL  *FvBlock,\r
  EFI_LBA                             Lba,\r
  UINTN                               BlockSize,\r
  UINTN                               NumberOfBlocks\r
  );\r
\r
/**\r
  Copy the content of spare block to working block. Size is FTW_BLOCK_SIZE.\r
  Spare block is accessed by FTW backup FVB protocol interface. LBA is\r
  FtwDevice->FtwSpareLba.\r
  Working block is accessed by FTW working FVB protocol interface. LBA is\r
  FtwDevice->FtwWorkBlockLba.\r
\r
  Since the working block header is important when FTW initializes, the\r
  state of the operation should be handled carefully. The Crc value is\r
  calculated without STATE element.\r
\r
  @param FtwDevice       The private data of FTW driver\r
\r
  @retval  EFI_SUCCESS               Spare block content is copied to target block\r
  @retval  EFI_OUT_OF_RESOURCES      Allocate memory error\r
  @retval  EFI_ABORTED               The function could not complete successfully\r
\r
**/\r
EFI_STATUS\r
FlushSpareBlockToWorkingBlock (\r
  EFI_FTW_DEVICE                      *FtwDevice\r
  );\r
\r
/**\r
  Copy the content of spare block to a boot block. Size is FTW_BLOCK_SIZE.\r
  Spare block is accessed by FTW working FVB protocol interface.\r
  Target block is accessed by FvBlock protocol interface.\r
\r
  FTW will do extra work on boot block update.\r
  FTW should depend on a protocol of EFI_ADDRESS_RANGE_SWAP_PROTOCOL,\r
  which is produced by a chipset driver.\r
  FTW updating boot block steps may be:\r
  1. GetRangeLocation(), if the Range is inside the boot block, FTW know\r
  that boot block will be update. It shall add a FLAG in the working block.\r
  2. When spare block is ready,\r
  3. SetSwapState(SWAPPED)\r
  4. erasing boot block,\r
  5. programming boot block until the boot block is ok.\r
  6. SetSwapState(UNSWAPPED)\r
  FTW shall not allow to update boot block when battery state is error.\r
\r
  @param FtwDevice       The private data of FTW driver\r
\r
  @retval EFI_SUCCESS             Spare block content is copied to boot block\r
  @retval EFI_INVALID_PARAMETER   Input parameter error\r
  @retval EFI_OUT_OF_RESOURCES    Allocate memory error\r
  @retval EFI_ABORTED             The function could not complete successfully\r
\r
**/\r
EFI_STATUS\r
FlushSpareBlockToBootBlock (\r
  EFI_FTW_DEVICE                      *FtwDevice\r
  );\r
\r
/**\r
  Update a bit of state on a block device. The location of the bit is\r
  calculated by the (Lba, Offset, bit). Here bit is determined by the\r
  the name of a certain bit.\r
\r
\r
  @param FvBlock         FVB Protocol interface to access SrcBlock and DestBlock\r
  @param BlockSize       The size of the block\r
  @param Lba             Lba of a block\r
  @param Offset          Offset on the Lba\r
  @param NewBit          New value that will override the old value if it can be change\r
\r
  @retval  EFI_SUCCESS    A state bit has been updated successfully\r
  @retval  Others         Access block device error.\r
                          Notes:\r
                          Assume all bits of State are inside the same BYTE.\r
  @retval  EFI_ABORTED    Read block fail\r
\r
**/\r
EFI_STATUS\r
FtwUpdateFvState (\r
  IN EFI_FIRMWARE_VOLUME_BLOCK_PROTOCOL  *FvBlock,\r
  IN UINTN                               BlockSize,\r
  IN EFI_LBA                             Lba,\r
  IN UINTN                               Offset,\r
  IN UINT8                               NewBit\r
  );\r
\r
/**\r
  Get the last Write Header pointer.\r
  The last write header is the header whose 'complete' state hasn't been set.\r
  After all, this header may be a EMPTY header entry for next Allocate.\r
\r
\r
  @param FtwWorkSpaceHeader Pointer of the working block header\r
  @param FtwWorkSpaceSize   Size of the work space\r
  @param FtwWriteHeader     Pointer to retrieve the last write header\r
\r
  @retval  EFI_SUCCESS      Get the last write record successfully\r
  @retval  EFI_ABORTED      The FTW work space is damaged\r
\r
**/\r
EFI_STATUS\r
FtwGetLastWriteHeader (\r
  IN EFI_FAULT_TOLERANT_WORKING_BLOCK_HEADER  *FtwWorkSpaceHeader,\r
  IN UINTN                                    FtwWorkSpaceSize,\r
  OUT EFI_FAULT_TOLERANT_WRITE_HEADER         **FtwWriteHeader\r
  );\r
\r
/**\r
  Get the last Write Record pointer. The last write Record is the Record\r
  whose DestinationCompleted state hasn't been set. After all, this Record\r
  may be a EMPTY record entry for next write.\r
\r
\r
  @param FtwWriteHeader  Pointer to the write record header\r
  @param FtwWriteRecord  Pointer to retrieve the last write record\r
\r
  @retval EFI_SUCCESS        Get the last write record successfully\r
  @retval EFI_ABORTED        The FTW work space is damaged\r
\r
**/\r
EFI_STATUS\r
FtwGetLastWriteRecord (\r
  IN EFI_FAULT_TOLERANT_WRITE_HEADER          *FtwWriteHeader,\r
  OUT EFI_FAULT_TOLERANT_WRITE_RECORD         **FtwWriteRecord\r
  );\r
\r
/**\r
  To check if FtwRecord is the first record of FtwHeader.\r
\r
  @param FtwHeader  Pointer to the write record header\r
  @param FtwRecord  Pointer to the write record\r
\r
  @retval TRUE      FtwRecord is the first Record of the FtwHeader\r
  @retval FALSE     FtwRecord is not the first Record of the FtwHeader\r
\r
**/\r
BOOLEAN\r
IsFirstRecordOfWrites (\r
  IN EFI_FAULT_TOLERANT_WRITE_HEADER    *FtwHeader,\r
  IN EFI_FAULT_TOLERANT_WRITE_RECORD    *FtwRecord\r
  );\r
\r
/**\r
  To check if FtwRecord is the last record of FtwHeader. Because the\r
  FtwHeader has NumberOfWrites & PrivateDataSize, the FtwRecord can be\r
  determined if it is the last record of FtwHeader.\r
\r
  @param FtwHeader  Pointer to the write record header\r
  @param FtwRecord  Pointer to the write record\r
\r
  @retval TRUE      FtwRecord is the last Record of the FtwHeader\r
  @retval FALSE     FtwRecord is not the last Record of the FtwHeader\r
\r
**/\r
BOOLEAN\r
IsLastRecordOfWrites (\r
  IN EFI_FAULT_TOLERANT_WRITE_HEADER    *FtwHeader,\r
  IN EFI_FAULT_TOLERANT_WRITE_RECORD    *FtwRecord\r
  );\r
\r
/**\r
  To check if FtwRecord is the first record of FtwHeader.\r
\r
  @param FtwHeader  Pointer to the write record header\r
  @param FtwRecord  Pointer to retrieve the previous write record\r
\r
  @retval EFI_ACCESS_DENIED  Input record is the first record, no previous record is return.\r
  @retval EFI_SUCCESS        The previous write record is found.\r
\r
**/\r
EFI_STATUS\r
GetPreviousRecordOfWrites (\r
  IN     EFI_FAULT_TOLERANT_WRITE_HEADER    *FtwHeader,\r
  IN OUT EFI_FAULT_TOLERANT_WRITE_RECORD    **FtwRecord\r
  );\r
\r
/**\r
\r
  Check whether a flash buffer is erased.\r
\r
  @param Buffer          Buffer to check\r
  @param BufferSize      Size of the buffer\r
\r
  @return A BOOLEAN value indicating erased or not.\r
\r
**/\r
BOOLEAN\r
IsErasedFlashBuffer (\r
  IN UINT8           *Buffer,\r
  IN UINTN           BufferSize\r
  );\r
/**\r
  Initialize a work space when there is no work space.\r
\r
  @param WorkingHeader   Pointer of working block header\r
\r
  @retval  EFI_SUCCESS    The function completed successfully\r
  @retval  EFI_ABORTED    The function could not complete successfully.\r
\r
**/\r
EFI_STATUS\r
InitWorkSpaceHeader (\r
  IN EFI_FAULT_TOLERANT_WORKING_BLOCK_HEADER *WorkingHeader\r
  );\r
/**\r
  Read from working block to refresh the work space in memory.\r
\r
  @param FtwDevice   Point to private data of FTW driver\r
\r
  @retval  EFI_SUCCESS    The function completed successfully\r
  @retval  EFI_ABORTED    The function could not complete successfully.\r
\r
**/\r
EFI_STATUS\r
WorkSpaceRefresh (\r
  IN EFI_FTW_DEVICE  *FtwDevice\r
  );\r
/**\r
  Check to see if it is a valid work space.\r
\r
\r
  @param WorkingHeader   Pointer of working block header\r
\r
  @retval  EFI_SUCCESS    The function completed successfully\r
  @retval  EFI_ABORTED    The function could not complete successfully.\r
\r
**/\r
BOOLEAN\r
IsValidWorkSpace (\r
  IN EFI_FAULT_TOLERANT_WORKING_BLOCK_HEADER *WorkingHeader\r
  );\r
/**\r
  Reclaim the work space on the working block.\r
\r
  @param FtwDevice       Point to private data of FTW driver\r
  @param PreserveRecord  Whether to preserve the working record is needed\r
\r
  @retval EFI_SUCCESS            The function completed successfully\r
  @retval EFI_OUT_OF_RESOURCES   Allocate memory error\r
  @retval EFI_ABORTED            The function could not complete successfully\r
\r
**/\r
EFI_STATUS\r
FtwReclaimWorkSpace (\r
  IN EFI_FTW_DEVICE  *FtwDevice,\r
  IN BOOLEAN         PreserveRecord\r
  );\r
\r
/**\r
\r
  Get firmware volume block by address.\r
\r
\r
  @param Address         Address specified the block\r
  @param FvBlock         The block caller wanted\r
\r
  @retval  EFI_SUCCESS    The protocol instance if found.\r
  @retval  EFI_NOT_FOUND  Block not found\r
\r
**/\r
EFI_HANDLE\r
GetFvbByAddress (\r
  IN  EFI_PHYSICAL_ADDRESS               Address,\r
  OUT EFI_FIRMWARE_VOLUME_BLOCK_PROTOCOL **FvBlock\r
  );\r
\r
/**\r
  Retrieve the proper Swap Address Range protocol interface.\r
\r
  @param[out] SarProtocol       The interface of SAR protocol\r
\r
  @retval EFI_SUCCESS           The SAR protocol instance was found and returned in SarProtocol.\r
  @retval EFI_NOT_FOUND         The SAR protocol instance was not found.\r
  @retval EFI_INVALID_PARAMETER SarProtocol is NULL.\r
\r
**/\r
EFI_STATUS\r
FtwGetSarProtocol (\r
  OUT VOID                                **SarProtocol\r
  );\r
  \r
/**\r
  Function returns an array of handles that support the FVB protocol\r
  in a buffer allocated from pool. \r
\r
  @param[out]  NumberHandles    The number of handles returned in Buffer.\r
  @param[out]  Buffer           A pointer to the buffer to return the requested\r
                                array of  handles that support FVB protocol.\r
\r
  @retval EFI_SUCCESS           The array of handles was returned in Buffer, and the number of\r
                                handles in Buffer was returned in NumberHandles.\r
  @retval EFI_NOT_FOUND         No FVB handle was found.\r
  @retval EFI_OUT_OF_RESOURCES  There is not enough pool memory to store the matching results.\r
  @retval EFI_INVALID_PARAMETER NumberHandles is NULL or Buffer is NULL.\r
\r
**/\r
EFI_STATUS\r
GetFvbCountAndBuffer (\r
  OUT UINTN                               *NumberHandles,\r
  OUT EFI_HANDLE                          **Buffer\r
  );\r
\r
\r
/**\r
  Allocate private data for FTW driver and initialize it.\r
\r
  @param[out] FtwData           Pointer to the FTW device structure\r
\r
  @retval EFI_SUCCESS           Initialize the FTW device successfully.\r
  @retval EFI_OUT_OF_RESOURCES  Allocate memory error\r
  @retval EFI_INVALID_PARAMETER Workspace or Spare block does not exist\r
\r
**/\r
EFI_STATUS\r
InitFtwDevice (\r
  OUT EFI_FTW_DEVICE               **FtwData \r
  );\r
\r
\r
/**\r
  Initialization for Fault Tolerant Write is done in this handler.\r
\r
  @param[in, out] FtwDevice     Pointer to the FTW device structure\r
\r
  @retval EFI_SUCCESS           Initialize the FTW protocol successfully.\r
  @retval EFI_NOT_FOUND         No proper FVB protocol was found.\r
  \r
**/\r
EFI_STATUS\r
InitFtwProtocol (\r
  IN OUT EFI_FTW_DEVICE               *FtwDevice\r
  );\r
\r
/**\r
  Initialize a local work space header.\r
\r
  Since Signature and WriteQueueSize have been known, Crc can be calculated out,\r
  then the work space header will be fixed.\r
**/\r
VOID\r
InitializeLocalWorkSpaceHeader (\r
  VOID\r
  );\r
\r
/**\r
  Read work space data from work block or spare block.\r
\r
  @param FvBlock        FVB Protocol interface to access the block.\r
  @param BlockSize      The size of the block.\r
  @param Lba            Lba of the block.\r
  @param Offset         The offset within the block.\r
  @param Length         The number of bytes to read from the block.\r
  @param Buffer         The data is read.\r
\r
  @retval EFI_SUCCESS   The function completed successfully.\r
  @retval EFI_ABORTED   The function could not complete successfully.\r
\r
**/\r
EFI_STATUS\r
ReadWorkSpaceData (\r
  IN EFI_FIRMWARE_VOLUME_BLOCK_PROTOCOL *FvBlock,\r
  IN UINTN                              BlockSize,\r
  IN EFI_LBA                            Lba,\r
  IN UINTN                              Offset,\r
  IN UINTN                              Length,\r
  OUT UINT8                             *Buffer\r
  );\r
\r
/**\r
  Write data to work block.\r
\r
  @param FvBlock        FVB Protocol interface to access the block.\r
  @param BlockSize      The size of the block.\r
  @param Lba            Lba of the block.\r
  @param Offset         The offset within the block to place the data.\r
  @param Length         The number of bytes to write to the block.\r
  @param Buffer         The data to write.\r
\r
  @retval EFI_SUCCESS   The function completed successfully.\r
  @retval EFI_ABORTED   The function could not complete successfully.\r
\r
**/\r
EFI_STATUS\r
WriteWorkSpaceData (\r
  IN EFI_FIRMWARE_VOLUME_BLOCK_PROTOCOL *FvBlock,\r
  IN UINTN                              BlockSize,\r
  IN EFI_LBA                            Lba,\r
  IN UINTN                              Offset,\r
  IN UINTN                              Length,\r
  IN UINT8                              *Buffer\r
  );\r
\r
#endif\r