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;

using log4net;

namespace AppLimit.NetSparkle

{

    public partial class NetSparkleDownloadProgress : Form

    {

        private static readonly ILog Log = LogManager.GetLogger(MethodBase.GetCurrentMethod().DeclaringType);

        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;

            if (e.Error != null)

            {

                Log.Error("Update download failed ",e.Error);

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

                lblSecurityHint.Text = "An error occured while downloading the update. Please try again later.";

                lblSecurityHint.Visible = true;

                BackColor = Color.Tomato;

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

                btnInstallAndReLaunch.Click-=btnInstallAndReLaunch_Click;

                btnInstallAndReLaunch.Text = "Close";

                btnInstallAndReLaunch.Click+=(s,args)=> Close();

                btnInstallAndReLaunch.Visible = true;

                return;

            }

            // 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)

        {

            try

            {

                // 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

                var extension = Path.GetExtension(_tempName).ToLower();

                switch (extension)

                {

                    case ".exe":

                        installerCMD = _tempName;

                        break;

                    case ".msi":

                        installerCMD = String.Format("msiexec /i \"{0}\"", _tempName);

                        break;

                    default:

                        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));

                using (var 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

                var process = new Process();

                process.StartInfo.FileName = cmd;

                process.StartInfo.WindowStyle = ProcessWindowStyle.Hidden;

                process.Start();

                // quit the app

                Environment.Exit(0);

            }

            catch (Exception exc)

            {

                Log.Error("Error while launching the update", exc);

                this.DialogResult = DialogResult.Cancel;

                MessageBox.Show("An error occured while executing the update.", "Update failed", MessageBoxButtons.OK, MessageBoxIcon.Exclamation);

            }

        }

    }

}