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

最新推荐文章于 2020-04-22 18:30:51 发布

weixin_34380948

最新推荐文章于 2020-04-22 18:30:51 发布

阅读量53

点赞数

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

最简单的方法:

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.method public hidebysig instance void  MyMethod() cil managed
{
  // 代码大小       53 (0x35)
  .maxstack  3
  IL_0000:  ldstr      "MyClass"
  IL_0005:  ldstr      "MyMethod"
  IL_000a:  ldstr      ""
  IL_000f:  call       void [TestAppLib]TestAppLib.MethodExecutionLogger::MethodStarted(string,
                                                                                        string,
                                                                                        string)
  .try
  {
    IL_0014:  nop
    IL_0015:  ldstr      "MyMethod"
    IL_001a:  call       void [mscorlib]System.Console::WriteLine(string)
    IL_001f:  nop
  }  // end .try
  finally
  {
    IL_0020:  ldstr      "MyClass"
    IL_0025:  ldstr      "MyMethod"
    IL_002a:  ldstr      ""
    IL_002f:  call       void [TestAppLib]TestAppLib.MethodExecutionLogger::MethodCompleted(string,
                                                                                            string,
                                                                                            string)
  }  // end handler
  IL_0034:  ret
} // end of method MyClass::MyMethod

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

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

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

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

ModifyCode        static void ModifyCode(ClassDef classDef, MethodDef methodDef, Method startLogMethod, Method endLogMethod)
        {
            string classNameString = MethodLoggerUtil.GetQualifiedClassName(classDef);
            string methodNameString = methodDef.Name();
            string paramsString = MethodLoggerUtil.GetParamsAsString(methodDef.GetParams());

            Param[] parms = methodDef.GetParams();

            // We'll be pushing typeName, methodName and parameters as string parameters, so set max stack size to 3.
            if (methodDef.GetMaxStack() < 3)
            {
                methodDef.SetMaxStack(3);
            }

            CILInstructions instructions = methodDef.GetCodeBuffer();

            instructions.StartInsert();
            instructions.ldstr(classNameString);
            instructions.ldstr(methodNameString);
            instructions.ldstr(paramsString);
            instructions.MethInst(MethodOp.call, startLogMethod);

            instructions.StartBlock();
            
            instructions.EndInsert();


            while (instructions.GetNextInstruction().GetPos() < instructions.NumInstructions() - 2) ;

            instructions.StartInsert();

            TryBlock tryBlock = instructions.EndTryBlock();
            instructions.StartBlock();
            instructions.ldstr(classNameString);
            instructions.ldstr(methodNameString);
            instructions.ldstr(paramsString);
            instructions.MethInst(MethodOp.call, endLogMethod);
            instructions.EndFinallyBlock(tryBlock);

            instructions.EndInsert();
        }