mirror of
https://github.com/proot-me/proot.git
synced 2025-08-28 16:43:49 +02:00
464d242f8cWhen the ptracer waits on a non-immediate child, the information is only available in the zombies list and we need to handle those ourselves. Expands the tracee lookup of the wait enter to include zombies of the tracee to handle this. The new test mimicing the testing operation from gdb's `linux_check_ptrace_features`.
144 lines
2.9 KiB
C
144 lines
2.9 KiB
C
//
|
|
|
|
#include <signal.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <sys/ptrace.h>
|
|
#include <sys/wait.h>
|
|
#include <unistd.h>
|
|
|
|
#include "../src/compat.h"
|
|
|
|
static int fork_to_function(int (*function)(void))
|
|
{
|
|
int child_pid = fork();
|
|
|
|
if (child_pid == -1) {
|
|
perror("fork()");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
if (child_pid == 0)
|
|
return function();
|
|
|
|
return child_pid;
|
|
}
|
|
|
|
static int grandchild_function(void)
|
|
{
|
|
exit(0);
|
|
}
|
|
|
|
static int child_function(void)
|
|
{
|
|
ptrace(PTRACE_TRACEME, 0, 0, 0);
|
|
kill(getpid(), SIGSTOP);
|
|
|
|
fork_to_function(grandchild_function);
|
|
|
|
exit(0);
|
|
}
|
|
|
|
static void kill_child(int child_pid)
|
|
{
|
|
pid_t got_pid;
|
|
int kill_status;
|
|
|
|
if (kill(child_pid, SIGKILL) != 0) {
|
|
perror("kill()");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
// Sleep some time to let proot handle the kill event
|
|
// in order to make sure a delayed wait on a dead ptracee works correctly.
|
|
sleep(2);
|
|
|
|
got_pid = waitpid(child_pid, &kill_status, 0);
|
|
if (got_pid != child_pid) {
|
|
fprintf(stderr, "waitpid: unexpected result %d.", (int)got_pid);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
if (!WIFSIGNALED(kill_status)) {
|
|
fprintf(stderr, "waitpid: unexpected status %d.", kill_status);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
}
|
|
|
|
static void test_tracefork(int child_pid)
|
|
{
|
|
int ret, status;
|
|
long second_pid;
|
|
|
|
ret = ptrace(PTRACE_SETOPTIONS, child_pid, 0, PTRACE_O_TRACEFORK);
|
|
if (ret != 0) {
|
|
// Skipped
|
|
perror("ptrace(PTRACE_SETOPTIONS, PTRACE_O_TRACEFORK)");
|
|
kill_child(child_pid);
|
|
exit(125);
|
|
}
|
|
|
|
ret = ptrace(PTRACE_CONT, child_pid, 0, 0);
|
|
if (ret != 0) {
|
|
perror("ptrace(PTRACE_CONT)");
|
|
kill_child(child_pid);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
ret = waitpid(child_pid, &status, 0);
|
|
/* Check if we received a fork event notification. */
|
|
if (ret == child_pid && WIFSTOPPED(status) && (status >> 16) == PTRACE_EVENT_FORK) {
|
|
/* We did receive a fork event notification. Make sure its PID
|
|
is reported. */
|
|
second_pid = 0;
|
|
ret = ptrace(PTRACE_GETEVENTMSG, child_pid, 0, &second_pid);
|
|
if (ret == 0 && second_pid != 0) {
|
|
int second_status;
|
|
|
|
/* Do some cleanup and kill the grandchild. */
|
|
waitpid(second_pid, &second_status, 0);
|
|
kill_child(second_pid);
|
|
}
|
|
else {
|
|
perror("ptrace(PTRACE_GETEVENTMSG)");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
}
|
|
else {
|
|
fprintf(stderr, "Unexpected result from waitpid: pid=%d, status=%d\n", ret, status);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
}
|
|
|
|
static void check_ptrace_features(int do_fork)
|
|
{
|
|
int child_pid, ret, status;
|
|
|
|
child_pid = fork_to_function(child_function);
|
|
|
|
ret = waitpid(child_pid, &status, 0);
|
|
if (ret == -1) {
|
|
perror("waitpid()");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
else if (ret != child_pid) {
|
|
fprintf(stderr, "waitpid: unexpected result %d.", ret);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
else if (!WIFSTOPPED(status)) {
|
|
fprintf(stderr, "waitpid: unexpected status %d.", status);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
if (do_fork)
|
|
test_tracefork(child_pid);
|
|
|
|
kill_child(child_pid);
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
(void)argv;
|
|
check_ptrace_features(argc > 1);
|
|
return 0;
|
|
}
|