forked from PowerShell/PSScriptAnalyzer
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathReflectionTypeAnalysis.cs
More file actions
212 lines (197 loc) · 9.35 KB
/
Copy pathReflectionTypeAnalysis.cs
File metadata and controls
212 lines (197 loc) · 9.35 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
//
// Copyright (c) Microsoft Corporation.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.
//
using System;
using System.Collections.Generic;
using System.Globalization;
using System.IO;
using System.Reflection;
using System.Reflection.Metadata;
using System.Reflection.PortableExecutable;
namespace NanoServerCompliance
{
public class ReflectionTypeAnalysis
{
public Dictionary<string, List<string>> typeMethod { get; set; }
public Dictionary<string, List<string>> typeProperties { get; set; }
public Dictionary<string, List<string>> typeFields { get; set; }
private static object syncRoot = new Object();
private static ReflectionTypeAnalysis instance = null;
/// <summary>
/// The helper instance that handles utility functions
/// </summary>
public static ReflectionTypeAnalysis Instance
{
get
{
if (instance == null)
{
lock (syncRoot)
{
if (instance == null)
instance = new ReflectionTypeAnalysis();
}
}
return instance;
}
}
/// <summary>
/// Get available types from reading the assembly
/// </summary>
/// <param name="assemblyDir"></param>
public void GetTypeFromAssembly(string assemblyDir)
{
//Initalize the type dictionaries
typeMethod = new Dictionary<string, List<string>>(StringComparer.OrdinalIgnoreCase);
typeProperties = new Dictionary<string, List<string>>(StringComparer.OrdinalIgnoreCase);
typeFields = new Dictionary<string, List<string>>(StringComparer.OrdinalIgnoreCase);
foreach (string filePath in Directory.GetFiles(assemblyDir))
{
if (!filePath.EndsWith(".METADATA_DLL", StringComparison.OrdinalIgnoreCase))
{
string message = string.Format(CultureInfo.CurrentCulture, "Unexpected file path");
//throw new InvalidOperationException(message);
}
using (Stream stream = File.OpenRead(filePath))
using (PEReader peReader = new PEReader(stream))
{
MetadataReader metadataReader = peReader.GetMetadataReader();
foreach (TypeDefinitionHandle typeHandle in metadataReader.TypeDefinitions)
{
// We only care about public types
TypeDefinition typeDefinition = metadataReader.GetTypeDefinition(typeHandle);
// The visibility mask is used to mask out the bits that contain the visibility.
// The visibilities are not combineable, e.g. you can't be both public and private, which is why these aren't independent powers of two.
TypeAttributes visibilityBits = typeDefinition.Attributes & TypeAttributes.VisibilityMask;
if (visibilityBits != TypeAttributes.Public && visibilityBits != TypeAttributes.NestedPublic)
{
continue;
}
string typeName = GetTypeFullName(metadataReader, typeDefinition);
List<string> methodNames = GetTypeMethod(typeDefinition, metadataReader);
List<string> propertyNames = GetTypeProperties(typeDefinition, metadataReader);
List<string> fieldNames = GetTypeFields(typeDefinition, metadataReader);
if (typeMethod.ContainsKey(typeName))
{
List<string> typeMethodNames = typeMethod[typeName];
typeMethodNames.AddRange(methodNames);
List<string> typePropertyNames = typeProperties[typeName];
typePropertyNames.AddRange(propertyNames);
List<string> typeFieldNames = typeFields[typeName];
typeFieldNames.AddRange(fieldNames);
}
else
{
typeMethod.Add(typeName, methodNames);
typeProperties.Add(typeName, propertyNames);
typeFields.Add(typeName, fieldNames);
}
}
}
}
}
/// <summary>
/// Get the full name of a Type.
/// </summary>
private string GetTypeFullName(MetadataReader metadataReader, TypeDefinition typeDefinition)
{
string fullName = String.Empty;
string typeName = metadataReader.GetString(typeDefinition.Name);
string nsName = metadataReader.GetString(typeDefinition.Namespace);
// Get the enclosing type if the type is nested
TypeDefinitionHandle declaringTypeHandle = typeDefinition.GetDeclaringType();
if (declaringTypeHandle.IsNil)
{
fullName = string.Format(CultureInfo.InvariantCulture, "{0}.{1}", nsName, typeName);
}
else
{
fullName = typeName;
while (!declaringTypeHandle.IsNil)
{
TypeDefinition declaringTypeDef = metadataReader.GetTypeDefinition(declaringTypeHandle);
declaringTypeHandle = declaringTypeDef.GetDeclaringType();
if (declaringTypeHandle.IsNil)
{
fullName = string.Format(CultureInfo.InvariantCulture, "{0}.{1}+{2}",
metadataReader.GetString(declaringTypeDef.Namespace),
metadataReader.GetString(declaringTypeDef.Name),
fullName);
}
else
{
fullName = string.Format(CultureInfo.InvariantCulture, "{0}+{1}",
metadataReader.GetString(declaringTypeDef.Name),
fullName);
}
}
}
return fullName;
}
/// <summary>
/// Get all public methods from a given type
/// </summary>
/// <param name="typeDefinition"></param>
/// <param name="metadataReader"></param>
/// <returns></returns>
public List<string> GetTypeMethod(TypeDefinition typeDefinition, MetadataReader metadataReader)
{
List<string> methodNames = new List<string>();
MethodDefinitionHandleCollection methods = typeDefinition.GetMethods();
foreach (MethodDefinitionHandle method in methods)
{
MethodDefinition methodDef = metadataReader.GetMethodDefinition(method);
StringHandle methodName = methodDef.Name;
string name = metadataReader.GetString(methodName);
methodNames.Add(name);
}
return methodNames;
}
/// <summary>
/// Get all properties from a type
/// </summary>
/// <param name="typeDefinition"></param>
/// <param name="metadataReader"></param>
/// <returns></returns>
public List<string> GetTypeProperties(TypeDefinition typeDefinition, MetadataReader metadataReader)
{
List<string> propertyNames = new List<string>();
PropertyDefinitionHandleCollection properties = typeDefinition.GetProperties();
foreach (PropertyDefinitionHandle property in properties)
{
PropertyDefinition propertyDef = metadataReader.GetPropertyDefinition(property);
StringHandle propertyName = propertyDef.Name;
string name = metadataReader.GetString(propertyName);
propertyNames.Add(name);
}
return propertyNames;
}
/// <summary>
/// Get all fields from a given type
/// </summary>
/// <param name="typeDefinition"></param>
/// <param name="metadataReader"></param>
/// <returns></returns>
public List<string> GetTypeFields(TypeDefinition typeDefinition, MetadataReader metadataReader)
{
List<string> fieldNames = new List<string>();
FieldDefinitionHandleCollection fields = typeDefinition.GetFields();
foreach (FieldDefinitionHandle field in fields)
{
FieldDefinition fieldDef = metadataReader.GetFieldDefinition(field);
StringHandle fieldName = fieldDef.Name;
string name = metadataReader.GetString(fieldName);
fieldNames.Add(name);
}
return fieldNames;
}
}
}