aboutsummaryrefslogtreecommitdiff
path: root/vendor/github.com/checkpoint-restore/go-criu/test/piggie.c
blob: 1dc0801c0ddd3fd88ffde4b425211685005fc646 (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
#define _GNU_SOURCE
#include <stdio.h>
#include <signal.h>
#include <unistd.h>
#include <sys/mman.h>
#include <fcntl.h>
#include <sched.h>

#define STKS	(4*4096)

#ifndef CLONE_NEWPID
#define CLONE_NEWPID    0x20000000
#endif

static int do_test(void *logf)
{
	int fd, i = 0;

	setsid();

	close(0);
	close(1);
	close(2);

	fd = open("/dev/null", O_RDONLY);
	if (fd != 0) {
		dup2(fd, 0);
		close(fd);
	}

	fd = open(logf, O_WRONLY | O_TRUNC | O_CREAT, 0600);
	dup2(fd, 1);
	dup2(fd, 2);
	if (fd != 1 && fd != 2)
		close(fd);

	while (1) {
		sleep(1);
		printf("%d\n", i++);
		fflush(stdout);
	}

	return 0;
}

int main(int argc, char **argv)
{
	int pid;
	void *stk;

	stk = mmap(NULL, STKS, PROT_READ | PROT_WRITE,
			MAP_PRIVATE | MAP_ANON | MAP_GROWSDOWN, 0, 0);
	pid = clone(do_test, stk + STKS, SIGCHLD | CLONE_NEWPID, argv[1]);
	printf("Child forked, pid %d\n", pid);

	return 0;
}