392 lines
8.8 KiB
C#
392 lines
8.8 KiB
C#
|
//
|
||
|
// 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 MonoFN.Collections.Generic;
|
||
|
using System;
|
||
|
|
||
|
namespace MonoFN.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:
|
||
|
if (type.Module.Assembly == null)
|
||
|
return null;
|
||
|
|
||
|
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;
|
||
|
|
||
|
return AreSame (a.DeclaringType, b.DeclaringType);
|
||
|
}
|
||
|
}
|
||
|
}
|