pidjail/main.c

171 lines
4.9 KiB
C
Raw Normal View History

2021-01-06 04:42:04 +01:00
// we need this so sched.h exports unshare and CLONE_*
#define _GNU_SOURCE
#include <sched.h>
#include <errno.h>
#include <err.h>
2021-01-06 04:42:04 +01:00
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <unistd.h>
#include <signal.h>
2021-01-15 20:22:22 +01:00
static
pid_t pid_child;
2021-01-06 04:42:04 +01:00
2021-01-15 20:22:22 +01:00
static
2021-01-15 20:22:22 +01:00
void drop_root(void)
{
/// Drop root privileges
// First group then user because we might not
// be able to drop group once we dropped user
gid_t gid = getgid();
if (setresgid(-1,gid,gid) == -1)
err(errno, "Failed to drop root privileges with setresgid");
uid_t uid = getuid();
if (setresuid(-1,uid,uid) == -1)
err(errno, "Failed to drop root privileges with setresuid");
// sanity check
2021-01-15 20:22:22 +01:00
if (seteuid(0) != -1)
errx(1, "Sanity check failed. Able to regain root");
}
2021-01-06 04:42:04 +01:00
2021-01-15 20:22:22 +01:00
static
void forward_signal(int sig)
{
2021-01-15 20:22:22 +01:00
if (kill(pid_child, sig) == -1)
{
if (sig == SIGTERM)
exit(1);
}
}
2021-01-15 20:22:22 +01:00
int main(int argc, char* const* argv)
2021-01-06 04:42:04 +01:00
{
2021-01-15 20:22:22 +01:00
struct sigaction forward_signal_descriptor;
forward_signal_descriptor.sa_flags = SA_RESTART;
forward_signal_descriptor.sa_handler = &forward_signal;
2021-01-15 20:22:22 +01:00
if (argc == 1)
{
2021-01-15 20:22:22 +01:00
fprintf(stderr,"Usage: %s PROGRAM ARGUMENTS...\n"
"Run command within its own pid namespace. Integrated init process.\n",
argv[0]);
2021-01-15 20:22:22 +01:00
return 0;
}
2021-01-06 04:42:04 +01:00
// next fork shall be in a new pid namespace
if (unshare(CLONE_NEWPID) != 0)
{
err(errno, "Failed to unshare pid namespace");
2021-01-06 04:42:04 +01:00
}
// Drop root privileges, we only needed those for the unshare call.
drop_root();
2021-01-06 04:42:04 +01:00
pid_t pid = fork();
if (pid == -1)
{
err(errno, "Failed to fork");
2021-01-06 04:42:04 +01:00
}
if (pid != 0)
{
2021-01-15 20:22:22 +01:00
/// Head process
// Wait for the init process in the PID namespace to terminate and forward its exit code.
// Also forward SIGTERM signals towards that init process.
// Setup signal handler to forward SIGTERM
pid_child = pid;
if (sigaction(SIGTERM, &forward_signal_descriptor, NULL) == -1)
2021-01-15 20:22:22 +01:00
{
2021-01-15 20:22:22 +01:00
int saved_errno = errno;
// Have to kill child here, otherwise that gets orphaned and runs anyway.
// Use SIGKILL here because it might forward SIGTERM to its child and that
// decides not to stop.
kill(pid_child, SIGKILL);
// Restore errno as it might've been overwritten by kill
2021-01-15 20:22:22 +01:00
errno = saved_errno;
err(errno, "Unable to set up signal handler in head process");
2021-01-15 20:22:22 +01:00
}
2021-01-06 04:42:04 +01:00
// parent waits for child then exits
// Could be interrupt due to a signal. Retry in that case.
2021-01-06 04:42:04 +01:00
int status;
if (waitpid(pid, &status, 0) == -1)
2021-01-06 04:42:04 +01:00
{
err(errno, "Failed to wait for init process");
2021-01-06 04:42:04 +01:00
}
return WEXITSTATUS(status);
}
else
{
// Child should be in new pid namespace and
// functions as the the init process
// it needs to fork again then wait for any child.
// if the forked child exits then exit.
pid = fork();
2021-01-15 20:22:22 +01:00
if (pid == -1)
{
err(errno, "Failed to fork in init process");
2021-01-15 20:22:22 +01:00
}
2021-01-06 04:42:04 +01:00
if (pid != 0)
{
2021-01-15 20:22:22 +01:00
/// Init process
// This part of the program runs as first process in the pid namespace
// When this terminates then Linux terminates all remaining processes
// in the PID namespace. As we want this to happen when our first child
// terminates, we wait for our first child to terminate before terminating
// ourselves.
// As first process in the PID namespace, this also functions as adopting parent
// for orphaned processes in the PID namespace and therefore has to wait for
// any child process and then check if the a child process that has terminated
// is the one we were waiting for.
2021-01-06 04:42:04 +01:00
pid_t first_child = pid;
pid_t exited_child;
int child_status;
// Setup forward for SIGTERM
pid_child = first_child;
if (sigaction(SIGTERM, &forward_signal_descriptor, NULL) == -1)
2021-01-15 20:22:22 +01:00
{
err(1, "Unable to setup signal forward in init");
2021-01-15 20:22:22 +01:00
}
int wait_errno;
// wait could be interrupt due to a signal. In that case just call wait again.
2021-01-06 04:42:04 +01:00
do {
exited_child = wait(&child_status);
wait_errno = errno;
} while (!(exited_child == first_child || (exited_child == -1 && wait_errno == ECHILD)));
2021-01-06 04:42:04 +01:00
if (exited_child == -1)
{
err(wait_errno, "Error while waiting for subprocess");
2021-01-06 04:42:04 +01:00
}
else
{
2021-01-15 20:22:22 +01:00
return WEXITSTATUS(child_status);
2021-01-06 04:42:04 +01:00
}
}
else
{
// First child of init process. do exec here
// use cli arguments for subprocess. skip 0 as it's our programs name.
2021-01-15 20:22:22 +01:00
argv++;
2021-01-06 04:42:04 +01:00
2021-01-15 20:22:22 +01:00
if (execvp(argv[0], argv) == -1)
2021-01-06 04:42:04 +01:00
{
err(errno, "Failed to exec");
2021-01-06 04:42:04 +01:00
}
}
}
}