1 /* libunwind - a platform-independent unwind library
2    Copyright (C) 2003-2004 Hewlett-Packard Co
3 	Contributed by David Mosberger-Tang <davidm@hpl.hp.com>
4 
5 Permission is hereby granted, free of charge, to any person obtaining
6 a copy of this software and associated documentation files (the
7 "Software"), to deal in the Software without restriction, including
8 without limitation the rights to use, copy, modify, merge, publish,
9 distribute, sublicense, and/or sell copies of the Software, and to
10 permit persons to whom the Software is furnished to do so, subject to
11 the following conditions:
12 
13 The above copyright notice and this permission notice shall be
14 included in all copies or substantial portions of the Software.
15 
16 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
17 EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
18 MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
19 NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
20 LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
21 OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
22 WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.  */
23 
24 /*  Verify that unw_resume() restores the signal mask at proper time.  */
25 
26 #ifdef HAVE_CONFIG_H
27 # include "config.h"
28 #endif
29 
30 #include "compiler.h"
31 
32 #include <libunwind.h>
33 #include <signal.h>
34 #include <stdio.h>
35 #include <stdlib.h>
36 #include <string.h>
37 #include <unistd.h>
38 #include <errno.h>
39 
40 #ifdef HAVE_IA64INTRIN_H
41 # include <ia64intrin.h>
42 #endif
43 
44 #define panic(args...)						\
45 	do { fprintf (stderr, args); ++nerrors; } while (0)
46 
47 int verbose;
48 int nerrors;
49 int got_usr1, got_usr2;
50 char *sigusr1_sp;
51 
52 uintptr_t
get_bsp(void)53 get_bsp (void)
54 {
55 #if UNW_TARGET_IA64
56 # ifdef __INTEL_COMPILER
57   return __getReg (_IA64_REG_AR_BSP);
58 # else
59   return (uintptr_t) __builtin_ia64_bsp ();
60 # endif
61 #else
62   return 0;
63 #endif
64 }
65 
66 #ifdef TEST_WITH_SIGINFO
67 void
handler(int sig,siginfo_t * si UNUSED,void * ucontext UNUSED)68 handler (int sig,
69          siginfo_t *si UNUSED,
70          void *ucontext UNUSED)
71 #else
72 void
73 handler (int sig)
74 #endif
75 {
76   unw_word_t ip;
77   sigset_t mask, oldmask;
78   unw_context_t uc;
79   unw_cursor_t c;
80   char foo;
81   int ret;
82 
83 #if UNW_TARGET_IA64
84   if (verbose)
85     printf ("bsp = %llx\n", (unsigned long long) get_bsp ());
86 #endif
87 
88   if (verbose)
89     printf ("got signal %d\n", sig);
90 
91   if (sig == SIGUSR1)
92     {
93       ++got_usr1;
94       sigusr1_sp = &foo;
95 
96       sigemptyset (&mask);
97       sigaddset (&mask, SIGUSR2);
98       sigprocmask (SIG_BLOCK, &mask, &oldmask);
99       kill (getpid (), SIGUSR2);	/* pend SIGUSR2 */
100 
101       signal (SIGUSR1, SIG_IGN);
102 
103       if ((ret = unw_getcontext (&uc)) < 0)
104 	panic ("unw_getcontext() failed: ret=%d\n", ret);
105 #if UNW_TARGET_X86_64
106       /* unw_getcontext() doesn't save signal mask to avoid a syscall */
107       uc.uc_sigmask = oldmask;
108 #endif
109       if ((ret = unw_init_local (&c, &uc)) < 0)
110 	panic ("unw_init_local() failed: ret=%d\n", ret);
111 
112       if ((ret = unw_step (&c)) < 0)		/* step to signal trampoline */
113 	panic ("unw_step(1) failed: ret=%d\n", ret);
114 
115       if ((ret = unw_step (&c)) < 0)		/* step to kill() */
116 	panic ("unw_step(2) failed: ret=%d\n", ret);
117 
118       if ((ret = unw_get_reg (&c, UNW_REG_IP, &ip)) < 0)
119 	panic ("unw_get_reg(IP) failed: ret=%d\n", ret);
120       if (verbose)
121 	printf ("resuming at 0x%lx, with SIGUSR2 pending\n",
122 		(unsigned long) ip);
123       unw_resume (&c);
124     }
125   else if (sig == SIGUSR2)
126     {
127       ++got_usr2;
128       if (got_usr1)
129 	{
130 	  if (verbose)
131 	    printf ("OK: stack still at %p\n", &foo);
132 	}
133       signal (SIGUSR2, SIG_IGN);
134     }
135   else
136     panic ("Got unexpected signal %d\n", sig);
137 }
138 
139 int
main(int argc,char ** argv UNUSED)140 main (int argc, char **argv UNUSED)
141 {
142   struct sigaction sa;
143   float d = 1.0;
144   int n = 0;
145 
146   if (argc > 1)
147     verbose = 1;
148 
149   memset (&sa, 0, sizeof(sa));
150 #ifdef TEST_WITH_SIGINFO
151   sa.sa_sigaction = handler;
152   sa.sa_flags = SA_SIGINFO;
153 #else
154   sa.sa_handler = handler;
155 #endif
156 
157   if (sigaction (SIGUSR1, &sa, NULL) != 0 ||
158       sigaction (SIGUSR2, &sa, NULL) != 0)
159     {
160       fprintf (stderr, "sigaction() failed: %s\n", strerror (errno));
161       return -1;
162     }
163 
164   /* Use the FPU a bit; otherwise we get spurious errors should the
165      signal handler need to use the FPU for any reason.  This seems to
166      happen on x86-64.  */
167   while (d > 0.0)
168     {
169       d /= 2.0;
170       ++n;
171     }
172   if (n > 9999)
173     return -1;	/* can't happen, but don't tell the compiler... */
174 
175   if (verbose)
176     printf ("sending SIGUSR1\n");
177   kill (getpid (), SIGUSR1);
178 
179   if (!got_usr2)
180     panic ("failed to get SIGUSR2\n");
181 
182   if (nerrors)
183     {
184       fprintf (stderr, "FAILURE: detected %d errors\n", nerrors);
185       exit (-1);
186     }
187 
188   if (verbose)
189     printf ("SUCCESS\n");
190   return 0;
191 }
192