306 lines
12 KiB
C#
306 lines
12 KiB
C#
// Copyright (c) 2011, Outercurve Foundation.
|
|
// 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.
|
|
//
|
|
// - Neither the name of the Outercurve Foundation nor the names of its
|
|
// contributors may be used to endorse or promote products derived from this
|
|
// software without specific prior written permission.
|
|
//
|
|
// 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.Text;
|
|
using System.Management;
|
|
using System.Diagnostics;
|
|
|
|
namespace WebsitePanel.HyperV.Utils
|
|
{
|
|
internal class Wmi
|
|
{
|
|
string nameSpace = null;
|
|
string computerName = null;
|
|
ManagementScope scope = null;
|
|
|
|
public Wmi(string nameSpace)
|
|
: this(nameSpace, null)
|
|
{
|
|
}
|
|
|
|
public Wmi(string computerName, string nameSpace)
|
|
{
|
|
this.nameSpace = nameSpace;
|
|
this.computerName = computerName;
|
|
}
|
|
|
|
internal ManagementObjectCollection ExecuteWmiQuery(string query, params object[] args)
|
|
{
|
|
if (args != null && args.Length > 0)
|
|
query = String.Format(query, args);
|
|
|
|
ManagementObjectSearcher searcher = new ManagementObjectSearcher(GetScope(),
|
|
new ObjectQuery(query));
|
|
return searcher.Get();
|
|
}
|
|
|
|
internal ManagementObject GetWmiObject(string className, string filter, params object[] args)
|
|
{
|
|
ManagementObjectCollection col = GetWmiObjects(className, filter, args);
|
|
ManagementObjectCollection.ManagementObjectEnumerator enumerator = col.GetEnumerator();
|
|
return enumerator.MoveNext() ? (ManagementObject)enumerator.Current : null;
|
|
}
|
|
|
|
internal ManagementObject GetWmiObject(string className)
|
|
{
|
|
return GetWmiObject(className, null);
|
|
}
|
|
|
|
internal ManagementObjectCollection GetWmiObjects(string className, string filter, params object[] args)
|
|
{
|
|
string query = "select * from " + className;
|
|
if (!String.IsNullOrEmpty(filter))
|
|
query += " where " + filter;
|
|
return ExecuteWmiQuery(query, args);
|
|
}
|
|
|
|
internal ManagementObjectCollection GetWmiObjects(string className)
|
|
{
|
|
return GetWmiObjects(className, null);
|
|
}
|
|
|
|
internal ManagementObject GetWmiObjectByPath(string path)
|
|
{
|
|
return new ManagementObject(GetScope(), new ManagementPath(path), null);
|
|
}
|
|
|
|
internal ManagementClass GetWmiClass(string className)
|
|
{
|
|
return new ManagementClass(GetScope(), new ManagementPath(className), null);
|
|
}
|
|
|
|
internal ManagementObject GetRelatedWmiObject(ManagementObject obj, string className)
|
|
{
|
|
ManagementObjectCollection col = obj.GetRelated(className);
|
|
ManagementObjectCollection.ManagementObjectEnumerator enumerator = col.GetEnumerator();
|
|
enumerator.MoveNext();
|
|
return (ManagementObject)enumerator.Current;
|
|
}
|
|
|
|
internal void Dump(ManagementBaseObject obj)
|
|
{
|
|
#if DEBUG
|
|
foreach (PropertyData prop in obj.Properties)
|
|
{
|
|
string typeName = prop.Value == null ? "null" : prop.Value.GetType().ToString();
|
|
Debug.WriteLine(prop.Name + ": " + prop.Value + " (" + typeName + ")");
|
|
}
|
|
#endif
|
|
}
|
|
|
|
// Converts a given datetime in DMTF format to System.DateTime object.
|
|
internal System.DateTime ToDateTime(string dmtfDate)
|
|
{
|
|
System.DateTime initializer = System.DateTime.MinValue;
|
|
int year = initializer.Year;
|
|
int month = initializer.Month;
|
|
int day = initializer.Day;
|
|
int hour = initializer.Hour;
|
|
int minute = initializer.Minute;
|
|
int second = initializer.Second;
|
|
long ticks = 0;
|
|
string dmtf = dmtfDate;
|
|
System.DateTime datetime = System.DateTime.MinValue;
|
|
string tempString = string.Empty;
|
|
if (String.IsNullOrEmpty(dmtf))
|
|
{
|
|
return DateTime.MinValue;
|
|
}
|
|
else if ((dmtf.Length != 25))
|
|
{
|
|
throw new System.ArgumentOutOfRangeException();
|
|
}
|
|
try
|
|
{
|
|
tempString = dmtf.Substring(0, 4);
|
|
if (("****" != tempString))
|
|
{
|
|
year = int.Parse(tempString);
|
|
}
|
|
tempString = dmtf.Substring(4, 2);
|
|
if (("**" != tempString))
|
|
{
|
|
month = int.Parse(tempString);
|
|
}
|
|
tempString = dmtf.Substring(6, 2);
|
|
if (("**" != tempString))
|
|
{
|
|
day = int.Parse(tempString);
|
|
}
|
|
tempString = dmtf.Substring(8, 2);
|
|
if (("**" != tempString))
|
|
{
|
|
hour = int.Parse(tempString);
|
|
}
|
|
tempString = dmtf.Substring(10, 2);
|
|
if (("**" != tempString))
|
|
{
|
|
minute = int.Parse(tempString);
|
|
}
|
|
tempString = dmtf.Substring(12, 2);
|
|
if (("**" != tempString))
|
|
{
|
|
second = int.Parse(tempString);
|
|
}
|
|
tempString = dmtf.Substring(15, 6);
|
|
if (("******" != tempString))
|
|
{
|
|
ticks = (long.Parse(tempString) * ((long)((System.TimeSpan.TicksPerMillisecond / 1000))));
|
|
}
|
|
if (((((((((year < 0)
|
|
|| (month < 0))
|
|
|| (day < 0))
|
|
|| (hour < 0))
|
|
|| (minute < 0))
|
|
|| (minute < 0))
|
|
|| (second < 0))
|
|
|| (ticks < 0)))
|
|
{
|
|
throw new System.ArgumentOutOfRangeException();
|
|
}
|
|
}
|
|
catch (System.Exception e)
|
|
{
|
|
throw new System.ArgumentOutOfRangeException(null, e.Message);
|
|
}
|
|
|
|
if (year == 0
|
|
&& month == 0
|
|
&& day == 0
|
|
&& hour == 0
|
|
&& minute == 0
|
|
&& second == 0
|
|
&& ticks == 0)
|
|
return DateTime.MinValue;
|
|
|
|
datetime = new System.DateTime(year, month, day, hour, minute, second, 0);
|
|
datetime = datetime.AddTicks(ticks);
|
|
System.TimeSpan tickOffset = System.TimeZone.CurrentTimeZone.GetUtcOffset(datetime);
|
|
int UTCOffset = 0;
|
|
int OffsetToBeAdjusted = 0;
|
|
long OffsetMins = ((long)((tickOffset.Ticks / System.TimeSpan.TicksPerMinute)));
|
|
tempString = dmtf.Substring(22, 3);
|
|
if ((tempString != "******"))
|
|
{
|
|
tempString = dmtf.Substring(21, 4);
|
|
try
|
|
{
|
|
UTCOffset = int.Parse(tempString);
|
|
}
|
|
catch (System.Exception e)
|
|
{
|
|
throw new System.ArgumentOutOfRangeException(null, e.Message);
|
|
}
|
|
OffsetToBeAdjusted = ((int)((OffsetMins - UTCOffset)));
|
|
datetime = datetime.AddMinutes(((double)(OffsetToBeAdjusted)));
|
|
}
|
|
return datetime;
|
|
}
|
|
|
|
// Converts a given System.DateTime object to DMTF datetime format.
|
|
internal string ToDmtfDateTime(System.DateTime date)
|
|
{
|
|
string utcString = string.Empty;
|
|
System.TimeSpan tickOffset = System.TimeZone.CurrentTimeZone.GetUtcOffset(date);
|
|
long OffsetMins = ((long)((tickOffset.Ticks / System.TimeSpan.TicksPerMinute)));
|
|
if ((System.Math.Abs(OffsetMins) > 999))
|
|
{
|
|
date = date.ToUniversalTime();
|
|
utcString = "+000";
|
|
}
|
|
else
|
|
{
|
|
if ((tickOffset.Ticks >= 0))
|
|
{
|
|
utcString = string.Concat("+", ((System.Int64)((tickOffset.Ticks / System.TimeSpan.TicksPerMinute))).ToString().PadLeft(3, '0'));
|
|
}
|
|
else
|
|
{
|
|
string strTemp = ((System.Int64)(OffsetMins)).ToString();
|
|
utcString = string.Concat("-", strTemp.Substring(1, (strTemp.Length - 1)).PadLeft(3, '0'));
|
|
}
|
|
}
|
|
string dmtfDateTime = ((System.Int32)(date.Year)).ToString().PadLeft(4, '0');
|
|
dmtfDateTime = string.Concat(dmtfDateTime, ((System.Int32)(date.Month)).ToString().PadLeft(2, '0'));
|
|
dmtfDateTime = string.Concat(dmtfDateTime, ((System.Int32)(date.Day)).ToString().PadLeft(2, '0'));
|
|
dmtfDateTime = string.Concat(dmtfDateTime, ((System.Int32)(date.Hour)).ToString().PadLeft(2, '0'));
|
|
dmtfDateTime = string.Concat(dmtfDateTime, ((System.Int32)(date.Minute)).ToString().PadLeft(2, '0'));
|
|
dmtfDateTime = string.Concat(dmtfDateTime, ((System.Int32)(date.Second)).ToString().PadLeft(2, '0'));
|
|
dmtfDateTime = string.Concat(dmtfDateTime, ".");
|
|
System.DateTime dtTemp = new System.DateTime(date.Year, date.Month, date.Day, date.Hour, date.Minute, date.Second, 0);
|
|
long microsec = ((long)((((date.Ticks - dtTemp.Ticks)
|
|
* 1000)
|
|
/ System.TimeSpan.TicksPerMillisecond)));
|
|
string strMicrosec = ((System.Int64)(microsec)).ToString();
|
|
if ((strMicrosec.Length > 6))
|
|
{
|
|
strMicrosec = strMicrosec.Substring(0, 6);
|
|
}
|
|
dmtfDateTime = string.Concat(dmtfDateTime, strMicrosec.PadLeft(6, '0'));
|
|
dmtfDateTime = string.Concat(dmtfDateTime, utcString);
|
|
return dmtfDateTime;
|
|
}
|
|
|
|
public ManagementScope GetScope()
|
|
{
|
|
if (scope != null)
|
|
return scope;
|
|
|
|
// create new scope
|
|
if (String.IsNullOrEmpty(computerName))
|
|
{
|
|
// local
|
|
scope = new ManagementScope(nameSpace);
|
|
}
|
|
else
|
|
{
|
|
// remote
|
|
ConnectionOptions options = new ConnectionOptions();
|
|
|
|
string path = String.Format(@"\\{0}\{1}", computerName, nameSpace);
|
|
scope = new ManagementScope(path, options);
|
|
}
|
|
|
|
// connect
|
|
try
|
|
{
|
|
scope.Connect();
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
string computer = "";
|
|
if (!String.IsNullOrEmpty(computerName))
|
|
computer = " on computer \"" + computerName + "\"";
|
|
throw new Exception("Hyper-V service could not be connected" + computer, ex);
|
|
}
|
|
return scope;
|
|
}
|
|
}
|
|
}
|