Fix VS2005 build error

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@7181 6f19259b-4bc3-4df7-8a09-765794883524

1 files changed, 7 insertions, 8 deletions

diff --git a/MdeModulePkg/Universal/FirmwareVolume/FaultTolerantWriteDxe/FtwLite.c b/MdeModulePkg/Universal/FirmwareVolume/FaultTolerantWriteDxe/FtwLite.c
index 76e3cb3..9a999c9 100644
--- a/MdeModulePkg/Universal/FirmwareVolume/FaultTolerantWriteDxe/FtwLite.c
+++ b/MdeModulePkg/Universal/FirmwareVolume/FaultTolerantWriteDxe/FtwLite.c
@@ -622,13 +622,7 @@ InitializeFtwLite (
 

   FtwLiteDevice = NULL;

   FtwLiteDevice = AllocatePool (sizeof (EFI_FTW_LITE_DEVICE) + Length);

-  if (FtwLiteDevice != NULL) {

-    Status = EFI_SUCCESS;

-  } else {

-    Status = EFI_OUT_OF_RESOURCES;

-  }

-

-  ASSERT_EFI_ERROR (Status);

+  ASSERT (FtwLiteDevice != NULL);

 

   ZeroMem (FtwLiteDevice, sizeof (EFI_FTW_LITE_DEVICE));

   FtwLiteDevice->Signature = FTW_LITE_DEVICE_SIGNATURE;

@@ -791,7 +785,8 @@ InitializeFtwLite (
       (FtwLiteDevice->FtwSpareLba == (EFI_LBA) (-1))

       ) {

     DEBUG ((EFI_D_ERROR, "FtwLite: Working or spare FVB not ready\n"));

-    ASSERT_EFI_ERROR (Status);

+    FreePool (FtwLiteDevice);

+    return EFI_ABORTED;

   }

   //

   // Refresh workspace data from working block

@@ -831,6 +826,7 @@ InitializeFtwLite (
       //

       Status = WorkSpaceRefresh (FtwLiteDevice);

       if (EFI_ERROR (Status)) {

+        FreePool (FtwLiteDevice);

         return EFI_ABORTED;

       }

     } else {

@@ -850,6 +846,7 @@ InitializeFtwLite (
       Status = FtwReclaimWorkSpace (FtwLiteDevice, FALSE);

 

       if (EFI_ERROR (Status)) {

+        FreePool (FtwLiteDevice);

         return EFI_ABORTED;

       }

     }

@@ -869,6 +866,7 @@ InitializeFtwLite (
                   &FtwLiteDevice->FtwLiteInstance

                   );

   if (EFI_ERROR (Status)) {

+    FreePool (FtwLiteDevice);

     return EFI_ABORTED;

   }

   //

@@ -912,6 +910,7 @@ InitializeFtwLite (
     Status = FtwReclaimWorkSpace (FtwLiteDevice, TRUE);

     if (EFI_ERROR (Status)) {

       DEBUG ((EFI_D_FTW_LITE, "FtwLite: Workspace reclaim - %r\n", Status));

+      FreePool (FtwLiteDevice);

       return EFI_ABORTED;

     }

   }