将代码注入到没有自定义属性的所有方法和属性的最简单方法

Question

AOP 有很多关于 .NET 位于 Stack Overflow 上，经常提到 PostSharp 和其他第三方产品。因此，.NET 和 C# 世界中似乎有相当多的 AOP 选项。但其中每一个都有其限制，在下载了有前途的 PostSharp 之后，我在他们的文档中发现“方法必须是虚拟的”以便能够注入代码（编辑：请参阅 ChrisWue 的答案和我的评论 - 我想，虚拟约束一定是在竞争者之一身上）。我没有进一步调查这个陈述的准确性，但它的明确性让我回到了 StackOverflow。

所以我想得到这个非常具体的问题的答案：

我想注入简单“if（某些条件）Console.WriteLine”样式代码到每个方法和属性 strong>（静态、密封、内部、虚拟、非虚拟，无关紧要）在我的项目中没有自定义注释，以便在运行时动态测试我的软件。此注入的代码不应保留在发布版本中，它仅用于开发期间的动态测试（与线程相关）。

做到这一点最简单的方法是什么？我偶然发现了 Mono.Cecil，它看起来很理想，只是你似乎必须编写代码您想要将其注入 IL 中。这不是一个大问题，使用 Mono.Cecil 很容易获得用 C# 编写的代码的 IL 版本。但尽管如此，如果有更简单的东西，最好甚至内置到 .NET 中（我仍然使用 .NET 3.5），我想知道。 [更新：如果建议的工具不是 .NET Framework 的一部分，那么如果它是开源的，例如 Mono.Cecil，或者免费提供，那就太好了]

Answer 1

我能够使用 Mono.Cecil 解决这个问题。我仍然对它的易学、易用和强大感到惊讶。几乎完全缺乏文档并没有改变这一点。

这些是我使用的 3 个文档来源：

• static -method-interception-in-net-with-c-and-monocecil
• 迁移到0.9
• 源代码本身

第一个链接提供了非常温和的介绍，但由于它描述了 Cecil 的旧版本 - 并且在此期间发生了很多变化 - 第二个链接对于将介绍翻译为 Cecil 0.9 非常有帮助。开始使用后，源代码（也未记录）非常宝贵，并回答了我的每一个问题 - 可能除了有关 .NET 平台的一般问题外，但我确信网上有大量关于这方面的书籍和材料。

我现在可以获取 DLL 或 EXE 文件，对其进行修改，然后将其写回磁盘。我唯一还没有做的事情是弄清楚如何保留调试信息——文件名、行号等，目前在写入 DLL 或 EXE 文件后会丢失。我的背景不是 .NET，所以我在这里猜测，我的猜测是我需要查看 mono.cecil.pdb 来解决这个问题。稍后的某个地方——现在对我来说并不是那么重要。我正在创建这个 EXE 文件，运行该应用程序 - 这是一个复杂的 GUI 应用程序，经过多年的发展，带有您期望在这样一个软件中找到的所有包袱 - 它会检查内容并记录错误我。

这是我的代码的要点：

DefaultAssemblyResolver assemblyResolver = new DefaultAssemblyResolver();
// so it won't complain about not finding assemblies sitting in the same directory as the dll/exe we are going to patch
assemblyResolver.AddSearchDirectory(assemblyDirectory);
var readerParameters = new ReaderParameters { AssemblyResolver = assemblyResolver };

AssemblyDefinition assembly = AssemblyDefinition.ReadAssembly(assemblyFilename, readerParameters);

foreach (var moduleDefinition in assembly.Modules)
{
    foreach (var type in ModuleDefinitionRocks.GetAllTypes(moduleDefinition))
    {
        foreach (var method in type.Methods)
        {
            if (!HasAttribute("MyCustomAttribute", method.method.CustomAttributes)
            {
              ILProcessor ilProcessor = method.Body.GetILProcessor();
              ilProcessor.InsertBefore(method.Body.Instructions.First(), ilProcessor.Create(OpCodes.Call, threadCheckerMethod));
// ...

private static bool HasAttribute(string attributeName, IEnumerable<CustomAttribute> customAttributes)
{
    return GetAttributeByName(attributeName, customAttributes) != null;
}

private static CustomAttribute GetAttributeByName(string attributeName, IEnumerable<CustomAttribute> customAttributes)
{
    foreach (var attribute in customAttributes)
        if (attribute.AttributeType.FullName == attributeName)
            return attribute;
    return null;
}

如果有人知道如何更简单地完成此操作，我仍然对答案感兴趣，并且我不会将其标记为解决方案 - 除非没有更简单的解决方案出现。

Answer 2

我不确定您从哪里得知方法必须是虚拟的。我们使用 Postsharp 来计时并记录对 WCF 服务接口实现的调用，利用 OnMethodBoundaryAspect 来创建我们可以用来装饰类的属性。快速示例：

[Serializable]
public class LogMethodCallAttribute : OnMethodBoundaryAspect
{
    public Type FilterAttributeType { get; set; }

    public LogMethodCallAttribute(Type filterAttributeType)
    {
        FilterAttributeType = filterAttributeType;
    }

    public override void OnEntry(MethodExecutionEventArgs eventArgs)
    {
        if (!Proceed(eventArgs)) return;
        Console.WriteLine(GetMethodName(eventArgs));
    }

    public override void OnException(MethodExecutionEventArgs eventArgs)
    {
        if (!Proceed(eventArgs)) return;
        Console.WriteLine(string.Format("Exception at {0}:\n{1}", 
                GetMethodName(eventArgs), eventArgs.Exception));
    }

    public override void OnExit(MethodExecutionEventArgs eventArgs)
    {
        if (!Proceed(eventArgs)) return;
         Console.WriteLine(string.Format("{0} returned {1}", 
                GetMethodName(eventArgs), eventArgs.ReturnValue));
    }
    private string GetMethodName(MethodExecutionEventArgs eventArgs)
    {
        return string.Format("{0}.{1}", eventArgs.Method.DeclaringType, eventArgs.Method.Name);
    }
    private bool Proceed(MethodExecutionEventArgs eventArgs)
    {
         return Attribute.GetCustomAttributes(eventArgs.Method, FilterAttributeType).Length == 0;
    }
}

然后像这样使用它：

 [LogMethodCallAttribute(typeof(MyCustomAttribute))]
 class MyClass
 {
      public class LogMe()
      {
      }

      [MyCustomAttribute]
      public class DoNotLogMe()
      {
      }
 }

在 Postsharp 1.5.6 中像魅力一样工作，无需使任何方法虚拟。也许他们已经在 2.x 中更改了它，但我当然不希望如此 - 这会使它变得不那么有用。

更新：我不确定您是否可以轻松说服 Postsharp 仅根据装饰的属性将代码注入到某些方法中。如果您查看本教程< /a> 它仅显示过滤类型和方法名称的方法。我们通过将要检查的类型传递到属性中解决了这个问题，然后在 OnEntry 中您可以使用反射来查找属性并决定是否记录。其结果会被缓存，因此您只需在第一次调用时执行此操作。

我调整了上面的代码来演示这个想法。