1 files changed, 161 insertions, 0 deletions

diff --git a/src/WixToolset.Core.Burn/Bundles/ProcessPayloadsCommand.cs b/src/WixToolset.Core.Burn/Bundles/ProcessPayloadsCommand.cs
new file mode 100644
index 00000000..5dbd6aaa
--- /dev/null
+++ b/src/WixToolset.Core.Burn/Bundles/ProcessPayloadsCommand.cs
@@ -0,0 +1,161 @@
+// 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.Burn.Bundles
+{
+    using System;
+    using System.Collections.Generic;
+    using System.Diagnostics;
+    using System.IO;
+    using System.Security.Cryptography;
+    using System.Security.Cryptography.X509Certificates;
+    using System.Text;
+    using WixToolset.Bind;
+    using WixToolset.Data;
+    using WixToolset.Data.Bind;
+    using WixToolset.Data.Rows;
+
+    internal class ProcessPayloadsCommand
+    {
+        private static readonly Version EmptyVersion = new Version(0, 0, 0, 0);
+
+        public IEnumerable<WixBundlePayloadRow> Payloads { private get; set; }
+
+        public PackagingType DefaultPackaging { private get; set; }
+
+        public string LayoutDirectory { private get; set; }
+
+        public IEnumerable<FileTransfer> FileTransfers { get; private set; }
+
+        public void Execute()
+        {
+            List<FileTransfer> fileTransfers = new List<FileTransfer>();
+
+            foreach (WixBundlePayloadRow payload in this.Payloads)
+            {
+                string normalizedPath = payload.Name.Replace('\\', '/');
+                if (normalizedPath.StartsWith("../", StringComparison.Ordinal) || normalizedPath.Contains("/../"))
+                {
+                    Messaging.Instance.OnMessage(WixErrors.PayloadMustBeRelativeToCache(payload.SourceLineNumbers, "Payload", "Name", payload.Name));
+                }
+
+                // Embedded files (aka: files from binary .wixlibs) are not content files (because they are hidden
+                // in the .wixlib).
+                ObjectField field = (ObjectField)payload.Fields[2];
+                payload.ContentFile = !field.EmbeddedFileIndex.HasValue;
+
+                this.UpdatePayloadPackagingType(payload);
+
+                if (String.IsNullOrEmpty(payload.SourceFile))
+                {
+                    // Remote payloads obviously cannot be embedded.
+                    Debug.Assert(PackagingType.Embedded != payload.Packaging);
+                }
+                else // not a remote payload so we have a lot more to update.
+                {
+                    this.UpdatePayloadFileInformation(payload);
+
+                    this.UpdatePayloadVersionInformation(payload);
+
+                    // External payloads need to be transfered.
+                    if (PackagingType.External == payload.Packaging)
+                    {
+                        FileTransfer transfer;
+                        if (FileTransfer.TryCreate(payload.FullFileName, Path.Combine(this.LayoutDirectory, payload.Name), false, "Payload", payload.SourceLineNumbers, out transfer))
+                        {
+                            fileTransfers.Add(transfer);
+                        }
+                    }
+                }
+            }
+
+            this.FileTransfers = fileTransfers;
+        }
+
+        private void UpdatePayloadPackagingType(WixBundlePayloadRow payload)
+        {
+            if (PackagingType.Unknown == payload.Packaging)
+            {
+                if (YesNoDefaultType.Yes == payload.Compressed)
+                {
+                    payload.Packaging = PackagingType.Embedded;
+                }
+                else if (YesNoDefaultType.No == payload.Compressed)
+                {
+                    payload.Packaging = PackagingType.External;
+                }
+                else
+                {
+                    payload.Packaging = this.DefaultPackaging;
+                }
+            }
+
+            // Embedded payloads that are not assigned a container already are placed in the default attached
+            // container.
+            if (PackagingType.Embedded == payload.Packaging && String.IsNullOrEmpty(payload.Container))
+            {
+                payload.Container = Compiler.BurnDefaultAttachedContainerId;
+            }
+        }
+
+        private void UpdatePayloadFileInformation(WixBundlePayloadRow payload)
+        {
+            FileInfo fileInfo = new FileInfo(payload.SourceFile);
+
+            if (null != fileInfo)
+            {
+                payload.FileSize = (int)fileInfo.Length;
+
+                payload.Hash = Common.GetFileHash(fileInfo.FullName);
+
+                // Try to get the certificate if the payload is a signed file and we're not suppressing signature validation.
+                if (payload.EnableSignatureValidation)
+                {
+                    X509Certificate2 certificate = null;
+                    try
+                    {
+                        certificate = new X509Certificate2(fileInfo.FullName);
+                    }
+                    catch (CryptographicException) // we don't care about non-signed files.
+                    {
+                    }
+
+                    // If there is a certificate, remember its hashed public key identifier and thumbprint.
+                    if (null != certificate)
+                    {
+                        byte[] publicKeyIdentifierHash = new byte[128];
+                        uint publicKeyIdentifierHashSize = (uint)publicKeyIdentifierHash.Length;
+
+                        WixToolset.Core.Native.NativeMethods.HashPublicKeyInfo(certificate.Handle, publicKeyIdentifierHash, ref publicKeyIdentifierHashSize);
+                        StringBuilder sb = new StringBuilder(((int)publicKeyIdentifierHashSize + 1) * 2);
+                        for (int i = 0; i < publicKeyIdentifierHashSize; ++i)
+                        {
+                            sb.AppendFormat("{0:X2}", publicKeyIdentifierHash[i]);
+                        }
+
+                        payload.PublicKey = sb.ToString();
+                        payload.Thumbprint = certificate.Thumbprint;
+                    }
+                }
+            }
+        }
+
+        private void UpdatePayloadVersionInformation(WixBundlePayloadRow payload)
+        {
+            FileVersionInfo versionInfo = FileVersionInfo.GetVersionInfo(payload.SourceFile);
+
+            if (null != versionInfo)
+            {
+                // Use the fixed version info block for the file since the resource text may not be a dotted quad.
+                Version version = new Version(versionInfo.ProductMajorPart, versionInfo.ProductMinorPart, versionInfo.ProductBuildPart, versionInfo.ProductPrivatePart);
+
+                if (ProcessPayloadsCommand.EmptyVersion != version)
+                {
+                    payload.Version = version.ToString();
+                }
+
+                payload.Description = versionInfo.FileDescription;
+                payload.DisplayName = versionInfo.ProductName;
+            }
+        }
+    }
+}