2017-01-16 20:59:50 +01:00
/ * Copyright ( c ) Citrix Systems , Inc .
2013-06-24 13:41:48 +02:00
* 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 ;
2013-11-12 13:44:19 +01:00
using XenAdmin.XenSearch ;
2013-06-24 13:41:48 +02:00
using XenAPI ;
2013-11-12 13:44:19 +01:00
using XenAdmin.Actions ;
2013-06-24 13:41:48 +02:00
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 ;
2013-11-12 13:44:19 +01:00
private ActionBase MainWindowActionAtNavigateTime = null ;
2013-06-24 13:41:48 +02:00
/// <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 ;
2013-11-12 13:44:19 +01:00
Browser . ProgressChanged + = Browser_ProgressChanged ;
2013-06-24 13:41:48 +02:00
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 ( ) ;
2013-11-12 13:44:19 +01:00
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 ) ;
}
}
2013-06-24 13:41:48 +02:00
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 } ;
}
}
2013-11-12 13:44:19 +01:00
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 ) ;
}
}
}
}
2013-06-24 13:41:48 +02:00
[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 ) ;
2013-09-25 12:10:28 +02:00
if ( ! string . IsNullOrEmpty ( secret_uuid ) )
2013-06-24 13:41:48 +02:00
{
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 ;
2016-01-15 02:57:36 +01:00
request . UserAgent = Branding . BRAND_CONSOLE + "\\Plugin" ;
2013-06-24 13:41:48 +02:00
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?
}
}
}