Linux下HOOK动态链接库中API的方法

        2012年，我写了一篇介绍Windows系统下Ring3层API的hook方案——《一种注册表沙箱的思路、实现——Hook Nt函数》，其在底层使用了微软的Detours库。5年后，我又遇到这么一个问题，但是系统变成了Linux。我最开始的想法是找一个Linux下的Detours库，于是找到了subhook。其原理是：修改被Hook函数起始地址处的汇编代码，让执行流程跳到我们定义的函数中。但是在实际使用中，我发现通过该库调用原始函数有错误——地址违例，导致进程崩溃，所以最终放弃了subhook的方案。（转载请指明出于breaksoftware的csdn博客）

        后来发现，Linux用户层Hook非常简单。我们只要定义一个和被Hook的API相同名称、参数、返回值的函数即可。比如我们需要Hook获取用户UID的函数getuid(原来是在libc.so中实现的)，则需要定义如下函数：

uid_t getuid(void) {    return 800;}

        我们在main函数中调用之

int main() {    printf("get_uid:%d\n", getuid());}

        函数返回

        我使用work账户登录的，其真实uid是502。而我们重写了程序中的getuid，则返回的是我们“指定”的800。

        如果我们希望在被hook中的函数中调用原始函数，怎么做呢？这儿有个比较尴尬的问题，那就是我们定义的getuid地址将对应于符号getuid，那么原始的getuid（以后称libc中的getuid）地址将对应什么符号？我们怎么找到它？

        可以想象libc中的getuid对应的符号不会因为我们的程序而被改变，那么就意味着程序运行中，将有两个getuid。事实也的确如此。

        第一个getuid就是我们重定义的hook的函数体，第二个是动态链接库libc.so中的。于是我们在重定义的函数体中，使用

dlsym(RTLD_NEXT, "getuid")

        就可以获得原始的函数地址。

        所以这种方案的精髓就是RTLD_NEXT参数。我们看下dlsym函数参数的说明：

        There are two special pseudo-handles, RTLD_DEFAULT and RTLD_NEXT. The former will find the first occurrence of the desired symbol using the default library search order. The latter will find the next occurrence of a function in the search order after the current library. This allows one to provide a wrapper around a function in another shared library.
        这段文字意思是：在默认的库查找顺序下，RTLD_DEFAULT是用于查找第一个符号匹配的函数地址，RTLD_NEXT是用于查找第二个符号匹配的函数地址。这种方式就提供了一种针对动态链接库中函数替换的功能。

        以我们例子，RTLD_DEFAULT将找到我们自己定义的getuid，而RTLD_NEXT将找到libc.so中的。
        为了方便使用这种方式，我封装了相关调用

#ifndef HOOK_BASE#define HOOK_BASE#ifndef _GNU_SOURCE#define _GNU_SOURCE#endif#include <dlfcn.h>#define HOOK_FUNC_TEMPLATE(function_name) function_name##_func_t#define HOOK_FUNC_ORI_NAME(function_name) function_name##_ori#define HOOK_FUNC_INIT(function_name) static HOOK_FUNC_TEMPLATE(function_name) HOOK_FUNC_ORI_NAME(function_name);#define HOOK_FUNC(function_name) \    if (!HOOK_FUNC_ORI_NAME(function_name)) {\        HOOK_FUNC_ORI_NAME(function_name) = (HOOK_FUNC_TEMPLATE(function_name)) dlsym(RTLD_NEXT, #function_name);\    }\#define ORIGINAL_FUNC(function_name) ((HOOK_FUNC_TEMPLATE(function_name)) HOOK_FUNC_ORI_NAME(function_name))#endif

        我们只要关注HOOK_FUNC_INIT、HOOK_FUNC和ORIGINAL_FUNC三个宏。HOOK_FUNC_INIT方法声明了一个全局函数指针变量，其在HOOK_FUNC宏中被指定为被HOOK函数的原始地址。ORIGINAL_FUNC则是将这个指针进行类型转换，从而方便调用。

        下一步我们要定义被HOOK的函数的类型

#ifndef HOOK_DEF#define HOOK_DEF#include "hook_base.h"#include <stdio.h>#include <sys/types.h>#include <unistd.h>typedef uid_t (*HOOK_FUNC_TEMPLATE(getuid))(void);#endif

        然后重定义我们要HOOK的函数

#include "hook_def.h"#include <stdio.h>#include <sys/types.h>HOOK_FUNC_INIT(getuid);uid_t getuid(void) {    HOOK_FUNC(getuid);    int uid = ORIGINAL_FUNC(getuid)();    printf("getuid original:%d\n", uid);    return 800;}

        这段代码，我们先调用原始的getuid函数，并打印出它的值。最后才返回一个我们定义的值——800。

        在main函数中，我们只调用getuid。并使用 gcc src/*.c -ldl -o main 编译

#include <stdio.h>#include <unistd.h>#include <sys/types.h>#include "hook_def.h"int main() {    printf("get_uid:%d\n", getuid());    return 0;}

        其返回结果如下