MemCacheDManager/WmiHelper.cs at master · BackTrak/MemCacheDManager · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
#region --- Libraries ---
using System;
using System.Collections.Generic;
using System.Text;
using System.Management;
using System.Net;
using System.Text.RegularExpressions;
#endregion

//From: http://blog.jrboelens.com/?p=8
namespace MemCacheDManager
{
	public class WmiHelper
	{

		#region Connect
		public static ManagementScope Connect(string machineName)
		{
			return Connect(machineName, null, null);
		}

		public static ManagementScope Connect(string machineName, string userNameAndDomainName, string password)
		{
			ConnectionOptions options = new ConnectionOptions();

			if (userNameAndDomainName != null && password != null)
			{
				string userName;
				string domainName;

				if (userNameAndDomainName.Contains("\\") == true)
				{
					domainName = userNameAndDomainName.Split('\\')[0];
					userName = userNameAndDomainName.Split('\\')[1];
				}
				else
				{
					domainName = null;
					userName = userNameAndDomainName;
				}

				if (NetworkHelper.IsMachineNameLocal(machineName) == false)
				{
					options.Authentication = AuthenticationLevel.PacketPrivacy;
					options.Username = userName;
					options.Password = password;

					if(domainName != null)
						options.Authority = "ntlmdomain:" + domainName;
				}
			}

			string path = "\\\\{0}\\root\\cimv2";
			path = String.Format(path, machineName);
			ManagementScope scope = new ManagementScope(path, options);
			scope.Connect();
			return scope;
		}
		#endregion

		#region GetInstanceByName
		private static ManagementObject GetInstanceByName(string machineName, string className, string name)
		{
			return GetInstanceByName(machineName, null, null, className, name);
		}

		private static ManagementObject GetInstanceByName(string machineName, string userName, string password, string className, string name)
		{
			ManagementScope scope = Connect(machineName, userName, password);
			ObjectQuery query = new ObjectQuery("SELECT * FROM " + className + " WHERE Name = '" + name + "'");
			ManagementObjectSearcher searcher = new ManagementObjectSearcher(scope, query);
			ManagementObjectCollection results = searcher.Get();
			foreach (ManagementObject manObject in results)
				return manObject;

			return null;
		}
		#endregion

		#region GetStaticByName
		private static ManagementClass GetStaticByName(string machineName, string className)
		{
			return GetStaticByName(machineName, null, null, className);
		}

		private static ManagementClass GetStaticByName(string machineName, string userName, string password, string className)
		{
			ManagementScope scope = Connect(machineName, userName, password);
			ObjectGetOptions getOptions = new ObjectGetOptions();
			ManagementPath path = new ManagementPath(className);
			ManagementClass manClass = new ManagementClass(scope, path, getOptions);
			return manClass;
		}
		#endregion

		#region InvokeInstanceMethod
		public static int InvokeInstanceMethod(string machineName, string className, string name, string methodName)
		{
			return InvokeInstanceMethod(machineName, className, name, methodName, null);
		}

		public static int InvokeInstanceMethod(string machineName, string className, string name, string methodName, object[] parameters)
		{
			return InvokeInstanceMethod(machineName, null, null, className, name, methodName, parameters);
		}

		public static int InvokeInstanceMethod(string machineName, string userName, string password, string className, string name, string methodName, object[] parameters)
		{
			try
			{
				ManagementObject manObject = GetInstanceByName(machineName, userName, password, className, name);
				object result = manObject.InvokeMethod(methodName, parameters);
				return Convert.ToInt32(result);
			}
			catch
			{
				return -1;
			}
		}

		#endregion

		#region InvokeStaticMethod
		public static int InvokeStaticMethod(string machineName, string className, string methodName)
		{
			return InvokeStaticMethod(machineName, className, methodName, null);
		}
		public static int InvokeStaticMethod(string machineName, string className, string methodName, object[] parameters)
		{
			return InvokeStaticMethod(machineName, null, null, className, methodName, parameters);
		}

		public static int InvokeStaticMethod(string machineName, string userName, string password, string className, string methodName, object[] parameters)
		{
			try
			{
				ManagementClass manClass = GetStaticByName(machineName, userName, password, className);
				object result = manClass.InvokeMethod(methodName, parameters);
				return Convert.ToInt32(result);
			}
			catch
			{
				return -1;
			}
		}
		#endregion
	}
}