public void RegisterAssemblySpecificSchema()
{
SecurityHelper.UnmanagedCode.Demand();
Type[] instrumentedTypes = InstrumentedAttribute.GetInstrumentedTypes(this.assembly);
StringCollection stringCollections = new StringCollection();
StringCollection stringCollections1 = new StringCollection();
StringCollection stringCollections2 = new StringCollection();
string[] mofFormat = new string[(int)instrumentedTypes.Length];
CodeWriter codeWriter = new CodeWriter();
ReferencesCollection referencesCollection = new ReferencesCollection();
codeWriter.AddChild(referencesCollection.UsingCode);
referencesCollection.Add(typeof(object));
referencesCollection.Add(typeof(ManagementClass));
referencesCollection.Add(typeof(Marshal));
referencesCollection.Add(typeof(SuppressUnmanagedCodeSecurityAttribute));
referencesCollection.Add(typeof(FieldInfo));
referencesCollection.Add(typeof(Hashtable));
codeWriter.Line();
CodeWriter codeWriter1 = codeWriter.AddChild("public class WMINET_Converter");
codeWriter1.Line("public static Hashtable mapTypeToConverter = new Hashtable();");
CodeWriter codeWriter2 = codeWriter1.AddChild("static WMINET_Converter()");
Hashtable hashtables = new Hashtable();
for (int i = 0; i < (int)instrumentedTypes.Length; i++)
{
hashtables[instrumentedTypes[i]] = string.Concat("ConvertClass_", i);
}
bool beCompared = this.IsSchemaToBeCompared();
bool flag = false;
if (!beCompared)
{
flag = !this.IsAssemblyRegistered();
}
for (int j = 0; j < (int)instrumentedTypes.Length; j++)
{
SchemaMapping schemaMapping = new SchemaMapping(instrumentedTypes[j], this, hashtables);
codeWriter2.Line(string.Format("mapTypeToConverter[typeof({0})] = typeof({1});", schemaMapping.ClassType.FullName.Replace('+', '.'), schemaMapping.CodeClassName));
if (beCompared && !this.IsClassAlreadyPresentInRepository(schemaMapping.NewClass))
{
flag = true;
}
SchemaNaming.ReplaceClassIfNecessary(schemaMapping.ClassPath, schemaMapping.NewClass);
mofFormat[j] = SchemaNaming.GetMofFormat(schemaMapping.NewClass);
codeWriter.AddChild(schemaMapping.Code);
InstrumentationType instrumentationType = schemaMapping.InstrumentationType;
switch (instrumentationType)
{
case InstrumentationType.Instance:
{
stringCollections1.Add(schemaMapping.ClassName);
break;
}
case InstrumentationType.Event:
{
stringCollections.Add(schemaMapping.ClassName);
break;
}
case InstrumentationType.Abstract:
{
stringCollections2.Add(schemaMapping.ClassName);
break;
}
}
}
this.RegisterAssemblySpecificDecoupledProviderInstance();
this.RegisterProviderAsEventProvider(stringCollections);
this.RegisterProviderAsInstanceProvider();
this.RegisterAssemblyAsInstrumented();
Directory.CreateDirectory(this.DataDirectory);
using (StreamWriter streamWriter = new StreamWriter(this.CodePath, false, Encoding.Unicode))
{
streamWriter.WriteLine(codeWriter);
streamWriter.WriteLine(string.Concat("class IWOA\r\n{\r\nprotected const string DllName = \"wminet_utils.dll\";\r\nprotected const string EntryPointName = \"UFunc\";\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"GetPropertyHandle\")] public static extern int GetPropertyHandle_f27(int vFunc, IntPtr pWbemClassObject, [In][MarshalAs(UnmanagedType.LPWStr)] string wszPropertyName, [Out] out Int32 pType, [Out] out Int32 plHandle);\r\n//[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"WritePropertyValue\")] public static extern int WritePropertyValue_f28(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [In] Int32 lNumBytes, [In] ref Byte aData);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"ReadPropertyValue\")] public static extern int ReadPropertyValue_f29(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [In] Int32 lBufferSize, [Out] out Int32 plNumBytes, [Out] out Byte aData);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"ReadDWORD\")] public static extern int ReadDWORD_f30(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [Out] out UInt32 pdw);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"WriteDWORD\")] public static extern int WriteDWORD_f31(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [In] UInt32 dw);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"ReadQWORD\")] public static extern int ReadQWORD_f32(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [Out] out UInt64 pqw);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"WriteQWORD\")] public static extern int WriteQWORD_f33(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [In] UInt64 pw);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"GetPropertyInfoByHandle\")] public static extern int GetPropertyInfoByHandle_f34(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [Out][MarshalAs(UnmanagedType.BStr)] out string pstrName, [Out] out Int32 pType);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"Lock\")] public static extern int Lock_f35(int vFunc, IntPtr pWbemClassObject, [In] Int32 lFlags);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"Unlock\")] public static extern int Unlock_f36(int vFunc, IntPtr pWbemClassObject, [In] Int32 lFlags);\r\n\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"Put\")] public static extern int Put_f5(int vFunc, IntPtr pWbemClassObject, [In][MarshalAs(UnmanagedType.LPWStr)] string wszName, [In] Int32 lFlags, [In] ref object pVal, [In] Int32 Type);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"WritePropertyValue\")] public static extern int WritePropertyValue_f28(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [In] Int32 lNumBytes, [In][MarshalAs(UnmanagedType.LPWStr)] string str);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"WritePropertyValue\")] public static extern int WritePropertyValue_f28(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [In] Int32 lNumBytes, [In] ref Byte n);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"WritePropertyValue\")] public static extern int WritePropertyValue_f28(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [In] Int32 lNumBytes, [In] ref SByte n);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"WritePropertyValue\")] public static extern int WritePropertyValue_f28(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [In] Int32 lNumBytes, [In] ref Int16 n);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"WritePropertyValue\")] public static extern int WritePropertyValue_f28(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [In] Int32 lNumBytes, [In] ref UInt16 n);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"WritePropertyValue\", CharSet=CharSet.Unicode)] public static extern int WritePropertyValue_f28(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [In] Int32 lNumBytes, [In] ref Char c);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"WriteDWORD\")] public static extern int WriteDWORD_f31(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [In] Int32 dw);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"WriteSingle\")] public static extern int WriteDWORD_f31(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [In] Single dw);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"WriteQWORD\")] public static extern int WriteQWORD_f33(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [In] Int64 pw);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"WriteDouble\")] public static extern int WriteQWORD_f33(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [In] Double pw);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"Clone\")] public static extern int Clone_f(int vFunc, IntPtr pWbemClassObject, [Out] out IntPtr ppCopy);\r\n}\r\ninterface IWmiConverter\r\n{\r\n void ToWMI(object obj);\r\n ManagementObject GetInstance();\r\n}\r\nclass SafeAssign\r\n{\r\n public static UInt16 boolTrue = 0xffff;\r\n public static UInt16 boolFalse = 0;\r\n static Hashtable validTypes = new Hashtable();\r\n static SafeAssign()\r\n {\r\n validTypes.Add(typeof(SByte), null);\r\n validTypes.Add(typeof(Byte), null);\r\n validTypes.Add(typeof(Int16), null);\r\n validTypes.Add(typeof(UInt16), null);\r\n validTypes.Add(typeof(Int32), null);\r\n validTypes.Add(typeof(UInt32), null);\r\n validTypes.Add(typeof(Int64), null);\r\n validTypes.Add(typeof(UInt64), null);\r\n validTypes.Add(typeof(Single), null);\r\n validTypes.Add(typeof(Double), null);\r\n validTypes.Add(typeof(Boolean), null);\r\n validTypes.Add(typeof(String), null);\r\n validTypes.Add(typeof(Char), null);\r\n validTypes.Add(typeof(DateTime), null);\r\n validTypes.Add(typeof(TimeSpan), null);\r\n validTypes.Add(typeof(ManagementObject), null);\r\n nullClass.SystemProperties [\"__CLASS\"].Value = \"nullInstance\";\r\n }\r\n public static object GetInstance(object o)\r\n {\r\n if(o is ManagementObject)\r\n return o;\r\n return null;\r\n }\r\n static ManagementClass nullClass = new ManagementClass(new ManagementPath(@\"", this.NamespaceName, "\"));\r\n \r\n public static ManagementObject GetManagementObject(object o)\r\n {\r\n if(o != null && o is ManagementObject)\r\n return o as ManagementObject;\r\n // Must return empty instance\r\n return nullClass.CreateInstance();\r\n }\r\n public static object GetValue(object o)\r\n {\r\n Type t = o.GetType();\r\n if(t.IsArray)\r\n t = t.GetElementType();\r\n if(validTypes.Contains(t))\r\n return o;\r\n return null;\r\n }\r\n public static string WMITimeToString(DateTime dt)\r\n {\r\n TimeSpan ts = dt.Subtract(dt.ToUniversalTime());\r\n int diffUTC = (ts.Minutes + ts.Hours * 60);\r\n if(diffUTC >= 0)\r\n return String.Format(\"{0:D4}{1:D2}{2:D2}{3:D2}{4:D2}{5:D2}.{6:D3}000+{7:D3}\", dt.Year, dt.Month, dt.Day, dt.Hour, dt.Minute, dt.Second, dt.Millisecond, diffUTC);\r\n return String.Format(\"{0:D4}{1:D2}{2:D2}{3:D2}{4:D2}{5:D2}.{6:D3}000-{7:D3}\", dt.Year, dt.Month, dt.Day, dt.Hour, dt.Minute, dt.Second, dt.Millisecond, -diffUTC);\r\n }\r\n public static string WMITimeToString(TimeSpan ts)\r\n {\r\n return String.Format(\"{0:D8}{1:D2}{2:D2}{3:D2}.{4:D3}000:000\", ts.Days, ts.Hours, ts.Minutes, ts.Seconds, ts.Milliseconds);\r\n }\r\n public static string[] WMITimeArrayToStringArray(DateTime[] dates)\r\n {\r\n string[] strings = new string[dates.Length];\r\n for(int i=0;i<dates.Length;i++)\r\n strings[i] = WMITimeToString(dates[i]);\r\n return strings;\r\n }\r\n public static string[] WMITimeArrayToStringArray(TimeSpan[] timeSpans)\r\n {\r\n string[] strings = new string[timeSpans.Length];\r\n for(int i=0;i<timeSpans.Length;i++)\r\n strings[i] = WMITimeToString(timeSpans[i]);\r\n return strings;\r\n }\r\n}\r\n"));
}
using (StreamWriter streamWriter1 = new StreamWriter(this.MofPath, false, Encoding.Unicode))
{
streamWriter1.WriteLine(this.GenerateMof(mofFormat));
}
if (flag)
{
SchemaNaming.RegisterSchemaUsingMofcomp(this.MofPath);
}
}
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 void RegisterAssemblySpecificSchema()
{
SecurityHelper.UnmanagedCode.Demand(); // Bug#112640 - Close off any potential use from anything but fully trusted code
Type[] types = InstrumentedAttribute.GetInstrumentedTypes(assembly);
StringCollection events = new StringCollection();
StringCollection instances = new StringCollection();
StringCollection abstracts = new StringCollection();
string[] mofs = new string[types.Length];
CodeWriter code = new CodeWriter();
ReferencesCollection references = new ReferencesCollection();
// Add the node with all the 'using' statements at the top
code.AddChild(references.UsingCode);
references.Add(typeof(Object));
references.Add(typeof(ManagementClass));
references.Add(typeof(Marshal));
references.Add(typeof(System.Security.SuppressUnmanagedCodeSecurityAttribute));
references.Add(typeof(System.Reflection.FieldInfo));
references.Add(typeof(Hashtable));
// Add a blank line
code.Line();
// Add master converter class
CodeWriter codeWMIConverter = code.AddChild("public class WMINET_Converter");
// Add master map of types to converters
codeWMIConverter.Line("public static Hashtable mapTypeToConverter = new Hashtable();");
// Add master CCTOR
CodeWriter codeCCTOR = codeWMIConverter.AddChild("static WMINET_Converter()");
// Make mapping of types to converter class names
Hashtable mapTypeToConverterClassName = new Hashtable();
for (int i = 0; i < types.Length; i++)
{
mapTypeToConverterClassName[types[i]] = "ConvertClass_" + i;
}
// [Microsoft] VSUQFE#2248 (VSWhidbey 231885)
bool bSchemaToBeCompared = IsSchemaToBeCompared();
bool bNewClassToCompile = false;
// Mof compilation is dictated by, whether the assembly is registered as instrumented or not
if (bSchemaToBeCompared == false)
{
bNewClassToCompile = !IsAssemblyRegistered();
}
for (int i = 0; i < types.Length; i++)
{
SchemaMapping mapping = new SchemaMapping(types[i], this, mapTypeToConverterClassName);
codeCCTOR.Line(String.Format("mapTypeToConverter[typeof({0})] = typeof({1});", mapping.ClassType.FullName.Replace('+', '.'), mapping.CodeClassName)); // bug#92918 - watch for nested classes
// [Microsoft] VSUQFE#2248 (VSWhidbey 231885)
if (bSchemaToBeCompared == true && IsClassAlreadyPresentInRepository(mapping.NewClass) == false)
{
bNewClassToCompile = true;
}
ReplaceClassIfNecessary(mapping.ClassPath, mapping.NewClass);
mofs[i] = GetMofFormat(mapping.NewClass);
code.AddChild(mapping.Code);
switch (mapping.InstrumentationType)
{
case InstrumentationType.Event:
events.Add(mapping.ClassName);
break;
case InstrumentationType.Instance:
instances.Add(mapping.ClassName);
break;
case InstrumentationType.Abstract:
abstracts.Add(mapping.ClassName);
break;
default:
break;
}
}
RegisterAssemblySpecificDecoupledProviderInstance();
RegisterProviderAsEventProvider(events);
RegisterProviderAsInstanceProvider();
RegisterAssemblyAsInstrumented();
Directory.CreateDirectory(DataDirectory);
using (StreamWriter log = new StreamWriter(CodePath, false, Encoding.Unicode))
{
log.WriteLine(code);
log.WriteLine(iwoaDef + "new ManagementPath(@\"" + this.NamespaceName + "\")" + iwoaDefEnd);
}
// Always generate the MOF in unicode
using (StreamWriter log = new StreamWriter(MofPath, false, Encoding.Unicode))
{
log.WriteLine(GenerateMof(mofs));
}
// [Microsoft] VSUQFE#2248 (VSWhidbey 231885)
// Write the mof to a file and compile it only if there are any new classes , apart from
// what is there in the repository
if (bNewClassToCompile == true)
{
RegisterSchemaUsingMofcomp(MofPath);
//WMICapabilities.AddAutorecoverMof(MofPath);
}
}
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 void RegisterAssemblySpecificSchema()
{
SecurityHelper.UnmanagedCode.Demand(); // Bug#112640 - Close off any potential use from anything but fully trusted code
Type[] types = InstrumentedAttribute.GetInstrumentedTypes(assembly);
StringCollection events = new StringCollection();
StringCollection instances = new StringCollection();
StringCollection abstracts = new StringCollection();
string[] mofs = new string[types.Length];
CodeWriter code = new CodeWriter();
ReferencesCollection references = new ReferencesCollection();
// Add the node with all the 'using' statements at the top
code.AddChild(references.UsingCode);
references.Add(typeof(Object));
references.Add(typeof(ManagementClass));
references.Add(typeof(Marshal));
references.Add(typeof(System.Security.SuppressUnmanagedCodeSecurityAttribute));
references.Add(typeof(System.Reflection.FieldInfo));
references.Add(typeof(Hashtable));
// Add a blank line
code.Line();
// Add master converter class
CodeWriter codeWMIConverter = code.AddChild("public class WMINET_Converter");
// Add master map of types to converters
codeWMIConverter.Line("public static Hashtable mapTypeToConverter = new Hashtable();");
// Add master CCTOR
CodeWriter codeCCTOR = codeWMIConverter.AddChild("static WMINET_Converter()");
// Make mapping of types to converter class names
Hashtable mapTypeToConverterClassName = new Hashtable();
for(int i=0;i<types.Length;i++)
mapTypeToConverterClassName[types[i]] = "ConvertClass_" + i;
// [[....]] VSUQFE#2248 (VSWhidbey 231885)
bool bSchemaToBeCompared = IsSchemaToBeCompared();
bool bNewClassToCompile = false;
// Mof compilation is dictated by, whether the assembly is registered as instrumented or not
if (bSchemaToBeCompared == false)
{
bNewClassToCompile = !IsAssemblyRegistered();
}
for(int i=0;i<types.Length;i++)
{
SchemaMapping mapping = new SchemaMapping(types[i], this, mapTypeToConverterClassName);
codeCCTOR.Line(String.Format("mapTypeToConverter[typeof({0})] = typeof({1});", mapping.ClassType.FullName.Replace('+', '.'), mapping.CodeClassName)); // bug#92918 - watch for nested classes
// [[....]] VSUQFE#2248 (VSWhidbey 231885)
if (bSchemaToBeCompared == true && IsClassAlreadyPresentInRepository(mapping.NewClass) == false)
{
bNewClassToCompile = true;
}
ReplaceClassIfNecessary(mapping.ClassPath, mapping.NewClass);
mofs[i] = GetMofFormat(mapping.NewClass);
code.AddChild(mapping.Code);
switch(mapping.InstrumentationType)
{
case InstrumentationType.Event:
events.Add(mapping.ClassName);
break;
case InstrumentationType.Instance:
instances.Add(mapping.ClassName);
break;
case InstrumentationType.Abstract:
abstracts.Add(mapping.ClassName);
break;
default:
break;
}
}
RegisterAssemblySpecificDecoupledProviderInstance();
RegisterProviderAsEventProvider(events);
RegisterProviderAsInstanceProvider();
RegisterAssemblyAsInstrumented();
Directory.CreateDirectory(DataDirectory);
using(StreamWriter log = new StreamWriter(CodePath, false, Encoding.Unicode))
{
log.WriteLine(code);
log.WriteLine(iwoaDef+"new ManagementPath(@\""+this.NamespaceName+"\")"+iwoaDefEnd);
}
// Always generate the MOF in unicode
using(StreamWriter log = new StreamWriter(MofPath, false, Encoding.Unicode))
{
log.WriteLine(GenerateMof(mofs));
}
// [[....]] VSUQFE#2248 (VSWhidbey 231885)
// Write the mof to a file and compile it only if there are any new classes , apart from
// what is there in the repository
if (bNewClassToCompile == true)
{
RegisterSchemaUsingMofcomp ( MofPath ) ;
//WMICapabilities.AddAutorecoverMof(MofPath);
}
}
public void RegisterAssemblySpecificSchema()
{
SecurityHelper.UnmanagedCode.Demand();
Type[] instrumentedTypes = InstrumentedAttribute.GetInstrumentedTypes(this.assembly);
StringCollection events = new StringCollection();
StringCollection strings2 = new StringCollection();
StringCollection strings3 = new StringCollection();
string[] mofs = new string[instrumentedTypes.Length];
CodeWriter writer = new CodeWriter();
ReferencesCollection referencess = new ReferencesCollection();
writer.AddChild(referencess.UsingCode);
referencess.Add(typeof(object));
referencess.Add(typeof(ManagementClass));
referencess.Add(typeof(Marshal));
referencess.Add(typeof(SuppressUnmanagedCodeSecurityAttribute));
referencess.Add(typeof(FieldInfo));
referencess.Add(typeof(Hashtable));
writer.Line();
CodeWriter writer2 = writer.AddChild("public class WMINET_Converter");
writer2.Line("public static Hashtable mapTypeToConverter = new Hashtable();");
CodeWriter writer3 = writer2.AddChild("static WMINET_Converter()");
Hashtable mapTypeToConverterClassName = new Hashtable();
for (int i = 0; i < instrumentedTypes.Length; i++)
{
mapTypeToConverterClassName[instrumentedTypes[i]] = "ConvertClass_" + i;
}
bool flag = this.IsSchemaToBeCompared();
bool flag2 = false;
if (!flag)
{
flag2 = !this.IsAssemblyRegistered();
}
for (int j = 0; j < instrumentedTypes.Length; j++)
{
SchemaMapping mapping = new SchemaMapping(instrumentedTypes[j], this, mapTypeToConverterClassName);
writer3.Line(string.Format("mapTypeToConverter[typeof({0})] = typeof({1});", mapping.ClassType.FullName.Replace('+', '.'), mapping.CodeClassName));
if (flag && !this.IsClassAlreadyPresentInRepository(mapping.NewClass))
{
flag2 = true;
}
ReplaceClassIfNecessary(mapping.ClassPath, mapping.NewClass);
mofs[j] = GetMofFormat(mapping.NewClass);
writer.AddChild(mapping.Code);
switch (mapping.InstrumentationType)
{
case InstrumentationType.Instance:
strings2.Add(mapping.ClassName);
break;
case InstrumentationType.Event:
events.Add(mapping.ClassName);
break;
case InstrumentationType.Abstract:
strings3.Add(mapping.ClassName);
break;
}
}
this.RegisterAssemblySpecificDecoupledProviderInstance();
this.RegisterProviderAsEventProvider(events);
this.RegisterProviderAsInstanceProvider();
this.RegisterAssemblyAsInstrumented();
Directory.CreateDirectory(this.DataDirectory);
using (StreamWriter writer4 = new StreamWriter(this.CodePath, false, Encoding.Unicode))
{
writer4.WriteLine(writer);
writer4.WriteLine("class IWOA\r\n{\r\nprotected const string DllName = \"wminet_utils.dll\";\r\nprotected const string EntryPointName = \"UFunc\";\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"GetPropertyHandle\")] public static extern int GetPropertyHandle_f27(int vFunc, IntPtr pWbemClassObject, [In][MarshalAs(UnmanagedType.LPWStr)] string wszPropertyName, [Out] out Int32 pType, [Out] out Int32 plHandle);\r\n//[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"WritePropertyValue\")] public static extern int WritePropertyValue_f28(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [In] Int32 lNumBytes, [In] ref Byte aData);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"ReadPropertyValue\")] public static extern int ReadPropertyValue_f29(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [In] Int32 lBufferSize, [Out] out Int32 plNumBytes, [Out] out Byte aData);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"ReadDWORD\")] public static extern int ReadDWORD_f30(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [Out] out UInt32 pdw);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"WriteDWORD\")] public static extern int WriteDWORD_f31(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [In] UInt32 dw);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"ReadQWORD\")] public static extern int ReadQWORD_f32(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [Out] out UInt64 pqw);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"WriteQWORD\")] public static extern int WriteQWORD_f33(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [In] UInt64 pw);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"GetPropertyInfoByHandle\")] public static extern int GetPropertyInfoByHandle_f34(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [Out][MarshalAs(UnmanagedType.BStr)] out string pstrName, [Out] out Int32 pType);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"Lock\")] public static extern int Lock_f35(int vFunc, IntPtr pWbemClassObject, [In] Int32 lFlags);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"Unlock\")] public static extern int Unlock_f36(int vFunc, IntPtr pWbemClassObject, [In] Int32 lFlags);\r\n\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"Put\")] public static extern int Put_f5(int vFunc, IntPtr pWbemClassObject, [In][MarshalAs(UnmanagedType.LPWStr)] string wszName, [In] Int32 lFlags, [In] ref object pVal, [In] Int32 Type);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"WritePropertyValue\")] public static extern int WritePropertyValue_f28(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [In] Int32 lNumBytes, [In][MarshalAs(UnmanagedType.LPWStr)] string str);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"WritePropertyValue\")] public static extern int WritePropertyValue_f28(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [In] Int32 lNumBytes, [In] ref Byte n);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"WritePropertyValue\")] public static extern int WritePropertyValue_f28(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [In] Int32 lNumBytes, [In] ref SByte n);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"WritePropertyValue\")] public static extern int WritePropertyValue_f28(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [In] Int32 lNumBytes, [In] ref Int16 n);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"WritePropertyValue\")] public static extern int WritePropertyValue_f28(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [In] Int32 lNumBytes, [In] ref UInt16 n);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"WritePropertyValue\", CharSet=CharSet.Unicode)] public static extern int WritePropertyValue_f28(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [In] Int32 lNumBytes, [In] ref Char c);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"WriteDWORD\")] public static extern int WriteDWORD_f31(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [In] Int32 dw);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"WriteSingle\")] public static extern int WriteDWORD_f31(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [In] Single dw);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"WriteQWORD\")] public static extern int WriteQWORD_f33(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [In] Int64 pw);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"WriteDouble\")] public static extern int WriteQWORD_f33(int vFunc, IntPtr pWbemClassObject, [In] Int32 lHandle, [In] Double pw);\r\n[SuppressUnmanagedCodeSecurity, DllImport(DllName, EntryPoint=\"Clone\")] public static extern int Clone_f(int vFunc, IntPtr pWbemClassObject, [Out] out IntPtr ppCopy);\r\n}\r\ninterface IWmiConverter\r\n{\r\n void ToWMI(object obj);\r\n ManagementObject GetInstance();\r\n}\r\nclass SafeAssign\r\n{\r\n public static UInt16 boolTrue = 0xffff;\r\n public static UInt16 boolFalse = 0;\r\n static Hashtable validTypes = new Hashtable();\r\n static SafeAssign()\r\n {\r\n validTypes.Add(typeof(SByte), null);\r\n validTypes.Add(typeof(Byte), null);\r\n validTypes.Add(typeof(Int16), null);\r\n validTypes.Add(typeof(UInt16), null);\r\n validTypes.Add(typeof(Int32), null);\r\n validTypes.Add(typeof(UInt32), null);\r\n validTypes.Add(typeof(Int64), null);\r\n validTypes.Add(typeof(UInt64), null);\r\n validTypes.Add(typeof(Single), null);\r\n validTypes.Add(typeof(Double), null);\r\n validTypes.Add(typeof(Boolean), null);\r\n validTypes.Add(typeof(String), null);\r\n validTypes.Add(typeof(Char), null);\r\n validTypes.Add(typeof(DateTime), null);\r\n validTypes.Add(typeof(TimeSpan), null);\r\n validTypes.Add(typeof(ManagementObject), null);\r\n nullClass.SystemProperties [\"__CLASS\"].Value = \"nullInstance\";\r\n }\r\n public static object GetInstance(object o)\r\n {\r\n if(o is ManagementObject)\r\n return o;\r\n return null;\r\n }\r\n static ManagementClass nullClass = new ManagementClass(new ManagementPath(@\"" + this.NamespaceName + "\"));\r\n \r\n public static ManagementObject GetManagementObject(object o)\r\n {\r\n if(o != null && o is ManagementObject)\r\n return o as ManagementObject;\r\n // Must return empty instance\r\n return nullClass.CreateInstance();\r\n }\r\n public static object GetValue(object o)\r\n {\r\n Type t = o.GetType();\r\n if(t.IsArray)\r\n t = t.GetElementType();\r\n if(validTypes.Contains(t))\r\n return o;\r\n return null;\r\n }\r\n public static string WMITimeToString(DateTime dt)\r\n {\r\n TimeSpan ts = dt.Subtract(dt.ToUniversalTime());\r\n int diffUTC = (ts.Minutes + ts.Hours * 60);\r\n if(diffUTC >= 0)\r\n return String.Format(\"{0:D4}{1:D2}{2:D2}{3:D2}{4:D2}{5:D2}.{6:D3}000+{7:D3}\", dt.Year, dt.Month, dt.Day, dt.Hour, dt.Minute, dt.Second, dt.Millisecond, diffUTC);\r\n return String.Format(\"{0:D4}{1:D2}{2:D2}{3:D2}{4:D2}{5:D2}.{6:D3}000-{7:D3}\", dt.Year, dt.Month, dt.Day, dt.Hour, dt.Minute, dt.Second, dt.Millisecond, -diffUTC);\r\n }\r\n public static string WMITimeToString(TimeSpan ts)\r\n {\r\n return String.Format(\"{0:D8}{1:D2}{2:D2}{3:D2}.{4:D3}000:000\", ts.Days, ts.Hours, ts.Minutes, ts.Seconds, ts.Milliseconds);\r\n }\r\n public static string[] WMITimeArrayToStringArray(DateTime[] dates)\r\n {\r\n string[] strings = new string[dates.Length];\r\n for(int i=0;i<dates.Length;i++)\r\n strings[i] = WMITimeToString(dates[i]);\r\n return strings;\r\n }\r\n public static string[] WMITimeArrayToStringArray(TimeSpan[] timeSpans)\r\n {\r\n string[] strings = new string[timeSpans.Length];\r\n for(int i=0;i<timeSpans.Length;i++)\r\n strings[i] = WMITimeToString(timeSpans[i]);\r\n return strings;\r\n }\r\n}\r\n");
}
using (StreamWriter writer5 = new StreamWriter(this.MofPath, false, Encoding.Unicode))
{
writer5.WriteLine(this.GenerateMof(mofs));
}
if (flag2)
{
RegisterSchemaUsingMofcomp(this.MofPath);
}
}
public void RegisterAssemblySpecificSchema()
{
SecurityHelper.UnmanagedCode.Demand(); // Bug#112640 - Close off any potential use from anything but fully trusted code
Type[] types = InstrumentedAttribute.GetInstrumentedTypes(assembly);
StringCollection events = new StringCollection();
StringCollection instances = new StringCollection();
StringCollection abstracts = new StringCollection();
string[] mofs = new string[types.Length];
CodeWriter code = new CodeWriter();
ReferencesCollection references = new ReferencesCollection();
// Add the node with all the 'using' statements at the top
code.AddChild(references.UsingCode);
references.Add(typeof(Object));
references.Add(typeof(ManagementClass));
references.Add(typeof(Marshal));
references.Add(typeof(System.Security.SuppressUnmanagedCodeSecurityAttribute));
references.Add(typeof(System.Reflection.FieldInfo));
references.Add(typeof(Hashtable));
// Add a blank line
code.Line();
// Add master converter class
CodeWriter codeWMIConverter = code.AddChild("public class WMINET_Converter");
// Add master map of types to converters
codeWMIConverter.Line("public static Hashtable mapTypeToConverter = new Hashtable();");
// Add master CCTOR
CodeWriter codeCCTOR = codeWMIConverter.AddChild("static WMINET_Converter()");
// Make mapping of types to converter class names
Hashtable mapTypeToConverterClassName = new Hashtable();
for (int i = 0; i < types.Length; i++)
{
mapTypeToConverterClassName[types[i]] = "ConvertClass_" + i;
}
for (int i = 0; i < types.Length; i++)
{
SchemaMapping mapping = new SchemaMapping(types[i], this, mapTypeToConverterClassName);
codeCCTOR.Line(String.Format("mapTypeToConverter[typeof({0})] = typeof({1});", mapping.ClassType.FullName.Replace('+', '.'), mapping.CodeClassName)); // bug#92918 - watch for nested classes
ReplaceClassIfNecessary(mapping.ClassPath, mapping.NewClass);
mofs[i] = GetMofFormat(mapping.NewClass);
code.AddChild(mapping.Code);
switch (mapping.InstrumentationType)
{
case InstrumentationType.Event:
events.Add(mapping.ClassName);
break;
case InstrumentationType.Instance:
instances.Add(mapping.ClassName);
break;
case InstrumentationType.Abstract:
abstracts.Add(mapping.ClassName);
break;
default:
break;
}
}
RegisterAssemblySpecificDecoupledProviderInstance();
RegisterProviderAsEventProvider(events);
RegisterProviderAsInstanceProvider();
RegisterAssemblyAsInstrumented();
Directory.CreateDirectory(DataDirectory);
using (StreamWriter log = new StreamWriter(CodePath, false, Encoding.Unicode))
{
log.WriteLine(code);
log.WriteLine(iwoaDef);
}
// Always generate the MOF in unicode
using (StreamWriter log = new StreamWriter(MofPath, false, Encoding.Unicode))
{
log.WriteLine(GenerateMof(mofs));
}
RegisterSchemaUsingMofcomp(MofPath);
//WMICapabilities.AddAutorecoverMof(MofPath);
}