专栏首页orientlulinux 链接器 库打桩

linux 链接器 库打桩

@(linux 编程)

翻看 CSAPP 看到库函数打桩,记录下。

linux 链接器支持库打桩(library interpositioning), 允许我们截获共享库的调用,执行自己的代码,通过这个机制,可以给程序调试带来很多便利。

库打桩实现有三种:

  • 编译时打桩
  • 链接时打桩
  • 运行时打桩

以下,参照书中例子,以 malloc 和 free 两个库函数的调用作为例子, 添加调用该函数时打印调试信息,以上述提到的三种方式实现打桩 ubuntu下测试代码

测试目标代码, 申请内存,赋值后答应,释放内存

#include<stdio.h>
#include"malloc.h"
int main()
{
    int *p = malloc(sizeof(int));
    *p = 12;
    printf("p = %d\n", *p);
    free(p);
    return 0;
}

编译时打桩

编译时打桩通过在编译时指定 include 路径,告诉C预处理器在搜索系统目录前,先查看当前目录,由于当前目录有malloc.h, 停止继续搜索

实现桩代码:

malloc.h

#ifndef _MALLOC_H
#define _MALLOC_H

#define malloc(size) mymalloc(size)
#define free(ptr) myfree(ptr)
void *mymalloc(size_t size);
void myfree(void *ptr);
#endif

mymalloc.c

#ifdef COMPILELINK
#include<stdio.h>
#include<malloc.h>
void *mymalloc(size_t size)
{
    void *ptr = malloc(size);
    printf("[debug] malloc size %d\n", (int)size);
    return ptr;
}
void myfree(void *ptr)
{
    free(ptr);
    printf("[debug] free %p\n", ptr);
}
#endif

具体实现如下makefile

all:out
out: main.c mymalloc.o
    # -I . : so will use mymalloc
    #  编译最终运行程序时指定include优先检索当前目录,所以会读取当前目录的头文件malloc.h
    #  替代系统库的
    gcc -I . -o out main.c mymalloc.o
mymalloc.o: mymalloc.c
    # no -I ., will use std malloc
    # 没有指定include当前目录,使用的是系统malloc
    gcc -DCOMPILELINK -c mymalloc.c

.PHONY : clean
clean:
    @rm -rf out *.o

链接时打桩

链接时打桩通过在链接时传递标志 -wl, --wrap f 给链接器,告诉链接器把符号 f__real_f解析为 __wrap_f,实现替换。 同样,实现替换的函数

mymalloc.c

#ifdef LINKTIME
#include<stdio.h>
#include<malloc.h>
//std malloc
//试了直接调用malloc,编译链接ok,但是运行时core
void *__real_malloc(size_t size);
void __real_free(void *ptr);
void *__wrap_malloc(size_t size)
{
    void *ptr = __real_malloc(size);
    printf("[debug] malloc size %d\n", (int)size);
    return ptr;
}
void __wrap_free(void *ptr)
{
    __real_free(ptr);
    printf("[debug] free %p\n", ptr);
}
#endif

链接时实现入makefile所示, 在链接时指定覆盖的函数

all:out
out: main.c mymalloc.o
    # __wrap_malloc and __wrap_free
    gcc -Wl,--wrap,malloc -Wl,--wrap,free -o out main.c mymalloc.o
    
mymalloc.o: mymalloc.c
    gcc -DLINKTIME -c mymalloc.c

.PHONY : clean
clean:
    @rm -rf out *.o

运行时打桩

以上两种需要有源文件的情况下实现,而对于运行时打桩,只需要可以访问执行文件,利用动态链接器的LD_PRELOAD环境变量实现。 当加载程序时,解析未定义的引用时,动态链接器会先搜索LD_PRELOAD指定的库,然后才搜索其他,因此,通过把自己实现的动态库设置到这个环境变量,动态链接器加载时搜索的该库内有对应实现的函数,就会直接使用该函数而不会再搜索其他系统库。

实现自己的动态库,包含需要替代的函数 mymalloc.c

#ifdef RUNTIME
#define _GNU_SOURCE
#include<stdio.h>
#include<stdlib.h>
#include<dlfcn.h>

void *malloc(size_t size)
{
    void *(*mallocp)(size_t size);
    char *error;
    // 查找标准库的实现
    mallocp = dlsym(RTLD_NEXT, "malloc");
    if ((error = dlerror()) != NULL) {
        fputs(error, stderr);
        exit(1);
    }
    void *ptr = mallocp(size);
    printf("[debug] malloc size %d\n", (int)size);
    return ptr;
}

void free(void *ptr)
{
    void (*freep)(void *ptr);
    char *error;
    freep = dlsym(RTLD_NEXT, "free");
    if ((error = dlerror()) != NULL) {
        fputs(error, stderr);
        exit(1);
    }
    freep(ptr);
    printf("[debug] free %p\n", ptr);
}
#endif

编译动态库,然后在运行时设定环境变量即可。

all:out
out: main.c mymalloc.o
    gcc -o out main.c

## 编译共享库
mymalloc.o: mymalloc.c
    gcc -DRUNTIME --share -fpic -o mymalloc.so mymalloc.c -ldl

.PHONY : clean run
run:
    # 指定运行时加载的库
    #setenv LD_PRELOAD "./mymalloc.SO"; ./out; unsetenv LD_PRELOAD
    ## 设定环境
    export LD_PRELOAD="./mymalloc.so"; ./out; unset LD_PRELOAD
    ## 其他任何的可执行程序都可以打桩
    export LD_PRELOAD="./mymalloc.so"; uptime; unset LD_PRELOAD

clean:
    @rm -rf out *.so

参考

  • <深入理解计算机系统>

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • 【系统设置】CentOS 修改机器名

    ken.io
  • 《动物魔法学校》儿童学编程Scratch之“外观”部分

    导读:本文通过一个案例《动物魔法学校》来学习Scratch语言的“外观”部分。之后通过一系列其他功能的综合运用对作品功能进行了扩展。

    一石匠人
  • SQL中GROUP BY用法示例

    GROUP BY我们可以先从字面上来理解,GROUP表示分组,BY后面写字段名,就表示根据哪个字段进行分组,如果有用Excel比较多的话,GROUP BY比较类...

    Awesome_Tang
  • 复杂业务下向Mysql导入30万条数据代码优化的踩坑记录

    从毕业到现在第一次接触到超过30万条数据导入MySQL的场景(有点low),就是在顺丰公司接入我司EMM产品时需要将AD中的员工数据导入MySQL中,因此楼主负...

    haifeiWu
  • 我不是算命先生,却对占卜有了疑惑——如何论证“占卜前提”的正确与否

    事出有因,我对《周易》感兴趣了很多年。只是觉得特别有趣,断断续续学习了一些皮毛。这几天又偶然接触到了《梅花易数》,觉得很是精彩,将五行八卦天干地支都串联了起来。...

    一石匠人
  • 一张图理清《梅花易数》梗概

    学《易经》的目的不一定是为了卜卦,但是了解卜卦绝对能够让你更好地了解易学。今天用一张思维导图对《梅花易数》的主要内容进行概括,希望能够给学友们提供帮助。

    一石匠人
  • 声音功能让儿童编程更有创造性

    导读:Scratch中声音功能非常强大,除了常规的音效,你甚至可以模拟各种乐器的各个发音、设置节拍、休止……如果你愿意,甚至可以用它创作一个交响乐。我们可以引导...

    一石匠人
  • 天干地支五行八卦的对应关系

    一石匠人
  • 什么样的人生才是有意义的人生——没有标准的标准答案

    【导读】其实我们可以跳出这个小圈圈去更加科客观地看一下这个世界。在夜晚的时候我们仰望天空,浩瀚的宇宙中整个地球只是一粒浮尘,何况地球上一个小小的人类?在漫长的历...

    一石匠人
  • 儿童创造力教育与编程教育的碰撞——MIT雷斯尼克教授最新理论梗概

    儿童编程教育已经在我国各一线二线城市疯狂出现,颇有“烂大街”的趋势。我们不禁要问很多很多问题:

    一石匠人

扫码关注云+社区

领取腾讯云代金券