From cd921db764df9578733c85c29e8c6c368f4c7e78 Mon Sep 17 00:00:00 2001
From: Sean Hall <r.sean.hall@gmail.com>
Date: Mon, 3 May 2021 12:23:31 -0500
Subject: Enforce payload and container verification.

---
 src/burn/engine/apply.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'src/burn/engine/apply.cpp')

diff --git a/src/burn/engine/apply.cpp b/src/burn/engine/apply.cpp
index b4fa9dac..4d527409 100644
--- a/src/burn/engine/apply.cpp
+++ b/src/burn/engine/apply.cpp
@@ -1410,12 +1410,12 @@ static HRESULT AcquireContainerOrPayload(
             fMinimumFileSize = TRUE;
             qwFileSize = pContainer->qwAttachedOffset + pContainer->qwFileSize;
         }
-        else if (pContainer->pbHash && pContext->wzLayoutDirectory)
+        else
         {
             qwFileSize = pContainer->qwFileSize;
         }
     }
-    else if (pPayload->pbHash)
+    else if (BURN_PAYLOAD_VERIFICATION_HASH == pPayload->verification || BURN_PAYLOAD_VERIFICATION_UPDATE_BUNDLE == pPayload->verification)
     {
         qwFileSize = pPayload->qwFileSize;
     }
-- 
cgit v1.2.3-55-g6feb