Changed to GPLv3.
[gnupg.git] / common / signal.c
1 /* signal.c - signal handling
2  * Copyright (C) 1998, 1999, 2000, 2001, 2002,
3  *               2005 Free Software Foundation, Inc.
4  *
5  * This file is part of GnuPG.
6  *
7  * GnuPG is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License as published by
9  * the Free Software Foundation; either version 3 of the License, or
10  * (at your option) any later version.
11  *
12  * GnuPG is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License
18  * along with this program; if not, see <http://www.gnu.org/licenses/>.
19  */
20
21 #include <config.h>
22 #include <stdio.h>
23 #include <stdlib.h>
24 #include <signal.h>
25 #include <unistd.h>
26 #include <string.h>
27 #include <errno.h>
28 #include <assert.h>
29
30 #include "util.h"
31
32
33 #ifndef HAVE_DOSISH_SYSTEM
34 static volatile int caught_fatal_sig;
35 static volatile int caught_sigusr1;
36 #endif
37 static void (*cleanup_fnc)(void);
38
39
40 #ifndef HAVE_DOSISH_SYSTEM
41 static void
42 init_one_signal (int sig, RETSIGTYPE (*handler)(int), int check_ign )
43 {
44 # ifdef HAVE_SIGACTION
45   struct sigaction oact, nact;
46   
47   if (check_ign)
48     {
49       /* we don't want to change an IGN handler */
50       sigaction (sig, NULL, &oact );
51       if (oact.sa_handler == SIG_IGN )
52         return;
53     }
54
55   nact.sa_handler = handler;
56   sigemptyset (&nact.sa_mask);
57   nact.sa_flags = 0;
58   sigaction ( sig, &nact, NULL);
59 # else 
60   RETSIGTYPE (*ohandler)(int);
61   
62   ohandler = signal (sig, handler);
63   if (check_ign && ohandler == SIG_IGN) 
64     {
65       /* Change it back if it was already set to IGN */
66       signal (sig, SIG_IGN);
67     }
68 # endif
69 }
70 #endif /*!HAVE_DOSISH_SYSTEM*/
71
72 #ifndef HAVE_DOSISH_SYSTEM
73 static const char *
74 get_signal_name( int signum )
75 {
76   /* Note that we can't use strsignal(), because it is not
77      reentrant. */
78 #if HAVE_DECL_SYS_SIGLIST && defined(NSIG)
79   return (signum >= 0 && signum < NSIG) ? sys_siglist[signum] : "?";
80 #else
81   return NULL;
82 #endif
83 }
84 #endif /*!HAVE_DOSISH_SYSTEM*/
85
86 #ifndef HAVE_DOSISH_SYSTEM
87 static RETSIGTYPE
88 got_fatal_signal (int sig)
89 {
90   const char *s;
91
92   if (caught_fatal_sig)
93     raise (sig);
94   caught_fatal_sig = 1;
95   
96   if (cleanup_fnc)
97     cleanup_fnc ();
98   /* Better don't translate these messages. */
99   write (2, "\n", 1 );
100   s = log_get_prefix (NULL);
101   if (s)
102     write(2, s, strlen (s));
103   write (2, ": signal ", 9 );
104   s = get_signal_name(sig);
105   if (s)
106     write (2, s, strlen(s) );
107   else
108     {
109       /* We are in a signal handler so we can't use any kind of printf
110          even not sprintf.  So we use a straightforward algorithm.  We
111          got a report that on one particular system, raising a signal
112          while in this handler, the parameter SIG get sclobbered and
113          things are messed up because we modify its value.  Although
114          this is a bug in that system, we will protect against it.  */
115       if (sig < 0 || sig >= 100000)
116         write (2, "?", 1);
117       else 
118         {
119           int i, value, any=0;
120
121           for (value=sig,i=10000; i; i /= 10)
122             {
123               if (value >= i || ((any || i==1) && !(value/i)))
124                 {
125                   write (2, "0123456789"+(value/i), 1);
126                   if ((value/i))
127                     any = 1;
128                   value %= i;
129                 }
130             }
131         }
132     }
133   write (2, " caught ... exiting\n", 20);
134   
135   /* Reset action to default action and raise signal again */
136   init_one_signal (sig, SIG_DFL, 0);
137   /* Fixme: remove_lockfiles ();*/
138 #ifdef __riscos__
139   close_fds ();
140 #endif /* __riscos__ */
141   raise( sig );
142 }
143 #endif /*!HAVE_DOSISH_SYSTEM*/
144
145 #ifndef HAVE_DOSISH_SYSTEM
146 static RETSIGTYPE
147 got_usr_signal (int sig)
148 {
149   caught_sigusr1 = 1;
150 }
151 #endif /*!HAVE_DOSISH_SYSTEM*/
152
153 void
154 gnupg_init_signals (int mode, void (*fast_cleanup)(void))
155 {
156   assert (!mode);
157
158   cleanup_fnc = fast_cleanup;
159 #ifndef HAVE_DOSISH_SYSTEM
160   init_one_signal (SIGINT, got_fatal_signal, 1 );
161   init_one_signal (SIGHUP, got_fatal_signal, 1 );
162   init_one_signal (SIGTERM, got_fatal_signal, 1 );
163   init_one_signal (SIGQUIT, got_fatal_signal, 1 );
164   init_one_signal (SIGSEGV, got_fatal_signal, 1 );
165   init_one_signal (SIGUSR1, got_usr_signal, 0 );
166   init_one_signal (SIGPIPE, SIG_IGN, 0 );
167 #endif
168 }
169
170 void
171 gnupg_pause_on_sigusr (int which)
172 {
173 #ifndef HAVE_DOSISH_SYSTEM
174 # ifdef HAVE_SIGPROCMASK
175   sigset_t mask, oldmask;
176
177   assert (which == 1);
178   sigemptyset( &mask );
179   sigaddset( &mask, SIGUSR1 );
180   
181   sigprocmask( SIG_BLOCK, &mask, &oldmask );
182   while (!caught_sigusr1)
183     sigsuspend (&oldmask);
184   caught_sigusr1 = 0;
185   sigprocmask (SIG_UNBLOCK, &mask, NULL);
186 # else 
187   assert (which == 1);
188   sighold (SIGUSR1);
189   while (!caught_sigusr1)
190     sigpause(SIGUSR1);
191   caught_sigusr1 = 0;
192   sigrelease(SIGUSR1);
193 # endif /*!HAVE_SIGPROCMASK*/
194 #endif
195 }
196
197
198 static void
199 do_block( int block )
200 {
201 #ifndef HAVE_DOSISH_SYSTEM
202   static int is_blocked;
203 #ifdef HAVE_SIGPROCMASK
204   static sigset_t oldmask;
205
206   if (block)
207     {
208       sigset_t newmask;
209
210       if (is_blocked)
211         log_bug ("signals are already blocked\n");
212       sigfillset( &newmask );
213       sigprocmask( SIG_BLOCK, &newmask, &oldmask );
214       is_blocked = 1;
215     }
216   else
217     {
218       if (!is_blocked)
219         log_bug("signals are not blocked\n");
220       sigprocmask (SIG_SETMASK, &oldmask, NULL);
221       is_blocked = 0;
222     }
223 #else /*!HAVE_SIGPROCMASK*/
224   static void (*disposition[MAXSIG])();
225   int sig;
226
227   if (block)
228     {
229       if (is_blocked)
230         log_bug("signals are already blocked\n");
231       for (sig=1; sig < MAXSIG; sig++)
232         {
233           disposition[sig] = sigset (sig, SIG_HOLD);
234         }
235       is_blocked = 1;
236     }
237   else
238     {
239       if (!is_blocked)
240         log_bug ("signals are not blocked\n");
241       for (sig=1; sig < MAXSIG; sig++) {
242         sigset (sig, disposition[sig]);
243       }
244       is_blocked = 0;
245     }
246 #endif /*!HAVE_SIGPROCMASK*/
247 #endif /*HAVE_DOSISH_SYSTEM*/
248 }
249
250
251 void
252 gnupg_block_all_signals ()
253 {
254   do_block(1);
255 }
256
257 void
258 gnupg_unblock_all_signals ()
259 {
260   do_block(0);
261 }