預加載我的庫的一些功能，同時使用LD_PRELOAD的其他人使用原件

Question

我已經編寫了open()系統調用的包裝並使用LD_PRELOAD環境變量預加載它。我只想要程序的一些功能使用修改的open()而其他人會使用原始的。在兩個程序中分離函數不是一個可以調用另一個的選項。如何做呢？

Answer 1

首先，必須有一種確定的方式來將待預加載的開放區域從默認開放區域中分離出來。這可以使用助手庫（必須動態加載）完成，該助手庫提供該特殊打開的另一個打包版本。通過預加載該庫的變體來替代該變體。

Answer 2

在以下示例中使用函數插入與this answer類似。

該示例提供了一個調用原始write()的包裝函數write()。請注意，您不能直接調用原始的write()，因爲它將被解釋爲對包裝器的調用。在main()中使用函數指針說明了如何避免混淆你正在調用哪個write()。

代碼： test.c的

#define _GNU_SOURCE 
#include <stdio.h> 
#include <string.h> 
#include <dlfcn.h> 

size_t write(int fd, const void *buf, size_t count) 
{ 
    static size_t (*write_func)(int, const void *, size_t) = NULL; 

    /* get reference to original (libc provided) write */ 
    if (!write_func) 
    { 
     write_func = (size_t(*)(int, const void *, size_t)) dlsym(RTLD_NEXT, "write"); 
    } 

    /* perform wrapper specific actions */ 
    /* ... */ 

    /* call original write() */ 
    return write_func(fd, buf, count); 
} 

int main(int argc, char *argv[]) 
{ 
    size_t (*wrap_write)(int, const void *, size_t); 
    size_t (*orig_write)(int, const void *, size_t); 
    char buf1[] = "write() wrapper called\n"; 
    char buf2[] = "orignial write() called\n"; 

    /* set pointer to write() wrapper to differentiate */ 
    wrap_write = write; 
    /* get reference to original (libc provided) write() */ 
    orig_write = (size_t(*)(int, const void *, size_t)) dlsym(RTLD_NEXT, "write"); 

    /* call write() wrapper */ 
    wrap_write(1, buf1, strlen(buf1));  
    /* call original write() */ 
    orig_write(1, buf2, strlen(buf2)); 

    return 0; 
} 

輸出：

$ GCC -Wall -Werror -ldl test.c的-o測試
$。/test
write（）包裝程序調用
orignial寫（）稱爲
$