如何判断在vfork（）处理过的过程中是否调用过函数？

Question

我有一个用于C的插入器库。无法插入vfork()因为vfork() （插入函数）在调用了真正的vfork()之后无法返回到调用函数。 但是我要插入_exit()因为我需要知道进程何时完成。 当然，我要插入exec*()函数。 我的问题是，当插入_exit() ，某些事情我想在正常进程调用_exit()时执行，但在该进程是受限制的vfork()进程时却不希望执行。

我该如何在C程序中判断我的进程是vfork()的进程，何时无法访问vfork()返回的进程ID？

插入器库：

/* COMPILE: gcc -shared -ldl -fPIC -o libso_interposer.so so_interposer.c -std=c99 */
/* RUN:     LD_PRELOAD=./libso_interposer.so so_interposer_test */
#define _GNU_SOURCE
#include <stdio.h>
#include <dlfcn.h>
#include <unistd.h>


static void (*_exit_R)(int) = (void *) 0;
static pid_t (*fork_R)(void) = (void *) 0;
static void teardown_interposer() {
    fprintf(stderr, "Destructing so_interposer.\n");
    /* Concurrency protected code to be executed only once
     * when the process finishes! */
    /* Must not be executed if/when vfork() process finishes. */
}

pid_t fork(void) {
    *(pid_t **) (&fork_R) = dlsym(RTLD_NEXT, "fork");
    /* Code to prepare for a new process.
    * More preparation in exec* interposing functions.*/
    pid_t pid = fork_R();
    return pid;
}
__attribute__((noreturn)) void _exit(int status) {
    *(void **) (&_exit_R) = dlsym(RTLD_NEXT, "_exit");
    fprintf(stderr, "Process '%lld' called _exit(%i).\n", (signed long long int) getpid(), status);
    teardown_interposer();
    _exit_R(status);
}

测试二进制文件：

/* COMPILE: gcc -std=c99 -D FORK=vfork -o so_interposer_test so_interposer_test.c */
#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#if !defined(FORK)
#error "Define FORK: 'fork' or 'vfork'!"
#endif
int main(void) {
    pid_t pid = FORK();
    if(pid == 0) {
        _exit(EXIT_SUCCESS);
    } else if(pid > 0) {
        _exit(EXIT_SUCCESS);
    }
    return EXIT_SUCCESS;
}

Answer 1

对于最新版本的gcc，您应该可以将vfork包装为：

typedef pid_t (*vfork_t)(void);
extern vfork_t wrap_vfork();

pid_t vfork(void) {
    vfork_t f = wrap_vfork();
    return f();
}

其中wrap_vfork完成所有vfork包装工作，并返回指向实际vfork的指针（不调用它）。 Gcc 6.3.0（-O3）编译为：

    .globl  vfork
    .type   vfork, @function
vfork:
.LFB11:
    .cfi_startproc
    subq    $8, %rsp
    .cfi_def_cfa_offset 16
    xorl    %eax, %eax
    call    wrap_vfork
    addq    $8, %rsp
    .cfi_def_cfa_offset 8
    jmp     *%rax
    .cfi_endproc
.LFE11:
    .size   vfork, .-vfork

需要注意的重要一点是，它直接跳转到实际的vfork函数而不是调用它，因此它本身不需要返回（真正的vfork将直接返回给该函数的调用者）

如果您不满意依靠编译器为您执行尾调用优化，则可以直接在asm中而不是在C中编写这个小例程。