summaryrefslogtreecommitdiffstats
path: root/regress/sys/kern/signal/sigaltstack_fork/sigaltstack_fork.c
blob: a33e717fa380317e8edec672f3691f0c99e89593 (plain) (blame)
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
64
65
66
67
/*	$OpenBSD: sigaltstack_fork.c,v 1.2 2011/11/26 04:11:34 guenther Exp $	*/

/*
 * Public domain.  2011, Joshua Elsasser
 *
 * Test if child processes inherit an alternate signal stack.
 */

#include <sys/types.h>
#include <sys/wait.h>

#include <err.h>
#include <signal.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

void
check_stack(void *buf, const char *label)
{
	struct sigaltstack ss;

	bzero(&ss, sizeof(ss));
	if (sigaltstack(NULL, &ss) != 0)
		err(1, "failed to get sigaltstack in %s", label);
	if (ss.ss_sp != buf ||
	    ss.ss_size != SIGSTKSZ ||
	    ss.ss_flags != 0)
		errx(1, "bad sigaltstack value in %s: "
		    "buf=%p ss_sp=%p ss_size=%zu ss_flags=0x%x",
		    label, buf, ss.ss_sp, ss.ss_size, ss.ss_flags);
}

int
main(int argc, char *argv[])
{
	struct sigaltstack ss;
	int status;
	pid_t kid;
	void *buf;

	if ((buf = malloc(SIGSTKSZ)) == NULL)
		err(1, "malloc failed");

	bzero(&ss, sizeof(ss));
	ss.ss_sp = buf;
	ss.ss_size = SIGSTKSZ;
	if (sigaltstack(&ss, NULL) != 0)
		err(1, "failed to set sigaltstack");

	check_stack(buf, "parent");

	if ((kid = fork()) == -1)
		err(1, "fork failed");

	if (kid == 0) {
		check_stack(buf, "child");
		_exit(0);
	}

	if (waitpid(kid, &status, 0) != kid)
		err(1, "waitpid failed");
	if (!WIFEXITED(status))
		errx(1, "child did not exit normally");

	return (WEXITSTATUS(status));
}