1 files changed, 30 insertions, 7 deletions

diff --git a/src/api/burn/WixToolset.Mba.Core/BalUtil.cs b/src/api/burn/WixToolset.Mba.Core/BalUtil.cs
index f478eca4..4c949a5a 100644
--- a/src/api/burn/WixToolset.Mba.Core/BalUtil.cs
+++ b/src/api/burn/WixToolset.Mba.Core/BalUtil.cs
@@ -7,16 +7,39 @@ namespace WixToolset.Mba.Core
 
     internal static class BalUtil
     {
-        [DllImport("mbanative.dll", ExactSpelling = true, PreserveSig = false)]
-        internal static extern IBootstrapperEngine InitializeFromCreateArgs(
-            IntPtr pArgs,
-            ref Command pCommand
+        [DllImport("mbanative.dll", ExactSpelling = true)]
+        internal static extern int BalEscapeStringFromEngine(
+            [MarshalAs(UnmanagedType.Interface)] IBootstrapperEngine pEngine,
+            [MarshalAs(UnmanagedType.LPWStr)] string wzIn,
+            ref StrUtil.StrHandle psczOut
+            );
+
+        [DllImport("mbanative.dll", ExactSpelling = true)]
+        internal static extern int BalFormatStringFromEngine(
+            [MarshalAs(UnmanagedType.Interface)] IBootstrapperEngine pEngine,
+            [MarshalAs(UnmanagedType.LPWStr)] string wzFormat,
+            ref StrUtil.StrHandle psczOut
+            );
+
+        [DllImport("mbanative.dll", ExactSpelling = true)]
+        internal static extern int BalGetStringVariableFromEngine(
+            [MarshalAs(UnmanagedType.Interface)] IBootstrapperEngine pEngine,
+            [MarshalAs(UnmanagedType.LPWStr)] string wzVariable,
+            ref StrUtil.StrHandle psczOut
+            );
+
+        [DllImport("mbanative.dll", ExactSpelling = true)]
+        internal static extern int BalGetVersionVariableFromEngine(
+            [MarshalAs(UnmanagedType.Interface)] IBootstrapperEngine pEngine,
+            [MarshalAs(UnmanagedType.LPWStr)] string wzVariable,
+            ref StrUtil.StrHandle psczOut
             );
 
         [DllImport("mbanative.dll", ExactSpelling = true)]
-        internal static extern void StoreBAInCreateResults(
-            IntPtr pResults,
-            [MarshalAs(UnmanagedType.Interface)] IBootstrapperApplication pBA
+        [return: MarshalAs(UnmanagedType.Bool)]
+        internal static extern bool BalVariableExistsFromEngine(
+            [MarshalAs(UnmanagedType.Interface)] IBootstrapperEngine pEngine,
+            [MarshalAs(UnmanagedType.LPWStr)] string wzVariable
             );
     }
 }