xref: /xnu-11417.140.69/tests/signal_stack.c (revision 43a90889846e00bfb5cf1d255cdc0a701a1e05a4)
1*43a90889SApple OSS Distributions #include <signal.h>
2*43a90889SApple OSS Distributions #include <stdio.h>
3*43a90889SApple OSS Distributions #include <stdlib.h>
4*43a90889SApple OSS Distributions #include <unistd.h>
5*43a90889SApple OSS Distributions #include <errno.h>
6*43a90889SApple OSS Distributions 
7*43a90889SApple OSS Distributions #include <darwintest.h>
8*43a90889SApple OSS Distributions 
9*43a90889SApple OSS Distributions static uint64_t stack_base, stack_end;
10*43a90889SApple OSS Distributions 
11*43a90889SApple OSS Distributions static void
signal_handler(int __unused signum,struct __siginfo * __unused info,void * __unused uap)12*43a90889SApple OSS Distributions signal_handler(int  __unused signum, struct __siginfo * __unused info, void * __unused uap)
13*43a90889SApple OSS Distributions {
14*43a90889SApple OSS Distributions 	T_LOG("In signal handler\n");
15*43a90889SApple OSS Distributions 	uint64_t signal_stack = (uint64_t)__builtin_frame_address(0);
16*43a90889SApple OSS Distributions 	T_ASSERT_LE(stack_base, signal_stack, NULL);
17*43a90889SApple OSS Distributions 	T_ASSERT_LE(signal_stack, stack_end, NULL);
18*43a90889SApple OSS Distributions 	T_END;
19*43a90889SApple OSS Distributions }
20*43a90889SApple OSS Distributions 
21*43a90889SApple OSS Distributions T_DECL(signalstack, "Check that the signal stack is set up correctly", T_META_ASROOT(YES), T_META_TAG_VM_PREFERRED)
22*43a90889SApple OSS Distributions {
23*43a90889SApple OSS Distributions 	void* stack_allocation = malloc(SIGSTKSZ);
24*43a90889SApple OSS Distributions 
25*43a90889SApple OSS Distributions 	stack_base = (uint64_t)stack_allocation;
26*43a90889SApple OSS Distributions 	stack_end = stack_base + SIGSTKSZ;
27*43a90889SApple OSS Distributions 
28*43a90889SApple OSS Distributions 	T_LOG("stack base = 0x%llx\n", stack_base);
29*43a90889SApple OSS Distributions 	T_LOG("stack end = 0x%llx\n", stack_end);
30*43a90889SApple OSS Distributions 
31*43a90889SApple OSS Distributions 	stack_t alt_stack;
32*43a90889SApple OSS Distributions 	alt_stack.ss_sp = stack_allocation;
33*43a90889SApple OSS Distributions 	alt_stack.ss_size = SIGSTKSZ;
34*43a90889SApple OSS Distributions 	alt_stack.ss_flags = 0;
35*43a90889SApple OSS Distributions 
36*43a90889SApple OSS Distributions 	if (sigaltstack(&alt_stack, NULL) < 0) {
37*43a90889SApple OSS Distributions 		T_FAIL("error: sigaltstack failed\n");
38*43a90889SApple OSS Distributions 	}
39*43a90889SApple OSS Distributions 
40*43a90889SApple OSS Distributions 	sigset_t signal_mask;
41*43a90889SApple OSS Distributions 	sigemptyset(&signal_mask);
42*43a90889SApple OSS Distributions 
43*43a90889SApple OSS Distributions 	struct sigaction sig_action;
44*43a90889SApple OSS Distributions 	sig_action.sa_sigaction = signal_handler;
45*43a90889SApple OSS Distributions 	sig_action.sa_mask = signal_mask;
46*43a90889SApple OSS Distributions 	sig_action.sa_flags = SA_ONSTACK;
47*43a90889SApple OSS Distributions 
48*43a90889SApple OSS Distributions 	if (sigaction(SIGUSR1, &sig_action, NULL) != 0) {
49*43a90889SApple OSS Distributions 		T_FAIL("error: sigaction failed\n");
50*43a90889SApple OSS Distributions 	}
51*43a90889SApple OSS Distributions 
52*43a90889SApple OSS Distributions 	T_LOG("Sending a SIGUSR1\n");
53*43a90889SApple OSS Distributions 	kill(getpid(), SIGUSR1);
54*43a90889SApple OSS Distributions 
55*43a90889SApple OSS Distributions 	return;
56*43a90889SApple OSS Distributions }
57