/// <summary>
/// Updates engine with attached container information and adds attached container again.
/// </summary>
/// <param name="bundleFile">Bundle with attached container.</param>
/// <param name="signedEngineFile">Signed bundle engine.</param>
/// <param name="outputFile">Signed engine with attached container.</param>
/// <returns>True if bundle was updated.</returns>
public bool InscribeBundle(string bundleFile, string signedEngineFile, string outputFile)
{
bool inscribed = false;
string tempFile = Path.Combine(this.TempFilesLocation, "bundle_engine_signed.exe");
using (BurnReader reader = BurnReader.Open(bundleFile))
{
File.Copy(signedEngineFile, tempFile, true);
// If there was an attached container on the original (unsigned) bundle, put it back.
if (reader.AttachedContainerSize > 0)
{
reader.Stream.Seek(reader.AttachedContainerAddress, SeekOrigin.Begin);
using (BurnWriter writer = BurnWriter.Open(tempFile))
{
writer.RememberThenResetSignature();
writer.AppendContainer(reader.Stream, reader.AttachedContainerSize, BurnCommon.Container.Attached);
inscribed = true;
}
}
}
Directory.CreateDirectory(Path.GetDirectoryName(outputFile));
if (File.Exists(outputFile))
{
File.Delete(outputFile);
}
File.Move(tempFile, outputFile);
WixToolset.Core.Native.NativeMethods.ResetAcls(new string[] { outputFile }, 1);
return(inscribed);
}
/// <summary>
/// Opens a Burn reader.
/// </summary>
/// <param name="fileExe">Path to file.</param>
/// <param name="messageHandler">Message handler.</param>
/// <returns>Burn reader.</returns>
public static BurnReader Open(string fileExe, IMessageHandler messageHandler)
{
BurnReader reader = new BurnReader(fileExe, messageHandler);
reader.binaryReader = new BinaryReader(File.Open(fileExe, FileMode.Open, FileAccess.Read, FileShare.Read | FileShare.Delete));
if (!reader.Initialize(reader.binaryReader))
{
reader.invalidBundle = true;
}
return(reader);
}
/// <summary>
/// Extracts engine from attached container and updates engine with detached container signatures.
/// </summary>
/// <param name="bundleFile">Bundle with attached container.</param>
/// <param name="outputFile">Bundle engine only.</param>
/// <returns>True if bundle was updated.</returns>
public bool InscribeBundleEngine(string bundleFile, string outputFile)
{
string tempFile = Path.Combine(this.TempFilesLocation, "bundle_engine_unsigned.exe");
using (BurnReader reader = BurnReader.Open(bundleFile))
using (FileStream writer = File.Open(tempFile, FileMode.Create, FileAccess.Write, FileShare.Read | FileShare.Delete))
{
reader.Stream.Seek(0, SeekOrigin.Begin);
byte[] buffer = new byte[4 * 1024];
int total = 0;
int read = 0;
do
{
read = Math.Min(buffer.Length, (int)reader.EngineSize - total);
read = reader.Stream.Read(buffer, 0, read);
writer.Write(buffer, 0, read);
total += read;
} while (total < reader.EngineSize && 0 < read);
if (total != reader.EngineSize)
{
throw new InvalidOperationException("Failed to copy engine out of bundle.");
}
// TODO: update writer with detached container signatures.
}
Directory.CreateDirectory(Path.GetDirectoryName(outputFile));
if (File.Exists(outputFile))
{
File.Delete(outputFile);
}
File.Move(tempFile, outputFile);
WixToolset.Core.Native.NativeMethods.ResetAcls(new string[] { outputFile }, 1);
return(true);
}
