aboutsummaryrefslogtreecommitdiff
path: root/main.c
blob: 7957be3ac5dad8e03f11dd08cce662de5d487e0e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
/* uroot - User's root
 *   main.c Source file with main function
 *
 * Copyright (C) 2018 Karel Kočí
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 */
#define _GNU_SOURCE
#include <stdio.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <sched.h>
#include <unistd.h>
#include <stdint.h>
#include <stdbool.h>
#include "child.h"
#include "utils.h"
#include "sigpipe.h"
#include "conf.h"


int main(int argc, char **argv) {
	// TODO arguments parsing

	struct child_args chargs = (struct child_args) {
		.ppid = getpid(),
		.argc = argc,
		.argv = argv,
		.sigpipe = sigpipe_new()
	};

	uint8_t stack[STACK_SIZE];
	pid_t chpid = clone(child_main, stack + STACK_SIZE,
			SIGCHLD | CLONE_NEWNS | CLONE_NEWUSER | CLONE_NEWPID,
			&chargs);

#define FAIL(MSG) do { fputs(MSG, stderr); kill(chpid, SIGKILL); return 1; } while (false);
	if (new_map_id("uid", chpid, getuid()))
		FAIL("Mapping of uid failed!\n");
	if (new_map_id("gid", chpid, getgid()))
		FAIL("Mapping of gid failed!\n");

	sigpipe_signal(chargs.sigpipe);

	int stat;
	waitpid(chpid, &stat, 0);
	if (WIFEXITED(stat))
		return WEXITSTATUS(stat);
	else
		return WIFSIGNALED(stat) || WCOREDUMP(stat);
}