xenadmin/XenAdmin/Plugins/Features/TabPageFeature.cs
Konstantina Chremmou 9de85b8772 CP-6267: Restored status strip on the main window.
Signed-off-by: Konstantina Chremmou <konstantina.chremmou@citrix.com>
2013-11-12 12:44:19 +00:00

1012 lines
41 KiB
C#

/* 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.IO;
using System.Net;
using System.Resources;
using System.Runtime.InteropServices;
using System.Text;
using System.Text.RegularExpressions;
using System.Threading;
using System.Windows.Forms;
using System.Xml;
using XenAdmin.Core;
using XenAdmin.Network;
using XenAdmin.XenSearch;
using XenAPI;
using XenAdmin.Actions;
namespace XenAdmin.Plugins
{
/// <summary>
/// Class describing a plugin feature which loads a URL to display as an extra tab inside XenCenter; this can be used
/// to allow access to web management consoles or to add additional user interface features into XenCenter.
/// </summary>
internal class TabPageFeature : Feature
{
/*
* Each TabPageFeature has an associated WebBrowser2 instance. There is just one per feature, and therefore they are shared if a
* feature appears on multiple XenObjects. Each time the user switches XenObject, the state of the browser for that particular
* XenObject is persisted in BrowserStates.
*
*
* We have a fairly complicated scheme for managing credentials for webpages:
*
* Normally, when you hit a page that requires authorization, IE will pop up a prompt. We intercept this prompt, and instead we
* persist credentials on the XenServer host. We also prompt if we see HTTP response 401 Authorization Required or 403 Forbidden,
* on the assumption that the password on the server has changed.
*
* Credentials are persisted server-side as Secrets, with a reference to the appropriate Secret stored in Pool.gui_config (see
* Pool.*XCPluginSecret). If the user chooses not to persist credentials, we still persist an empty string in Pool.gui_config,
* to record the fact that the user has made that choice.
*
* When we see an authentication request, we enter Browser_AuthenticationPrompt on the UI thread. We check the server for
* persisted credentials, which has to occur on a background thread, but we can't return from Browser_AuthenticationPrompt until
* we have the credentials. To handle this, we spawn a thread in TriggerGetSecret, and then use Application.DoEvents to wait on
* the UI thread without blocking redraws. See CompleteGetSecret. The background thread will set BrowserState.Credentials when
* it is finished, releasing the UI thread.
*
* If there's nothing in Pool.gui_config for us, or if there is the empty string (indicating that the user has chosen not to
* persist credentials), then we need to prompt for the credentials from the user. In that case, it's back onto the UI thread
* to show a TabPageCredentialsDialog before returning to the background thread to persist the new credentials.
*/
private static readonly log4net.ILog log = log4net.LogManager.GetLogger(System.Reflection.MethodBase.GetCurrentMethod().DeclaringType);
private const char CREDENTIALS_SEPARATOR = '\x0294';
private readonly string Url; // required - "url" attribute, the local or remote url of the HTML page to load
private readonly bool ContextEnabled; // optional - "context-menu" attribute, if set the context menu will be disabled
private readonly bool XenCenterOnly; // optional - "xencenter-only" attribute, if set this tabpage will appear on the XenCenter node and no where else
private readonly bool RelativeUrl; // optional - "relative" attribute, if set the url will be resolved relative to the XenCenter directory
private readonly string HelpLink; // optional - "help-link" attribute, if set when the help button is pressed this url will be loaded in a separate browser
private readonly bool Credentials; // optional - "credentials" attribute, if set XenCenter will duplicate a session for use by the htmls JavaScript
private readonly bool Console; // optional - "console" attribute on the "TabPage" tag.
public const string ELEMENT_NAME = "TabPage";
public const string ATT_XC_ONLY = "xencenter-only";
public const string ATT_CONTEXT_MENU = "context-menu";
public const string ATT_HELP_LINK = "help-link";
public const string ATT_RELATIVE = "relative";
public const string ATT_CREDENTIALS = "credentials";
public const string ATT_CONSOLE = "console";
public const string ATT_URL = "url";
private readonly Dictionary<IXenObject, BrowserState> BrowserStates = new Dictionary<IXenObject, BrowserState>();
private BrowserState lastBrowserState = null;
private TabControl tabControl;
private IXenObject selectedXenObject;
private IXenObject lastXenModelObject;
private TabPage _tabPage;
private WebBrowser2 Browser;
private bool UrlIsLoaded;
private ActionBase MainWindowActionAtNavigateTime = null;
/// <summary>
/// If true, this indicates that the most recent Navigation was an error.
/// </summary>
private bool navigationError;
public bool HasHelp
{
get
{
return !string.IsNullOrEmpty(HelpLink);
}
}
public TabPage TabPage
{
get
{
return _tabPage;
}
}
public IXenObject SelectedXenObject
{
get
{
return selectedXenObject;
}
set
{
selectedXenObject = value;
_tabControl_SelectedIndexChanged(null, EventArgs.Empty);
}
}
public TabPageFeature(ResourceManager resourceManager, XmlNode node, PluginDescriptor plugin)
: base(resourceManager, node, plugin)
{
XenCenterOnly = Helpers.GetBoolXmlAttribute(node, ATT_XC_ONLY, false);
ContextEnabled = Helpers.GetBoolXmlAttribute(node, ATT_CONTEXT_MENU, false);
// plugins v2
HelpLink = Helpers.GetStringXmlAttribute(node, ATT_HELP_LINK, "");
RelativeUrl = Helpers.GetBoolXmlAttribute(node, ATT_RELATIVE, false);
Credentials = Helpers.GetBoolXmlAttribute(node, ATT_CREDENTIALS, false);
// indicates that this tab-page is a replacement for the console tab.
Console = Helpers.GetBoolXmlAttribute(node, ATT_CONSOLE, false);
string urlString = Helpers.GetStringXmlAttribute(node, ATT_URL);
Url = urlString == null ? "" : string.Format("{0}{1}", RelativeUrl ? string.Format("{0}/", Application.StartupPath) : "", urlString);
}
public override string CheckForError()
{
if (Url == "")
return string.Format(Messages.CANNOT_PARSE_NODE_PARAM, node.Name, ATT_URL);
return base.CheckForError();
}
public override void Initialize()
{
_tabPage = new TabPage();
_tabPage.SuspendLayout();
_tabPage.Text = ToString();
_tabPage.ToolTipText = Tooltip ?? "";
if (Program.MainWindow != null) // for unit tests
{
_tabPage.HelpRequested += Program.MainWindow.MainWindow_HelpRequested;
}
CreateBrowser();
_tabPage.Controls.Add(Browser);
_tabPage.Tag = this; // Tag the tab page so we can reload when the tab is focussed.
_tabPage.ResumeLayout();
_tabPage.ParentChanged += TabPage_ParentChanged;
}
private void CreateBrowser()
{
Browser = new WebBrowser2();
Browser.Dock = DockStyle.Fill;
Browser.IsWebBrowserContextMenuEnabled = ContextEnabled;
Browser.ProgressChanged += Browser_ProgressChanged;
Browser.DocumentCompleted += Browser_DocumentCompleted;
Browser.Navigating += Browser_Navigating;
Browser.Navigated += Browser_Navigated;
Browser.NavigateError += Browser_NavigateError;
Browser.WindowClosed += Browser_WindowClosed;
Browser.PreviewKeyDown += Browser_PreviewKeyDown;
Browser.AuthenticationPrompt += Browser_AuthenticationPrompt;
// Navigate to about:blank to work around http://support.microsoft.com/kb/320153.
Browser.Navigate("about:blank");
}
void Browser_WindowClosed(object sender, EventArgs e)
{
if (!Program.Exiting && Enabled && !_tabPage.Disposing)
{
_tabPage.Controls.Remove(Browser);
CreateBrowser();
_tabPage.Controls.Add(Browser);
if (SelectedXenObject != null)
{
BrowserStates.Remove(SelectedXenObject);
SetUrl();
}
}
}
public bool ShowTab
{
get
{
try
{
if (_tabPage == null || Program.MainWindow.SearchMode)
{
return false;
}
if (XenCenterOnly)
{
return SelectedXenObject == null;
}
if (SelectedXenObject == null || !Enabled || !Placeholders.UriValid(Url, SelectedXenObject))
{
return false;
}
return Search == null || Search.Query.Match(SelectedXenObject);
}
catch (UriFormatException e)
{
log.Debug(string.Format("Not displaying tab '{0}' for plugin '{1}'. Invalid properties in url '{2}'", ToString(), PluginDescriptor.Name, Url), e);
}
return false;
}
}
private void SetUrl()
{
if (UrlIsLoaded && XenCenterOnly) // Never update XenCenter node tabs.
return;
BrowserState state;
if (selectedXenObject == null)
{
// XenCenter node is selected, the placeholder code will sub in "null" for all placeholders
// After this point we will never update this url again for this node, so there is no need to store a browser state
state = new BrowserState(Placeholders.SubstituteUri(Url, selectedXenObject), selectedXenObject, Browser);
}
else if (BrowserStates.ContainsKey(selectedXenObject) && !BrowserStates[selectedXenObject].IsError)
{
// if there wasn't an error with navigation then use the stored browser-state. Otherwise try again.
state = BrowserStates[selectedXenObject];
}
else
{
state = new BrowserState(Placeholders.SubstituteUri(Url, selectedXenObject), selectedXenObject, Browser);
BrowserStates[selectedXenObject] = state;
}
if (lastBrowserState == state)
return;
try
{
if (state.ObjectForScripting != null)
{
if (Credentials)
state.ObjectForScripting.LoginSession();
Browser.ObjectForScripting = state.ObjectForScripting;
}
Browser.Navigate(state.Urls);
lastBrowserState = state;
}
catch (Exception e)
{
log.Error(string.Format("Failed to set TabPage url to '{0}' for plugin '{1}'", string.Join(",", state.Urls.ConvertAll(u => u.ToString()).ToArray()), PluginDescriptor.Name), e);
}
}
private void Browser_DocumentCompleted(object sender, WebBrowserDocumentCompletedEventArgs e)
{
Program.AssertOnEventThread();
if (tabControl != null && tabControl.SelectedTab != null && tabControl.SelectedTab.Tag == this)
{
if (Program.MainWindow.StatusBarAction == null
|| Program.MainWindow.StatusBarAction.IsCompleted && MainWindowActionAtNavigateTime.Equals(Program.MainWindow.StatusBarAction))
{
// we still have 'control' of the status bar
Program.MainWindow.SetProgressBar(false, 0);
Program.MainWindow.SetStatusBar(null, null);
}
}
if (!XenCenterOnly && lastBrowserState != null)
{
log.DebugFormat("url for '{0}' set to '{1}'", Helpers.GetName(lastBrowserState.Obj), e.Url);
lastBrowserState.Urls = new List<Uri> { e.Url };
}
}
private void Browser_ProgressChanged(object sender, WebBrowserProgressChangedEventArgs e)
{
Program.AssertOnEventThread();
if (tabControl != null && tabControl.SelectedTab != null && tabControl.SelectedTab.Tag == this)
{
if (e.MaximumProgress != 0)
{
if (Program.MainWindow.StatusBarAction == null || Program.MainWindow.StatusBarAction.IsCompleted)
{
MainWindowActionAtNavigateTime = Program.MainWindow.StatusBarAction;
int progr = Convert.ToInt32((e.CurrentProgress * 100L) / (e.MaximumProgress));
Program.MainWindow.SetProgressBar(true, progr);
}
if (Browser.Url != null)
{
string text = e.CurrentProgress >= e.MaximumProgress
? ""
: string.Format(Messages.LOADING, Browser.Url.ToString().Ellipsise(50));
Program.MainWindow.SetStatusBar(null, text);
}
}
}
}
[DllImport("wininet.dll", SetLastError = true)]
private static extern long DeleteUrlCacheEntry(string url);
private void Browser_Navigating(object sender, WebBrowserNavigatingEventArgs e)
{
Program.AssertOnEventThread();
UrlIsLoaded |= e.Url.AbsoluteUri != "about:blank";
navigationError = false;
if (XenCenterOnly || e.TargetFrameName != "" || lastBrowserState == null)
return;
log.DebugFormat("url for '{0}' set to '{1}'", Helpers.GetName(lastBrowserState.Obj), e.Url);
if (Console)
{
// delete this page from the cache.... that we can be sure that if the page stops working then
// the user reliably gets the real console back.
DeleteUrlCacheEntry(e.Url.AbsoluteUri);
}
lastBrowserState.Urls = new List<Uri> { e.Url };
}
/// <summary>
/// Nothrow guarantee.
/// </summary>
void Browser_NavigateError(object sender, WebBrowserNavigateErrorEventArgs e)
{
Program.AssertOnEventThread();
navigationError = true;
try
{
log.DebugFormat("Navigate error ({0}): {1}", e.StatusCode, e.Url);
if ((e.StatusCode == 401 || e.StatusCode == 403))
{
log.Warn("Clearing secret and re-prompting, since we've seen a 401/403.");
BrowserState.BrowserCredentials creds = lastBrowserState.Credentials;
bool persisting = creds == null ? true : creds.PersistCredentials;
CompleteClearSecret(lastBrowserState);
TriggerGetSecret(lastBrowserState);
}
}
catch (Exception exn)
{
log.Error(exn, exn);
}
}
private void TabPage_ParentChanged(object sender, EventArgs e)
{
TabControl tabControl = _tabPage.Parent as TabControl;
if (this.tabControl != null)
{
this.tabControl.SelectedIndexChanged -= _tabControl_SelectedIndexChanged;
}
this.tabControl = tabControl;
if (this.tabControl != null)
{
this.tabControl.SelectedIndexChanged += _tabControl_SelectedIndexChanged;
}
}
private void _tabControl_SelectedIndexChanged(object sender, EventArgs e)
{
if (Console)
{
// if this is a console replacement window, then update the browser object with every selection change. This way the console
// tabs will always stay up-to-date
lastXenModelObject = SelectedXenObject;
if (ShowTab)
SetUrl();
}
else
{
// if this isn't a console replacement window, then only update when this tab is selected.
if (tabControl != null && tabControl.SelectedTab != null && tabControl.SelectedTab.Tag == this)
{
lastXenModelObject = SelectedXenObject;
if (ShowTab)
SetUrl();
}
else if (lastXenModelObject != null)
{
lastXenModelObject = null;
}
}
}
private void Browser_Navigated(object sender, WebBrowserNavigatedEventArgs e)
{
if (!XenCenterOnly && lastBrowserState != null)
{
log.DebugFormat("url for '{0}' set to '{1}'", Helpers.GetName(lastBrowserState.Obj), e.Url);
lastBrowserState.Urls = new List<Uri> { e.Url };
if (lastBrowserState.IsError != navigationError)
{
lastBrowserState.IsError = navigationError;
if (Console)
{
// if this plugin tab-page is a console replacement and the error-state of the navigation has changed
// then update the tabs. This ensures that user gets the real console tab back.
Program.MainWindow.UpdateToolbars();
}
}
}
}
private void Browser_PreviewKeyDown(object sender, PreviewKeyDownEventArgs e)
{
if (e.KeyCode == Keys.F1 && Browser.Focused)
{
Program.MainWindow.MainWindow_HelpRequested(null, null);
}
}
/// <summary>
/// Nothrow guarantee.
/// </summary>
void Browser_AuthenticationPrompt(WebBrowser2 sender, WebBrowserAuthenticationPromptEventArgs e)
{
try
{
Program.AssertOnEventThread();
log.Debug("Prompting for authentication...");
CompleteClearSecret(lastBrowserState);
CompleteGetSecret(lastBrowserState);
BrowserState.BrowserCredentials creds = lastBrowserState.Credentials;
if (creds != null && creds.Valid)
{
e.Username = creds.Username;
e.Password = creds.Password;
e.Success = true;
log.Debug("Prompt for authentication successful.");
}
else
{
e.Success = false;
log.Debug("Prompt for authentication cancelled / failed.");
}
}
catch (Exception exn)
{
log.Error("Prompt for authentication failed", exn);
e.Success = false;
}
}
private void CompleteGetSecret(BrowserState state)
{
if (state.Credentials != null)
return;
TriggerGetSecret(state);
while (state.Credentials == null)
Application.DoEvents();
}
/// <summary>
/// Load the secret where we've saved credentials. This will be placed on a background thread, and we'll poll for
/// its results.
/// </summary>
private void TriggerGetSecret(BrowserState state)
{
Thread t = new Thread(GetSecret);
t.IsBackground = true;
t.Start(state);
}
private delegate void BrowserStateDelegate(bool persistByDefault, BrowserState obj);
/// <summary>
/// Get the persisted secret from the server, or prompt for new credentials and persist those back to the server.
/// Completion of this thread is indicated by state.Credentials being set.
///
/// Nothrow guarantee.
/// </summary>
/// <param name="obj"></param>
private void GetSecret(object obj)
{
Program.AssertOffEventThread();
BrowserState state = (BrowserState)obj;
try
{
Session session = state.Obj.Connection.Session;
do
{
Pool pool = Helpers.GetPoolOfOne(state.Obj.Connection);
if (pool == null)
{
log.Warn("Failed to get Pool!");
// Sleep and retry.
Thread.Sleep(5000);
continue;
}
string secret_uuid = pool.GetXCPluginSecret(PluginDescriptor.Name, state.Obj);
if (secret_uuid == null)
{
log.Debug("Nothing persisted. Prompting for new credentials.");
Program.Invoke(Program.MainWindow, (BrowserStateDelegate)PromptForUsernamePassword, true, state);
MaybePersistCredentials(session, pool, state);
return;
}
else if (secret_uuid == "")
{
log.Debug("User chose not to persist these credentials.");
Program.Invoke(Program.MainWindow, (BrowserStateDelegate)PromptForUsernamePassword, false, state);
MaybePersistCredentials(session, pool, state);
return;
}
else
{
log.Debug("Found a secret.");
XenRef<Secret> secret = null;
try
{
secret = Secret.get_by_uuid(session, secret_uuid);
}
catch (Failure exn)
{
log.Warn(string.Format("Secret {0} for {1} on plugin {2} has disappeared!",
secret_uuid, Helpers.GetName(state.Obj), PluginDescriptor.Name),
exn);
TryToRemoveSecret(pool, session, PluginDescriptor.Name, state.Obj);
// Retry.
continue;
}
string val = Secret.get_value(session, secret);
string[] bits = val.Split(CREDENTIALS_SEPARATOR);
if (bits.Length != 2)
{
log.WarnFormat("Corrupt secret {0} at {1} for {2} on plugin {3}! Deleting.", val, secret_uuid,
Helpers.GetName(state.Obj), PluginDescriptor.Name);
TryToDestroySecret(session, secret.opaque_ref);
TryToRemoveSecret(pool, session, PluginDescriptor.Name, state.Obj);
// Retry.
continue;
}
log.Debug("Secret successfully read.");
BrowserState.BrowserCredentials creds = new BrowserState.BrowserCredentials();
creds.Username = bits[0];
creds.Password = bits[1];
creds.PersistCredentials = true;
creds.Valid = true;
state.Credentials = creds;
return;
}
// Unreachable. Should either have returned, or continued (to retry).
} while (true);
}
catch (Exception exn)
{
log.Warn("Ignoring exception when trying to get secret", exn);
// Note that it's essential that we set state.Credentials before leaving this function, because other threads are waiting
// for that value to appear.
BrowserState.BrowserCredentials creds = new BrowserState.BrowserCredentials();
creds.Valid = false;
state.Credentials = creds;
}
}
private void MaybePersistCredentials(Session session, Pool pool, BrowserState state)
{
BrowserState.BrowserCredentials creds = state.Credentials;
if (creds != null && creds.Valid)
{
string val =
creds.PersistCredentials ?
CreateSecret(session, creds.Username, creds.Password) :
"";
pool.SetXCPluginSecret(session, PluginDescriptor.Name, state.Obj, val);
}
}
private string CreateSecret(Session session, string username, string password)
{
string val = string.Format("{0}{1}{2}", username, CREDENTIALS_SEPARATOR, password);
return Secret.CreateSecret(session, val);
}
/// <summary>
/// Prompt for credentials, and set state.Credentials appropriately.
/// </summary>
/// <param name="defaultPersist">Whether the "persist these credentials" checkbox is checked on entry to the dialog.</param>
/// <param name="state"></param>
private void PromptForUsernamePassword(bool defaultPersist, BrowserState state)
{
Program.AssertOnEventThread();
TabPageCredentialsDialog d = new TabPageCredentialsDialog();
d.ServiceName = Label;
d.DefaultPersist = defaultPersist;
BrowserState.BrowserCredentials creds = new BrowserState.BrowserCredentials();
if (DialogResult.OK == d.ShowDialog(Program.MainWindow))
{
creds.Username = d.Username;
creds.Password = d.Password;
creds.PersistCredentials = d.PersistCredentials;
creds.Valid = true;
}
else
{
creds.Valid = false;
}
state.Credentials = creds;
}
private void CompleteClearSecret(BrowserState state)
{
if (state.Credentials == null)
return;
TriggerClearSecret(state);
while (state.Credentials != null)
Application.DoEvents();
}
private void TriggerClearSecret(BrowserState state)
{
Thread t = new Thread(ClearSecret);
t.IsBackground = true;
t.Start(state);
}
/// <summary>
/// Clear the persisted secret from the server.
/// Completion of this thread is indicated by state.Credentials being set to null.
///
/// Nothrow guarantee.
/// </summary>
/// <param name="obj"></param>
private void ClearSecret(object obj)
{
Program.AssertOffEventThread();
BrowserState state = (BrowserState)obj;
try
{
Session session = state.Obj.Connection.Session;
Pool pool = Helpers.GetPoolOfOne(state.Obj.Connection);
if (pool == null)
return;
string secret_uuid = pool.GetXCPluginSecret(PluginDescriptor.Name, state.Obj);
if (!string.IsNullOrEmpty(secret_uuid))
{
XenRef<Secret> secret_ref = Secret.get_by_uuid(session, secret_uuid);
TryToDestroySecret(session, secret_ref.opaque_ref);
TryToRemoveSecret(pool, session, PluginDescriptor.Name, state.Obj);
}
}
catch (Exception exn)
{
log.Warn("Ignoring exception when trying to clear secret", exn);
}
state.Credentials = null;
}
/// <summary>
/// Nothrow guarantee.
/// </summary>
private static void TryToDestroySecret(Session session, string opaque_ref)
{
try
{
Secret.destroy(session, opaque_ref);
log.DebugFormat("Successfully removed secret {0}", opaque_ref);
}
catch (Exception exn)
{
log.Error(string.Format("Failed to destroy secret {0}", opaque_ref), exn);
}
}
/// <summary>
/// Nothrow guarantee.
/// </summary>
private static void TryToRemoveSecret(Pool pool, Session session, string name, IXenObject obj)
{
try
{
pool.RemoveXCPluginSecret(session, name, obj);
log.DebugFormat("Successfully removed secret for {0}/{1}", name, Helpers.GetName(obj));
}
catch (Exception exn)
{
log.Error(string.Format("Failed to remove secret for {0}/{1}", name, obj), exn);
}
}
protected override void Dispose(bool disposing)
{
base.Dispose(disposing);
BrowserStates.Clear();
if (Browser != null)
{
Browser.Dispose();
Browser = null;
}
if (_tabPage != null)
{
_tabPage.Dispose();
_tabPage = null;
}
}
public void DisposeURL(IXenObject xmo)
{
if (BrowserStates.ContainsKey(xmo))
{
BrowserStates.Remove(xmo);
}
}
public void LaunchHelp()
{
if (HasHelp)
{
Program.OpenURL(HelpLink);
}
}
/// <summary>
/// Gets a value indicating whether this instance should replace the console tab.
/// </summary>
public bool IsConsoleReplacement
{
get
{
return Console;
}
}
/// <summary>
/// Gets a value indicating whether the most recent navigation for the selected xen object
/// resulted in an error.
/// </summary>
public bool IsError
{
get
{
if (selectedXenObject != null)
{
if (BrowserStates.ContainsKey(selectedXenObject))
{
return BrowserStates[selectedXenObject].IsError;
}
}
return true;
}
}
private class BrowserState
{
/// <summary>
/// May be null, if this is the XenCenter node.
/// </summary>
public readonly ScriptingObject ObjectForScripting;
public List<Uri> Urls;
/// <summary>
/// May be null, if this is the XenCenter node.
/// </summary>
public IXenObject Obj;
/// <summary>
/// May be null.
/// </summary>
public BrowserCredentials Credentials = null;
/// <summary>
/// Indicates that the browser is currently in an error state i.e. the page couldn't be loaded. This is used when the console is being replaced
/// by this tab-page so that the user can get the real console back.
/// </summary>
public bool IsError = true;
public BrowserState(List<Uri> urls, IXenObject obj, WebBrowser2 browser)
{
Urls = new List<Uri>(urls);
Obj = obj;
ObjectForScripting = obj == null ? null : new ScriptingObject(browser, obj);
}
public class BrowserCredentials
{
/// <summary>
/// True if the other fields here are valid (i.e. the user pressed OK, not Cancel, when prompted for credentials).
/// </summary>
internal bool Valid;
internal string Username;
internal string Password;
internal bool PersistCredentials;
}
}
}
[ComVisible(true)]
public class ScriptingObject
{
private static readonly log4net.ILog log = log4net.LogManager.GetLogger(System.Reflection.MethodBase.GetCurrentMethod().DeclaringType);
public const string JAVASCRIPT_CALLBACK_METHOD = "XenCenterXML";
public string SessionUuid;
public string SessionUrl;
private IXenConnection connection;
public WebBrowser2 browser;
public string SelectedObjectType;
public string SelectedObjectRef;
public ScriptingObject(WebBrowser2 Browser, IXenObject XenObject)
{
browser = Browser;
connection = XenObject.Connection;
if (connection != null)
{
connection.ConnectionResult += new EventHandler<ConnectionResultEventArgs>(connection_ConnectionResult);
}
SetObject(XenObject);
}
public void SetObject(IXenObject XenObject)
{
// Annoyingly all of our classes start with an uppercase character, whereas the servers only uppercase abbreviations
List<string> abbreviations = new List<string>(new string[]{"SR", "VDI", "VBD", "VM", "PIF", "VIF", "PBD"});
SelectedObjectType = XenObject.GetType().Name;
if (abbreviations.Find(delegate(string s) { return SelectedObjectType.StartsWith(s); }) == null)
{
string firstLetter = SelectedObjectType.Substring(0, 1);
SelectedObjectType = firstLetter.ToLowerInvariant() + SelectedObjectType.Substring(1, SelectedObjectType.Length - 1);
}
SelectedObjectRef = XenObject.opaque_ref;
}
void connection_ConnectionResult(object sender, ConnectionResultEventArgs e)
{
if (connection.IsConnected)
{
// If the connection has been reconnected then we need to update the session ref, as the old one is invalid
LoginSession();
Program.Invoke(Program.MainWindow, delegate
{
if (!browser.IsDisposed)
browser.Document.InvokeScript("RefreshPage");
else
log.Debug("Tried to access disposed webbrowser, ignoring refresh.");
});
}
}
public void LoginSession()
{
if (connection != null)
{
SessionUrl = connection.Session.Url;
SessionUuid = connection.DuplicateSession().uuid;
}
}
/// <summary>
/// This is the method called by any javascript logic in the webpage to make an api call.
/// We take the data and post it straight through to xapi on a background thread to allow
/// the calling method in the JS to exit and the page to continue drawing.
/// </summary>
/// <param name="jsCallback">Method name we will call in the JS to return the server response</param>
/// <param name="data"></param>
public void WriteXML(string jsCallback, string data)
{
Thread t = new Thread(writeXML);
t.IsBackground = true;
t.Start(new string[] { jsCallback, data });
}
private void writeXML(object context)
{
try
{
// We post to the json url so that xapi returns the information in a way that is easy to consume by the JS
string[] jsCallbackAndData = context as string[];
HttpWebRequest request = (HttpWebRequest)WebRequest.Create(string.Format("{0}/json", SessionUrl));
request.Method = "POST";
request.ContentType = "xml";
request.ContentLength = Encoding.UTF8.GetBytes(jsCallbackAndData[1]).Length;
request.UserAgent = "XenCenter\\Plugin";
using (StreamWriter xmlstream = new StreamWriter(request.GetRequestStream()))
{
xmlstream.Write(jsCallbackAndData[1]);
}
WebResponse response = request.GetResponse();
StringBuilder outputBuilder = new StringBuilder();
using (StreamReader reader = new StreamReader(response.GetResponseStream()))
{
while (reader.Peek() != -1)
{
outputBuilder.Append(reader.ReadLine());
}
}
// The xmlrpc bit of jquery expects a function object, which turns up here as a string that you could eval to get the function
// We just want the name, so we filter it out.
Regex functionNameReg = new Regex("^function(.*)\\(");
string funcName = "";
Match m = functionNameReg.Match(jsCallbackAndData[0]);
if (m.Success)
{
funcName = m.Groups[1].Value.Trim();
}
Program.Invoke(Program.MainWindow, delegate
{
if (browser.IsDisposed || browser.Disposing)
{
log.DebugFormat("Browser has been disposed, cannot return message to plugin: {0}", outputBuilder.ToString());
}
else if (browser.ObjectForScripting != this)
{
// If you don't do this, you can get old data re-entering the javascript execution after you have switched to a new object
log.DebugFormat("Scripting object has been changed, discarding message to plugin: {0}", outputBuilder.ToString());
}
else
{
browser.Document.InvokeScript(JAVASCRIPT_CALLBACK_METHOD, new string[] { funcName, outputBuilder.ToString() });
}
});
}
catch (Exception e)
{
log.Error(e);
}
//TODO: Whats the sensible way to let the JS know that there has been an error? Invoke an method with the info? How does this work with regard to message callback?
}
}
}