/* 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.ComponentModel; using System.Drawing; using System.Linq; using System.Threading; using System.Windows.Forms; using XenAdmin.Actions; using XenAdmin.Controls; using XenAdmin.Core; using XenAdmin.Diagnostics.Checks; using XenAdmin.Diagnostics.Problems; using XenAdmin.Diagnostics.Problems.HostProblem; using XenAdmin.Dialogs; using XenAdmin.Properties; using XenAPI; namespace XenAdmin.Wizards.PatchingWizard { public partial class PatchingWizard_PrecheckPage : XenTabPage { private static readonly log4net.ILog log = log4net.LogManager.GetLogger(System.Reflection.MethodBase.GetCurrentMethod().DeclaringType); private BackgroundWorker _worker = null; public List SelectedServers = new List(); public List ProblemsResolvedPreCheck = new List(); public bool IsInAutomaticMode { get; set; } private AsyncAction resolvePrechecksAction = null; protected List SelectedPools { get { return SelectedServers.Select(host => Helpers.GetPool(host.Connection)).Where(pool => pool != null).Distinct().ToList(); } } public PatchingWizard_PrecheckPage() { InitializeComponent(); dataGridView1.BackgroundColor = dataGridView1.DefaultCellStyle.BackColor; } public override string PageTitle { get { return Messages.PATCHINGWIZARD_PRECHECKPAGE_TITLE; } } public override string Text { get { return Messages.PATCHINGWIZARD_PRECHECKPAGE_TEXT; } } public override string HelpID { get { return "UpdatePrechecks"; } } void Connection_ConnectionStateChanged(object sender, EventArgs e) { Program.Invoke(Program.MainWindow, RefreshRechecks); } private void RegisterEventHandlers() { foreach (Host selectedServer in SelectedServers) { selectedServer.Connection.ConnectionStateChanged += Connection_ConnectionStateChanged; } } private void DeregisterEventHandlers() { foreach (Host selectedServer in SelectedServers) { selectedServer.Connection.ConnectionStateChanged -= Connection_ConnectionStateChanged; } } public override void PageLoaded(PageLoadedDirection direction) { base.PageLoaded(direction); try { RegisterEventHandlers(); if (direction == PageLoadedDirection.Back) return; if (IsInAutomaticMode) { labelPrechecksFirstLine.Text = Messages.PATCHINGWIZARD_PRECHECKPAGE_FIRSTLINE_AUTOMATIC_MODE; } else { labelPrechecksFirstLine.Text = Patch != null ? string.Format(Messages.PATCHINGWIZARD_PRECHECKPAGE_FIRSTLINE, Patch.Name) : Messages.PATCHINGWIZARD_PRECHECKPAGE_FIRSTLINE_NO_PATCH_NAME; } RefreshRechecks(); } catch (Exception e) { log.Error(e, e); throw;//better throw an exception rather than closing the wizard suddenly and silently } } protected void RefreshRechecks() { buttonResolveAll.Enabled = buttonReCheckProblems.Enabled = checkBoxViewPrecheckFailuresOnly.Enabled = false; _worker = null; _worker = new BackgroundWorker(); _worker.DoWork += new DoWorkEventHandler(worker_DoWork); _worker.WorkerReportsProgress = true; _worker.WorkerSupportsCancellation = true; _worker.ProgressChanged += new ProgressChangedEventHandler(worker_ProgressChanged); _worker.RunWorkerCompleted += new RunWorkerCompletedEventHandler(_worker_RunWorkerCompleted); _worker.RunWorkerAsync(Patch); } private void _worker_RunWorkerCompleted(object sender, RunWorkerCompletedEventArgs e) { if (!e.Cancelled) OnPageUpdated(); progressBar1.Value = 100; labelProgress.Text = string.Empty; bool showResolveAllButton = false; foreach (PreCheckGridRow row in dataGridView1.Rows) { PreCheckHostRow hostRow = row as PreCheckHostRow; //CA-65508: if the problem cannot be solved immediately there's no point in enabling the Resolve All button //CA-136211: Changed the code below to enable the Resolve All button only when there is at least one problem and all the problems have solution/fix. if (hostRow != null && hostRow.IsProblem) { if (!hostRow.IsFixable) { showResolveAllButton = false; break; } else { showResolveAllButton = true; } } } buttonResolveAll.Enabled = showResolveAllButton; buttonReCheckProblems.Enabled = checkBoxViewPrecheckFailuresOnly.Enabled = true; } private void AddRowToGridView(DataGridViewRow row) { lock (_update_grid_lock) { dataGridView1.Rows.Add(row); } } private void worker_ProgressChanged(object sender, ProgressChangedEventArgs e) { try { PreCheckHostRow rowHost = e.UserState as PreCheckHostRow; if (rowHost != null) { if (checkBoxViewPrecheckFailuresOnly.Checked && rowHost.Problem != null || !checkBoxViewPrecheckFailuresOnly.Checked) AddRowToGridView(rowHost); } else { var row = e.UserState as DataGridViewRow; if (row != null && !dataGridView1.Rows.Contains(row)) { AddRowToGridView(row); } } int step = (int)((1.0 / ((float)_numberChecks)) * e.ProgressPercentage); progressBar1.Value += (step + progressBar1.Value) > 100 ? 0 : step; } catch (Exception) { } } private bool IsCheckInProgress { get { return _worker != null && _worker.IsBusy; } } private bool IsResolveActionInProgress { get { return resolvePrechecksAction != null && !resolvePrechecksAction.IsCompleted; } } private List ExecuteCheck(Check check) { var rows = new List(); var problems = check.RunAllChecks(); if (problems.Count == 0) { rows.Add(new PreCheckHostRow(check)); return rows; } foreach (var pr in problems) { var problem = pr; // we need this line because we sometimes reassign it below if (problem is HostNotLive) { // this host is no longer live -> remove all previous problems regarding this host Problem curProblem = problem; ProblemsResolvedPreCheck.RemoveAll(p => p.Check.Host == curProblem.Check.Host); } if (ProblemsResolvedPreCheck.Contains(problem)) { Problem curProblem = problem; problem = ProblemsResolvedPreCheck.Find(p => p.Equals(curProblem)); } else ProblemsResolvedPreCheck.Add(problem); rows.Add(new PreCheckHostRow(problem)); } return rows; } private int _numberChecks = 0; private readonly object _lock = new object(); private readonly object _update_grid_lock = new object(); private void worker_DoWork(object sender, DoWorkEventArgs e) { lock (_lock) { Program.Invoke(this, () => { dataGridView1.Rows.Clear(); progressBar1.Value = 0; labelProgress.Text = Messages.PATCHING_WIZARD_RUNNING_PRECHECKS; }); Pool_patch patch = e.Argument as Pool_patch; List>> checks = GenerateChecks(patch); _numberChecks = checks.Count; for (int i = 0; i < checks.Count; i++) { if (_worker.CancellationPending) { e.Cancel = true; return; } List checkGroup = checks[i].Value; PreCheckHeaderRow headerRow = new PreCheckHeaderRow(string.Format(Messages.PATCHING_WIZARD_PRECHECK_STATUS, checks[i].Key)); _worker.ReportProgress(5, headerRow); PreCheckResult precheckResult = PreCheckResult.OK; for (int j = 0; j < checkGroup.Count; j++) { if (_worker.CancellationPending) { e.Cancel = true; return; } Check check = checkGroup[j]; foreach (PreCheckHostRow row in ExecuteCheck(check)) { if (precheckResult != PreCheckResult.Failed && row.Problem != null) precheckResult = row.IsProblem ? PreCheckResult.Failed : PreCheckResult.Warning; _worker.ReportProgress(PercentageSelectedServers(j + 1), row); } } lock (_update_grid_lock) { headerRow.UpdateDescription(precheckResult); } } } } public Dictionary LivePatchCodesByHost { get; set; } protected virtual List>> GenerateChecks(Pool_patch patch) { List>> checks = new List>>(); LivePatchCodesByHost = new Dictionary(); //HostLivenessCheck checks checks.Add(new KeyValuePair>(Messages.CHECKING_HOST_LIVENESS_STATUS, new List())); List checkGroup = checks[checks.Count - 1].Value; foreach (Host host in SelectedServers) { checkGroup.Add(new HostLivenessCheck(host)); } //HA checks checks.Add(new KeyValuePair>(Messages.CHECKING_HA_STATUS, new List())); checkGroup = checks[checks.Count - 1].Value; foreach (Host host in SelectedServers) { if (Helpers.HostIsMaster(host)) checkGroup.Add(new HAOffCheck(host)); } //PBDsPluggedCheck checks.Add(new KeyValuePair>(Messages.CHECKING_STORAGE_CONNECTIONS_STATUS, new List())); checkGroup = checks[checks.Count - 1].Value; foreach (Host host in SelectedServers) { checkGroup.Add(new PBDsPluggedCheck(host)); } //Checking other things if (patch != null) { checks.Add(new KeyValuePair>(Messages.CHECKING_SERVER_SIDE_STATUS, new List())); checkGroup = checks[checks.Count - 1].Value; foreach (Host host in SelectedServers) { List poolPatches = new List(host.Connection.Cache.Pool_patches); Pool_patch poolPatchFromHost = poolPatches.Find(otherPatch => string.Equals(otherPatch.uuid, patch.uuid, StringComparison.OrdinalIgnoreCase)); checkGroup.Add(new PatchPrecheckCheck(host, poolPatchFromHost, LivePatchCodesByHost)); } } //Checking can evacuate host //CA-97061 - evacuate host -> suspended VMs. This is only needed for restartHost //Also include this check for the supplemental packs (patch == null), as their guidance is restartHost if (patch == null || patch.after_apply_guidance.Contains(after_apply_guidance.restartHost)) { checks.Add(new KeyValuePair>(Messages.CHECKING_CANEVACUATE_STATUS, new List())); checkGroup = checks[checks.Count - 1].Value; foreach (Host host in SelectedServers) { checkGroup.Add(new AssertCanEvacuateCheck(host, LivePatchCodesByHost)); } } if (IsInAutomaticMode) { checks.Add(new KeyValuePair>(Messages.PATCHINGWIZARD_PRECHECKPAGE_CHECKING_DISK_SPACE, new List())); checkGroup = checks[checks.Count - 1].Value; foreach (Pool pool in SelectedPools) { var us = Updates.GetUpgradeSequence(pool.Connection); foreach (Host host in us.Keys) { checkGroup.Add( new DiskSpaceForBatchUpdatesCheck( host, host.IsMaster() ? us[host].Sum(p => p.InstallationSize) + us.Values.SelectMany(a => a).Max(p => p.InstallationSize) // master: all updates on master + largest update in pool : us[host].Sum(p => p.InstallationSize) + us[host].Max(p => p.InstallationSize) // non-master: all updates on this host + largest on this host )); } } } return checks; } [DefaultValue(true)] protected bool ManualUpgrade { set; get; } private int PercentageSelectedServers(int i) { return (int)(((float)i / (float)(SelectedServers.Count)) * 100); } public override void PageCancelled() { DeregisterEventHandlers(); if (_worker != null) _worker.CancelAsync(); if (resolvePrechecksAction != null && !resolvePrechecksAction.IsCompleted) resolvePrechecksAction.Cancel(); } public override void PageLeave(PageLoadedDirection direction, ref bool cancel) { DeregisterEventHandlers(); if (direction == PageLoadedDirection.Back && _worker != null) { _worker.CancelAsync(); _worker = null; } base.PageLeave(direction, ref cancel); } public override bool EnablePrevious() { return !IsCheckInProgress && !IsResolveActionInProgress; } public override bool EnableNext() { bool problemsFound = false; foreach (DataGridViewRow row in dataGridView1.Rows) { PreCheckHostRow preCheckHostRow = row as PreCheckHostRow; if (preCheckHostRow != null && preCheckHostRow.IsProblem) { problemsFound = true; break; } } UpdateControls(problemsFound); return !IsCheckInProgress && !IsResolveActionInProgress && !problemsFound; } public UpdateType SelectedUpdateType { private get; set; } public Pool_patch Patch { private get; set; } internal enum PreCheckResult { OK, Warning, Failed } private abstract class PreCheckGridRow : XenAdmin.Controls.DataGridViewEx.DataGridViewExRow { protected DataGridViewImageCell _iconCell = new DataGridViewImageCell(); protected DataGridViewTextBoxCell _descriptionCell = new DataGridViewTextBoxCell(); protected DataGridViewCell _solutionCell = null; protected PreCheckGridRow(DataGridViewCell solutionCell) { _solutionCell = solutionCell; Cells.Add(_iconCell); Cells.Add(_descriptionCell); Cells.Add(_solutionCell); } } private class PreCheckHeaderRow : PreCheckGridRow { private string description; public PreCheckHeaderRow(string text) : base(new DataGridViewTextBoxCell()) { _iconCell.Value = new Bitmap(1, 1); description = text; _descriptionCell.Value = text; _descriptionCell.Style.Font = new Font(Program.DefaultFont, FontStyle.Bold); } public void UpdateDescription(PreCheckResult precheckResult) { string result = precheckResult == PreCheckResult.OK ? Messages.GENERAL_STATE_OK : precheckResult == PreCheckResult.Warning ? Messages.WARNING : Messages.FAILED; _descriptionCell.Value = string.Format("{0} {1}", description, result); } } private class PreCheckHostRow : PreCheckGridRow { private Problem _problem = null; private Check _check = null; public PreCheckHostRow(Problem problem) : base(new DataGridViewTextBoxCell()) { _problem = problem; _check = problem.Check; UpdateRowFields(); } public PreCheckHostRow(Check check) : base(new DataGridViewTextBoxCell()) { _check = check; UpdateRowFields(); } public bool IsFixable { get { return Problem != null && Problem.IsFixable && !string.IsNullOrEmpty(this.Solution); } } private void UpdateRowFields() { _iconCell.Value = Problem == null ? Resources._000_Tick_h32bit_16 : Problem is Warning ? Resources._000_Alert2_h32bit_16 : Resources._000_Abort_h32bit_16; string description = string.Empty; if (Problem != null) description = Problem.Description; else if (_check != null) description = String.Format(Messages.PATCHING_WIZARD_HOST_CHECK_OK, _check.Host.Name, _check.Description); if (description != string.Empty) _descriptionCell.Value = String.Format(Messages.PATCHING_WIZARD_DESC_CELL_INDENT, null, description); _solutionCell.Value = Problem == null ? string.Empty : Problem.HelpMessage; if (Problem is WarningWithInformationUrl) _solutionCell.Value = (Problem as WarningWithInformationUrl).LinkText; if (Problem is ProblemWithInformationUrl) _solutionCell.Value = (Problem as ProblemWithInformationUrl).LinkText; UpdateSolutionCellStyle(); } private void UpdateSolutionCellStyle() { if (_solutionCell == null) return; if (Enabled) { _solutionCell.Style.Font = new Font(Program.DefaultFont, FontStyle.Underline); _solutionCell.Style.ForeColor = Color.Blue; } else _solutionCell.Style = DefaultCellStyle; } public Problem Problem { get { return _problem; } } public string Solution { get { return (string)_solutionCell.Value; } } public bool IsProblem { get { return _problem != null && !(_problem is Warning); } } public override bool Equals(object obj) { PreCheckHostRow other = obj as PreCheckHostRow; if (other != null && other.Problem != null && _problem != null) return _problem.CompareTo(other.Problem) == 0; return false; } public override int GetHashCode() { return (_problem != null ? _problem.GetHashCode() : 0); } public override bool Enabled { get { return base.Enabled; } set { base.Enabled = value; UpdateSolutionCellStyle(); } } } private void dataGridView1_CellContentClick(object sender, DataGridViewCellEventArgs e) { PreCheckHostRow preCheckHostRow = dataGridView1.Rows[e.RowIndex] as PreCheckHostRow; if (preCheckHostRow != null && preCheckHostRow.Enabled && e.ColumnIndex == 2) { ExecuteSolution(preCheckHostRow); } } private void dataGridView1_KeyPress(object sender, KeyPressEventArgs e) { if (e.KeyChar == (char)Keys.Enter && dataGridView1.CurrentCell != null) { PreCheckHostRow preCheckHostRow = dataGridView1.CurrentCell.OwningRow as PreCheckHostRow; int columnIndex = dataGridView1.CurrentCell.ColumnIndex; if (preCheckHostRow != null && preCheckHostRow.Enabled && columnIndex == 2) ExecuteSolution(preCheckHostRow); } } private void ExecuteSolution(PreCheckHostRow preCheckHostRow) { bool cancelled; resolvePrechecksAction = preCheckHostRow.Problem.SolveImmediately(out cancelled); if (resolvePrechecksAction != null) { // disable all problems foreach (DataGridViewRow row in dataGridView1.Rows) { PreCheckHostRow preCheckRow = row as PreCheckHostRow; if (preCheckRow != null && preCheckRow.Problem != null) { preCheckRow.Enabled = false; } } StartResolvePrechecksAction(); } else { if (preCheckHostRow.Problem is WarningWithInformationUrl) (preCheckHostRow.Problem as WarningWithInformationUrl).LaunchUrlInBrowser(); else if (preCheckHostRow.Problem is ProblemWithInformationUrl) (preCheckHostRow.Problem as ProblemWithInformationUrl).LaunchUrlInBrowser(); else if (!cancelled) using (var dlg = new ThreeButtonDialog(new ThreeButtonDialog.Details(SystemIcons.Information, string.Format(Messages.PATCHING_WIZARD_SOLVE_MANUALLY, preCheckHostRow.Problem.Description).Replace("\\n", "\n"), Messages.PATCHINGWIZARD_PRECHECKPAGE_TEXT))) { dlg.ShowDialog(this); } } } private void buttonReCheckProblems_Click(object sender, EventArgs e) { RefreshRechecks(); } private void buttonResolveAll_Click(object sender, EventArgs e) { List actions = new List(); foreach (DataGridViewRow row in dataGridView1.Rows) { PreCheckHostRow preCheckHostRow = row as PreCheckHostRow; if (preCheckHostRow != null && preCheckHostRow.Problem != null) { bool cancelled; AsyncAction action = preCheckHostRow.Problem.SolveImmediately(out cancelled); if (action != null) { preCheckHostRow.Enabled = false; actions.Add(action); } } } resolvePrechecksAction = new ParallelAction(Messages.PATCHINGWIZARD_PRECHECKPAGE_RESOLVING_ALL, Messages.PATCHINGWIZARD_PRECHECKPAGE_RESOLVING_ALL, Messages.COMPLETED, actions, true, false); StartResolvePrechecksAction(); } private void resolvePrecheckAction_Changed(object sender) { var action = sender as AsyncAction; if (action == null) return; Program.Invoke(this, () => UpdateActionProgress(action)); } private void resolvePrecheckAction_Completed(object sender) { var action = sender as AsyncAction; if (action == null) return; action.Changed -= resolvePrecheckAction_Changed; action.Completed -= resolvePrecheckAction_Completed; Program.Invoke(this, () => { UpdateControls(); RefreshRechecks(); }); } private void StartResolvePrechecksAction() { if (resolvePrechecksAction == null) return; resolvePrechecksAction.Changed += resolvePrecheckAction_Changed; resolvePrechecksAction.Completed += resolvePrecheckAction_Completed; resolvePrechecksAction.RunAsync(); UpdateActionProgress(resolvePrechecksAction); UpdateControls(); OnPageUpdated(); } private void UpdateActionProgress(AsyncAction action) { progressBar1.Value = action == null ? 0 : action.PercentComplete; labelProgress.Text = action == null ? string.Empty : action.Description; } private void UpdateControls(bool problemsFound = false) { bool actionInProgress = IsResolveActionInProgress; bool checkInProgress = IsCheckInProgress; buttonResolveAll.Enabled = buttonReCheckProblems.Enabled = checkBoxViewPrecheckFailuresOnly.Enabled = !actionInProgress && !checkInProgress; labelProgress.Visible = actionInProgress || checkInProgress || !problemsFound; pictureBoxIssues.Visible = labelIssues.Visible = problemsFound && !actionInProgress && !checkInProgress; } private void checkBox1_CheckedChanged(object sender, EventArgs e) { RefreshRechecks(); } private void dataGridView1_CellMouseMove(object sender, DataGridViewCellMouseEventArgs e) { PreCheckHostRow preCheckHostRow = dataGridView1.Rows[e.RowIndex] as PreCheckHostRow; if (preCheckHostRow != null && preCheckHostRow.Enabled && e.ColumnIndex == 2 && !string.IsNullOrEmpty(preCheckHostRow.Solution)) Cursor = Cursors.Hand; else Cursor = Cursors.Arrow; } } }