internal static Type GetBaseInstrumentationType(Type type) { if (InstrumentationClassAttribute.GetAttribute(type.BaseType) == null) { return(null); } else { return(type.BaseType); } }
internal static string GetBaseClassName(Type type) { InstrumentationClassAttribute attribute = InstrumentationClassAttribute.GetAttribute(type); string managedBaseClassName = attribute.ManagedBaseClassName; if (managedBaseClassName != null) { return managedBaseClassName; } if (InstrumentationClassAttribute.GetAttribute(type.BaseType) == null) { switch (attribute.InstrumentationType) { case InstrumentationType.Instance: return null; case InstrumentationType.Event: return "__ExtrinsicEvent"; case InstrumentationType.Abstract: return null; } } return GetMemberName(type.BaseType); }
internal static string GetBaseClassName(Type type) { InstrumentationClassAttribute attr = InstrumentationClassAttribute.GetAttribute(type); string name = attr.ManagedBaseClassName; if (name != null) { return(name); } // Get managed base type's attribute InstrumentationClassAttribute attrParent = InstrumentationClassAttribute.GetAttribute(type.BaseType); // If the base type does not have a InstrumentationClass attribute, // return a base type based on the InstrumentationType if (null == attrParent) { switch (attr.InstrumentationType) { case InstrumentationType.Abstract: return(null); case InstrumentationType.Instance: return(null); case InstrumentationType.Event: return("__ExtrinsicEvent"); default: break; } } // Our parent was also a managed provider type. Use it's managed name. return(GetMemberName(type.BaseType)); }
internal static string GetBaseClassName(Type type) { InstrumentationClassAttribute attribute = InstrumentationClassAttribute.GetAttribute(type); string managedBaseClassName = attribute.ManagedBaseClassName; if (managedBaseClassName == null) { InstrumentationClassAttribute instrumentationClassAttribute = InstrumentationClassAttribute.GetAttribute(type.BaseType); if (instrumentationClassAttribute == null) { InstrumentationType instrumentationType = attribute.InstrumentationType; switch (instrumentationType) { case InstrumentationType.Instance: { return(null); } case InstrumentationType.Event: { return("__ExtrinsicEvent"); } case InstrumentationType.Abstract: { return(null); } } } return(ManagedNameAttribute.GetMemberName(type.BaseType)); } else { return(managedBaseClassName); } }
private static bool IsInstrumentationClass(Type type) { return(null != InstrumentationClassAttribute.GetAttribute(type)); }
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 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); } } }