xenadmin/XenModel/Actions/Pool_Patch/UploadPatchAction.cs

/* Copyright (c) Citrix Systems Inc. 
 * All rights reserved. 
 * 
 * Redistribution and use in source and binary forms, 
 * with or without modification, are permitted provided 
 * that the following conditions are met: 
 * 
 * *   Redistributions of source code must retain the above 
 *     copyright notice, this list of conditions and the 
 *     following disclaimer. 
 * *   Redistributions in binary form must reproduce the above 
 *     copyright notice, this list of conditions and the 
 *     following disclaimer in the documentation and/or other 
 *     materials provided with the distribution. 
 * 
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND 
 * CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, 
 * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF 
 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 
 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR 
 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, 
 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR 
 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, 
 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 
 * SUCH DAMAGE.
 */

using System;
using System.Collections.Generic;
using System.Text;
using System.Net;
using System.IO;
using XenAdmin.Network;
using XenAdmin.Core;
using XenAPI;


namespace XenAdmin.Actions
{
    public class UploadPatchAction : AsyncAction
    {
        private static readonly log4net.ILog log = log4net.LogManager.GetLogger(System.Reflection.MethodBase.GetCurrentMethod().DeclaringType);

        private Dictionary<Host, Pool_patch> patches = new Dictionary<Host, Pool_patch>();

        private readonly IList<Host> embeddedHosts;
        private readonly string embeddedPatchPath;

        private readonly IList<Host> retailHosts;
        private readonly string retailPatchPath;

        /// <summary>
        /// This constructor is used to upload a single 'normal' patch
        /// </summary>
        /// <param name="connection"></param>
        /// <param name="path"></param>
        public UploadPatchAction(IXenConnection connection, string path)
            : base(connection, null, Messages.UPLOADING_PATCH)
        {
            Host master = Helpers.GetMaster(connection);
            if (master == null)
                throw new NullReferenceException();

            ApiMethodsToRoleCheck.Add("pool.sync_database");
            ApiMethodsToRoleCheck.Add("http/put_oem_patch_stream");
            ApiMethodsToRoleCheck.Add("http/put_pool_patch_upload");

            if (master.isOEM)
            {
                embeddedHosts = new List<Host>(connection.Cache.Hosts);
                embeddedPatchPath = path;

                retailHosts = new List<Host>();
                retailPatchPath = string.Empty;
            }
            else
            {

                retailHosts = new List<Host>(new Host[] { master });
                retailPatchPath = path;

                embeddedHosts = new List<Host>();
                embeddedPatchPath = string.Empty;

            }

            Host = master;
        }

        public Dictionary<Host, Pool_patch> PatchRefs
        {
            get
            {
                return patches;
            }
        }

        private static long FileSize(string path)
        {
            FileInfo fileInfo = new FileInfo(path);
            return fileInfo.Length;
        }

        private long TotalSize = 0;
        private long TotalUploaded = 0;

        protected override void Run()
        {
            SafeToExit = false;
            TotalSize =
                (embeddedHosts.Count > 0 ? embeddedHosts.Count * FileSize(embeddedPatchPath) : 0)
                + (retailHosts.Count > 0 ? retailHosts.Count * FileSize(retailPatchPath) : 0);

            foreach (Host host in embeddedHosts)
            {
                UploadEmbeddedPatch(host);
            }

            foreach (Host host in retailHosts)
            {
                try
                {
                    patches[host] = UploadRetailPatch(host);
                }
                catch (Failure f)
                {
                    // Need to check if the patch already exists.  
                    // If it does, and we're upload multiple patches
                    // (ie from the wizard) then we don't care about 
                    // throwing the error
                    if (f.ErrorDescription != null
                        && f.ErrorDescription.Count > 1
                        && f.ErrorDescription[0] == XenAPI.Failure.PATCH_ALREADY_EXISTS
                        && embeddedHosts.Count + retailHosts.Count > 1)
                    {
                        string uuid = f.ErrorDescription[1];
                        Session session = host.Connection.DuplicateSession();
                        patches[host] = Connection.Resolve(Pool_patch.get_by_uuid(session, uuid));
                    }
                    else
                        throw;
                }
            }

            if (embeddedHosts.Count + retailHosts.Count > 1)
                this.Description = Messages.ALL_UPDATES_UPLOADED;
        }

        public override void RecomputeCanCancel()
        {
            CanCancel = !Cancelling;
        }

        private void UploadEmbeddedPatch(Host host)
        {
            this.Description = String.Format(Messages.UPLOADING_PATCH_TO, host.Name);

            long size = FileSize(embeddedPatchPath);

            Session session = NewSession();

            HTTP.UpdateProgressDelegate progressDelegate = delegate(int percent)
                {
                    int actionPercent = (int)(((TotalUploaded * 90) + (size * percent)) / TotalSize);
                    this.Tick(actionPercent, this.Description);
                };

            RelatedTask = XenAPI.Task.create(session, "uploadTask", host.address);
            Connection = host.Connection;

            try
            {
                HTTPHelper.Put(progressDelegate, GetCancelling, true, Connection, RelatedTask, ref session, embeddedPatchPath,
                    host.address, (HTTP_actions.put_ss)HTTP_actions.put_oem_patch_stream, session.uuid);
            }
            finally
            {
                Task.destroy(session, RelatedTask);

                Connection = null;
                RelatedTask = null;
            }

            TotalUploaded += size;

            try
            {
                RelatedTask = XenAPI.Pool.async_sync_database(session);
                PollToCompletion(90, 100);
            }
            catch (Exception exn)
            {
                log.Warn("Exception during pool.sync_database", exn);
            }

            this.Description = String.Format(Messages.PATCH_UPLOADED, host.Name);
        }

        private Pool_patch UploadRetailPatch(Host host)
        {
            this.Description = String.Format(Messages.UPLOADING_PATCH_TO, host.Name);

            long size = FileSize(retailPatchPath);

            Session session = NewSession();

            try
            {
                Host h = Helpers.GetMaster(Connection);

                HTTP.UpdateProgressDelegate progressDelegate = delegate(int percent)
                    {
                        int actionPercent = (int)(((TotalUploaded * 90) + (size * percent)) / TotalSize);
                        this.Tick(actionPercent, this.Description);
                    };

                RelatedTask = XenAPI.Task.create(session, "uploadTask", retailPatchPath);

                String result;

                try
                {
                    result = HTTPHelper.Put(progressDelegate, GetCancelling, true, Connection, RelatedTask, ref session, retailPatchPath,
                        h.address, (HTTP_actions.put_ss)HTTP_actions.put_pool_patch_upload, session.uuid);
                }
                finally
                {
                    Task.destroy(session, RelatedTask);
                    RelatedTask = null;
                }

                TotalUploaded += size;
                return Connection.WaitForCache(new XenRef<Pool_patch>(result));
            }
            finally
            {
                Description = String.Format(Messages.PATCH_UPLOADED, host.Name);
            }
        }
    }
}