9

ptrace用来跟踪进程的系统调用。分叉进程后,我使用PTRACE_TRACEME开始跟踪进程。代码如下所示:

while (true) {
    int status;
    int gotPid;
    gotPid = waitpid(pid, &status, 0);

    if (WIFEXITED(status) || WIFSIGNALED(status)) {
        break;
    }

    if (WIFSTOPPED(status)) {
        handleTrace();
    }
}

然后是handleTrace函数,看起来像这样。

long syscall;
syscall = ptrace(PTRACE_PEEKUSER,
     pid, 8 * ORIG_RAX, NULL);

// do something with the syscall

// continue program
ptrace(PTRACE_SYSCALL, pid, NULL, NULL);

这一切都很好,但是如果程序分叉(或创建一个新线程),我还想跟踪跟踪的进程创建的子进程(以及该进程创建的线程)。我知道它可以使用PTRACE_O_TRACEFORK,PTRACE_O_TRACEVFORK和来完成PTRACE_O_TRACECLONE,但是从man文档中很难弄清楚它是如何完成的。我需要一些例子。

编辑:

我在这里发现了一个类似的问题:How to ptrace a multi-threaded application?我用下面的代码试了一下。此代码跟踪启动进程的系统调用,它也应该跟踪分叉进程。它fork()在父进程中的 a 之后运行(子进程调用 aPTRACE_TRACEME和 an exec())。

编辑2:

我对代码进行了更多修改,并取得了更多进展。

long orig_eax;
int status;
int numPrograms = 1;

while(1) {
    int pid;
    CHECK_ERROR_VALUE(pid = waitpid(-1, &status, __WALL));
    std::cout << pid << ": Got event." << std::endl;
    if(WIFEXITED(status) || WIFSIGNALED(status)) {
        std::cout << pid << ": Program exited." << std::endl;
        if (--numPrograms == 0) {
            break;
        }
        continue;
    }

    if (status >> 16 == PTRACE_EVENT_FORK || status >> 16 == PTRACE_EVENT_VFORK ||
            status >> 16 == PTRACE_EVENT_CLONE) {
        int newpid;
        CHECK_ERROR_VALUE(ptrace(PTRACE_GETEVENTMSG, child, NULL, (long) &newpid));
        std::cout << pid << ": Attached to offspring " << newpid << std::endl;
        boost::this_thread::sleep(boost::posix_time::millisec(100));
        CHECK_ERROR_VALUE(ptrace(PTRACE_SETOPTIONS,
                newpid, NULL, PTRACE_O_TRACEFORK | PTRACE_O_TRACEVFORK | PTRACE_O_TRACECLONE));
        CHECK_ERROR_VALUE(ptrace(PTRACE_SYSCALL, newpid, NULL, NULL));
        ++numPrograms;
    } else {
        CHECK_ERROR_VALUE(ptrace(PTRACE_SETOPTIONS,
                pid, NULL, PTRACE_O_TRACEFORK  | PTRACE_O_TRACEVFORK | PTRACE_O_TRACECLONE));
        CHECK_ERROR_VALUE(orig_eax = ptrace(PTRACE_PEEKUSER,
                pid, 8 * ORIG_RAX, NULL));
        std::cout << pid << ": Syscall called: " <<
                SyscallMap::instance().get().right.at(orig_eax) <<
                std::endl;
        CHECK_ERROR_VALUE(ptrace(PTRACE_SYSCALL,
                pid, NULL, NULL));
    }

}

CHECK_ERROR_VALUE只是一个检查结果代码并抛出异常的宏,其中的描述errno

显然,当我收到 fork/clone 事件时,新进程还不存在,如果我尝试对其进行 ptrace,则会收到“进程不存在”错误消息。如果我在尝试 ptrace 新进程之前进入睡眠状态,我不会收到错误消息。现在当程序到达fork/clone点时,它开始跟踪新进程,但它永远不会到达clone()父进程中系统调用的返回点,这意味着子进程成功完成,但父进程挂起它的分叉点。

4

1 回答 1

1

Strace 执行此操作,其 README-linux-ptrace 文件包含有关该主题的一些信息:

https://github.com/strace/strace/blob/master/README-linux-ptrace

它似乎确实在抱怨这个平台中的内核错误,所以 YMMV。

该代码解释了如何获取孩子的 pid。但是,由于在它调用的二进制文件上设置了 setuid 或 setgid 位,孩子可能会获得另一个用户 ID。所以答案是您在子 PID 上调用 ptrace 并查看您是否可以访问。

这是相关部分:

PTRACE_EVENT跟踪器观察到停止,因为 waitpid 返回 WIFSTOPPED(status) == true, WSTOPSIG(status) == SIGTRAP。附加位设置在状态字的高字节中:值((status >> 8) & 0xffff) 将是(SIGTRAP | PTRACE_EVENT_foo << 8)。存在以下事件:

  • PTRACE_EVENT_VFORK- 在从 vfork/clone+CLONE_VFORK 返回之前停止。在此之后继续跟踪时,它将等待子进程退出/执行,然后再继续执行(IOW:vfork 上的通常行为)。
  • PTRACE_EVENT_FORK- 在从 fork/clone+SIGCHLD 返回之前停止
  • PTRACE_EVENT_CLONE- 在从克隆返回之前停止
  • PTRACE_EVENT_VFORK_DONE- 在从 vfork/clone+CLONE_VFORK 返回之前停止,但在 vfork 子进程通过退出或执行解除阻塞之后。

对于上述所有四个停止:停止发生在父线程中,而不是在新创建的线程中。PTRACE_GETEVENTMSG可用于检索新线程的 tid。

于 2012-11-25T08:20:19.467 回答