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