Enhance FtwReclaimWorkSpace() so that it can be called when the working space header...

[mirror_edk2.git] / MdeModulePkg / Universal / FirmwareVolume / FaultTolerantWriteDxe / FtwLite.h

/** @file\r
\r
  The internal header file includes the common header files, defines\r
  internal structure and functions used by FtwLite module.\r
\r
Copyright (c) 2006 - 2008, Intel Corporation                                                         \r
All rights reserved. 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_LITE_H_\r
#define _EFI_FAULT_TOLERANT_WRITE_LITE_H_\r
\r
\r
#include <PiDxe.h>\r
\r
#include <Protocol/PciRootBridgeIo.h>\r
#include <Guid/SystemNvDataGuid.h>\r
#include <Protocol/FaultTolerantWriteLite.h>\r
#include <Protocol/FirmwareVolumeBlock.h>\r
\r
#include <Library/PcdLib.h>\r
#include <Library/DebugLib.h>\r
#include <Library/UefiDriverEntryPoint.h>\r
#include <Library/BaseMemoryLib.h>\r
#include <Library/MemoryAllocationLib.h>\r
#include <Library/UefiBootServicesTableLib.h>\r
\r
#include <WorkingBlockHeader.h>\r
\r
#define EFI_D_FTW_LITE  EFI_D_ERROR\r
#define EFI_D_FTW_INFO  EFI_D_INFO\r
\r
//\r
// Flash erase polarity is 1\r
//\r
#define FTW_ERASE_POLARITY  1\r
\r
#define FTW_VALID_STATE     0\r
#define FTW_INVALID_STATE   1\r
\r
#define FTW_ERASED_BYTE     ((UINT8) (255))\r
#define FTW_POLARITY_REVERT ((UINT8) (255))\r
\r
typedef struct {\r
  UINT8         WriteAllocated : 1;\r
  UINT8         SpareCompleted : 1;\r
  UINT8         WriteCompleted : 1;\r
  UINT8         Reserved : 5;\r
#define WRITE_ALLOCATED 0x1\r
#define SPARE_COMPLETED 0x2\r
#define WRITE_COMPLETED 0x4\r
\r
  EFI_DEV_PATH  DevPath;\r
  EFI_LBA       Lba;\r
  UINTN         Offset;\r
  UINTN         NumBytes;\r
  //\r
  // UINTN           SpareAreaOffset;\r
  //\r
} EFI_FTW_LITE_RECORD;\r
\r
#define FTW_LITE_DEVICE_SIGNATURE EFI_SIGNATURE_32 ('F', 'T', 'W', 'L')\r
\r
//\r
// MACRO for Block size.\r
// Flash Erasing will do in block granularity.\r
//\r
#ifdef FV_BLOCK_SIZE\r
#define FTW_BLOCK_SIZE  FV_BLOCK_SIZE\r
#else\r
#define FV_BLOCK_SIZE   0x10000\r
#define FTW_BLOCK_SIZE  FV_BLOCK_SIZE\r
#endif\r
//\r
// MACRO for FTW WORK SPACE Base & Size\r
//\r
#ifdef EFI_FTW_WORKING_OFFSET\r
#define FTW_WORK_SPACE_BASE EFI_FTW_WORKING_OFFSET\r
#else\r
#define FTW_WORK_SPACE_BASE 0x00E000\r
#endif\r
\r
#ifdef EFI_FTW_WORKING_LENGTH\r
#define FTW_WORK_SPACE_SIZE EFI_FTW_WORKING_LENGTH\r
#else\r
#define FTW_WORK_SPACE_SIZE 0x002000\r
#endif\r
//\r
// MACRO for FTW header and record\r
//\r
#define FTW_WORKING_QUEUE_SIZE  (FTW_WORK_SPACE_SIZE - sizeof (EFI_FAULT_TOLERANT_WORKING_BLOCK_HEADER))\r
#define FTW_LITE_RECORD_SIZE    (sizeof (EFI_FTW_LITE_RECORD))\r
#define WRITE_TOTAL_SIZE        FTW_LITE_RECORD_SIZE\r
\r
//\r
// EFI Fault tolerant protocol private data structure\r
//\r
typedef struct {\r
  UINTN                                   Signature;\r
  EFI_HANDLE                              Handle;\r
  EFI_FTW_LITE_PROTOCOL                   FtwLiteInstance;\r
  EFI_PHYSICAL_ADDRESS                    WorkSpaceAddress;\r
  UINTN                                   WorkSpaceLength;\r
  EFI_PHYSICAL_ADDRESS                    SpareAreaAddress;\r
  UINTN                                   SpareAreaLength;\r
  UINTN                                   NumberOfSpareBlock; // Number of the blocks in spare block\r
  UINTN                                   SizeOfSpareBlock;   // Block size in bytes of the blocks in spare block\r
  EFI_FAULT_TOLERANT_WORKING_BLOCK_HEADER *FtwWorkSpaceHeader;\r
  EFI_FTW_LITE_RECORD                     *FtwLastRecord;\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;\r
  EFI_LBA                                 FtwWorkBlockLba;    // Start LBA of working block\r
  EFI_LBA                                 FtwWorkSpaceLba;    // Start LBA of working space\r
  UINTN                                   FtwWorkSpaceBase;   // Offset from LBA start addr\r
  UINTN                                   FtwWorkSpaceSize;\r
  UINT8                                   *FtwWorkSpace;\r
  //\r
  // Following a buffer of FtwWorkSpace[FTW_WORK_SPACE_SIZE],\r
  // Allocated with EFI_FTW_LITE_DEVICE.\r
  //\r
} EFI_FTW_LITE_DEVICE;\r
\r
#define FTW_LITE_CONTEXT_FROM_THIS(a) CR (a, EFI_FTW_LITE_DEVICE, FtwLiteInstance, FTW_LITE_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
\r
  @param ImageHandle     EFI_HANDLE: A handle for the image that is initializing\r
                         this driver\r
  @param SystemTable     EFI_SYSTEM_TABLE: A pointer to the EFI system table\r
\r
  @retval  EFI_SUCCESS            FTW has finished the initialization\r
  @retval  EFI_ABORTED            FTW initialization error\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
InitializeFtwLite (\r
  IN EFI_HANDLE                 ImageHandle,\r
  IN EFI_SYSTEM_TABLE           *SystemTable\r
  )\r
;\r
\r
//\r
// Fault Tolerant Write Protocol API\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 FvbHandle       The handle of FVB protocol that provides services for\r
                         reading, writing, and erasing the target block.\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 NumBytes        The number of bytes to write to the target block.\r
  @param Buffer          The data to write.\r
\r
  @retval  EFI_SUCCESS           The function completed successfully\r
  @retval  EFI_BAD_BUFFER_SIZE   The write would span a target block, which is not\r
                                 a valid action.\r
  @retval  EFI_ACCESS_DENIED     No writes have been allocated.\r
  @retval  EFI_NOT_FOUND         Cannot find FVB by handle.\r
  @retval  EFI_OUT_OF_RESOURCES  Cannot allocate memory.\r
  @retval  EFI_ABORTED           The function could not complete successfully.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
FtwLiteWrite (\r
  IN EFI_FTW_LITE_PROTOCOL                      *This,\r
  IN EFI_HANDLE                                 FvbHandle,\r
  IN EFI_LBA                                    Lba,\r
  IN UINTN                                      Offset,\r
  IN OUT UINTN                                 *NumBytes,\r
  IN VOID                                       *Buffer\r
  )\r
;\r
\r
//\r
// Internal functions\r
//\r
/**\r
  Restarts a previously interrupted write. The caller must provide the\r
  block protocol needed to complete the interrupted write.\r
\r
\r
  @param FtwLiteDevice   The private data of FTW_LITE driver\r
                         FvbHandle           - 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
FtwRestart (\r
  IN EFI_FTW_LITE_DEVICE    *FtwLiteDevice\r
  )\r
;\r
\r
/**\r
  Aborts all previous allocated writes.\r
\r
\r
  @param FtwLiteDevice   The private data of FTW_LITE driver\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
FtwAbort (\r
  IN EFI_FTW_LITE_DEVICE    *FtwLiteDevice\r
  )\r
;\r
\r
\r
/**\r
  Write a record with fault tolerant mannaer.\r
  Since the content has already backuped in spare block, the write is\r
  guaranteed to be completed with fault tolerant manner.\r
\r
\r
  @param FtwLiteDevice   The private data of FTW_LITE driver\r
  @param Fvb             The 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_ABORTED          The function could not complete successfully\r
\r
**/\r
EFI_STATUS\r
FtwWriteRecord (\r
  IN EFI_FTW_LITE_DEVICE                   *FtwLiteDevice,\r
  IN EFI_FIRMWARE_VOLUME_BLOCK_PROTOCOL    *Fvb\r
  )\r
;\r
\r
/**\r
  To Erase one block. The size is FTW_BLOCK_SIZE\r
\r
\r
  @param FtwLiteDevice   Calling context\r
  @param FvBlock         FVB Protocol interface\r
  @param Lba             Lba of the firmware block\r
\r
  @retval  EFI_SUCCESS    Block LBA is Erased successfully\r
  @retval  Others         Error occurs\r
\r
**/\r
EFI_STATUS\r
FtwEraseBlock (\r
  IN EFI_FTW_LITE_DEVICE              *FtwLiteDevice,\r
  EFI_FIRMWARE_VOLUME_BLOCK_PROTOCOL  *FvBlock,\r
  EFI_LBA                             Lba\r
  )\r
;\r
\r
/**\r
\r
  Erase spare block.\r
\r
\r
  @param FtwLiteDevice   Calling context\r
\r
  @retval EFI_SUCCESS The erase request was successfully\r
                      completed.\r
  \r
  @retval EFI_ACCESS_DENIED   The firmware volume is in the\r
                              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
EFI_STATUS\r
FtwEraseSpareBlock (\r
  IN EFI_FTW_LITE_DEVICE   *FtwLiteDevice\r
  )\r
;\r
\r
/**\r
  Retrive 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
\r
  Get firmware 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_STATUS\r
GetFvbByAddress (\r
  IN  EFI_PHYSICAL_ADDRESS               Address,\r
  OUT EFI_FIRMWARE_VOLUME_BLOCK_PROTOCOL **FvBlock\r
  )\r
;\r
\r
/**\r
\r
  Is it in working block?\r
\r
\r
  @param FtwLiteDevice   Calling context\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
IsInWorkingBlock (\r
  EFI_FTW_LITE_DEVICE                 *FtwLiteDevice,\r
  EFI_FIRMWARE_VOLUME_BLOCK_PROTOCOL  *FvBlock,\r
  EFI_LBA                             Lba\r
  )\r
;\r
\r
/**\r
\r
  Check whether the block is a boot block.\r
\r
\r
  @param FtwLiteDevice   Calling context\r
  @param FvBlock         Fvb protocol instance\r
  @param Lba             Lba value\r
\r
  @retval FALSE           This is a boot block.\r
  @retval TRUE            This is not a boot block.\r
\r
**/\r
BOOLEAN\r
IsBootBlock (\r
  EFI_FTW_LITE_DEVICE                 *FtwLiteDevice,\r
  EFI_FIRMWARE_VOLUME_BLOCK_PROTOCOL  *FvBlock,\r
  EFI_LBA                             Lba\r
  )\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. LBA is\r
  FtwLiteDevice->FtwSpareLba.\r
  Target block is accessed by FvBlock protocol interface. LBA is Lba.\r
\r
\r
  @param FtwLiteDevice   The private data of FTW_LITE driver\r
  @param FvBlock         FVB Protocol interface to access target block\r
  @param Lba             Lba of the target block\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_LITE_DEVICE                 *FtwLiteDevice,\r
  EFI_FIRMWARE_VOLUME_BLOCK_PROTOCOL  *FvBlock,\r
  EFI_LBA                             Lba\r
  )\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
  FtwLiteDevice->FtwSpareLba.\r
  Working block is accessed by FTW working FVB protocol interface. LBA is\r
  FtwLiteDevice->FtwWorkBlockLba.\r
\r
\r
  @param FtwLiteDevice   The private data of FTW_LITE 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
                                     Notes:\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
**/\r
EFI_STATUS\r
FlushSpareBlockToWorkingBlock (\r
  EFI_FTW_LITE_DEVICE                 *FtwLiteDevice\r
  )\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 backup FVB protocol interface. LBA is\r
  FtwLiteDevice->FtwSpareLba.\r
  Boot block is accessed by BootFvb protocol interface. LBA is 0.\r
\r
\r
  @param FtwLiteDevice   The private data of FTW_LITE 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
                                     Notes:\r
\r
**/\r
EFI_STATUS\r
FlushSpareBlockToBootBlock (\r
  EFI_FTW_LITE_DEVICE                 *FtwLiteDevice\r
  )\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 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 EFI_LBA                             Lba,\r
  IN UINTN                               Offset,\r
  IN UINT8                               NewBit\r
  )\r
;\r
\r
/**\r
  Get the last Write record pointer.\r
  The last record is the record 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 FtwLiteDevice   Private data of this driver\r
  @param FtwLastRecord   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
FtwGetLastRecord (\r
  IN  EFI_FTW_LITE_DEVICE  *FtwLiteDevice,\r
  OUT EFI_FTW_LITE_RECORD  **FtwLastRecord\r
  )\r
;\r
\r
/**\r
\r
  Check whether a flash buffer is erased.\r
\r
\r
  @param Polarity        All 1 or all 0\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 BOOLEAN         Polarity,\r
  IN UINT8           *Buffer,\r
  IN UINTN           BufferSize\r
  )\r
;\r
\r
/**\r
  Initialize a work space when there is no 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
EFI_STATUS\r
InitWorkSpaceHeader (\r
  IN EFI_FAULT_TOLERANT_WORKING_BLOCK_HEADER *WorkingHeader\r
  )\r
;\r
\r
/**\r
  Read from working block to refresh the work space in memory.\r
\r
\r
  @param FtwLiteDevice   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_LITE_DEVICE  *FtwLiteDevice\r
  )\r
;\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
\r
/**\r
  Reclaim the work space on the working block.\r
\r
\r
  @param FtwLiteDevice   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_LITE_DEVICE  *FtwLiteDevice,\r
  IN BOOLEAN              PreserveRecord\r
  )\r
;\r
\r
#endif\r