【Linux】LD_PRELOAD用法

LD_PRELOAD，是个环境变量，用于动态库的加载，动态库加载的优先级最高，一般情况下，其加载顺序为LD_PRELOAD>LD_LIBRARY_PATH>/etc/ld.so.cache>/lib>/usr/lib。程序中我们经常要调用一些外部库的函数，以malloc为例，如果我们有个自定义的malloc函数，把它编译成动态库后，通过LD_PRELOAD加载，当程序中调用malloc函数时，调用的其实是我们自定义的函数，下面以一个例子说明。

// test.c
#include <stdio.h>
#include <stdlib.h>

int main()
{
    int i = 0;
    for (; i < 5; ++i) {
        char *c = (char*)malloc(sizeof(char));
        if (NULL == c) {
            printf("malloc fails\n");
        }
        else {
            printf("malloc ok\n");
        }
    }

    return 0;
}

编译运行，结果如下：

$gcc -o test test.c
$./test
malloc ok
malloc ok
malloc ok
malloc ok
malloc ok

可以看出，程序运行没有问题，我们稍作修改，自定义malloc。

// preload.c
#include <stdio.h>
#include <stdlib.h>

void* malloc(size_t size)
{
    printf("%s size: %lu\n", __func__, size);
    return NULL;
}

然后把自定义的malloc打包为动态库。

$gcc -shared -fpic -o libpreload.so preload.c

接着使用LD_PRELOAD加载libpreload.so，看看情况如何：

$LD_PRELOAD=./libpreload.so ./test
malloc size: 1
malloc fails
malloc size: 1
malloc fails
malloc size: 1
malloc fails
malloc size: 1
malloc fails
malloc size: 1
malloc fails

看到了吧，5次malloc都失败了，如果不知道是 LD_PRELOAD在作怪，那可能分析很长时间都找不出原因所在。这个 LD_PRELOAD就是把双刃剑，用好了可以帮助我们，如果别有用心，那可能会有意外的惊喜。