} // END ArbitraryNameSpace FUNC
static void BuildEventConsumer()
{
try
{
// Derive base class
ManagementClass eventConsumerBase = new ManagementClass("Root\\Win32", "__EventConsumer", null);
ManagementClass newActiveScriptEventConsumer = eventConsumerBase.Derive("ActiveScriptEventConsumer");
// Add properties to turn base class into facimale of ActiveScriptEventConsumer class
newActiveScriptEventConsumer.Properties.Add("Name", CimType.String, false);
newActiveScriptEventConsumer.Properties["Name"].Qualifiers.Add("Key", true, false, true, true, false);
newActiveScriptEventConsumer.Properties.Add("ScriptingEngine", CimType.String, false);
newActiveScriptEventConsumer.Properties["ScriptingEngine"].Qualifiers.Add("not_null", true, false, false, false, true);
newActiveScriptEventConsumer.Properties["ScriptingEngine"].Qualifiers.Add("write", true, false, false, false, true);
newActiveScriptEventConsumer.Properties.Add("ScriptText", CimType.String, false);
newActiveScriptEventConsumer.Properties["ScriptText"].Qualifiers.Add("write", true, false, false, false, true);
newActiveScriptEventConsumer.Properties.Add("ScriptFilename", CimType.String, false);
newActiveScriptEventConsumer.Properties["ScriptFilename"].Qualifiers.Add("write", true, false, false, false, true);
newActiveScriptEventConsumer.Properties.Add("KillTimeout", CimType.UInt32, false);
newActiveScriptEventConsumer.Properties["KillTimeout"].Qualifiers.Add("write", true, false, false, false, true);
newActiveScriptEventConsumer.Put();
Console.WriteLine("[*] ActiveScriptEventConsumer created!");
// The new class needs to be bound to a provider and registed
// First we bind the class to a provider
ManagementScope scope = new ManagementScope(@"\\.\root\Win32");
ManagementObject newActiveScriptEventConsumerProviderBinding = null;
ManagementObject eventConsumerProviderRegistration = null;
newActiveScriptEventConsumerProviderBinding = new ManagementClass(scope, new ManagementPath("__Win32Provider"), null).CreateInstance();
newActiveScriptEventConsumerProviderBinding["Name"] = "ActiveScriptEventConsumer";
newActiveScriptEventConsumerProviderBinding["Clsid"] = "{266c72e7-62e8-11d1-ad89-00c04fd8fdff}";
newActiveScriptEventConsumerProviderBinding["PerUserInitialization"] = true;
newActiveScriptEventConsumerProviderBinding["HostingModel"] = "SelfHost";
newActiveScriptEventConsumerProviderBinding.Put();
Console.WriteLine("[*] New provider binding creatd!");
// Then we register the binding
string[] className = { "ActiveScriptEventConsumer" };
eventConsumerProviderRegistration = new ManagementClass(scope, new ManagementPath("__EventConsumerProviderRegistration"), null).CreateInstance();
eventConsumerProviderRegistration["provider"] = newActiveScriptEventConsumerProviderBinding;
eventConsumerProviderRegistration["ConsumerClassNames"] = className;
eventConsumerProviderRegistration.Put();
Console.WriteLine("[*] Consumer registered with provider!");
}
catch (Exception e)
{
Console.WriteLine(e);
} // END CATCH
} // END BuildEventConsumer FUNC
public static void Main()
{
ManagementClass existingClass =
new ManagementClass("CIM_Service");
ManagementClass newClass = existingClass.Derive("My_Service");
newClass.Put(); //to commit the new class to the WMI repository.
}
ManagementClass MakeProviderClass()
{
ManagementClass baseClass = new ManagementClass(DecoupledProviderClassPath);
ManagementClass newClass = baseClass.Derive(ProviderClassName);
PropertyDataCollection props = newClass.Properties;
props.Add("Assembly", CimType.String, false);
return(newClass);
}
ManagementClass MakeNamingClass()
{
ManagementClass baseClass = new ManagementClass(GlobalInstrumentationClassPath);
ManagementClass newClass = baseClass.Derive(NamingClassName);
newClass.Qualifiers.Add("abstract", true);
PropertyDataCollection props = newClass.Properties;
props.Add("InstrumentedAssembliesClassName", InstrumentedAssembliesClassName, CimType.String);
return(newClass);
}
ManagementClass MakeDecoupledProviderClass()
{
ManagementClass baseClass = new ManagementClass(Win32ProviderClassPath);
ManagementClass newClass = baseClass.Derive(DecoupledProviderClassName);
PropertyDataCollection props = newClass.Properties;
props.Add("HostingModel", "Decoupled:Com", CimType.String);
props.Add("SecurityDescriptor", CimType.String, false);
props.Add("Version", 1, CimType.UInt32);
props["CLSID"].Value = DecoupledProviderCLSID;
return(newClass);
}
ManagementClass MakeNamespaceRegistrationClass()
{
ManagementClass baseClass = new ManagementClass(GlobalInstrumentationClassPath);
ManagementClass newClass = baseClass.Derive(InstrumentedNamespacesClassName);
PropertyDataCollection props = newClass.Properties;
props.Add("NamespaceName", CimType.String, false);
PropertyData prop = props["NamespaceName"];
prop.Qualifiers.Add("key", true);
return(newClass);
}
ManagementClass MakeRegistrationClass()
{
ManagementClass baseClass = new ManagementClass(InstrumentationClassPath);
ManagementClass newClass = baseClass.Derive(InstrumentedAssembliesClassName);
PropertyDataCollection props = newClass.Properties;
props.Add("Name", CimType.String, false);
PropertyData prop = props["Name"];
prop.Qualifiers.Add("key", true);
props.Add("RegisteredBuild", CimType.String, false);
props.Add("FullName", CimType.String, false);
props.Add("PathToAssembly", CimType.String, false);
props.Add("Code", CimType.String, false);
props.Add("Mof", CimType.String, false);
return(newClass);
}
public SchemaMapping(Type type, SchemaNaming naming, Hashtable mapTypeToConverterClassName)
{
codeClassName = (string)mapTypeToConverterClassName[type];
classType = type;
bool hasGenericEmbeddedObject = false;
string baseClassName = ManagedNameAttribute.GetBaseClassName(type);
className = ManagedNameAttribute.GetMemberName(type);
instrumentationType = InstrumentationClassAttribute.GetAttribute(type).InstrumentationType;
classPath = naming.NamespaceName + ":" + className;
if (null == baseClassName)
{
newClass = new ManagementClass(naming.NamespaceName, "", null);
newClass.SystemProperties ["__CLASS"].Value = className;
}
else
{
ManagementClass baseClass = new ManagementClass(naming.NamespaceName + ":" + baseClassName);
if (instrumentationType == InstrumentationType.Instance)
{
bool baseAbstract = false;
try
{
QualifierData o = baseClass.Qualifiers["abstract"];
if (o.Value is bool)
{
baseAbstract = (bool)o.Value;
}
}
catch (ManagementException e)
{
if (e.ErrorCode != ManagementStatus.NotFound)
{
throw;
}
}
if (!baseAbstract)
{
throw new Exception(RC.GetString("CLASSINST_EXCEPT"));
}
}
newClass = baseClass.Derive(className);
}
// Create the converter class
CodeWriter codeClass = code.AddChild("public class " + codeClassName + " : IWmiConverter");
// Create code block for one line Members
CodeWriter codeOneLineMembers = codeClass.AddChild(new CodeWriter());
codeOneLineMembers.Line("static ManagementClass managementClass = new ManagementClass(@\"" + classPath + "\");");
codeOneLineMembers.Line("static IntPtr classWbemObjectIP;");
codeOneLineMembers.Line("static Guid iidIWbemObjectAccess = new Guid(\"49353C9A-516B-11D1-AEA6-00C04FB68820\");");
codeOneLineMembers.Line("internal ManagementObject instance = managementClass.CreateInstance();");
codeOneLineMembers.Line("object reflectionInfoTempObj = null ; ");
codeOneLineMembers.Line("FieldInfo reflectionIWbemClassObjectField = null ; ");
codeOneLineMembers.Line("IntPtr emptyWbemObject = IntPtr.Zero ; ");
//
// Reuters VSQFE#: 750 [marioh] see comments above
// Used as a temporary pointer to the newly created instance that we create to avoid re-using the same
// object causing unbound memory usage in IWbemClassObject implementation.
codeOneLineMembers.Line("IntPtr theClone = IntPtr.Zero;");
codeOneLineMembers.Line("public static ManagementObject emptyInstance = managementClass.CreateInstance();");
// TODO: Make these non-public
codeOneLineMembers.Line("public IntPtr instWbemObjectAccessIP;");
// Create static constructor to initialize handles
CodeWriter codeCCTOR = codeClass.AddChild("static " + codeClassName + "()");
codeCCTOR.Line("classWbemObjectIP = (IntPtr)managementClass;");
codeCCTOR.Line("IntPtr wbemObjectAccessIP;");
codeCCTOR.Line("Marshal.QueryInterface(classWbemObjectIP, ref iidIWbemObjectAccess, out wbemObjectAccessIP);");
codeCCTOR.Line("int cimType;");
// Create constructor
CodeWriter codeCTOR = codeClass.AddChild("public " + codeClassName + "()");
codeCTOR.Line("IntPtr wbemObjectIP = (IntPtr)instance;");
codeCTOR.Line("Marshal.QueryInterface(wbemObjectIP, ref iidIWbemObjectAccess, out instWbemObjectAccessIP);");
//
// Reuters VSQFE#: 750 [marioh]
// In the CCTOR we set things up only once:
// 1. We get the IWbemClassObjectFreeThreaded object '_wbemObject' from the ManagementObject instance
// 2. We then get the actual IntPtr to the underlying WMI object
// 3. Finally, the simple cast to IntPtr from the ManagementObject instance
// These fields will be used later during the ToWMI call.
codeCTOR.Line("FieldInfo tempField = instance.GetType().GetField ( \"_wbemObject\", BindingFlags.Instance | BindingFlags.NonPublic );");
codeCTOR.Line("if ( tempField == null )");
codeCTOR.Line("{");
codeCTOR.Line(" tempField = instance.GetType().GetField ( \"wbemObject\", BindingFlags.Instance | BindingFlags.NonPublic ) ;");
codeCTOR.Line("}");
codeCTOR.Line("reflectionInfoTempObj = tempField.GetValue (instance) ;");
codeCTOR.Line("reflectionIWbemClassObjectField = reflectionInfoTempObj.GetType().GetField (\"pWbemClassObject\", BindingFlags.Instance | BindingFlags.NonPublic );");
codeCTOR.Line("emptyWbemObject = (IntPtr) emptyInstance;");
// Create destructor that will be called at process cleanup
CodeWriter codeDTOR = codeClass.AddChild("~" + codeClassName + "()");
codeDTOR.AddChild("if(instWbemObjectAccessIP != IntPtr.Zero)").Line("Marshal.Release(instWbemObjectAccessIP);");
// Create method to convert from managed code to WMI
CodeWriter codeToWMI = codeClass.AddChild("public void ToWMI(object obj)");
//
// Reuters VSQFE#: 750 [marioh] see comments above
// Ensure the release of the WbemObjectAccess interface pointer.
codeToWMI.Line("if(instWbemObjectAccessIP != IntPtr.Zero)");
codeToWMI.Line("{");
codeToWMI.Line(" Marshal.Release(instWbemObjectAccessIP);");
codeToWMI.Line(" instWbemObjectAccessIP = IntPtr.Zero;");
codeToWMI.Line("}");
codeToWMI.Line("if(theClone != IntPtr.Zero)");
codeToWMI.Line("{");
codeToWMI.Line(" Marshal.Release(theClone);");
codeToWMI.Line(" theClone = IntPtr.Zero;");
codeToWMI.Line("}");
codeToWMI.Line("IWOA.Clone_f(12, emptyWbemObject, out theClone) ;");
codeToWMI.Line("Marshal.QueryInterface(theClone, ref iidIWbemObjectAccess, out instWbemObjectAccessIP) ;");
codeToWMI.Line("reflectionIWbemClassObjectField.SetValue ( reflectionInfoTempObj, theClone ) ;");
codeToWMI.Line(String.Format("{0} instNET = ({0})obj;", type.FullName.Replace('+', '.'))); // bug#92918 - watch for nested classes
// Explicit cast to IntPtr
CodeWriter codeIntPtrCast = codeClass.AddChild("public static explicit operator IntPtr(" + codeClassName + " obj)");
codeIntPtrCast.Line("return obj.instWbemObjectAccessIP;");
// Add GetInstance
codeOneLineMembers.Line("public ManagementObject GetInstance() {return instance;}");
PropertyDataCollection props = newClass.Properties;
// type specific info
switch (instrumentationType)
{
case InstrumentationType.Event:
break;
case InstrumentationType.Instance:
props.Add("ProcessId", CimType.String, false);
props.Add("InstanceId", CimType.String, false);
props["ProcessId"].Qualifiers.Add("key", true);
props["InstanceId"].Qualifiers.Add("key", true);
newClass.Qualifiers.Add("dynamic", true, false, false, false, true);
newClass.Qualifiers.Add("provider", naming.DecoupledProviderInstanceName, false, false, false, true);
break;
case InstrumentationType.Abstract:
newClass.Qualifiers.Add("abstract", true, false, false, false, true);
break;
default:
break;
}
int propCount = 0;
bool needsNullObj = false;
foreach (MemberInfo field in type.GetMembers())
{
if (!(field is FieldInfo || field is PropertyInfo))
{
continue;
}
if (field.GetCustomAttributes(typeof(IgnoreMemberAttribute), false).Length > 0)
{
continue;
}
if (field is FieldInfo)
{
FieldInfo fi = field as FieldInfo;
// We ignore statics
if (fi.IsStatic)
{
ThrowUnsupportedMember(field);
}
}
else if (field is PropertyInfo)
{
PropertyInfo pi = field as PropertyInfo;
// We must have a 'get' property accessor
if (!pi.CanRead)
{
ThrowUnsupportedMember(field);
}
// We ignore static properties
MethodInfo mi = pi.GetGetMethod();
if (null == mi || mi.IsStatic)
{
ThrowUnsupportedMember(field);
}
// We don't support parameters on properties
if (mi.GetParameters().Length > 0)
{
ThrowUnsupportedMember(field);
}
}
String propName = ManagedNameAttribute.GetMemberName(field);
#if SUPPORTS_ALTERNATE_WMI_PROPERTY_TYPE
Type t2 = ManagedTypeAttribute.GetManagedType(field);
#else
Type t2;
if (field is FieldInfo)
{
t2 = (field as FieldInfo).FieldType;
}
else
{
t2 = (field as PropertyInfo).PropertyType;
}
#endif
bool isArray = false;
if (t2.IsArray)
{
// We only support one dimensional arrays in this version
if (t2.GetArrayRank() != 1)
{
ThrowUnsupportedMember(field);
}
isArray = true;
t2 = t2.GetElementType();
}
string embeddedTypeName = null;
string embeddedConverterName = null;
if (mapTypeToConverterClassName.Contains(t2))
{
embeddedConverterName = (string)mapTypeToConverterClassName[t2];
embeddedTypeName = ManagedNameAttribute.GetMemberName(t2);
}
bool isGenericEmbeddedObject = false;
if (t2 == typeof(object))
{
isGenericEmbeddedObject = true;
if (hasGenericEmbeddedObject == false)
{
hasGenericEmbeddedObject = true;
// Add map
codeOneLineMembers.Line("static Hashtable mapTypeToConverter = new Hashtable();");
foreach (DictionaryEntry entry in mapTypeToConverterClassName)
{
codeCCTOR.Line(String.Format("mapTypeToConverter[typeof({0})] = typeof({1});", ((Type)entry.Key).FullName.Replace('+', '.'), (string)entry.Value)); // bug#92918 - watch for nested classes
}
}
}
string propFieldName = "prop_" + (propCount);
string handleFieldName = "handle_" + (propCount++);
// Add handle for field, which is static accross all instances
codeOneLineMembers.Line("static int " + handleFieldName + ";");
codeCCTOR.Line(String.Format("IWOA.GetPropertyHandle_f27(27, wbemObjectAccessIP, \"{0}\", out cimType, out {1});", propName, handleFieldName));
// Add PropertyData for field, which is specific to each instance
codeOneLineMembers.Line("PropertyData " + propFieldName + ";");
codeCTOR.Line(String.Format("{0} = instance.Properties[\"{1}\"];", propFieldName, propName));
if (isGenericEmbeddedObject)
{
CodeWriter codeNotNull = codeToWMI.AddChild(String.Format("if(instNET.{0} != null)", field.Name));
CodeWriter codeElse = codeToWMI.AddChild("else");
codeElse.Line(String.Format("{0}.Value = null;", propFieldName));
if (isArray)
{
codeNotNull.Line(String.Format("int len = instNET.{0}.Length;", field.Name));
codeNotNull.Line("ManagementObject[] embeddedObjects = new ManagementObject[len];");
codeNotNull.Line("IWmiConverter[] embeddedConverters = new IWmiConverter[len];");
CodeWriter codeForLoop = codeNotNull.AddChild("for(int i=0;i<len;i++)");
CodeWriter codeFoundType = codeForLoop.AddChild(String.Format("if((instNET.{0}[i] != null) && mapTypeToConverter.Contains(instNET.{0}[i].GetType()))", field.Name));
codeFoundType.Line(String.Format("Type type = (Type)mapTypeToConverter[instNET.{0}[i].GetType()];", field.Name));
codeFoundType.Line("embeddedConverters[i] = (IWmiConverter)Activator.CreateInstance(type);");
codeFoundType.Line(String.Format("embeddedConverters[i].ToWMI(instNET.{0}[i]);", field.Name));
codeFoundType.Line("embeddedObjects[i] = embeddedConverters[i].GetInstance();");
codeForLoop.AddChild("else").Line(String.Format("embeddedObjects[i] = SafeAssign.GetManagementObject(instNET.{0}[i]);", field.Name));
codeNotNull.Line(String.Format("{0}.Value = embeddedObjects;", propFieldName));
}
else
{
CodeWriter codeFoundType = codeNotNull.AddChild(String.Format("if(mapTypeToConverter.Contains(instNET.{0}.GetType()))", field.Name));
codeFoundType.Line(String.Format("Type type = (Type)mapTypeToConverter[instNET.{0}.GetType()];", field.Name));
codeFoundType.Line("IWmiConverter converter = (IWmiConverter)Activator.CreateInstance(type);");
codeFoundType.Line(String.Format("converter.ToWMI(instNET.{0});", field.Name));
codeFoundType.Line(String.Format("{0}.Value = converter.GetInstance();", propFieldName));
codeNotNull.AddChild("else").Line(String.Format("{0}.Value = SafeAssign.GetInstance(instNET.{1});", propFieldName, field.Name));
}
}
else if (embeddedTypeName != null)
{
// If this is an embedded struct, it cannot be null
CodeWriter codeNotNull;
if (t2.IsValueType)
{
codeNotNull = codeToWMI;
}
else
{
codeNotNull = codeToWMI.AddChild(String.Format("if(instNET.{0} != null)", field.Name));
CodeWriter codeElse = codeToWMI.AddChild("else");
codeElse.Line(String.Format("{0}.Value = null;", propFieldName));
}
if (isArray)
{
codeNotNull.Line(String.Format("int len = instNET.{0}.Length;", field.Name));
codeNotNull.Line("ManagementObject[] embeddedObjects = new ManagementObject[len];");
codeNotNull.Line(String.Format("{0}[] embeddedConverters = new {0}[len];", embeddedConverterName));
CodeWriter codeForLoop = codeNotNull.AddChild("for(int i=0;i<len;i++)");
codeForLoop.Line(String.Format("embeddedConverters[i] = new {0}();", embeddedConverterName));
// If this is a struct array, the elements are never null
if (t2.IsValueType)
{
codeForLoop.Line(String.Format("embeddedConverters[i].ToWMI(instNET.{0}[i]);", field.Name));
}
else
{
CodeWriter codeArrayElementNotNull = codeForLoop.AddChild(String.Format("if(instNET.{0}[i] != null)", field.Name));
codeArrayElementNotNull.Line(String.Format("embeddedConverters[i].ToWMI(instNET.{0}[i]);", field.Name));
}
codeForLoop.Line("embeddedObjects[i] = embeddedConverters[i].instance;");
codeNotNull.Line(String.Format("{0}.Value = embeddedObjects;", propFieldName));
}
else
{
// We cannot create an instance of 'embeddedConverterName' because it may be the
// same type as we are defining (in other words, a cyclic loop, such as class XXX
// having an instance of an XXX as a member). To prevent an infinite loop of constructing
// converter classes, we create a 'lazy' variable that is initialized to NULL, and the first
// time it is used, we set it to a 'new embeddedConverterName'.
codeOneLineMembers.Line(String.Format("{0} lazy_embeddedConverter_{1} = null;", embeddedConverterName, propFieldName));
CodeWriter codeConverterProp = codeClass.AddChild(String.Format("{0} embeddedConverter_{1}", embeddedConverterName, propFieldName));
CodeWriter codeGet = codeConverterProp.AddChild("get");
CodeWriter codeIf = codeGet.AddChild(String.Format("if(null == lazy_embeddedConverter_{0})", propFieldName));
codeIf.Line(String.Format("lazy_embeddedConverter_{0} = new {1}();", propFieldName, embeddedConverterName));
codeGet.Line(String.Format("return lazy_embeddedConverter_{0};", propFieldName));
codeNotNull.Line(String.Format("embeddedConverter_{0}.ToWMI(instNET.{1});", propFieldName, field.Name));
codeNotNull.Line(String.Format("{0}.Value = embeddedConverter_{0}.instance;", propFieldName));
}
}
else if (!isArray)
{
if (t2 == typeof(Byte) || t2 == typeof(SByte))
{
codeToWMI.Line(String.Format("IWOA.WritePropertyValue_f28(28, instWbemObjectAccessIP, {0}, 1, ref instNET.{1});", handleFieldName, field.Name));
}
else if (t2 == typeof(Int16) || t2 == typeof(UInt16) || t2 == typeof(Char))
{
codeToWMI.Line(String.Format("IWOA.WritePropertyValue_f28(28, instWbemObjectAccessIP, {0}, 2, ref instNET.{1});", handleFieldName, field.Name));
}
else if (t2 == typeof(UInt32) || t2 == typeof(Int32) || t2 == typeof(Single))
{
codeToWMI.Line(String.Format("IWOA.WriteDWORD_f31(31, instWbemObjectAccessIP, {0}, instNET.{1});", handleFieldName, field.Name));
}
else if (t2 == typeof(UInt64) || t2 == typeof(Int64) || t2 == typeof(Double))
{
codeToWMI.Line(String.Format("IWOA.WriteQWORD_f33(33, instWbemObjectAccessIP, {0}, instNET.{1});", handleFieldName, field.Name));
}
else if (t2 == typeof(Boolean))
{
// TODO: Fix this to use IWOA
// codeToWMI.Line(String.Format("{0}.Value = instNET.{1};", propFieldName, field.Name));
codeToWMI.Line(String.Format("if(instNET.{0})", field.Name));
codeToWMI.Line(String.Format(" IWOA.WritePropertyValue_f28(28, instWbemObjectAccessIP, {0}, 2, ref SafeAssign.boolTrue);", handleFieldName));
codeToWMI.Line("else");
codeToWMI.Line(String.Format(" IWOA.WritePropertyValue_f28(28, instWbemObjectAccessIP, {0}, 2, ref SafeAssign.boolFalse);", handleFieldName));
}
else if (t2 == typeof(String))
{
CodeWriter codeQuickString = codeToWMI.AddChild(String.Format("if(null != instNET.{0})", field.Name));
codeQuickString.Line(String.Format("IWOA.WritePropertyValue_f28(28, instWbemObjectAccessIP, {0}, (instNET.{1}.Length+1)*2, instNET.{1});", handleFieldName, field.Name));
// codeToWMI.AddChild("else").Line(String.Format("{0}.Value = instNET.{1};", propFieldName, field.Name));
codeToWMI.AddChild("else").Line(String.Format("IWOA.Put_f5(5, instWbemObjectAccessIP, \"{0}\", 0, ref nullObj, 8);", propName));
if (needsNullObj == false)
{
needsNullObj = true;
// Bug#111623 - This line used to say 'nullObj = null;' When nullObj was passed
// to IWOA.Put, this did NOT set the value of a string variable to NULL. The correct
// thing to do was to pass a reference to DBNull.Value to IWOA.Put instead.
codeOneLineMembers.Line("object nullObj = DBNull.Value;");
}
}
else if (t2 == typeof(DateTime) || t2 == typeof(TimeSpan))
{
codeToWMI.Line(String.Format("IWOA.WritePropertyValue_f28(28, instWbemObjectAccessIP, {0}, 52, SafeAssign.WMITimeToString(instNET.{1}));", handleFieldName, field.Name));
// codeToWMI.Line(String.Format("{0}.Value = SafeAssign.DateTimeToString(instNET.{1});", propFieldName, field.Name));
}
else
{
codeToWMI.Line(String.Format("{0}.Value = instNET.{1};", propFieldName, field.Name));
}
}
else
{
// We have an array type
if (t2 == typeof(DateTime) || t2 == typeof(TimeSpan))
{
codeToWMI.AddChild(String.Format("if(null == instNET.{0})", field.Name)).Line(String.Format("{0}.Value = null;", propFieldName));
codeToWMI.AddChild("else").Line(String.Format("{0}.Value = SafeAssign.WMITimeArrayToStringArray(instNET.{1});", propFieldName, field.Name));
}
else
{
// This handles arrays of all primative types
codeToWMI.Line(String.Format("{0}.Value = instNET.{1};", propFieldName, field.Name));
}
}
CimType cimtype = CimType.String;
if (field.DeclaringType != type)
{
continue;
}
#if REQUIRES_EXPLICIT_DECLARATION_OF_INHERITED_PROPERTIES
if (InheritedPropertyAttribute.GetAttribute(field) != null)
{
continue;
}
#else
// See if this field already exists on the WMI class
// In other words, is it inherited from a base class
// TODO: Make this more efficient
// - If we have a null base class name, all property names
// should be new
// - We could get all base class property names into a
// hashtable, and look them up from there
bool propertyExists = true;
try
{
PropertyData prop = newClass.Properties[propName];
// HACK for bug#96863 - The above line used to throw a
// not found exception. This was changed with a recent
// checkin. If this functionality is not reverted, the
// following statement should force the necessary 'not
// found' exception that we are looking for.
CimType cimType = prop.Type;
// Make sure that if the property exists, it is inherited
// If it is local, they probably named two properties with
// the same name
if (prop.IsLocal)
{
throw new ArgumentException(String.Format(RC.GetString("MEMBERCONFLILCT_EXCEPT"), field.Name), field.Name);
}
}
catch (ManagementException e)
{
if (e.ErrorCode != ManagementStatus.NotFound)
{
throw;
}
else
{
propertyExists = false;
}
}
if (propertyExists)
{
continue;
}
#endif
if (embeddedTypeName != null)
{
cimtype = CimType.Object;
}
else if (isGenericEmbeddedObject)
{
cimtype = CimType.Object;
}
else if (t2 == typeof(ManagementObject))
{
cimtype = CimType.Object;
}
else if (t2 == typeof(SByte))
{
cimtype = CimType.SInt8;
}
else if (t2 == typeof(Byte))
{
cimtype = CimType.UInt8;
}
else if (t2 == typeof(Int16))
{
cimtype = CimType.SInt16;
}
else if (t2 == typeof(UInt16))
{
cimtype = CimType.UInt16;
}
else if (t2 == typeof(Int32))
{
cimtype = CimType.SInt32;
}
else if (t2 == typeof(UInt32))
{
cimtype = CimType.UInt32;
}
else if (t2 == typeof(Int64))
{
cimtype = CimType.SInt64;
}
else if (t2 == typeof(UInt64))
{
cimtype = CimType.UInt64;
}
else if (t2 == typeof(Single))
{
cimtype = CimType.Real32;
}
else if (t2 == typeof(Double))
{
cimtype = CimType.Real64;
}
else if (t2 == typeof(Boolean))
{
cimtype = CimType.Boolean;
}
else if (t2 == typeof(String))
{
cimtype = CimType.String;
}
else if (t2 == typeof(Char))
{
cimtype = CimType.Char16;
}
else if (t2 == typeof(DateTime))
{
cimtype = CimType.DateTime;
}
else if (t2 == typeof(TimeSpan))
{
cimtype = CimType.DateTime;
}
else
{
ThrowUnsupportedMember(field);
}
// HACK: The following line cause a strange System.InvalidProgramException when run through InstallUtil
// throw new Exception("Unsupported type for event member - " + t2.Name);
// TODO: if(t2 == typeof(Decimal))
#if SUPPORTS_WMI_DEFAULT_VAULES
Object defaultValue = ManagedDefaultValueAttribute.GetManagedDefaultValue(field);
// TODO: Is it safe to make this one line?
if (null == defaultValue)
{
props.Add(propName, cimtype, false);
}
else
{
props.Add(propName, defaultValue, cimtype);
}
#else
try
{
props.Add(propName, cimtype, isArray);
}
catch (ManagementException e)
{
ThrowUnsupportedMember(field, e);
}
#endif
// Must at 'interval' SubType on TimeSpans
if (t2 == typeof(TimeSpan))
{
PropertyData prop = props[propName];
prop.Qualifiers.Add("SubType", "interval", false, true, true, true);
}
if (embeddedTypeName != null)
{
PropertyData prop = props[propName];
prop.Qualifiers["CIMTYPE"].Value = "object:" + embeddedTypeName;
}
}
codeCCTOR.Line("Marshal.Release(wbemObjectAccessIP);");
// codeToWMI.Line("Console.WriteLine(instance.GetText(TextFormat.Mof));");
}
public static int Main(string[] args)
{
// Ensure we delete any previous copies of the class and its subclasses
try {
ManagementClass testSubClass = new ManagementClass("root/default:URTNEWSUBCLASS");
testSubClass.Delete();
} catch (ManagementException) {}
try {
ManagementClass testClass = new ManagementClass("root/default:URTNEWCLASS");
testClass.Delete();
} catch (ManagementException) {}
// Get an empty class
ManagementClass emptyClass = new ManagementClass("root/default", "", null);
// Set the class name
emptyClass.SystemProperties ["__CLASS"].Value = "URTNEWCLASS";
// Add some qualifiers
QualifierCollection qualSet = emptyClass.Qualifiers;
qualSet.Add("sint32", (int)132235);
qualSet.Add("string", "Floxihoxibellification");
qualSet.Add("real64", (double)-12.33434234);
qualSet.Add("boolean", true);
qualSet.Add("sint32Array", new int[] { 1, 2, 3 }, false, true, false, true);
qualSet.Add("stringArray", new string[] { "il", "spezzio" }, false, false, false, false);
qualSet.Add("real64Array", new double[] { 1.2, -1.3 }, true, true, true, true);
qualSet.Add("booleanArray", new bool[] { false, true }, true, false, true, true);
Console.WriteLine("sint32: " + qualSet["sint32"].Value);
Console.WriteLine("string: " + qualSet["string"].Value);
Console.WriteLine("real64: " + qualSet["real64"].Value);
Console.WriteLine("boolean: " + qualSet["boolean"].Value);
Console.WriteLine("sint32Array: " + ArrayToString((Array)qualSet["sint32Array"].Value));
Console.WriteLine("stringArray: " + ArrayToString((Array)qualSet["stringArray"].Value));
Console.WriteLine("real64Array: " + ArrayToString((Array)qualSet["real64Array"].Value));
Console.WriteLine("booleanArray: " + ArrayToString((Array)qualSet["booleanArray"].Value));
// Add some properties
PropertyCollection propSet = emptyClass.Properties;
// Use the simple Add that selects the right type
propSet.Add("uint8", (byte)10);
propSet.Add("uint16", (ushort)10);
propSet.Add("uint32", (uint)10);
propSet.Add("uint32x", (int)11);
propSet.Add("uint32xx", (UInt32)123);
propSet.Add("uint32xxx", "12");
propSet.Add("uint64", (ulong)10);
propSet.Add("sint8", (sbyte)10);
propSet.Add("sint16", (short)10);
propSet.Add("sint32", (int)10);
propSet.Add("sint64", (long)10);
propSet.Add("bool", true);
propSet.Add("string", "Wibble");
propSet.Add("real32", (float)10.23);
propSet.Add("real64", (double)11.2222);
ManagementClass embeddedClass = new ManagementClass("root/default", "", null);
embeddedClass.SystemProperties ["__CLASS"].Value = "URTEMBEDDEDCLASS";
propSet.Add("object", embeddedClass);
// For datetime/reference types, use the non-default Add
propSet.Add("datetime", "20000728044535.000000-420", CimType.DateTime);
propSet.Add("reference", "foo=10", CimType.Reference);
// Echo the non-null property values
Console.WriteLine("uint8: " + emptyClass["uint8"]);
Console.WriteLine("uint16: " + emptyClass["uint16"]);
Console.WriteLine("uint32: " + emptyClass["uint32"]);
Console.WriteLine("uint32x: " + emptyClass["uint32x"]);
Console.WriteLine("uint32xx: " + emptyClass["uint32xx"]);
Console.WriteLine("uint32xxx: " + emptyClass["uint32xxx"]);
Console.WriteLine("uint64: " + emptyClass["uint64"]);
Console.WriteLine("sint8: " + emptyClass["sint8"]);
Console.WriteLine("sint16: " + emptyClass["sint16"]);
Console.WriteLine("sint32: " + emptyClass["sint32"]);
Console.WriteLine("sint64: " + emptyClass["sint64"]);
Console.WriteLine("bool: " + emptyClass["bool"]);
Console.WriteLine("string: " + emptyClass["string"]);
Console.WriteLine("real32: " + emptyClass["real32"]);
Console.WriteLine("real64: " + emptyClass["real64"]);
Console.WriteLine("datetime: " + emptyClass["datetime"]);
Console.WriteLine("reference: " + emptyClass["reference"]);
Console.WriteLine("object: " + ((ManagementBaseObject)emptyClass["object"]).GetText(TextFormat.Mof));
// Add a null property and put some qualifiers on it
propSet.Add("referenceNull", CimType.Reference, false);
Property newProp = propSet["referenceNull"];
newProp.Qualifiers.Add("read", true);
// Add other simple null properties
propSet.Add("uint8Null", CimType.UInt8, false);
propSet.Add("uint16Null", CimType.UInt16, false);
propSet.Add("uint32Null", CimType.UInt32, false);
propSet.Add("uint64Null", CimType.UInt64, false);
propSet.Add("sint8Null", CimType.SInt8, false);
propSet.Add("sint16Null", CimType.SInt16, false);
propSet.Add("sint32Null", CimType.SInt32, false);
propSet.Add("sint64Null", CimType.SInt64, false);
propSet.Add("boolNull", CimType.Boolean, false);
propSet.Add("stringNull", CimType.String, false);
propSet.Add("real32Null", CimType.Real32, false);
propSet.Add("real64Null", CimType.Real64, false);
propSet.Add("objectNull", CimType.Object, false);
propSet.Add("datetimeNull", CimType.DateTime, false);
// Add array null properties
propSet.Add("uint8ArrayNull", CimType.UInt8, true);
propSet.Add("uint16ArrayNull", CimType.UInt16, true);
propSet.Add("uint32ArrayNull", CimType.UInt32, true);
propSet.Add("uint64ArrayNull", CimType.UInt64, true);
propSet.Add("sint8ArrayNull", CimType.SInt8, true);
propSet.Add("sint16ArrayNull", CimType.SInt16, true);
propSet.Add("sint32ArrayNull", CimType.SInt32, true);
propSet.Add("sint64ArrayNull", CimType.SInt64, true);
propSet.Add("boolArrayNull", CimType.Boolean, true);
propSet.Add("stringArrayNull", CimType.String, true);
propSet.Add("real32ArrayNull", CimType.Real32, true);
propSet.Add("real64ArrayNull", CimType.Real64, true);
propSet.Add("objectArrayNull", CimType.Object, true);
propSet.Add("datetimeArrayNull", CimType.DateTime, true);
// Add some array properties
propSet.Add("uint8Array", new byte[] { 10, 20 });
propSet.Add("uint16Array", new ushort[] { 10, 20 });
propSet.Add("uint32Array", new uint[] { 10, 20 });
propSet.Add("uint64Array", new ulong[] { 10, 20 });
propSet.Add("sint8Array", new sbyte[] { 10, 20 });
propSet.Add("sint16Array", new short[] { 10, 20 });
propSet.Add("sint32Array", new int[] { 10, 20 });
propSet.Add("sint64Array", new long[] { 10, 20 });
propSet.Add("boolArray", new bool[] { true, false, true });
propSet.Add("stringArray", new string[] { "Wibble", "Wobble" });
propSet.Add("real32Array", new float[] { (float)10.23, (float)111.22 });
propSet.Add("real64Array", new double[] { 11.2222, -23.32 });
propSet.Add("datetimeArray", new string[] { "20000728044535.000000-420" }, CimType.DateTime);
propSet.Add("referenceArray", new string[] { "Foo=10", "bar.Nonesuch=\"a\"" }, CimType.Reference);
// BUGBUG following causes a TypeMismatch - RAID 45235 against runtime
propSet.Add("objectArray", new ManagementClass[] { embeddedClass, embeddedClass });
// Echo the non-null array property values
Console.WriteLine("uint8Array: " + ArrayToString((Array)emptyClass["uint8Array"]));
Console.WriteLine("uint16Array: " + ArrayToString((Array)emptyClass["uint16Array"]));
Console.WriteLine("uint32Array: " + ArrayToString((Array)emptyClass["uint32Array"]));
Console.WriteLine("uint64Array: " + ArrayToString((Array)emptyClass["uint64Array"]));
Console.WriteLine("sint8Array: " + ArrayToString((Array)emptyClass["sint8Array"]));
Console.WriteLine("sint16Array: " + ArrayToString((Array)emptyClass["sint16Array"]));
Console.WriteLine("sint32Array: " + ArrayToString((Array)emptyClass["sint32Array"]));
Console.WriteLine("sint64Array: " + ArrayToString((Array)emptyClass["sint64Array"]));
Console.WriteLine("boolArray: " + ArrayToString((Array)emptyClass["boolArray"]));
Console.WriteLine("stringArray: " + ArrayToString((Array)emptyClass["stringArray"]));
Console.WriteLine("real32Array: " + ArrayToString((Array)emptyClass["real32Array"]));
Console.WriteLine("real64Array: " + ArrayToString((Array)emptyClass["real64Array"]));
Console.WriteLine("datetimeArray: " + ArrayToString((Array)emptyClass["datetimeArray"]));
Console.WriteLine("referenceArray: " + ArrayToString((Array)emptyClass["referenceArray"]));
Console.WriteLine("objectArray: " + ObjectArrayToString((Array)emptyClass["objectArray"]));
emptyClass.Put();
Console.WriteLine("Successfully saved new base class");
// Create a derived class
ManagementClass baseClass = new ManagementClass("root/default:URTNEWCLASS");
ManagementClass newClass = baseClass.Derive("URTNEWSUBCLASS");
newClass.Put();
Console.WriteLine("Successfully saved new subclass");
return(0);
}
public SchemaMapping(Type type, SchemaNaming naming, Hashtable mapTypeToConverterClassName)
{
this.codeClassName = (string)mapTypeToConverterClassName[type];
this.classType = type;
bool flag = false;
string baseClassName = ManagedNameAttribute.GetBaseClassName(type);
this.className = ManagedNameAttribute.GetMemberName(type);
this.instrumentationType = InstrumentationClassAttribute.GetAttribute(type).InstrumentationType;
this.classPath = naming.NamespaceName + ":" + this.className;
if (baseClassName == null)
{
this.newClass = new ManagementClass(naming.NamespaceName, "", null);
this.newClass.SystemProperties["__CLASS"].Value = this.className;
}
else
{
ManagementClass class2 = new ManagementClass(naming.NamespaceName + ":" + baseClassName);
if (this.instrumentationType == System.Management.Instrumentation.InstrumentationType.Instance)
{
bool flag2 = false;
try
{
QualifierData data = class2.Qualifiers["abstract"];
if (data.Value is bool)
{
flag2 = (bool)data.Value;
}
}
catch (ManagementException exception)
{
if (exception.ErrorCode != ManagementStatus.NotFound)
{
throw;
}
}
if (!flag2)
{
throw new Exception(RC.GetString("CLASSINST_EXCEPT"));
}
}
this.newClass = class2.Derive(this.className);
}
CodeWriter writer = this.code.AddChild("public class " + this.codeClassName + " : IWmiConverter");
CodeWriter writer2 = writer.AddChild(new CodeWriter());
writer2.Line("static ManagementClass managementClass = new ManagementClass(@\"" + this.classPath + "\");");
writer2.Line("static IntPtr classWbemObjectIP;");
writer2.Line("static Guid iidIWbemObjectAccess = new Guid(\"49353C9A-516B-11D1-AEA6-00C04FB68820\");");
writer2.Line("internal ManagementObject instance = managementClass.CreateInstance();");
writer2.Line("object reflectionInfoTempObj = null ; ");
writer2.Line("FieldInfo reflectionIWbemClassObjectField = null ; ");
writer2.Line("IntPtr emptyWbemObject = IntPtr.Zero ; ");
writer2.Line("IntPtr originalObject = IntPtr.Zero ; ");
writer2.Line("bool toWmiCalled = false ; ");
writer2.Line("IntPtr theClone = IntPtr.Zero;");
writer2.Line("public static ManagementObject emptyInstance = managementClass.CreateInstance();");
writer2.Line("public IntPtr instWbemObjectAccessIP;");
CodeWriter writer3 = writer.AddChild("static " + this.codeClassName + "()");
writer3.Line("classWbemObjectIP = (IntPtr)managementClass;");
writer3.Line("IntPtr wbemObjectAccessIP;");
writer3.Line("Marshal.QueryInterface(classWbemObjectIP, ref iidIWbemObjectAccess, out wbemObjectAccessIP);");
writer3.Line("int cimType;");
CodeWriter writer4 = writer.AddChild("public " + this.codeClassName + "()");
writer4.Line("IntPtr wbemObjectIP = (IntPtr)instance;");
writer4.Line("originalObject = (IntPtr)instance;");
writer4.Line("Marshal.QueryInterface(wbemObjectIP, ref iidIWbemObjectAccess, out instWbemObjectAccessIP);");
writer4.Line("FieldInfo tempField = instance.GetType().GetField ( \"_wbemObject\", BindingFlags.Instance | BindingFlags.NonPublic );");
writer4.Line("if ( tempField == null )");
writer4.Line("{");
writer4.Line(" tempField = instance.GetType().GetField ( \"wbemObject\", BindingFlags.Instance | BindingFlags.NonPublic ) ;");
writer4.Line("}");
writer4.Line("reflectionInfoTempObj = tempField.GetValue (instance) ;");
writer4.Line("reflectionIWbemClassObjectField = reflectionInfoTempObj.GetType().GetField (\"pWbemClassObject\", BindingFlags.Instance | BindingFlags.NonPublic );");
writer4.Line("emptyWbemObject = (IntPtr) emptyInstance;");
CodeWriter writer5 = writer.AddChild("~" + this.codeClassName + "()");
writer5.AddChild("if(instWbemObjectAccessIP != IntPtr.Zero)").Line("Marshal.Release(instWbemObjectAccessIP);");
writer5.Line("if ( toWmiCalled == true )");
writer5.Line("{");
writer5.Line("\tMarshal.Release (originalObject);");
writer5.Line("}");
CodeWriter writer6 = writer.AddChild("public void ToWMI(object obj)");
writer6.Line("toWmiCalled = true ;");
writer6.Line("if(instWbemObjectAccessIP != IntPtr.Zero)");
writer6.Line("{");
writer6.Line(" Marshal.Release(instWbemObjectAccessIP);");
writer6.Line(" instWbemObjectAccessIP = IntPtr.Zero;");
writer6.Line("}");
writer6.Line("if(theClone != IntPtr.Zero)");
writer6.Line("{");
writer6.Line(" Marshal.Release(theClone);");
writer6.Line(" theClone = IntPtr.Zero;");
writer6.Line("}");
writer6.Line("IWOA.Clone_f(12, emptyWbemObject, out theClone) ;");
writer6.Line("Marshal.QueryInterface(theClone, ref iidIWbemObjectAccess, out instWbemObjectAccessIP) ;");
writer6.Line("reflectionIWbemClassObjectField.SetValue ( reflectionInfoTempObj, theClone ) ;");
writer6.Line(string.Format("{0} instNET = ({0})obj;", type.FullName.Replace('+', '.')));
writer.AddChild("public static explicit operator IntPtr(" + this.codeClassName + " obj)").Line("return obj.instWbemObjectAccessIP;");
writer2.Line("public ManagementObject GetInstance() {return instance;}");
PropertyDataCollection properties = this.newClass.Properties;
switch (this.instrumentationType)
{
case System.Management.Instrumentation.InstrumentationType.Instance:
properties.Add("ProcessId", CimType.String, false);
properties.Add("InstanceId", CimType.String, false);
properties["ProcessId"].Qualifiers.Add("key", true);
properties["InstanceId"].Qualifiers.Add("key", true);
this.newClass.Qualifiers.Add("dynamic", true, false, false, false, true);
this.newClass.Qualifiers.Add("provider", naming.DecoupledProviderInstanceName, false, false, false, true);
break;
case System.Management.Instrumentation.InstrumentationType.Abstract:
this.newClass.Qualifiers.Add("abstract", true, false, false, false, true);
break;
}
int num = 0;
bool flag3 = false;
foreach (MemberInfo info in type.GetMembers())
{
if (((info is FieldInfo) || (info is PropertyInfo)) && (info.GetCustomAttributes(typeof(IgnoreMemberAttribute), false).Length <= 0))
{
Type fieldType;
if (info is FieldInfo)
{
FieldInfo info2 = info as FieldInfo;
if (info2.IsStatic)
{
ThrowUnsupportedMember(info);
}
}
else if (info is PropertyInfo)
{
PropertyInfo info3 = info as PropertyInfo;
if (!info3.CanRead)
{
ThrowUnsupportedMember(info);
}
MethodInfo getMethod = info3.GetGetMethod();
if ((null == getMethod) || getMethod.IsStatic)
{
ThrowUnsupportedMember(info);
}
if (getMethod.GetParameters().Length > 0)
{
ThrowUnsupportedMember(info);
}
}
string memberName = ManagedNameAttribute.GetMemberName(info);
if (info is FieldInfo)
{
fieldType = (info as FieldInfo).FieldType;
}
else
{
fieldType = (info as PropertyInfo).PropertyType;
}
bool isArray = false;
if (fieldType.IsArray)
{
if (fieldType.GetArrayRank() != 1)
{
ThrowUnsupportedMember(info);
}
isArray = true;
fieldType = fieldType.GetElementType();
}
string str3 = null;
string str4 = null;
if (mapTypeToConverterClassName.Contains(fieldType))
{
str4 = (string)mapTypeToConverterClassName[fieldType];
str3 = ManagedNameAttribute.GetMemberName(fieldType);
}
bool flag5 = false;
if (fieldType == typeof(object))
{
flag5 = true;
if (!flag)
{
flag = true;
writer2.Line("static Hashtable mapTypeToConverter = new Hashtable();");
foreach (DictionaryEntry entry in mapTypeToConverterClassName)
{
string introduced55 = ((Type)entry.Key).FullName.Replace('+', '.');
writer3.Line(string.Format("mapTypeToConverter[typeof({0})] = typeof({1});", introduced55, (string)entry.Value));
}
}
}
string str5 = "prop_" + num;
string str6 = "handle_" + num++;
writer2.Line("static int " + str6 + ";");
writer3.Line(string.Format("IWOA.GetPropertyHandle_f27(27, wbemObjectAccessIP, \"{0}\", out cimType, out {1});", memberName, str6));
writer2.Line("PropertyData " + str5 + ";");
writer4.Line(string.Format("{0} = instance.Properties[\"{1}\"];", str5, memberName));
if (flag5)
{
CodeWriter writer8 = writer6.AddChild(string.Format("if(instNET.{0} != null)", info.Name));
writer6.AddChild("else").Line(string.Format("{0}.Value = null;", str5));
if (isArray)
{
writer8.Line(string.Format("int len = instNET.{0}.Length;", info.Name));
writer8.Line("ManagementObject[] embeddedObjects = new ManagementObject[len];");
writer8.Line("IWmiConverter[] embeddedConverters = new IWmiConverter[len];");
CodeWriter writer10 = writer8.AddChild("for(int i=0;i<len;i++)");
CodeWriter writer11 = writer10.AddChild(string.Format("if((instNET.{0}[i] != null) && mapTypeToConverter.Contains(instNET.{0}[i].GetType()))", info.Name));
writer11.Line(string.Format("Type type = (Type)mapTypeToConverter[instNET.{0}[i].GetType()];", info.Name));
writer11.Line("embeddedConverters[i] = (IWmiConverter)Activator.CreateInstance(type);");
writer11.Line(string.Format("embeddedConverters[i].ToWMI(instNET.{0}[i]);", info.Name));
writer11.Line("embeddedObjects[i] = embeddedConverters[i].GetInstance();");
writer10.AddChild("else").Line(string.Format("embeddedObjects[i] = SafeAssign.GetManagementObject(instNET.{0}[i]);", info.Name));
writer8.Line(string.Format("{0}.Value = embeddedObjects;", str5));
}
else
{
CodeWriter writer12 = writer8.AddChild(string.Format("if(mapTypeToConverter.Contains(instNET.{0}.GetType()))", info.Name));
writer12.Line(string.Format("Type type = (Type)mapTypeToConverter[instNET.{0}.GetType()];", info.Name));
writer12.Line("IWmiConverter converter = (IWmiConverter)Activator.CreateInstance(type);");
writer12.Line(string.Format("converter.ToWMI(instNET.{0});", info.Name));
writer12.Line(string.Format("{0}.Value = converter.GetInstance();", str5));
writer8.AddChild("else").Line(string.Format("{0}.Value = SafeAssign.GetInstance(instNET.{1});", str5, info.Name));
}
}
else if (str3 != null)
{
CodeWriter writer13;
if (fieldType.IsValueType)
{
writer13 = writer6;
}
else
{
writer13 = writer6.AddChild(string.Format("if(instNET.{0} != null)", info.Name));
writer6.AddChild("else").Line(string.Format("{0}.Value = null;", str5));
}
if (isArray)
{
writer13.Line(string.Format("int len = instNET.{0}.Length;", info.Name));
writer13.Line("ManagementObject[] embeddedObjects = new ManagementObject[len];");
writer13.Line(string.Format("{0}[] embeddedConverters = new {0}[len];", str4));
CodeWriter writer15 = writer13.AddChild("for(int i=0;i<len;i++)");
writer15.Line(string.Format("embeddedConverters[i] = new {0}();", str4));
if (fieldType.IsValueType)
{
writer15.Line(string.Format("embeddedConverters[i].ToWMI(instNET.{0}[i]);", info.Name));
}
else
{
writer15.AddChild(string.Format("if(instNET.{0}[i] != null)", info.Name)).Line(string.Format("embeddedConverters[i].ToWMI(instNET.{0}[i]);", info.Name));
}
writer15.Line("embeddedObjects[i] = embeddedConverters[i].instance;");
writer13.Line(string.Format("{0}.Value = embeddedObjects;", str5));
}
else
{
writer2.Line(string.Format("{0} lazy_embeddedConverter_{1} = null;", str4, str5));
CodeWriter writer18 = writer.AddChild(string.Format("{0} embeddedConverter_{1}", str4, str5)).AddChild("get");
writer18.AddChild(string.Format("if(null == lazy_embeddedConverter_{0})", str5)).Line(string.Format("lazy_embeddedConverter_{0} = new {1}();", str5, str4));
writer18.Line(string.Format("return lazy_embeddedConverter_{0};", str5));
writer13.Line(string.Format("embeddedConverter_{0}.ToWMI(instNET.{1});", str5, info.Name));
writer13.Line(string.Format("{0}.Value = embeddedConverter_{0}.instance;", str5));
}
}
else if (!isArray)
{
if ((fieldType == typeof(byte)) || (fieldType == typeof(sbyte)))
{
writer6.Line(string.Format("{0} instNET_{1} = instNET.{1} ;", fieldType, info.Name));
writer6.Line(string.Format("IWOA.WritePropertyValue_f28(28, instWbemObjectAccessIP, {0}, 1, ref instNET_{1});", str6, info.Name));
}
else if (((fieldType == typeof(short)) || (fieldType == typeof(ushort))) || (fieldType == typeof(char)))
{
writer6.Line(string.Format("{0} instNET_{1} = instNET.{1} ;", fieldType, info.Name));
writer6.Line(string.Format("IWOA.WritePropertyValue_f28(28, instWbemObjectAccessIP, {0}, 2, ref instNET_{1});", str6, info.Name));
}
else if (((fieldType == typeof(uint)) || (fieldType == typeof(int))) || (fieldType == typeof(float)))
{
writer6.Line(string.Format("IWOA.WriteDWORD_f31(31, instWbemObjectAccessIP, {0}, instNET.{1});", str6, info.Name));
}
else if (((fieldType == typeof(ulong)) || (fieldType == typeof(long))) || (fieldType == typeof(double)))
{
writer6.Line(string.Format("IWOA.WriteQWORD_f33(33, instWbemObjectAccessIP, {0}, instNET.{1});", str6, info.Name));
}
else if (fieldType == typeof(bool))
{
writer6.Line(string.Format("if(instNET.{0})", info.Name));
writer6.Line(string.Format(" IWOA.WritePropertyValue_f28(28, instWbemObjectAccessIP, {0}, 2, ref SafeAssign.boolTrue);", str6));
writer6.Line("else");
writer6.Line(string.Format(" IWOA.WritePropertyValue_f28(28, instWbemObjectAccessIP, {0}, 2, ref SafeAssign.boolFalse);", str6));
}
else if (fieldType == typeof(string))
{
writer6.AddChild(string.Format("if(null != instNET.{0})", info.Name)).Line(string.Format("IWOA.WritePropertyValue_f28(28, instWbemObjectAccessIP, {0}, (instNET.{1}.Length+1)*2, instNET.{1});", str6, info.Name));
writer6.AddChild("else").Line(string.Format("IWOA.Put_f5(5, instWbemObjectAccessIP, \"{0}\", 0, ref nullObj, 8);", memberName));
if (!flag3)
{
flag3 = true;
writer2.Line("object nullObj = DBNull.Value;");
}
}
else if ((fieldType == typeof(DateTime)) || (fieldType == typeof(TimeSpan)))
{
writer6.Line(string.Format("IWOA.WritePropertyValue_f28(28, instWbemObjectAccessIP, {0}, 52, SafeAssign.WMITimeToString(instNET.{1}));", str6, info.Name));
}
else
{
writer6.Line(string.Format("{0}.Value = instNET.{1};", str5, info.Name));
}
}
else if ((fieldType == typeof(DateTime)) || (fieldType == typeof(TimeSpan)))
{
writer6.AddChild(string.Format("if(null == instNET.{0})", info.Name)).Line(string.Format("{0}.Value = null;", str5));
writer6.AddChild("else").Line(string.Format("{0}.Value = SafeAssign.WMITimeArrayToStringArray(instNET.{1});", str5, info.Name));
}
else
{
writer6.Line(string.Format("{0}.Value = instNET.{1};", str5, info.Name));
}
CimType propertyType = CimType.String;
if (info.DeclaringType == type)
{
bool flag6 = true;
try
{
PropertyData data2 = this.newClass.Properties[memberName];
CimType type1 = data2.Type;
if (data2.IsLocal)
{
throw new ArgumentException(string.Format(RC.GetString("MEMBERCONFLILCT_EXCEPT"), info.Name), info.Name);
}
}
catch (ManagementException exception2)
{
if (exception2.ErrorCode != ManagementStatus.NotFound)
{
throw;
}
flag6 = false;
}
if (!flag6)
{
if (str3 != null)
{
propertyType = CimType.Object;
}
else if (flag5)
{
propertyType = CimType.Object;
}
else if (fieldType == typeof(ManagementObject))
{
propertyType = CimType.Object;
}
else if (fieldType == typeof(sbyte))
{
propertyType = CimType.SInt8;
}
else if (fieldType == typeof(byte))
{
propertyType = CimType.UInt8;
}
else if (fieldType == typeof(short))
{
propertyType = CimType.SInt16;
}
else if (fieldType == typeof(ushort))
{
propertyType = CimType.UInt16;
}
else if (fieldType == typeof(int))
{
propertyType = CimType.SInt32;
}
else if (fieldType == typeof(uint))
{
propertyType = CimType.UInt32;
}
else if (fieldType == typeof(long))
{
propertyType = CimType.SInt64;
}
else if (fieldType == typeof(ulong))
{
propertyType = CimType.UInt64;
}
else if (fieldType == typeof(float))
{
propertyType = CimType.Real32;
}
else if (fieldType == typeof(double))
{
propertyType = CimType.Real64;
}
else if (fieldType == typeof(bool))
{
propertyType = CimType.Boolean;
}
else if (fieldType == typeof(string))
{
propertyType = CimType.String;
}
else if (fieldType == typeof(char))
{
propertyType = CimType.Char16;
}
else if (fieldType == typeof(DateTime))
{
propertyType = CimType.DateTime;
}
else if (fieldType == typeof(TimeSpan))
{
propertyType = CimType.DateTime;
}
else
{
ThrowUnsupportedMember(info);
}
try
{
properties.Add(memberName, propertyType, isArray);
}
catch (ManagementException exception3)
{
ThrowUnsupportedMember(info, exception3);
}
if (fieldType == typeof(TimeSpan))
{
PropertyData data3 = properties[memberName];
data3.Qualifiers.Add("SubType", "interval", false, true, true, true);
}
if (str3 != null)
{
PropertyData data4 = properties[memberName];
data4.Qualifiers["CIMTYPE"].Value = "object:" + str3;
}
}
}
}
}
writer3.Line("Marshal.Release(wbemObjectAccessIP);");
}
public SchemaMapping(Type type, SchemaNaming naming)
{
classType = type;
string baseClassName = ManagedNameAttribute.GetBaseClassName(type);
className = ManagedNameAttribute.GetMemberName(type);
instrumentationType = InstrumentationClassAttribute.GetAttribute(type).InstrumentationType;
classPath = naming.NamespaceName + ":" + className;
if (null == baseClassName)
{
newClass = new ManagementClass(naming.NamespaceName, "", null);
newClass.SystemProperties ["__CLASS"].Value = className;
}
else
{
ManagementClass baseClass = new ManagementClass(naming.NamespaceName + ":" + baseClassName);
newClass = baseClass.Derive(className);
}
PropertyDataCollection props = newClass.Properties;
// type specific info
switch (instrumentationType)
{
case InstrumentationType.Event:
break;
case InstrumentationType.Instance:
props.Add("ProcessId", CimType.String, false);
props.Add("InstanceId", CimType.String, false);
props["ProcessId"].Qualifiers.Add("key", true);
props["InstanceId"].Qualifiers.Add("key", true);
newClass.Qualifiers.Add("dynamic", true, false, false, false, true);
newClass.Qualifiers.Add("provider", naming.DecoupledProviderInstanceName, false, false, false, true);
break;
case InstrumentationType.Abstract:
newClass.Qualifiers.Add("abstract", true, false, false, false, true);
break;
default:
break;
}
foreach (MemberInfo field in type.GetFields())
{
if (!(field is FieldInfo || field is PropertyInfo))
{
continue;
}
if (field.DeclaringType != type)
{
continue;
}
if (field.GetCustomAttributes(typeof(IgnoreMemberAttribute), false).Length > 0)
{
continue;
}
String propName = ManagedNameAttribute.GetMemberName(field);
#if REQUIRES_EXPLICIT_DECLARATION_OF_INHERITED_PROPERTIES
if (InheritedPropertyAttribute.GetAttribute(field) != null)
{
continue;
}
#else
// See if this field already exists on the WMI class
// In other words, is it inherited from a base class
// TODO: Make this more efficient
// - If we have a null base class name, all property names
// should be new
// - We could get all base class property names into a
// hashtable, and look them up from there
bool propertyExists = true;
try
{
PropertyData prop = newClass.Properties[propName];
}
catch (ManagementException e)
{
if (e.ErrorCode != ManagementStatus.NotFound)
{
throw e;
}
else
{
propertyExists = false;
}
}
if (propertyExists)
{
continue;
}
#endif
#if SUPPORTS_ALTERNATE_WMI_PROPERTY_TYPE
Type t2 = ManagedTypeAttribute.GetManagedType(field);
#else
Type t2;
if (field is FieldInfo)
{
t2 = (field as FieldInfo).FieldType;
}
else
{
t2 = (field as PropertyInfo).PropertyType;
}
#endif
CimType cimtype = CimType.String;
if (t2 == typeof(SByte))
{
cimtype = CimType.SInt8;
}
else if (t2 == typeof(Byte))
{
cimtype = CimType.UInt8;
}
else if (t2 == typeof(Int16))
{
cimtype = CimType.SInt16;
}
else if (t2 == typeof(UInt16))
{
cimtype = CimType.UInt16;
}
else if (t2 == typeof(Int32))
{
cimtype = CimType.SInt32;
}
else if (t2 == typeof(UInt32))
{
cimtype = CimType.UInt32;
}
else if (t2 == typeof(Int64))
{
cimtype = CimType.SInt64;
}
else if (t2 == typeof(UInt64))
{
cimtype = CimType.UInt64;
}
else if (t2 == typeof(Single))
{
cimtype = CimType.Real32;
}
else if (t2 == typeof(Double))
{
cimtype = CimType.Real64;
}
else if (t2 == typeof(Boolean))
{
cimtype = CimType.Boolean;
}
else if (t2 == typeof(String))
{
cimtype = CimType.String;
}
else if (t2 == typeof(Char))
{
cimtype = CimType.Char16;
}
else if (t2 == typeof(DateTime))
{
cimtype = CimType.DateTime;
}
else if (t2 == typeof(TimeSpan))
{
cimtype = CimType.DateTime;
}
else
{
throw new Exception(String.Format("Unsupported type for event member - {0}", t2.Name));
}
// HACK: The following line cause a strange System.InvalidProgramException when run through InstallUtil
// throw new Exception("Unsupported type for event member - " + t2.Name);
// TODO: if(t2 == typeof(Decimal))
#if SUPPORTS_WMI_DEFAULT_VAULES
Object defaultValue = ManagedDefaultValueAttribute.GetManagedDefaultValue(field);
// TODO: Is it safe to make this one line?
if (null == defaultValue)
{
props.Add(propName, cimtype, false);
}
else
{
props.Add(propName, defaultValue, cimtype);
}
#else
props.Add(propName, cimtype, false);
#endif
// Must at 'interval' SubType on TimeSpans
if (t2 == typeof(TimeSpan))
{
PropertyData prop = props[propName];
prop.Qualifiers.Add("SubType", "interval", false, true, true, true);
}
}
}
internal void ExtensionProviderSetup()
{
Console.WriteLine("Creating WMI_extension");
using (ManagementClass __Win32Provider = new ManagementClass(@"ROOT\cimv2", "__Win32Provider", null))
{
using (ManagementClass WMI_extension = __Win32Provider.Derive("WMI_extension"))
{
WMI_extension.Properties["Name"].Value = null;
WMI_extension.Properties["ClsId"].Value = providerGuid;
WMI_extension.Properties["Version"].Value = 1;
WMI_extension.Properties["HostingModel"].Value = "Decoupled:COM";
WMI_extension.Properties["SecurityDescriptor"].Value = null;
WMI_extension.Properties.Add("AssemblyName", CimType.String, false);
WMI_extension.Properties.Add("AssemblyPath", CimType.String, false);
WMI_extension.Properties.Add("CLRVersion", CimType.String, false);
try
{
WMI_extension.Put();
}
catch (ManagementException ex)
{
Console.WriteLine("[-] {0}", ex.Message);
}
}
}
Console.WriteLine("Registering " + Assembly.GetExecutingAssembly().GetName().Name + " as a WMI_extension Instance");
ManagementPath managementPath = null;
using (ManagementClass WMI_extension = new ManagementClass(@"ROOT\cimv2", "WMI_extension", null))
{
using (ManagementObject managementObject = WMI_extension.CreateInstance())
{
managementObject.SetPropertyValue("AssemblyName", assembly.FullName);
managementObject.SetPropertyValue("AssemblyPath", "file:///C:/Windows/System32/wbem/" + Assembly.GetExecutingAssembly().GetName().Name + ".dll");
managementObject.SetPropertyValue("CLRVersion", assembly.ImageRuntimeVersion);
managementObject.SetPropertyValue("CLSID", providerGuid);
managementObject.SetPropertyValue("HostingModel", "LocalSystemHost:CLR2.0");
managementObject.SetPropertyValue("Name", assembly.FullName);
try
{
managementPath = managementObject.Put();
}
catch (ManagementException ex)
{
Console.WriteLine("WMI_extension: " + ex.Message);
}
}
}
Console.WriteLine("Registering " + providerDisplayName + " as an Instance Provider");
using (ManagementClass __InstanceProviderRegistration = new ManagementClass(@"ROOT\cimv2", "__InstanceProviderRegistration", null))
{
using (ManagementObject managementObject = __InstanceProviderRegistration.CreateInstance())
{
managementObject.SetPropertyValue("Provider", managementPath);
managementObject.SetPropertyValue("SupportsGet", true);
managementObject.SetPropertyValue("SupportsPut", true);
managementObject.SetPropertyValue("SupportsDelete", true);
managementObject.SetPropertyValue("SupportsEnumeration", true);
try
{
managementObject.Put();
}
catch (ManagementException ex)
{
Console.WriteLine("__InstanceProviderRegistration: " + ex.Message);
}
}
}
Console.WriteLine("Registering " + providerDisplayName + " as an Method Provider");
using (ManagementClass __MethodProviderRegistration = new ManagementClass(@"ROOT\cimv2", "__MethodProviderRegistration", null))
{
using (ManagementObject managementObject = __MethodProviderRegistration.CreateInstance())
{
managementObject.SetPropertyValue("Provider", managementPath);
try
{
managementObject.Put();
}
catch (ManagementException ex)
{
Console.WriteLine("__MethodProviderRegistration: " + ex.Message);
}
}
}
}
