Synnefo » Pithos MS Client

using System;

using System.Collections.Generic;

using System.ComponentModel;

using System.Data;

using System.Drawing;

using System.Linq;

using System.Text;

using System.Windows.Forms;

using System.Threading;

using System.Net;

using System.IO;

using System.Diagnostics;

using System.Reflection;

namespace AppLimit.NetSparkle

{

    public partial class NetSparkleDownloadProgress : Form

    {

        private String _tempName;

        private NetSparkleAppCastItem _item;

        private String _referencedAssembly;

        private Sparkle _sparkle;

        private Boolean _unattend;

        public NetSparkleDownloadProgress(Sparkle sparkle, NetSparkleAppCastItem item, String referencedAssembly, Image appIcon, Icon windowIcon, Boolean Unattend)

        {

            InitializeComponent();

            if (appIcon != null)

                imgAppIcon.Image = appIcon;

            if (windowIcon != null)

                Icon = windowIcon;

            // store the item

            _sparkle = sparkle;

            _item = item;

            _referencedAssembly = referencedAssembly;

            _unattend = Unattend;

            // init ui

            btnInstallAndReLaunch.Visible = false;

            lblHeader.Text = lblHeader.Text.Replace("APP", item.AppName + " " + item.Version);

            progressDownload.Maximum = 100;

            progressDownload.Minimum = 0;

            progressDownload.Step = 1;

            // show the right 

            Size = new Size(Size.Width, 107);

            lblSecurityHint.Visible = false;                

            // get the filename of the download lin

            String[] segments = item.DownloadLink.Split('/');

            String fileName = segments[segments.Length - 1];

            // get temp path

            _tempName = Environment.ExpandEnvironmentVariables("%temp%\\" + fileName);

            // start async download

            WebClient Client = new WebClient();

            Client.DownloadProgressChanged += new DownloadProgressChangedEventHandler(Client_DownloadProgressChanged);

            Client.DownloadFileCompleted += new AsyncCompletedEventHandler(Client_DownloadFileCompleted);

            Uri url = new Uri(item.DownloadLink);

            Client.DownloadFileAsync(url, _tempName);

        }

        private void Client_DownloadFileCompleted(object sender, AsyncCompletedEventArgs e)

        {

            progressDownload.Visible = false;

            btnInstallAndReLaunch.Visible = true;            

            // report message            

            _sparkle.ReportDiagnosticMessage("Finished downloading file to: " + _tempName);

            // check if we have a dsa signature in appcast            

            if (_item.DSASignature == null || _item.DSASignature.Length == 0)

            {

                _sparkle.ReportDiagnosticMessage("No DSA check needed");

            }

            else

            {

                Boolean bDSAOk = false;

            // report

            _sparkle.ReportDiagnosticMessage("Performing DSA check");

            // get the assembly

            if (File.Exists(_tempName))

            {

                // check if the file was downloaded successfully

                String absolutePath = Path.GetFullPath(_tempName);

                if (!File.Exists(absolutePath))

                    throw new FileNotFoundException();

                // get the assembly reference from which we start the update progress

                // only from this trusted assembly the public key can be used

                Assembly refassembly = System.Reflection.Assembly.GetEntryAssembly();

                if (refassembly != null)

                {

                    // Check if we found the public key in our entry assembly

                    if (NetSparkleDSAVerificator.ExistsPublicKey("NetSparkle_DSA.pub"))

                    {

                        // check the DSA Code and modify the back color            

                        NetSparkleDSAVerificator dsaVerifier = new NetSparkleDSAVerificator("NetSparkle_DSA.pub");

                        bDSAOk = dsaVerifier.VerifyDSASignature(_item.DSASignature, _tempName);

                    }

                }

            }

                if (!bDSAOk)

            {

                Size = new Size(Size.Width, 137);

                lblSecurityHint.Visible = true;

                BackColor = Color.Tomato;

        }

            }

            // Check the unattended mode

            if (_unattend)

                btnInstallAndReLaunch_Click(null, null);

        }

        private void Client_DownloadProgressChanged(object sender, DownloadProgressChangedEventArgs e)

        {

            progressDownload.Value = e.ProgressPercentage;            

        }

        private void btnInstallAndReLaunch_Click(object sender, EventArgs e)

        {

            // get the commandline 

            String cmdLine = Environment.CommandLine;

            String workingDir = Environment.CurrentDirectory;

            // generate the batch file path

            String cmd = Environment.ExpandEnvironmentVariables("%temp%\\" + Guid.NewGuid() + ".cmd");

            String installerCMD;

            // get the file type

            if (Path.GetExtension(_tempName).ToLower().Equals(".exe"))

            {

                // build the command line 

                installerCMD = _tempName;

            }

            else if (Path.GetExtension(_tempName).ToLower().Equals(".msi"))

            {                

                // buid the command line

                installerCMD = "msiexec /i \"" + _tempName + "\"";                

            }

            else

            {

                MessageBox.Show("Updater not supported, please execute " + _tempName + " manually", "Error", MessageBoxButtons.OK, MessageBoxIcon.Error);

                Environment.Exit(-1);

                return;

            }

            // generate the batch file                

            _sparkle.ReportDiagnosticMessage("Generating MSI batch in " + Path.GetFullPath(cmd));

            StreamWriter write = new StreamWriter(cmd);

            write.WriteLine(installerCMD);

            write.WriteLine("cd " + workingDir);

            write.WriteLine(cmdLine);

            write.Close();

            // report

            _sparkle.ReportDiagnosticMessage("Going to execute batch: " + cmd);

            // start the installer helper

            Process process = new Process();

            process.StartInfo.FileName = cmd;

            process.StartInfo.WindowStyle = ProcessWindowStyle.Hidden;

            process.Start();

            // quit the app

            Environment.Exit(0);

        }

    }

}