MethodLogger 修改dll 记录方法的开始和结束(Hook into a method)

如果你想记录一个方法的开始和结束，那么该怎么做呢？

最简单的方法:

public class MyClass
{
    public void MyMethod()
    {
        Log.EnterMethod();
        Console.WriteLine("MyMethod");
        Log.ExitMethod();
    }
}

 

可是，如果已经有了一个程序集，想在每个方法的开始和结束记录方法的执行，那么该如何做呢？

在CodeProject 上有篇文章就介绍了如何处理这种情况：

MethodLogger - Hook into method calls in .NET binaries

本文不打算做翻译，只是随便介绍下：How to hook into a method.

打开原文的project：

TestAppLib 主要是注入的方法:

namespace TestAppLib
{
    public class MethodExecutionLogger
    {
        public static void MethodStarted(string typeName, string methodName, string args)
        {
            Console.WriteLine("Start of" + typeName + "." + methodName + "(" + args + ")");
        }

        public static void MethodCompleted(string typeName, string methodName, string args)
        {
            Console.WriteLine("End of" + typeName + "." + methodName + "(" + args + ")");
        }
    }
}

可以看到主要是记录typeName,methodName,args,当然如果你记录了时间的话，那么做为性能分析也未尝不可.

原文有一个缺陷，就是Winform程序使用console，如果想要看到控制台消息的话，那么应该修改下：

运行程序，点击按钮可以看到效果：

 

打开src里面的项目

，修改命令行参数：

其中路径可能需要修改下自己的TestApp.exe对应的路径：

运行后，再次运行TestApp.exe，可以看到结果如下：

 

MyClass,MyMethod 的IL如下：

MyClass.MyMethod

有的同学觉得很神奇，其实这也比较简单，这就是所谓的元数据编程？，通过修改IL 来实现一些特殊的需要。

通过上面的IL可以看到，MethodLogger 通过修改IL的方式实现了方法的注入。

作者使用的修改IL的库是：PERWAPI (PE Read Write API)

 

最主要的一个方法是这个方法：

ModifyCode

主要也是使用PERWAPI 来读写IL，然后在每个方法中写如BeginMethod和EndMethod。

修改后的MyClass.MyMethod() 大致是这个样子的：