[韩小明]自动化测试框架：用AOP为每一个操作写Log

在写这个自动化测试框架的时候，我一直在留意各方面的需求。毕竟，我本人并没有做过真正的自动化测试。管理测试方面的领导，提出一个需求，就是在用例运行失败的时候，应该将过程记录下来，并形成报告，Email给相关人员。

个人认为这个需求是非常合理的。事实上，任何系统，如果没有输出，那么只能停留在程序员手里。有了报表，才叫真正解决了用户的目标需求。

在分析这个需求的过程，我提出了针对每一个操作接口的每一个方法，进行Log。而完成这个工作的第一方法，就想到了AOP，也就是Hook技术的应用。因为Delphi下面并没有对AOP的直接支持，所以考虑这个实现，变成了一个技术研究过程。

从技术上讲，本篇博客只适合了解VCL的Delphi程序员阅读。但其间的思想，相信大家都可以借鉴。下面我的描述过程，是以我的探索过程来进行讲述的。中间会带出相关技术点，供大家参考。

第一、接口的方法，是由类来实现的。框架中，已经对所有支持的类都进行了登记。那么，只需要在这些类型中，找到所有实现的接口的所有方法的地址，那么Hook就变得有可能了。

TObject有一个方法：GetInterfaceTable，可以获取所有接口列表。所有非常容易找到接口对应的VTable。VTable在Delphi中并没有明确的注释，但是可以知道VTable是一个指针列表，每一项都记录着一个方法的“实现地址”。

可惜的是，我发现VTable本身并没有告诉你，这个接口有多少个方法！另外，你也不能得到每一个方法的名称，以及参数等等描述。

第二、于是我考虑到接口的RTTI。接口的RTTI，我以前是没有使用过的。通过VCL的代码研究，发现接口中有一个非常特殊的接口定义：{$M+}IInvokable = interface;{$M-}。这个接口本身并没有添加什么服务，只是使用了编译指令M，来使得接口拥有了RTTI。

实现的时候，可以通过从IInvokable派生，或者直接添加编译指令，从而获得RTTI服务。

下面的问题是，如何使用RTTI？我们知道，Delphi中有一个单元叫TypInfo.pas，后来我发现，其实有另外一个单元叫：IntfInfo.pas。这里面有一个方法GetIntfMetaData可以帮助你获得RTTI。另外，值得一提的是，获取接口类型的PTypeInfo的方法是调用TypeInfo(IMyInterface);

第三、通过MetaData分析，我们可以知道接口的方法个数以及每一个方法的详细定义。那么，现在就是如何Hook了。下面是一个Object的对象实例事例图。

做左边，有Self标识的是对象的实例数据块。某一个几口指针IMyInterface指针，指向了一个VTable。而VTable中的每一个Method，都指向了一段代码，这段代码的前一部分，是为了计算EAX（保证将IMyInterface的地址，偏移到Self所在地址）。

分析上面的结构，再实际在CPU窗体中，调试以下接口方法的调用过程，发现，必然和Method地址有关。因此，Hook的目标，就非常自然地变成修改VTable中的Method1的地址值。

第四、如何修改代码？这里建议大家学习以下FastCode代码。简单一点，就是通过调用VirutalProtect方法，修改代码段中内存的访问属性，然后修改地址，最后再恢复回去。

显然在Hook之前，必须声明新的函数。

第五、新的函数并不是那么好声明的。关注一下，接口函数的调用代码，你会发现很多问题。下面举一个简单的例子。

IMyInterface = interface(IInvokable)

  procedure AAA;

end;

假设TMyIntfImpl类实现了上面的接口。那么oIntfObj: IMyInterface声明的对象，oIntfObj.AAA;的汇编代码是如下的样子：

上面是两段代码的图片，其中[dex+$0c]指的就是$004661FD,也就是第二段代码图片的首地址。大家可以再联系一下上面的示意图理解一下地址的关键。

好，言归正传。这里注意一下，我们要修改的是[dex+$0c]里面的值。但是由于这个是call过去的。所以在call之前，会在堆栈中压入函数返回地址。另外，在调用函数之前，还有函数参数的准备。比如说Self指针的传入到EAX中，如果本身方法还有参数的话，可能占用其他寄存器或者堆栈。

由于我们要求是Hook住所有的方法，并且所有方法的参数类型并不一定一样。所以在call之前的代码，是无法预计的。所以在新的函数中，必须考虑如何做到保存寄存器和做到ret时候的栈平衡。

通过我的实践，我的做法是通过先弹出当前的ret地址，保存到一个数据区中。等待调用完原先的代码后，再压栈。而调用Writelog的时候，先保存寄存器，调用完了之后，再恢复寄存器。这是因为寄存器也可能是返回值的地方。而且后续代码有可能优化使用。

第六、完成了汇编的编写，还有一个问题，那就是由于每一个函数的原地址不一样，所以必须为每一个函数，定义一个代理函数。由于这些函数的地址和个数都是未定的，所以，这里就必须要用到动态创建代码。

动态创建代码的方法看上去简单，申请一段空间，将那一段模板代码地址复制过来。但是，实际情况并非如此。

首先，申请控件的时候，使用VirutalAlloc，并指定EXECUTE_READWRITE属性。另外，要关注到原来的代码是在代码段执行的，所以有些函数的地址可能只是一个偏移地址。而后申请的代码，是在HEAP中运行的，所以，如果只是单纯地复制，函数调用就会报错了。

好了，上面讲了六点关键因素。如果你足够理解上面的过程，你也可以做到AOP了。这篇文章是一个纯技术的，可能关心测试的会非常失望，只能说sorry了。

Trackback: http://tb.blog.csdn.net/TrackBack.aspx?PostId=1632072