xenadmin/XenAdmin/Dialogs/Wlb/WlbCredentialsDialog.cs

/* Copyright (c) Cloud Software Group, Inc. 
 * 
 * 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.Net;
using System.Net.Sockets;
using System.Windows.Forms;
using XenAdmin.Actions.Wlb;
using XenAdmin.Core;
using XenAdmin.Wlb;
using XenAPI;
using XenCenterLib;

namespace XenAdmin.Dialogs.Wlb
{
    public partial class WlbCredentialsDialog : XenDialogBase
    {
        private static readonly log4net.ILog log = log4net.LogManager.GetLogger(System.Reflection.MethodBase.GetCurrentMethod().DeclaringType);

        private const int DEFAULT_WLB_PORT = 8012;
        private Pool _pool;

        public WlbCredentialsDialog(Pool pool)
        {
            InitializeComponent();
            checkboxUseCurrentXSCredentials.Text = string.Format(checkboxUseCurrentXSCredentials.Text, BrandManager.BrandConsole);
            LabelWLBServerNameBlurb.Text = string.Format(LabelWLBServerNameBlurb.Text, BrandManager.ProductBrand);
            LabelWLBServerCredsBlurb.Text = string.Format(LabelWLBServerCredsBlurb.Text, BrandManager.ProductBrand);
            LabelXenServerCredsBlurb.Text = string.Format(LabelXenServerCredsBlurb.Text, BrandManager.ProductBrand);
            decentGroupBoxXSCredentials.Text = string.Format(decentGroupBoxXSCredentials.Text, BrandManager.ProductBrand);

            _pool = pool;
            PopulateControls();
            SetControlState();
        }

        private void PopulateControls()
        {
            StringUtility.ParseHostnamePort(_pool.wlb_url, out var hostname, out var port);
            
            if (port == 0)
                port = DEFAULT_WLB_PORT;

            textboxWlbUrl.Text = hostname;
            textboxWLBPort.Text = port.ToString();
            textboxWlbUserName.Text = _pool.wlb_username;
        }

        private void SetControlState()
        {
            switch (WlbServerState.GetState(_pool))
            {
                case WlbServerState.ServerState.NotConfigured:
                case WlbServerState.ServerState.ConnectionError:
                    {
                        decentGroupBoxWLBServerAddress.Enabled = true;
                        break;
                    }
            }
        }

        private void checkboxUseCurrentXSCredentials_CheckedChanged(object sender, EventArgs e)
        {
            SetXSCredentials(checkboxUseCurrentXSCredentials.Checked);
        }

        private void SetXSCredentials(bool useCurrent)
        {
            if (useCurrent)
            {
                textboxXSUserName.Text = _pool.Connection.Username;
                textboxXSPassword.Text = _pool.Connection.Password;
                textboxXSUserName.Enabled = false;
                textboxXSPassword.Enabled = false;
            }
            else
            {
                textboxXSUserName.Text = String.Empty;
                textboxXSPassword.Text = String.Empty;
                textboxXSUserName.Enabled = true;
                textboxXSPassword.Enabled = true;
            }
        }

        private void buttonOK_Click(object sender, EventArgs e)
        {
            if (WlbServerState.GetState(_pool) == WlbServerState.ServerState.ConnectionError || WlbServerState.GetState(_pool) == WlbServerState.ServerState.NotConfigured)
            {
                try
                {
                    if (InitializeWLB())
                    {
                        this.DialogResult = DialogResult.OK;
                        this.Close();
                    }
                }
                catch
                {
                    log.Warn("Error Intializing WLB");
                }
            }
            else 
            {
                if (WlbServerState.GetState(_pool) == WlbServerState.ServerState.Disabled)
                {
                    if (EnableWlb())
                    {
                        this.DialogResult = DialogResult.OK;
                        this.Close();
                    }
                }
            }
        }

        private bool EnableWlb()
        {

            // Enable WLB.
            EnableWLBAction action = new EnableWLBAction(_pool);
            // We will need to re-enable buttons when the action completes
            action.Completed += Program.MainWindow.action_Completed;
            using (var dialog = new ActionProgressDialog(action, ProgressBarStyle.Blocks))
            {
                dialog.ShowCancel = true;
                dialog.ShowDialog(this);
            }

            return action.Succeeded;
        }

        private bool InitializeWLB()
        {
            //combine url and port 
            string wlbHost = textboxWlbUrl.Text;
            string wlbPort = textboxWLBPort.Text;
            IPAddress address;
            string wlbUrl = (IPAddress.TryParse(wlbHost, out address) && address.AddressFamily == AddressFamily.InterNetworkV6) ?
                ("[" + wlbHost + "]:" + wlbPort) : (wlbHost + ":" + wlbPort);

            //handle the wlb creds
            string wlbUserName = textboxWlbUserName.Text;
            string wlbPassword = textboxWlbPassword.Text;

            //handle the xenserver creds
            string xsUserName = textboxXSUserName.Text;
            string xsPassword = textboxXSPassword.Text;

            InitializeWLBAction action = new InitializeWLBAction(_pool, wlbUrl, wlbUserName, wlbPassword, xsUserName, xsPassword);
            using (var dialog = new ActionProgressDialog(action, ProgressBarStyle.Blocks))
            {
                dialog.ShowCancel = true;
                dialog.ShowDialog(this);
            }

            Program.MainWindow.UpdateToolbars();
            return action.Succeeded;
        }

        private void textboxWlbUrl_TextChanged(object sender, EventArgs e)
        {
            buttonOK.Enabled = checkEnabled_OkButton();
        }

        private bool checkEnabled_OkButton()
        {
            return textboxWlbUrl.Text.Length > 0 &&
                   IsValidServerAddress(textboxWlbUrl.Text) &&
                   textboxWLBPort.Text.Length > 0 &&
                   int.TryParse(textboxWLBPort.Text, out _) &&
                   textboxWlbUserName.Text.Length > 0 &&
                   textboxWlbPassword.Text.Length > 0 &&
                   textboxXSUserName.Text.Length > 0 &&
                   textboxXSPassword.Text.Length > 0;
        }

        private bool IsValidServerAddress(string addr)
        {
            // A valid server address should be an IPv4 / IPv6 address or a valid domain name

            if (IPAddress.TryParse(addr, out var address))
            {
                return address.AddressFamily == AddressFamily.InterNetwork || address.AddressFamily == AddressFamily.InterNetworkV6;
            }

            try
            {
                // use UriBuilder as a quick validator
                var ub = new UriBuilder($"http://{addr}:80/");
                return ub.Host == addr;
            }
            catch
            {
                return false;
            }
        }

        private void textboxWLBPort_TextChanged(object sender, EventArgs e)
        {
            buttonOK.Enabled = checkEnabled_OkButton();
        }

        private void textboxWlbUserName_TextChanged(object sender, EventArgs e)
        {
            buttonOK.Enabled = checkEnabled_OkButton();
        }

        private void textboxWlbPassword_TextChanged(object sender, EventArgs e)
        {
            buttonOK.Enabled = checkEnabled_OkButton();
        }

        private void textboxXSUserName_TextChanged(object sender, EventArgs e)
        {
            buttonOK.Enabled = checkEnabled_OkButton();
        }

        private void textboxXSPassword_TextChanged(object sender, EventArgs e)
        {
            buttonOK.Enabled = checkEnabled_OkButton();
        }
    }
}