Allow E_IMPL from BA/bext and check all cbSizes from BA/bext.

1 files changed, 23 insertions, 1 deletions

diff --git a/src/engine/burnextension.cpp b/src/engine/burnextension.cpp
index 157b082f..7568f75e 100644
--- a/src/engine/burnextension.cpp
+++ b/src/engine/burnextension.cpp
@@ -2,6 +2,14 @@
 
 #include "precomp.h"
 
+
+static HRESULT SendRequiredBextMessage(
+    __in BURN_EXTENSION* pExtension,
+    __in BUNDLE_EXTENSION_MESSAGE message,
+    __in const LPVOID pvArgs,
+    __inout LPVOID pvResults
+    );
+
 // function definitions
 
 /*******************************************************************
@@ -234,9 +242,23 @@ EXTERN_C BEEAPI BurnExtensionPerformSearch(
 
     results.cbSize = sizeof(results);
 
-    hr = pExtension->pfnBurnExtensionProc(BUNDLE_EXTENSION_MESSAGE_SEARCH, &args, &results, pExtension->pvBurnExtensionProcContext);
+    hr = SendRequiredBextMessage(pExtension, BUNDLE_EXTENSION_MESSAGE_SEARCH, &args, &results);
     ExitOnFailure(hr, "BundleExtension '%ls' Search '%ls' failed.", pExtension->sczId, wzSearchId);
 
 LExit:
     return hr;
 }
+
+static HRESULT SendRequiredBextMessage(
+    __in BURN_EXTENSION* pExtension,
+    __in BUNDLE_EXTENSION_MESSAGE message,
+    __in const LPVOID pvArgs,
+    __inout LPVOID pvResults
+    )
+{
+    HRESULT hr = S_OK;
+
+    hr = pExtension->pfnBurnExtensionProc(message, pvArgs, pvResults, pExtension->pvBurnExtensionProcContext);
+
+    return hr;
+}