Code clean for Ftw driver, remove the obsolete logic for boot block. Correct some...

[mirror_edk2.git] / MdeModulePkg / Universal / FirmwareVolume / FaultTolerantWriteDxe / FtwWorkSpace.c

diff --git a/MdeModulePkg/Universal/FirmwareVolume/FaultTolerantWriteDxe/FtwWorkSpace.c b/MdeModulePkg/Universal/FirmwareVolume/FaultTolerantWriteDxe/FtwWorkSpace.c
index 49e2b8efd7acab1c504504fd42ffff8ea7be94aa..6474c2d0a5846309c80b125900aac369023a3f3b 100644 (file)
--- a/MdeModulePkg/Universal/FirmwareVolume/FaultTolerantWriteDxe/FtwWorkSpace.c
+++ b/MdeModulePkg/Universal/FirmwareVolume/FaultTolerantWriteDxe/FtwWorkSpace.c
@@ -109,7 +109,7 @@ InitWorkSpaceHeader (
     &gEfiSystemNvDataFvGuid,\r
     sizeof (EFI_GUID)\r
     );\r
-  WorkingHeader->WriteQueueSize = FTW_WORKING_QUEUE_SIZE;\r
+  WorkingHeader->WriteQueueSize = (UINT64) (PcdGet32 (PcdFlashNvStorageFtwWorkingSize) - sizeof (EFI_FAULT_TOLERANT_WORKING_BLOCK_HEADER));\r
 \r
   //\r
   // Crc is calculated with all the fields except Crc and STATE\r
@@ -284,7 +284,7 @@ WorkSpaceRefresh (
   // If work space has error or Record is out of the workspace limit, THEN\r
   //   call reclaim.\r
   //\r
-  if (EFI_ERROR (Status) || (Offset + WRITE_TOTAL_SIZE >= FtwLiteDevice->FtwWorkSpaceSize)) {\r
+  if (EFI_ERROR (Status) || (Offset + FTW_LITE_RECORD_SIZE >= FtwLiteDevice->FtwWorkSpaceSize)) {\r
     //\r
     // reclaim work space in working block.\r
     //\r
@@ -388,7 +388,7 @@ FtwReclaimWorkSpace (
       CopyMem (\r
         (UINT8 *) Ptr + sizeof (EFI_FAULT_TOLERANT_WORKING_BLOCK_HEADER),\r
         Record,\r
-        WRITE_TOTAL_SIZE\r
+        FTW_LITE_RECORD_SIZE\r
         );\r
     }\r
   }\r