2011-09-28 04:06:43 +08:00
|
|
|
|
/*
|
2013-01-02 00:03:16 +08:00
|
|
|
|
Copyright (C) 2011-2013 de4dot@gmail.com
|
2011-09-28 04:06:43 +08:00
|
|
|
|
|
|
|
|
|
This file is part of de4dot.
|
|
|
|
|
|
|
|
|
|
de4dot is free software: you can redistribute it and/or modify
|
|
|
|
|
it under the terms of the GNU General Public License as published by
|
|
|
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
|
|
|
(at your option) any later version.
|
|
|
|
|
|
|
|
|
|
de4dot is distributed in the hope that it will be useful,
|
|
|
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
|
GNU General Public License for more details.
|
|
|
|
|
|
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
|
|
|
along with de4dot. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
using System;
|
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
using System.Reflection;
|
2012-12-20 09:06:09 +08:00
|
|
|
|
using dnlib.DotNet;
|
2011-09-28 04:06:43 +08:00
|
|
|
|
using de4dot.blocks;
|
|
|
|
|
|
|
|
|
|
namespace AssemblyData.methodsrewriter {
|
|
|
|
|
class TypeInstanceResolver {
|
|
|
|
|
Type type;
|
|
|
|
|
Dictionary<string, List<MethodBase>> methods;
|
|
|
|
|
Dictionary<string, List<FieldInfo>> fields;
|
|
|
|
|
|
2012-11-01 05:44:54 +08:00
|
|
|
|
public TypeInstanceResolver(Type type, ITypeDefOrRef typeRef) {
|
2013-01-19 20:03:57 +08:00
|
|
|
|
this.type = ResolverUtils.MakeInstanceType(type, typeRef);
|
2011-09-28 04:06:43 +08:00
|
|
|
|
}
|
|
|
|
|
|
2013-01-19 20:03:57 +08:00
|
|
|
|
public FieldInfo Resolve(IField fieldRef) {
|
|
|
|
|
InitFields();
|
2011-09-28 04:06:43 +08:00
|
|
|
|
|
|
|
|
|
List<FieldInfo> list;
|
2012-11-01 14:34:40 +08:00
|
|
|
|
if (!fields.TryGetValue(fieldRef.Name.String, out list))
|
2011-09-28 04:06:43 +08:00
|
|
|
|
return null;
|
|
|
|
|
|
2013-01-19 20:03:57 +08:00
|
|
|
|
fieldRef = GenericArgsSubstitutor.Create(fieldRef, fieldRef.DeclaringType.TryGetGenericInstSig());
|
2011-09-28 04:06:43 +08:00
|
|
|
|
|
|
|
|
|
foreach (var field in list) {
|
2013-01-19 20:03:57 +08:00
|
|
|
|
if (ResolverUtils.CompareFields(field, fieldRef))
|
2011-09-28 04:06:43 +08:00
|
|
|
|
return field;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return null;
|
|
|
|
|
}
|
|
|
|
|
|
2013-01-19 20:03:57 +08:00
|
|
|
|
void InitFields() {
|
2011-09-28 04:06:43 +08:00
|
|
|
|
if (fields != null)
|
|
|
|
|
return;
|
|
|
|
|
fields = new Dictionary<string, List<FieldInfo>>(StringComparer.Ordinal);
|
|
|
|
|
|
|
|
|
|
var flags = BindingFlags.DeclaredOnly | BindingFlags.Public | BindingFlags.NonPublic | BindingFlags.Static | BindingFlags.Instance;
|
|
|
|
|
foreach (var field in type.GetFields(flags)) {
|
|
|
|
|
List<FieldInfo> list;
|
|
|
|
|
if (!fields.TryGetValue(field.Name, out list))
|
|
|
|
|
fields[field.Name] = list = new List<FieldInfo>();
|
|
|
|
|
list.Add(field);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2013-01-19 20:03:57 +08:00
|
|
|
|
public MethodBase Resolve(IMethod methodRef) {
|
|
|
|
|
InitMethods();
|
2011-09-28 04:06:43 +08:00
|
|
|
|
|
|
|
|
|
List<MethodBase> list;
|
2012-11-01 14:34:40 +08:00
|
|
|
|
if (!methods.TryGetValue(methodRef.Name.String, out list))
|
2011-09-28 04:06:43 +08:00
|
|
|
|
return null;
|
|
|
|
|
|
2013-01-19 20:03:57 +08:00
|
|
|
|
methodRef = GenericArgsSubstitutor.Create(methodRef, methodRef.DeclaringType.TryGetGenericInstSig());
|
2011-09-28 04:06:43 +08:00
|
|
|
|
|
|
|
|
|
foreach (var method in list) {
|
2013-01-19 20:03:57 +08:00
|
|
|
|
if (ResolverUtils.CompareMethods(method, methodRef))
|
2011-09-28 04:06:43 +08:00
|
|
|
|
return method;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return null;
|
|
|
|
|
}
|
|
|
|
|
|
2013-01-19 20:03:57 +08:00
|
|
|
|
void InitMethods() {
|
2011-09-28 04:06:43 +08:00
|
|
|
|
if (methods != null)
|
|
|
|
|
return;
|
|
|
|
|
methods = new Dictionary<string, List<MethodBase>>(StringComparer.Ordinal);
|
|
|
|
|
|
|
|
|
|
var flags = BindingFlags.DeclaredOnly | BindingFlags.Public | BindingFlags.NonPublic | BindingFlags.Static | BindingFlags.Instance;
|
2013-01-19 20:03:57 +08:00
|
|
|
|
foreach (var method in ResolverUtils.GetMethodBases(type, flags)) {
|
2011-09-28 04:06:43 +08:00
|
|
|
|
List<MethodBase> list;
|
|
|
|
|
if (!methods.TryGetValue(method.Name, out list))
|
|
|
|
|
methods[method.Name] = list = new List<MethodBase>();
|
|
|
|
|
list.Add(method);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|