1 files changed, 87 insertions, 0 deletions

diff --git a/src/WixToolset.Core.WindowsInstaller/Bind/CreateDeltaPatchesCommand.cs b/src/WixToolset.Core.WindowsInstaller/Bind/CreateDeltaPatchesCommand.cs
new file mode 100644
index 00000000..767671b8
--- /dev/null
+++ b/src/WixToolset.Core.WindowsInstaller/Bind/CreateDeltaPatchesCommand.cs
@@ -0,0 +1,87 @@
+// Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information.
+
+namespace WixToolset.Core.WindowsInstaller.Databases
+{
+    using System;
+    using System.Collections.Generic;
+    using System.Globalization;
+    using System.IO;
+    using WixToolset.Core.Bind;
+    using WixToolset.Data;
+    using WixToolset.Data.Rows;
+
+    /// <summary>
+    /// Creates delta patches and updates the appropriate rows to point to the newly generated patches.
+    /// </summary>
+    internal class CreateDeltaPatchesCommand
+    {
+        public IEnumerable<FileFacade> FileFacades { private get; set; }
+
+        public Table WixPatchIdTable { private get; set; }
+
+        public string TempFilesLocation { private get; set; }
+
+        public void Execute()
+        {
+            bool optimizePatchSizeForLargeFiles = false;
+            PatchSymbolFlagsType apiPatchingSymbolFlags = 0;
+
+            if (null != this.WixPatchIdTable)
+            {
+                Row row = this.WixPatchIdTable.Rows[0];
+                if (null != row)
+                {
+                    if (null != row[2])
+                    {
+                        optimizePatchSizeForLargeFiles = (1 == Convert.ToUInt32(row[2], CultureInfo.InvariantCulture));
+                    }
+
+                    if (null != row[3])
+                    {
+                        apiPatchingSymbolFlags = (PatchSymbolFlagsType)Convert.ToUInt32(row[3], CultureInfo.InvariantCulture);
+                    }
+                }
+            }
+
+            foreach (FileFacade facade in this.FileFacades)
+            {
+                if (RowOperation.Modify == facade.File.Operation &&
+                    0 != (facade.WixFile.PatchAttributes & PatchAttributeType.IncludeWholeFile))
+                {
+                    string deltaBase = String.Concat("delta_", facade.File.File);
+                    string deltaFile = Path.Combine(this.TempFilesLocation, String.Concat(deltaBase, ".dpf"));
+                    string headerFile = Path.Combine(this.TempFilesLocation, String.Concat(deltaBase, ".phd"));
+
+                    bool retainRangeWarning = false;
+
+                    if (PatchAPI.PatchInterop.CreateDelta(
+                            deltaFile,
+                            facade.WixFile.Source,
+                            facade.DeltaPatchFile.Symbols,
+                            facade.DeltaPatchFile.RetainOffsets,
+                            new[] { facade.WixFile.PreviousSource },
+                            facade.DeltaPatchFile.PreviousSymbols.Split(new[] { ';' }),
+                            facade.DeltaPatchFile.PreviousIgnoreLengths.Split(new[] { ';' }),
+                            facade.DeltaPatchFile.PreviousIgnoreOffsets.Split(new[] { ';' }),
+                            facade.DeltaPatchFile.PreviousRetainLengths.Split(new[] { ';' }),
+                            facade.DeltaPatchFile.PreviousRetainOffsets.Split(new[] { ';' }),
+                            apiPatchingSymbolFlags,
+                            optimizePatchSizeForLargeFiles,
+                            out retainRangeWarning))
+                    {
+                        PatchAPI.PatchInterop.ExtractDeltaHeader(deltaFile, headerFile);
+
+                        facade.WixFile.Source = deltaFile;
+                        facade.WixFile.DeltaPatchHeaderSource = headerFile;
+                    }
+
+                    if (retainRangeWarning)
+                    {
+                        // TODO: get patch family to add to warning message for PatchWiz parity.
+                        Messaging.Instance.OnMessage(WixWarnings.RetainRangeMismatch(facade.File.SourceLineNumbers, facade.File.File));
+                    }
+                }
+            }
+        }
+    }
+}