524 lines
13 KiB
C#
524 lines
13 KiB
C#
namespace Microsoft.VSDesigner.WMI {
|
|
using System.Runtime.Serialization;
|
|
using System.Diagnostics;
|
|
using System;
|
|
using System.WinForms;
|
|
using System.Drawing;
|
|
using System.Collections;
|
|
using Microsoft.VSDesigner;
|
|
using Microsoft.VSDesigner.ServerExplorer;
|
|
using System.ComponentModel.Design;
|
|
using System.ComponentModel;
|
|
using System.Resources;
|
|
|
|
using WbemScripting;
|
|
|
|
|
|
// <doc>
|
|
// <desc>
|
|
// This represents the wmi classes node under a server node.
|
|
// </desc>
|
|
// </doc>
|
|
[UserContextAttribute("serveritem", "IntrinsicEventQuery")]
|
|
public class IntrinsicEventQueryNode : Node, ISerializable {
|
|
//
|
|
// FIELDS
|
|
//
|
|
|
|
enum SubscriptionState
|
|
{
|
|
Started,
|
|
Stopped,
|
|
Error
|
|
};
|
|
|
|
static Image icon;
|
|
public static readonly Type parentType = typeof(WMIIntrinsicEventsNode);
|
|
private string label = string.Empty;
|
|
|
|
private ResourceManager rm = null;
|
|
|
|
private string serverName = string.Empty;
|
|
private string nsName = string.Empty;
|
|
private string query = string.Empty;
|
|
|
|
private ISWbemServices wbemServices = null;
|
|
|
|
private SubscriptionState state = SubscriptionState.Stopped;
|
|
|
|
private SWbemSink theSink = null;
|
|
|
|
private IntrinsicEventQueryComponent browseObject = null;
|
|
|
|
//
|
|
// CONSTRUCTORS
|
|
//
|
|
|
|
// <doc>
|
|
// <desc>
|
|
// Main constructor.
|
|
// Parameters:
|
|
// string server (in, machine name)
|
|
// string pathIn (in, WMI path without the server name , e.g. "root\default:MyClass")
|
|
// </desc>
|
|
// </doc>
|
|
public IntrinsicEventQueryNode(string serverIn,
|
|
string nsIn,
|
|
string queryIn) {
|
|
try
|
|
{
|
|
|
|
serverName = serverIn;
|
|
nsName = nsIn;
|
|
query = queryIn;
|
|
|
|
state = SubscriptionState.Stopped;
|
|
|
|
try
|
|
{
|
|
rm = new ResourceManager( "Microsoft.VSDesigner.WMI.Res", // Name of the resource.
|
|
".", // Use current directory.
|
|
null);
|
|
}
|
|
catch (Exception)
|
|
{
|
|
//do nothing, will use static RM
|
|
}
|
|
|
|
//Get and cache wbemServices
|
|
ISWbemLocator wbemLocator = WmiHelper.WbemLocator;//(ISWbemLocator)(new SWbemLocator());
|
|
|
|
wbemServices = wbemLocator.ConnectServer(serverName,
|
|
nsName,
|
|
"", //user: blank defaults to current logged-on user
|
|
"", //password: blank defaults to current logged-on user
|
|
"", //locale: blank for current locale
|
|
"", //authority: NTLM or Kerberos. Blank lets DCOM negotiate.
|
|
0, //flags: reserved
|
|
null); //context info: not needed here
|
|
if (wbemServices == null)
|
|
{
|
|
throw new Exception(WMISys.GetString("WMISE_WMIConnectFailed"));
|
|
}
|
|
|
|
browseObject = new IntrinsicEventQueryComponent(serverName, nsName, query, this);
|
|
|
|
|
|
}
|
|
catch (Exception exc)
|
|
{
|
|
MessageBox.Show(WMISys.GetString("WMISE_Exception", exc.Message, exc.StackTrace));
|
|
throw (exc);
|
|
}
|
|
}
|
|
|
|
|
|
/// <summary>
|
|
/// The object retrieves its serialization info.
|
|
/// </summary>
|
|
public IntrinsicEventQueryNode(SerializationInfo info, StreamingContext context) {
|
|
|
|
try
|
|
{
|
|
try
|
|
{
|
|
rm = new ResourceManager( "Microsoft.VSDesigner.WMI.Res", // Name of the resource.
|
|
".", // Use current directory.
|
|
null);
|
|
}
|
|
catch (Exception)
|
|
{
|
|
//do nothing, will use static RM
|
|
}
|
|
|
|
serverName = info.GetString("Server");
|
|
nsName = info.GetString("NS");
|
|
query = info.GetString("Query");
|
|
|
|
state = SubscriptionState.Stopped;
|
|
|
|
rm = new ResourceManager( "Microsoft.VSDesigner.WMI.Res", // Name of the resource.
|
|
".", // Use current directory.
|
|
null);
|
|
|
|
//Get and cache wbemServices
|
|
ISWbemLocator wbemLocator = WmiHelper.WbemLocator;//(ISWbemLocator)(new SWbemLocator());
|
|
|
|
wbemServices = wbemLocator.ConnectServer(serverName,
|
|
nsName,
|
|
"", //user: blank defaults to current logged-on user
|
|
"", //password: blank defaults to current logged-on user
|
|
"", //locale: blank for current locale
|
|
"", //authority: NTLM or Kerberos. Blank lets DCOM negotiate.
|
|
0, //flags: reserved
|
|
null); //context info: not needed here
|
|
|
|
|
|
browseObject = new IntrinsicEventQueryComponent(serverName, nsName, query, this);
|
|
|
|
|
|
}
|
|
catch (Exception exc) {
|
|
MessageBox.Show(WMISys.GetString("WMISE_Exception", exc.Message, exc.StackTrace));
|
|
throw (exc);
|
|
}
|
|
}
|
|
|
|
|
|
//
|
|
// PROPERTIES
|
|
//
|
|
|
|
// <doc>
|
|
// <desc>
|
|
// Returns icon for this node.
|
|
// </desc>
|
|
// </doc>
|
|
|
|
public Image Icon {
|
|
override get {
|
|
//TODO: display different icons for associations, abstract, etc.
|
|
if (icon == null)
|
|
{
|
|
|
|
if (state == SubscriptionState.Started)
|
|
{
|
|
if (rm != null)
|
|
{
|
|
icon = (Image)rm.GetObject ("Microsoft.VSDesigner.WMI.Start.ico");
|
|
}
|
|
else
|
|
{
|
|
icon = new Icon(GetType(), "Start.ico").ToBitmap();
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (state == SubscriptionState.Stopped)
|
|
{
|
|
if (rm != null)
|
|
{
|
|
icon = (Image)rm.GetObject ("Microsoft.VSDesigner.WMI.Stop.ico");
|
|
}
|
|
else
|
|
{
|
|
icon = new Icon(GetType(), "Stop.ico").ToBitmap();
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (rm != null)
|
|
{
|
|
icon = (Image)rm.GetObject ("Microsoft.VSDesigner.WMI.ErrorNode.ico");
|
|
}
|
|
else
|
|
{
|
|
icon = new Icon(GetType(), "ErrorNode.ico").ToBitmap();
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
return icon;
|
|
|
|
}
|
|
}
|
|
|
|
|
|
/// <summary>
|
|
/// Context menu for a class object contains static method names
|
|
/// </summary>
|
|
public override ContextMenuItem[] GetContextMenuItems()
|
|
{
|
|
ContextMenuItem[] theMenu = new ContextMenuItem[4];
|
|
theMenu[0] = new ContextMenuItem(WMISys.GetString("WMISE_EventQueryStart"),
|
|
new EventHandler(OnStartReceiving),
|
|
(state != SubscriptionState.Started));
|
|
|
|
theMenu[1] = new ContextMenuItem(WMISys.GetString("WMISE_EventQueryStop"),
|
|
new EventHandler(OnStopReceiving),
|
|
(state == SubscriptionState.Started));
|
|
|
|
theMenu[2] = new ContextMenuItem(WMISys.GetString("WMISE_EventQueryPurge"),
|
|
new EventHandler(OnPurgeEvents));
|
|
//theMenu[3] = new ContextMenuItem("&Modify query...",
|
|
// new EventHandler(OnModifyQuery));
|
|
|
|
return theMenu;
|
|
}
|
|
|
|
public override bool CanDeleteNode()
|
|
{
|
|
return true;
|
|
}
|
|
|
|
public override bool ConfirmDeletingNode()
|
|
{
|
|
return true;
|
|
}
|
|
|
|
/*
|
|
/// <summary>
|
|
/// The method returns a NewChildNode that, when double clicked, triggers add filter.
|
|
/// </summary>
|
|
/// <returns>
|
|
/// A NewChildNode, a child node, that supports add server node.
|
|
/// </returns>
|
|
public override Node GetNewChildNode() {
|
|
|
|
if (newChildNode == null)
|
|
{
|
|
newChildNode = new NewChildNode();
|
|
newChildNode.Label = "<Create New Instance...>"; //VSDSys.GetString("SE_NewServerNodeLabel_AddServer");
|
|
newChildNode.WhenToSave = WhenToSaveCondition.Always;
|
|
newChildNode.SetIconImage((Image)rm.GetObject ("inst.bmp"));
|
|
newChildNode.DoubleClickHandler = new EventHandler(OnCreateNewInstance);
|
|
}
|
|
return newChildNode;
|
|
}
|
|
*/
|
|
// <doc>
|
|
// <desc>
|
|
// Returns label constant for this node.
|
|
// </desc>
|
|
// </doc>
|
|
public string Label {
|
|
override get {
|
|
if (label == null || label.Length == 0) {
|
|
label = query;
|
|
}
|
|
return label;
|
|
}
|
|
override set {
|
|
}
|
|
}
|
|
|
|
//
|
|
// METHODS
|
|
//
|
|
|
|
// <doc>
|
|
// <desc>
|
|
// Create instance nodes under this node.
|
|
// </desc>
|
|
// </doc>
|
|
public override Node[] CreateChildren() {
|
|
|
|
try
|
|
{
|
|
|
|
Node[] nodes = new Node[0];
|
|
return nodes;
|
|
|
|
}
|
|
catch(Exception exc)
|
|
{
|
|
MessageBox.Show(WMISys.GetString("WMISE_Exception", exc.Message, exc.StackTrace));
|
|
return null;
|
|
}
|
|
|
|
}
|
|
|
|
// <doc>
|
|
// <desc>
|
|
// This node is a singleton.
|
|
// </desc>
|
|
// </doc>
|
|
public override int CompareUnique(Node node) {
|
|
if (node is IntrinsicEventQueryNode &&
|
|
((IntrinsicEventQueryNode)node).serverName == this.serverName &&
|
|
((IntrinsicEventQueryNode)node).nsName == this.nsName &&
|
|
((IntrinsicEventQueryNode)node).query == this.query)
|
|
{
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
|
|
/// <summary>
|
|
/// Get component to be displayed in PB window
|
|
/// </summary>
|
|
public override Object GetBrowseComponent()
|
|
{
|
|
return browseObject;
|
|
}
|
|
|
|
|
|
/// <summary>
|
|
/// Subscribes for event notifications
|
|
/// </summary>
|
|
/// <param name="sender"> </param>
|
|
/// <param name="e"> </param>
|
|
private void OnStartReceiving(object sender, EventArgs e)
|
|
{
|
|
try
|
|
{
|
|
//TODO: enter subscription, change icon, disable "start" context menu
|
|
if (theSink != null)
|
|
{
|
|
theSink.Cancel();
|
|
theSink = null;
|
|
}
|
|
|
|
theSink = new SWbemSink();
|
|
theSink.OnObjectReady += new ISWbemSinkEvents_OnObjectReadyEventHandler(OnEventReady);
|
|
theSink.OnProgress += new ISWbemSinkEvents_OnProgressEventHandler(OnQueryProgress);
|
|
|
|
|
|
//MessageBox.Show("About to enter notification query: " + query);
|
|
|
|
wbemServices.ExecNotificationQueryAsync(theSink, query, "WQL",
|
|
(int)WbemFlagEnum.wbemFlagSendStatus,
|
|
null, null);
|
|
|
|
|
|
state = SubscriptionState.Started;
|
|
icon = (Image)rm.GetObject("Microsoft.VSDesigner.WMI.Start.ico");
|
|
this.GetNodeSite().UpdateIcon();
|
|
|
|
//GetNodeSite().AddChild(childNode);
|
|
|
|
}
|
|
catch(Exception exc)
|
|
{
|
|
state = SubscriptionState.Error;
|
|
icon = (Image)rm.GetObject("Microsoft.VSDesigner.WMI.ErrorNode.ico");
|
|
this.GetNodeSite().UpdateIcon();
|
|
MessageBox.Show(WMISys.GetString("WMISE_Exception", exc.Message, exc.StackTrace));
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Un-subscribes from event notifications
|
|
/// </summary>
|
|
/// <param name="sender"> </param>
|
|
/// <param name="e"> </param>
|
|
private void OnStopReceiving(object sender, EventArgs e)
|
|
{
|
|
try
|
|
{
|
|
//TODO: cancel subscription, change icon, disable "stop" context menu
|
|
if (theSink != null)
|
|
{
|
|
theSink.Cancel();
|
|
}
|
|
|
|
state = SubscriptionState.Stopped;
|
|
icon = (Image)rm.GetObject("Microsoft.VSDesigner.WMI.Stop.ico");
|
|
this.GetNodeSite().UpdateIcon();
|
|
|
|
}
|
|
catch(Exception exc)
|
|
{
|
|
MessageBox.Show(WMISys.GetString("WMISE_Exception", exc.Message, exc.StackTrace));
|
|
}
|
|
|
|
}
|
|
|
|
/// <summary>
|
|
/// Removes existing event entries
|
|
/// </summary>
|
|
/// <param name="sender"> </param>
|
|
/// <param name="e"> </param>
|
|
private void OnPurgeEvents(object sender, EventArgs e)
|
|
{
|
|
try
|
|
{
|
|
//TODO: purge event entries
|
|
|
|
}
|
|
catch(Exception exc)
|
|
{
|
|
MessageBox.Show(WMISys.GetString("WMISE_Exception", exc.Message, exc.StackTrace));
|
|
}
|
|
|
|
}
|
|
|
|
|
|
/// <summary>
|
|
/// To determine whether this node supports drag and drop.
|
|
/// </summary>
|
|
/// <returns>
|
|
/// TRUE to indicate ProcessNode supports drag and drop.
|
|
/// </returns>
|
|
public override bool HasDragComponents() {
|
|
return false; //change this!
|
|
}
|
|
|
|
public override IComponent[] CreateDragComponents(IDesignerHost designerHost) {
|
|
/*
|
|
Object comp = null;
|
|
|
|
if (pathNoServer.ToLower() == "root\\cimv2:win32_share")
|
|
{
|
|
comp = new Win32_Share (wmiObj);
|
|
}
|
|
else
|
|
{
|
|
comp = new WMIObjectComponent(wmiObj);
|
|
}
|
|
|
|
return new IComponent[] {(IComponent)comp};
|
|
*/
|
|
|
|
return new IComponent[0];
|
|
}
|
|
|
|
/// <summary>
|
|
/// This allows modification of the query string from outside,
|
|
/// e.g. from the browse component
|
|
/// </summary>
|
|
public String Query
|
|
{
|
|
get
|
|
{
|
|
return query;
|
|
}
|
|
set
|
|
{
|
|
if (state == SubscriptionState.Started)
|
|
{
|
|
OnStopReceiving(null, null);
|
|
}
|
|
|
|
query = value;
|
|
if (browseObject.Query != query)
|
|
{
|
|
browseObject.Query = query;
|
|
}
|
|
|
|
label = query;
|
|
|
|
GetNodeSite().StartRefresh();
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// The object should write the serialization info.
|
|
/// </summary>
|
|
public virtual void GetObjectData(SerializationInfo si, StreamingContext context) {
|
|
si.AddValue("Server", serverName);
|
|
si.AddValue("NS", nsName);
|
|
si.AddValue("Query", query);
|
|
}
|
|
|
|
private static void OnEventReady(ISWbemObject obj,
|
|
ISWbemNamedValueSet context)
|
|
{
|
|
MessageBox.Show("Event received: " + obj.GetObjectText_(0));
|
|
}
|
|
|
|
private static void OnQueryProgress(int numTasks,
|
|
int curTask,
|
|
string message,
|
|
ISWbemNamedValueSet context)
|
|
{
|
|
MessageBox.Show("OnQueryProgress received: " + message);
|
|
}
|
|
|
|
}
|
|
}
|