.NET 8中通过堆栈检测TearDownAttribute失效问题求助

1. InvokeStub 机制解释 #

.NET 8引入的**动态调用存根（InvokeStub）**是运行时的性能优化手段，专门用于提升反射调用的效率。当框架（比如NUnit）通过反射调用方法时，运行时会自动生成一个存根方法作为中间代理层：

• 存根会适配反射调用的统一参数约定（比如你看到的Object, Object, IntPtr*签名），避免每次反射调用都重复做参数适配的开销。
• 这个存根是运行时动态生成的临时方法，不会携带原方法的特性（比如[TearDown]），因为它只是代理执行原方法的中间层，并非原方法的副本。
• 你的场景中，NUnit通过反射调用TestBase.TearDown时，.NET 8生成了InvokeStub_TestBase.TearDown存根，导致堆栈中出现这个代理方法，而你的校验逻辑找不到[TearDown]特性，最终抛出异常。

2. 修复方案 #

针对这个问题，有几种可行的修复思路：

方案一：跳过堆栈中的存根方法 #

修改IsMethodWithAttributeOnStack方法，直接跳过名称以InvokeStub_开头的代理方法，继续查找后续堆栈帧中的原方法：

private static bool IsMethodWithAttributeOnStack<TAttribute>()
    where TAttribute : Attribute
{
    var stack = new StackTrace();
    foreach (var frame in stack.GetFrames())
    {
        var method = frame.GetMethod();
        if (method == null) continue;
        
        // 跳过InvokeStub代理方法，继续遍历堆栈
        if (method.DeclaringType?.Name.StartsWith("InvokeStub_") ?? false)
            continue;
            
        if (method.HasAttribute<TAttribute>())
            return true;
    }
    return false;
}

方案二：从存根中解析出原方法 #

如果需要更严谨的校验，可以通过解析存根方法的IL代码，找到它代理的原方法，再检查原方法的特性。注意这种方式依赖.NET内部实现，可能存在版本兼容性风险：

using System.Reflection.Emit;

private static MethodInfo GetOriginalMethodFromStub(MethodInfo stubMethod)
{
    var ilBytes = stubMethod.GetMethodBody()?.GetILAsByteArray();
    if (ilBytes == null || ilBytes.Length < 5) return null;
    
    // 查找IL中的call或callvirt指令，解析出原方法的元数据令牌
    if (ilBytes[0] == 0x28 || ilBytes[0] == 0x6F)
    {
        int token = BitConverter.ToInt32(ilBytes, 1);
        return stubMethod.Module.ResolveMethod(token);
    }
    return null;
}

// 修改后的校验逻辑
private static bool IsMethodWithAttributeOnStack<TAttribute>()
    where TAttribute : Attribute
{
    var stack = new StackTrace();
    foreach (var frame in stack.GetFrames())
    {
        var method = frame.GetMethod();
        if (method == null) continue;
        
        // 如果是存根方法，尝试获取它代理的原方法
        if (method.DeclaringType?.Name.StartsWith("InvokeStub_") ?? false)
        {
            method = GetOriginalMethodFromStub(method);
            if (method == null) continue;
        }
            
        if (method.HasAttribute<TAttribute>())
            return true;
    }
    return false;
}

方案三：从设计层面约束调用 #

既然堆栈校验依赖运行时实现，容易受版本更新影响，可以换一种方式避免非法调用：

• 将OnTearDown设为protected internal，仅允许TestBase类或内部标记了[TearDown]的方法调用。
• 强制要求客户通过继承TestBase来使用清理逻辑，而不是直接调用OnTearDown，从根源上杜绝非法调用场景。

方案四：临时禁用反射优化 #

如果需要快速临时解决问题，可以在项目中禁用.NET 8的动态调用存根优化，但这会影响反射调用的性能，仅作为应急方案：
在项目的.csproj文件中添加以下配置：

<PropertyGroup>
  <EnableDynamicInvokeStubs>false</EnableDynamicInvokeStubs>
</PropertyGroup>

内容的提问来源于stack exchange，提问作者Arnon Axelrod