path: root/process.c

/* See LICENSE file for copyright and license details. */
#include "common.h"


static struct process head;
static struct process tail;


void
init_process_list(void)
{
	head.next = &tail;
	tail.prev = &head;
}


struct process *
find_process(pid_t pid)
{
	struct process *p;
	for (p = head.next; p->next; p = p->next)
		if (p->pid == pid)
			return p;
	return NULL;
}


struct process *
add_process(pid_t pid, unsigned long int trace_options)
{
	struct process *proc;
	int status;

	proc = calloc(1, sizeof(*proc));
	if (!proc)
		eprintf("calloc: %s\n");
	proc->pid = pid;
	proc->next = &tail;
	proc->prev = tail.prev;
	proc->prev->next = proc;
	tail.prev = proc;

	while (waitpid(pid, &status, WSTOPPED) != pid) {
		if (errno == EINTR)
			continue;
		eprintf_and_kill(pid, "waitpid %ju <buffer> WSTOPPED:", (uintmax_t)pid);
	}

	switch (status) {
	case __W_STOPCODE(SIGSTOP):
		if (ptrace(PTRACE_SEIZE, pid, 0, trace_options))
			eprintf_and_kill(pid, "ptrace PTRACE_SEIZE %ju 0 ...:", (uintmax_t)pid);
		if (ptrace(PTRACE_INTERRUPT, pid, NULL, 0))
			eprintf_and_kill(pid, "ptrace PTRACE_INTERRUPT %ju NULL 0:", (uintmax_t)pid);
		if (kill(pid, SIGCONT) < 0)
			eprintf_and_kill(pid, "kill &ju SIGCONT:", (uintmax_t)pid);
		break;

	case __W_STOPCODE(SIGTRAP) | (PTRACE_EVENT_STOP << 16):
		if (ptrace(PTRACE_SYSCALL, pid, NULL, 0))
			eprintf_and_kill(pid, "ptrace PTRACE_SYSCALL %ju NULL 0:", (uintmax_t)pid);
		break;

	default:
		eprintf_and_kill(pid, "unexpected return of waitpid %ju <buffer> WSTOPPED: %#x\n", (uintmax_t)pid, status);
	}

	return proc;
}


void
remove_process(struct process *proc)
{
 	proc->prev->next = proc->next;
	proc->next->prev = proc->prev;
	if (proc->vfork_waiting_on)
		proc->vfork_waiting_on->continue_on_exit = NULL;
	if (proc->continue_on_exit)
		proc->continue_on_exit->vfork_waiting_on = NULL;
	free(proc);
}