MetadataResolver.cs 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370
  1. //
  2. // MetadataResolver.cs
  3. //
  4. // Author:
  5. // Jb Evain (jbevain@gmail.com)
  6. //
  7. // Copyright (c) 2008 - 2011 Jb Evain
  8. //
  9. // Permission is hereby granted, free of charge, to any person obtaining
  10. // a copy of this software and associated documentation files (the
  11. // "Software"), to deal in the Software without restriction, including
  12. // without limitation the rights to use, copy, modify, merge, publish,
  13. // distribute, sublicense, and/or sell copies of the Software, and to
  14. // permit persons to whom the Software is furnished to do so, subject to
  15. // the following conditions:
  16. //
  17. // The above copyright notice and this permission notice shall be
  18. // included in all copies or substantial portions of the Software.
  19. //
  20. // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  21. // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  22. // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  23. // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  24. // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  25. // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  26. // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  27. //
  28. using System;
  29. using Mono.Collections.Generic;
  30. namespace Mono.Cecil
  31. {
  32. public interface IAssemblyResolver
  33. {
  34. AssemblyDefinition Resolve(AssemblyNameReference name);
  35. AssemblyDefinition Resolve(AssemblyNameReference name, ReaderParameters parameters);
  36. AssemblyDefinition Resolve(string fullName);
  37. AssemblyDefinition Resolve(string fullName, ReaderParameters parameters);
  38. event AssemblyResolveEventHandler ResolveFailure;
  39. }
  40. public interface IMetadataResolver
  41. {
  42. TypeDefinition Resolve(TypeReference type);
  43. FieldDefinition Resolve(FieldReference field);
  44. MethodDefinition Resolve(MethodReference method);
  45. }
  46. public class ResolutionException : Exception
  47. {
  48. readonly MemberReference member;
  49. public MemberReference Member
  50. {
  51. get { return member; }
  52. }
  53. public ResolutionException(MemberReference member)
  54. : base("Failed to resolve " + member.FullName)
  55. {
  56. this.member = member;
  57. }
  58. }
  59. public class MetadataResolver : IMetadataResolver
  60. {
  61. readonly IAssemblyResolver assembly_resolver;
  62. public IAssemblyResolver AssemblyResolver
  63. {
  64. get { return assembly_resolver; }
  65. }
  66. public MetadataResolver(IAssemblyResolver assemblyResolver)
  67. {
  68. if (assemblyResolver == null)
  69. throw new ArgumentNullException("assemblyResolver");
  70. assembly_resolver = assemblyResolver;
  71. }
  72. public virtual TypeDefinition Resolve(TypeReference type)
  73. {
  74. if (type == null)
  75. throw new ArgumentNullException("type");
  76. type = type.GetElementType();
  77. var scope = type.Scope;
  78. switch (scope.MetadataScopeType)
  79. {
  80. case MetadataScopeType.AssemblyNameReference:
  81. var assembly = assembly_resolver.Resolve((AssemblyNameReference)scope);
  82. if (assembly == null)
  83. return null;
  84. return GetType(assembly.MainModule, type);
  85. case MetadataScopeType.ModuleDefinition:
  86. return GetType((ModuleDefinition)scope, type);
  87. case MetadataScopeType.ModuleReference:
  88. var modules = type.Module.Assembly.Modules;
  89. var module_ref = (ModuleReference)scope;
  90. for (int i = 0; i < modules.Count; i++)
  91. {
  92. var netmodule = modules[i];
  93. if (netmodule.Name == module_ref.Name)
  94. return GetType(netmodule, type);
  95. }
  96. break;
  97. }
  98. throw new NotSupportedException();
  99. }
  100. static TypeDefinition GetType(ModuleDefinition module, TypeReference reference)
  101. {
  102. var type = GetTypeDefinition(module, reference);
  103. if (type != null)
  104. return type;
  105. if (!module.HasExportedTypes)
  106. return null;
  107. var exported_types = module.ExportedTypes;
  108. for (int i = 0; i < exported_types.Count; i++)
  109. {
  110. var exported_type = exported_types[i];
  111. if (exported_type.Name != reference.Name)
  112. continue;
  113. if (exported_type.Namespace != reference.Namespace)
  114. continue;
  115. return exported_type.Resolve();
  116. }
  117. return null;
  118. }
  119. static TypeDefinition GetTypeDefinition(ModuleDefinition module, TypeReference type)
  120. {
  121. if (!type.IsNested)
  122. return module.GetType(type.Namespace, type.Name);
  123. var declaring_type = type.DeclaringType.Resolve();
  124. if (declaring_type == null)
  125. return null;
  126. return Mixin.GetNestedType(declaring_type, type.Name);
  127. }
  128. public virtual FieldDefinition Resolve(FieldReference field)
  129. {
  130. if (field == null)
  131. throw new ArgumentNullException("field");
  132. var type = Resolve(field.DeclaringType);
  133. if (type == null)
  134. return null;
  135. if (!type.HasFields)
  136. return null;
  137. return GetField(type, field);
  138. }
  139. FieldDefinition GetField(TypeDefinition type, FieldReference reference)
  140. {
  141. while (type != null)
  142. {
  143. var field = GetField(type.Fields, reference);
  144. if (field != null)
  145. return field;
  146. if (type.BaseType == null)
  147. return null;
  148. type = Resolve(type.BaseType);
  149. }
  150. return null;
  151. }
  152. static FieldDefinition GetField(Collection<FieldDefinition> fields, FieldReference reference)
  153. {
  154. for (int i = 0; i < fields.Count; i++)
  155. {
  156. var field = fields[i];
  157. if (field.Name != reference.Name)
  158. continue;
  159. if (!AreSame(field.FieldType, reference.FieldType))
  160. continue;
  161. return field;
  162. }
  163. return null;
  164. }
  165. public virtual MethodDefinition Resolve(MethodReference method)
  166. {
  167. if (method == null)
  168. throw new ArgumentNullException("method");
  169. var type = Resolve(method.DeclaringType);
  170. if (type == null)
  171. return null;
  172. method = method.GetElementMethod();
  173. if (!type.HasMethods)
  174. return null;
  175. return GetMethod(type, method);
  176. }
  177. MethodDefinition GetMethod(TypeDefinition type, MethodReference reference)
  178. {
  179. while (type != null)
  180. {
  181. var method = GetMethod(type.Methods, reference);
  182. if (method != null)
  183. return method;
  184. if (type.BaseType == null)
  185. return null;
  186. type = Resolve(type.BaseType);
  187. }
  188. return null;
  189. }
  190. public static MethodDefinition GetMethod(Collection<MethodDefinition> methods, MethodReference reference)
  191. {
  192. for (int i = 0; i < methods.Count; i++)
  193. {
  194. var method = methods[i];
  195. if (method.Name != reference.Name)
  196. continue;
  197. if (method.HasGenericParameters != reference.HasGenericParameters)
  198. continue;
  199. if (method.HasGenericParameters && method.GenericParameters.Count != reference.GenericParameters.Count)
  200. continue;
  201. if (!AreSame(method.ReturnType, reference.ReturnType))
  202. continue;
  203. if (method.HasParameters != reference.HasParameters)
  204. continue;
  205. if (!method.HasParameters && !reference.HasParameters)
  206. return method;
  207. if (!AreSame(method.Parameters, reference.Parameters))
  208. continue;
  209. return method;
  210. }
  211. return null;
  212. }
  213. static bool AreSame(Collection<ParameterDefinition> a, Collection<ParameterDefinition> b)
  214. {
  215. var count = a.Count;
  216. if (count != b.Count)
  217. return false;
  218. if (count == 0)
  219. return true;
  220. for (int i = 0; i < count; i++)
  221. if (!AreSame(a[i].ParameterType, b[i].ParameterType))
  222. return false;
  223. return true;
  224. }
  225. static bool AreSame(TypeSpecification a, TypeSpecification b)
  226. {
  227. if (!AreSame(a.ElementType, b.ElementType))
  228. return false;
  229. if (a.IsGenericInstance)
  230. return AreSame((GenericInstanceType)a, (GenericInstanceType)b);
  231. if (a.IsRequiredModifier || a.IsOptionalModifier)
  232. return AreSame((IModifierType)a, (IModifierType)b);
  233. if (a.IsArray)
  234. return AreSame((ArrayType)a, (ArrayType)b);
  235. return true;
  236. }
  237. static bool AreSame(ArrayType a, ArrayType b)
  238. {
  239. if (a.Rank != b.Rank)
  240. return false;
  241. // TODO: dimensions
  242. return true;
  243. }
  244. static bool AreSame(IModifierType a, IModifierType b)
  245. {
  246. return AreSame(a.ModifierType, b.ModifierType);
  247. }
  248. static bool AreSame(GenericInstanceType a, GenericInstanceType b)
  249. {
  250. if (a.GenericArguments.Count != b.GenericArguments.Count)
  251. return false;
  252. for (int i = 0; i < a.GenericArguments.Count; i++)
  253. if (!AreSame(a.GenericArguments[i], b.GenericArguments[i]))
  254. return false;
  255. return true;
  256. }
  257. static bool AreSame(GenericParameter a, GenericParameter b)
  258. {
  259. return a.Position == b.Position;
  260. }
  261. static bool AreSame(TypeReference a, TypeReference b)
  262. {
  263. if (ReferenceEquals(a, b))
  264. return true;
  265. if (a == null || b == null)
  266. return false;
  267. if (a.etype != b.etype)
  268. return false;
  269. if (a.IsGenericParameter)
  270. return AreSame((GenericParameter)a, (GenericParameter)b);
  271. if (Mixin.IsTypeSpecification(a))
  272. return AreSame((TypeSpecification)a, (TypeSpecification)b);
  273. if (a.Name != b.Name || a.Namespace != b.Namespace)
  274. return false;
  275. //TODO: check scope
  276. return AreSame(a.DeclaringType, b.DeclaringType);
  277. }
  278. }
  279. }