使用反射查找枚举数的方法

时间:2019-11-16 04:27:46

标签: c# reflection cil enumerator ienumerator

背景

我正在使用Harmony Library修改现有代码。现有的C#代码遵循以下结构:

public class ToModify
{
    public override void Update()
    {
        foreach (StatusItemGroup.Entry entry in collection)
        {
            // I am trying to alter an operation at the end of this loop.
        }
    }
}

public class StatusItemGroup
{
    public IEnumerator<Entry> GetEnumerator()
    {
        return items.GetEnumerator();
    }

    private List<Entry> items = new List<Entry>();

    public struct Entry { }
}

由于这种情况,我必须修改正在生成的IL代码,为此,我必须获取目标操作数的MethodInfo。这是目标:

IL_12B6: callvirt  instance bool [mscorlib]System.Collections.IEnumerator::MoveNext()

问题:

如何为枚举数的MethodInfo方法获得MoveNext

我尝试过的事情:

我能想到的一切都产生了null个结果。这是我最基本的尝试:

MethodInfo targetMethod = typeof(IEnumerator<StatusItemGroup.Entry>).GetMethod("MoveNext");

我不明白为什么这行不通,而且我不知道我需要做些什么才能正确获得MethodInfo

2 个答案:

答案 0 :(得分:3)

MoveNext不是在IEnumerator<T>上定义的,而是在IEnumerator继承的非通用IEnumerator<T>上定义的。

接口继承与反射结合起来有点奇怪,因此您需要直接从定义接口的基本接口获取方法信息:

MethodInfo targetMethod = typeof(System.Collections.IEnumerator).GetMethod("MoveNext");

答案 1 :(得分:1)

使用免费的LinqPad,我使用Harmony 2.0 RC2创建它。如您所见,我使用传递后缀来更改枚举器并将其包装。还有其他方法,我怀疑您实际上在某个地方有IEnumeration。通过直接在返回IEnumeration的原始方法上使用传递后缀,可以更轻松地进行修补。在这种情况下,无需包装枚举器。

但是我不知道您的完整用例,所以现在,这是可行的示例:

void Main()
{
    var harmony = new Harmony("test");
    harmony.PatchAll();

    var group = new StatusItemGroup();
    var items = new List<StatusItemGroup.Entry>() { StatusItemGroup.Entry.Make("A"), StatusItemGroup.Entry.Make("B") };
    Traverse.Create(group).Field("items").SetValue(items);

    var enumerator = group.GetEnumerator();
    while(enumerator.MoveNext())
        Console.WriteLine(enumerator.Current.id);
}

[HarmonyPatch]
class Patch
{
    public class ProxyEnumerator<T> : IEnumerable<T>
    {
        public IEnumerator<T> enumerator;
        public Func<T, T> transformer;
        IEnumerator IEnumerable.GetEnumerator() { return GetEnumerator(); }
        public IEnumerator<T> GetEnumerator()
        {
            while(enumerator.MoveNext())
                yield return transformer(enumerator.Current);
        }
    } 

    [HarmonyPatch(typeof(StatusItemGroup), "GetEnumerator")]
    static IEnumerator<StatusItemGroup.Entry> Postfix(IEnumerator<StatusItemGroup.Entry> enumerator)
    {
        StatusItemGroup.Entry Transform(StatusItemGroup.Entry entry)
        {
            entry.id += "+";
            return entry;
        }

        var myEnumerator = new ProxyEnumerator<StatusItemGroup.Entry>()
        {
            enumerator = enumerator,
            transformer = Transform
        };
        return myEnumerator.GetEnumerator();
    }
}

public class StatusItemGroup
{
    public IEnumerator<Entry> GetEnumerator()
    {
        return items.GetEnumerator();
    }

    private List<Entry> items = new List<Entry>();

    public struct Entry
    {
        public string id;
        public static Entry Make(string id) { return new Entry() { id = id }; }
    }
}