4 files changed, 10 insertions, 16 deletions

diff --git a/src/api/burn/balutil/inc/BalBaseBAFunctions.h b/src/api/burn/balutil/inc/BalBaseBAFunctions.h
index 8d9bddca..7f52f76e 100644
--- a/src/api/burn/balutil/inc/BalBaseBAFunctions.h
+++ b/src/api/burn/balutil/inc/BalBaseBAFunctions.h
@@ -583,6 +583,7 @@ public: // IBootstrapperApplication
         __in DWORD /*cFiles*/,
         __in_ecount_z(cFiles) LPCWSTR* /*rgwzFiles*/,
         __in int /*nRecommendation*/,
+        __in BOOTSTRAPPER_FILES_IN_USE_TYPE /*source*/,
         __inout int* /*pResult*/
         )
     {
diff --git a/src/api/burn/balutil/inc/BalBaseBootstrapperApplication.h b/src/api/burn/balutil/inc/BalBaseBootstrapperApplication.h
index 4d043dfe..5665fee3 100644
--- a/src/api/burn/balutil/inc/BalBaseBootstrapperApplication.h
+++ b/src/api/burn/balutil/inc/BalBaseBootstrapperApplication.h
@@ -738,6 +738,7 @@ public: // IBootstrapperApplication
         __in DWORD /*cFiles*/,
         __in_ecount_z(cFiles) LPCWSTR* /*rgwzFiles*/,
         __in int /*nRecommendation*/,
+        __in BOOTSTRAPPER_FILES_IN_USE_TYPE /*source*/,
         __inout int* pResult
         )
     {
diff --git a/src/api/burn/balutil/inc/BalBaseBootstrapperApplicationProc.h b/src/api/burn/balutil/inc/BalBaseBootstrapperApplicationProc.h
index 59bfc1f8..2292cd64 100644
--- a/src/api/burn/balutil/inc/BalBaseBootstrapperApplicationProc.h
+++ b/src/api/burn/balutil/inc/BalBaseBootstrapperApplicationProc.h
@@ -438,7 +438,7 @@ static HRESULT BalBaseBAProcOnExecuteFilesInUse(
     __inout BA_ONEXECUTEFILESINUSE_RESULTS* pResults
     )
 {
-    return pBA->OnExecuteFilesInUse(pArgs->wzPackageId, pArgs->cFiles, pArgs->rgwzFiles, pArgs->nRecommendation, &pResults->nResult);
+    return pBA->OnExecuteFilesInUse(pArgs->wzPackageId, pArgs->cFiles, pArgs->rgwzFiles, pArgs->nRecommendation, pArgs->source, &pResults->nResult);
 }
 
 static HRESULT BalBaseBAProcOnExecutePackageComplete(
diff --git a/src/api/burn/balutil/inc/IBootstrapperApplication.h b/src/api/burn/balutil/inc/IBootstrapperApplication.h
index be9b7b6e..2ba1f503 100644
--- a/src/api/burn/balutil/inc/IBootstrapperApplication.h
+++ b/src/api/burn/balutil/inc/IBootstrapperApplication.h
@@ -488,27 +488,19 @@ DECLARE_INTERFACE_IID_(IBootstrapperApplication, IUnknown, "53C31D56-49C0-426B-A
         __inout int* pResult
         ) = 0;
 
-    // OnExecuteFilesInUse - called when the engine encounters files in use while
-    //                       executing a package.
+    // OnExecuteFilesInUse - called when the engine receives a files in use message
+    //                       while executing a package.
     //
-    // Return:
-    //  IDOK instructs the engine to let the Restart Manager attempt to close the
-    //       applications to avoid a restart.
-    //
-    //  IDCANCEL instructs the engine to abort the execution and start rollback.
-    //
-    //  IDIGNORE instructs the engine to ignore the running applications. A restart will be
-    //           required.
-    //
-    //  IDRETRY instructs the engine to check if the applications are still running again.
-    //
-    //  IDNOACTION is equivalent to ignoring the running applications. A restart will be
-    //             required.
+    // Return value depends on the source:
+    //  BOOTSTRAPPER_FILES_IN_USE_TYPE_MSI: https://docs.microsoft.com/en-us/windows/win32/msi/installvalidate-action
+    //  BOOTSTRAPPER_FILES_IN_USE_TYPE_MSI_RM: https://docs.microsoft.com/en-us/windows/win32/msi/using-restart-manager-with-an-external-ui-
+    //  BOOTSTRAPPER_FILES_IN_USE_TYPE_NETFX: https://docs.microsoft.com/en-us/dotnet/framework/deployment/how-to-get-progress-from-the-dotnet-installer
     STDMETHOD(OnExecuteFilesInUse)(
         __in_z LPCWSTR wzPackageId,
         __in DWORD cFiles,
         __in_ecount_z(cFiles) LPCWSTR* rgwzFiles,
         __in int nRecommendation,
+        __in BOOTSTRAPPER_FILES_IN_USE_TYPE source,
         __inout int* pResult
         ) = 0;