/Mono.Cecil/MetadataResolver.cs
http://github.com/jbevain/cecil · C# · 391 lines · 278 code · 102 blank · 11 comment · 105 complexity · 91e96837f10f1b3070c7141f179ddeb1 MD5 · raw file
- //
- // Author:
- // Jb Evain (jbevain@gmail.com)
- //
- // Copyright (c) 2008 - 2015 Jb Evain
- // Copyright (c) 2008 - 2011 Novell, Inc.
- //
- // Licensed under the MIT/X11 license.
- //
- using System;
- using Mono.Collections.Generic;
- namespace Mono.Cecil {
- public interface IAssemblyResolver : IDisposable {
- AssemblyDefinition Resolve (AssemblyNameReference name);
- AssemblyDefinition Resolve (AssemblyNameReference name, ReaderParameters parameters);
- }
- public interface IMetadataResolver {
- TypeDefinition Resolve (TypeReference type);
- FieldDefinition Resolve (FieldReference field);
- MethodDefinition Resolve (MethodReference method);
- }
- #if !NET_CORE
- [Serializable]
- #endif
- public sealed class ResolutionException : Exception {
- readonly MemberReference member;
- public MemberReference Member {
- get { return member; }
- }
- public IMetadataScope Scope {
- get {
- var type = member as TypeReference;
- if (type != null)
- return type.Scope;
- var declaring_type = member.DeclaringType;
- if (declaring_type != null)
- return declaring_type.Scope;
- throw new NotSupportedException ();
- }
- }
- public ResolutionException (MemberReference member)
- : base ("Failed to resolve " + member.FullName)
- {
- if (member == null)
- throw new ArgumentNullException ("member");
- this.member = member;
- }
- public ResolutionException (MemberReference member, Exception innerException)
- : base ("Failed to resolve " + member.FullName, innerException)
- {
- if (member == null)
- throw new ArgumentNullException ("member");
- this.member = member;
- }
- #if !NET_CORE
- ResolutionException (
- System.Runtime.Serialization.SerializationInfo info,
- System.Runtime.Serialization.StreamingContext context)
- : base (info, context)
- {
- }
- #endif
- }
- public class MetadataResolver : IMetadataResolver {
- readonly IAssemblyResolver assembly_resolver;
- public IAssemblyResolver AssemblyResolver {
- get { return assembly_resolver; }
- }
- public MetadataResolver (IAssemblyResolver assemblyResolver)
- {
- if (assemblyResolver == null)
- throw new ArgumentNullException ("assemblyResolver");
- assembly_resolver = assemblyResolver;
- }
- public virtual TypeDefinition Resolve (TypeReference type)
- {
- Mixin.CheckType (type);
- type = type.GetElementType ();
- var scope = type.Scope;
- if (scope == null)
- return null;
- switch (scope.MetadataScopeType) {
- case MetadataScopeType.AssemblyNameReference:
- var assembly = assembly_resolver.Resolve ((AssemblyNameReference) scope);
- if (assembly == null)
- return null;
- return GetType (assembly.MainModule, type);
- case MetadataScopeType.ModuleDefinition:
- return GetType ((ModuleDefinition) scope, type);
- case MetadataScopeType.ModuleReference:
- var modules = type.Module.Assembly.Modules;
- var module_ref = (ModuleReference) scope;
- for (int i = 0; i < modules.Count; i++) {
- var netmodule = modules [i];
- if (netmodule.Name == module_ref.Name)
- return GetType (netmodule, type);
- }
- break;
- }
- throw new NotSupportedException ();
- }
- static TypeDefinition GetType (ModuleDefinition module, TypeReference reference)
- {
- var type = GetTypeDefinition (module, reference);
- if (type != null)
- return type;
- if (!module.HasExportedTypes)
- return null;
- var exported_types = module.ExportedTypes;
- for (int i = 0; i < exported_types.Count; i++) {
- var exported_type = exported_types [i];
- if (exported_type.Name != reference.Name)
- continue;
- if (exported_type.Namespace != reference.Namespace)
- continue;
- return exported_type.Resolve ();
- }
- return null;
- }
- static TypeDefinition GetTypeDefinition (ModuleDefinition module, TypeReference type)
- {
- if (!type.IsNested)
- return module.GetType (type.Namespace, type.Name);
- var declaring_type = type.DeclaringType.Resolve ();
- if (declaring_type == null)
- return null;
- return declaring_type.GetNestedType (type.TypeFullName ());
- }
- public virtual FieldDefinition Resolve (FieldReference field)
- {
- Mixin.CheckField (field);
- var type = Resolve (field.DeclaringType);
- if (type == null)
- return null;
- if (!type.HasFields)
- return null;
- return GetField (type, field);
- }
- FieldDefinition GetField (TypeDefinition type, FieldReference reference)
- {
- while (type != null) {
- var field = GetField (type.Fields, reference);
- if (field != null)
- return field;
- if (type.BaseType == null)
- return null;
- type = Resolve (type.BaseType);
- }
- return null;
- }
- static FieldDefinition GetField (Collection<FieldDefinition> fields, FieldReference reference)
- {
- for (int i = 0; i < fields.Count; i++) {
- var field = fields [i];
- if (field.Name != reference.Name)
- continue;
- if (!AreSame (field.FieldType, reference.FieldType))
- continue;
- return field;
- }
- return null;
- }
- public virtual MethodDefinition Resolve (MethodReference method)
- {
- Mixin.CheckMethod (method);
- var type = Resolve (method.DeclaringType);
- if (type == null)
- return null;
- method = method.GetElementMethod ();
- if (!type.HasMethods)
- return null;
- return GetMethod (type, method);
- }
- MethodDefinition GetMethod (TypeDefinition type, MethodReference reference)
- {
- while (type != null) {
- var method = GetMethod (type.Methods, reference);
- if (method != null)
- return method;
- if (type.BaseType == null)
- return null;
- type = Resolve (type.BaseType);
- }
- return null;
- }
- public static MethodDefinition GetMethod (Collection<MethodDefinition> methods, MethodReference reference)
- {
- for (int i = 0; i < methods.Count; i++) {
- var method = methods [i];
- if (method.Name != reference.Name)
- continue;
- if (method.HasGenericParameters != reference.HasGenericParameters)
- continue;
- if (method.HasGenericParameters && method.GenericParameters.Count != reference.GenericParameters.Count)
- continue;
- if (!AreSame (method.ReturnType, reference.ReturnType))
- continue;
- if (method.IsVarArg () != reference.IsVarArg ())
- continue;
- if (method.IsVarArg () && IsVarArgCallTo (method, reference))
- return method;
- if (method.HasParameters != reference.HasParameters)
- continue;
- if (!method.HasParameters && !reference.HasParameters)
- return method;
- if (!AreSame (method.Parameters, reference.Parameters))
- continue;
- return method;
- }
- return null;
- }
- static bool AreSame (Collection<ParameterDefinition> a, Collection<ParameterDefinition> b)
- {
- var count = a.Count;
- if (count != b.Count)
- return false;
- if (count == 0)
- return true;
- for (int i = 0; i < count; i++)
- if (!AreSame (a [i].ParameterType, b [i].ParameterType))
- return false;
- return true;
- }
- static bool IsVarArgCallTo (MethodDefinition method, MethodReference reference)
- {
- if (method.Parameters.Count >= reference.Parameters.Count)
- return false;
- if (reference.GetSentinelPosition () != method.Parameters.Count)
- return false;
- for (int i = 0; i < method.Parameters.Count; i++)
- if (!AreSame (method.Parameters [i].ParameterType, reference.Parameters [i].ParameterType))
- return false;
- return true;
- }
- static bool AreSame (TypeSpecification a, TypeSpecification b)
- {
- if (!AreSame (a.ElementType, b.ElementType))
- return false;
- if (a.IsGenericInstance)
- return AreSame ((GenericInstanceType) a, (GenericInstanceType) b);
- if (a.IsRequiredModifier || a.IsOptionalModifier)
- return AreSame ((IModifierType) a, (IModifierType) b);
- if (a.IsArray)
- return AreSame ((ArrayType) a, (ArrayType) b);
- return true;
- }
- static bool AreSame (ArrayType a, ArrayType b)
- {
- if (a.Rank != b.Rank)
- return false;
- // TODO: dimensions
- return true;
- }
- static bool AreSame (IModifierType a, IModifierType b)
- {
- return AreSame (a.ModifierType, b.ModifierType);
- }
- static bool AreSame (GenericInstanceType a, GenericInstanceType b)
- {
- if (a.GenericArguments.Count != b.GenericArguments.Count)
- return false;
- for (int i = 0; i < a.GenericArguments.Count; i++)
- if (!AreSame (a.GenericArguments [i], b.GenericArguments [i]))
- return false;
- return true;
- }
- static bool AreSame (GenericParameter a, GenericParameter b)
- {
- return a.Position == b.Position;
- }
- static bool AreSame (TypeReference a, TypeReference b)
- {
- if (ReferenceEquals (a, b))
- return true;
- if (a == null || b == null)
- return false;
- if (a.etype != b.etype)
- return false;
- if (a.IsGenericParameter)
- return AreSame ((GenericParameter) a, (GenericParameter) b);
- if (a.IsTypeSpecification ())
- return AreSame ((TypeSpecification) a, (TypeSpecification) b);
- if (a.Name != b.Name || a.Namespace != b.Namespace)
- return false;
- //TODO: check scope
- return AreSame (a.DeclaringType, b.DeclaringType);
- }
- }
- }