如何在LD_PRELOAD加载的共享库构造函数中创建线程？

嘿，你这个思路完全可行，但这里有几个容易踩的坑得提前留意。先看看你的原始代码，咱们一步步梳理问题和优化方案：

#include <stdio.h>
#include <stdlib.h>
#include <pthread.h>
#include <unistd.h>

void* worker(void* t) {
    sleep(1);
    printf("Running worker..\n");
    return NULL;
}

__attribute__((constructor)) void spawn() {
    pthread_t t;
    printf("[INFO] In the constructor..\n");
    int rc = pthread_create(&t, NULL, worker, (void *)NULL);
    if (rc){
        printf("[FATAL] Failed spawn the ...");
    }
}

你当前代码存在的潜在问题 #

1. 构造函数执行时机的风险
   共享库构造函数是在main函数执行前、进程初始化阶段运行的，这时候进程的一些核心组件（比如stdio缓冲区、线程调度的内部结构）可能还没完全就绪，直接调用pthread_create在某些环境下可能触发未知问题。

2. 线程生命周期未管理
   如果主进程在构造函数执行完后快速进入main并退出，你的worker线程可能还没来得及执行就被强制终止——进程退出时会直接干掉所有未分离的线程。

3. 错误处理不完整
   你只打印了失败提示，但没有输出具体错误码，排查问题时会非常麻烦。

优化后的代码示例 #

#include <stdio.h>
#include <stdlib.h>
#include <pthread.h>
#include <unistd.h>
#include <string.h>

// 按需开启：是否要等待worker线程执行完成再进入main
#define WAIT_FOR_WORKER 1

void* worker(void* unused) {
    // 若遇到stdio初始化问题，可替换为write系统调用更可靠
    // write(STDOUT_FILENO, "Running worker..\n", 17);
    printf("Running worker..\n");
    sleep(1);
    return NULL;
}

__attribute__((constructor)) void spawn() {
    pthread_t t_id;
    pthread_attr_t attr;
    int rc;

    printf("[INFO] In the constructor..\n");

    // 初始化线程属性，设置为分离状态：线程结束后自动释放资源，无需主线程join
    rc = pthread_attr_init(&attr);
    if (rc != 0) {
        fprintf(stderr, "[FATAL] Failed to init thread attr: %s\n", strerror(rc));
        exit(EXIT_FAILURE);
    }
    rc = pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED);
    if (rc != 0) {
        fprintf(stderr, "[FATAL] Failed to set detach state: %s\n", strerror(rc));
        pthread_attr_destroy(&attr);
        exit(EXIT_FAILURE);
    }

    // 创建线程
    rc = pthread_create(&t_id, &attr, worker, NULL);
    if (rc != 0) {
        fprintf(stderr, "[FATAL] Failed to spawn thread: %s\n", strerror(rc));
        pthread_attr_destroy(&attr);
        exit(EXIT_FAILURE);
    }

    pthread_attr_destroy(&attr);

#if WAIT_FOR_WORKER
    // 若需要确保worker在main前执行完，可开启join；否则注释掉让线程后台运行
    rc = pthread_join(t_id, NULL);
    if (rc != 0) {
        fprintf(stderr, "[WARN] Failed to join thread: %s\n", strerror(rc));
    }
#endif
}

关键优化点说明 #

• 分离线程属性：用PTHREAD_CREATE_DETACHED标记线程，避免进程退出时留下僵尸线程资源，也省去了主线程手动join的麻烦。
• 完善错误处理：通过strerror把错误码转为可读信息，方便调试定位问题。
• stdio替代方案：如果在构造函数中用printf出现输出异常，可直接使用write系统调用，绕开未完全初始化的stdio缓冲区。
• 线程等待选项：通过宏控制是否等待线程完成，平衡程序启动速度和线程执行完整性的需求。

编译与使用提示 #

编译共享库时记得链接pthread库：

gcc -shared -fPIC -o my_preload_lib.so my_preload_lib.c -lpthread

运行目标程序时加载共享库：

LD_PRELOAD=./my_preload_lib.so ./your_target_program

内容的提问来源于stack exchange，提问作者budchan chao