如何安全的Post-Hook一个函数-技术开发专区

如何安全的Post-Hook一个函数

作者：来自ITPUB论坛编辑： cynthia 2008-02-15 00:00

　　【IT168 技术文档】本文讨论了参数和返回值的安全有效的处理方式.

　　This HOWTO deals with post-hooks. For details on pre-hooks, see 如何安全的Hook一个函数.

　　For more information on the actual hooking of functions, see 如何Hook一个函数.

　　你通常这样使用么

　　local orig_foo = foo
　　function foo(a1, a2)
　　local r1, r2, r3 = orig_foo(a1, a2)
　　-- some code that looks at a1
　　return r1, r2, r3
　　end

　　问题在于这个方法只能处理固定数目的参数, 如果方法的API改变了, 将导致无法使用. 幸运的是我们有办法使他继续工作.

　　Blizzard's APIs do change from time to time!

　　使用安全的方式

　　local orig_foo = foo
　　function posthook(a1, ...)
　　--do something with a1
　　return ... -- this returns all the original's returns, which were passed to us

　　end
　　function foo(a1, ...)
　　return posthook(a1, orig_foo(a1, ...))
　　end

　　这样确保了所有的参数会传递到原始方法中, 即便你不知道具体有多少个参数. 同样确保了所有返回值都能正确返回. 另一个好处是, 我们使用了局部变量来保存原始方法并做了一个适当的尾调用可以带来更好的性能, 从而为我们的hook做了最小化的付出.

　　会带来巨大的性能影响么?

　　在WoW-2.0以前的设计中, 使用unpack(), 在每次hook被调用时创建一个垃圾回收表. 在新的设计中改进了, 使用'...'变量, 去掉了垃圾回收这部分源码. 在Lua5.1中, 在每次hook调用时包括传参和返回值都不会浪费表的内存.

　　更安全的使用方式

　　function foo(a1)
　　-- do something with a1
　　end
　　hooksecurefunc("SomeBlizzardFunction", foo);

　　这和上面的不同之处在于使用了Blizzard提供的方法hooksecurefunc() 来创建安全的hook, 这意味着没有缺陷代码. 这里的关键是应用程序的原始hook方法会在一个安全的路径下执行. 如果不使用hooksecurefunc()可能会发生因Blizzard API失效而引发错误. 而hooksecurefunc()的代价是较昂贵的因此, 最好在没有问题的情况下使用上面的提到的方式.

关注我们