ee55f1917c395f306bc50c7630bb40fae1e9a498
[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 #ifdef HAVE_SIGNAL_H
25 # include <signal.h>
26 #endif
27 #include <unistd.h>
28 #include <string.h>
29 #include <errno.h>
30 #include <assert.h>
31
32 #include "util.h"
33
34
35 #ifndef HAVE_DOSISH_SYSTEM
36 static volatile int caught_fatal_sig;
37 static volatile int caught_sigusr1;
38 #endif
39 static void (*cleanup_fnc)(void);
40
41
42 #ifndef HAVE_DOSISH_SYSTEM
43 static void
44 init_one_signal (int sig, RETSIGTYPE (*handler)(int), int check_ign )
45 {
46 # ifdef HAVE_SIGACTION
47   struct sigaction oact, nact;
48   
49   if (check_ign)
50     {
51       /* we don't want to change an IGN handler */
52       sigaction (sig, NULL, &oact );
53       if (oact.sa_handler == SIG_IGN )
54         return;
55     }
56
57   nact.sa_handler = handler;
58   sigemptyset (&nact.sa_mask);
59   nact.sa_flags = 0;
60   sigaction ( sig, &nact, NULL);
61 # else 
62   RETSIGTYPE (*ohandler)(int);
63   
64   ohandler = signal (sig, handler);
65   if (check_ign && ohandler == SIG_IGN) 
66     {
67       /* Change it back if it was already set to IGN */
68       signal (sig, SIG_IGN);
69     }
70 # endif
71 }
72 #endif /*!HAVE_DOSISH_SYSTEM*/
73
74 #ifndef HAVE_DOSISH_SYSTEM
75 static const char *
76 get_signal_name( int signum )
77 {
78   /* Note that we can't use strsignal(), because it is not
79      reentrant. */
80 #if HAVE_DECL_SYS_SIGLIST && defined(NSIG)
81   return (signum >= 0 && signum < NSIG) ? sys_siglist[signum] : "?";
82 #else
83   return NULL;
84 #endif
85 }
86 #endif /*!HAVE_DOSISH_SYSTEM*/
87
88 #ifndef HAVE_DOSISH_SYSTEM
89 static RETSIGTYPE
90 got_fatal_signal (int sig)
91 {
92   /* Dummy result variable to suppress gcc warning.  */
93   int res;
94   const char *s;
95
96   if (caught_fatal_sig)
97     raise (sig);
98   caught_fatal_sig = 1;
99   
100   if (cleanup_fnc)
101     cleanup_fnc ();
102   /* Better don't translate these messages. */
103   res = write (2, "\n", 1 );
104   s = log_get_prefix (NULL);
105   if (s)
106     res = write(2, s, strlen (s));
107   res = write (2, ": signal ", 9 );
108   s = get_signal_name(sig);
109   if (s)
110     res = write (2, s, strlen(s) );
111   else
112     {
113       /* We are in a signal handler so we can't use any kind of printf
114          even not sprintf.  So we use a straightforward algorithm.  We
115          got a report that on one particular system, raising a signal
116          while in this handler, the parameter SIG get sclobbered and
117          things are messed up because we modify its value.  Although
118          this is a bug in that system, we will protect against it.  */
119       if (sig < 0 || sig >= 100000)
120         res = write (2, "?", 1);
121       else 
122         {
123           int i, value, any=0;
124
125           for (value=sig,i=10000; i; i /= 10)
126             {
127               if (value >= i || ((any || i==1) && !(value/i)))
128                 {
129                   res = write (2, "0123456789"+(value/i), 1);
130                   if ((value/i))
131                     any = 1;
132                   value %= i;
133                 }
134             }
135         }
136     }
137   res = write (2, " caught ... exiting\n", 20);
138   
139   /* Reset action to default action and raise signal again */
140   init_one_signal (sig, SIG_DFL, 0);
141   /* Fixme: remove_lockfiles ();*/
142 #ifdef __riscos__
143   close_fds ();
144 #endif /* __riscos__ */
145   raise( sig );
146 }
147 #endif /*!HAVE_DOSISH_SYSTEM*/
148
149 #ifndef HAVE_DOSISH_SYSTEM
150 static RETSIGTYPE
151 got_usr_signal (int sig)
152 {
153   (void)sig;
154   caught_sigusr1 = 1;
155 }
156 #endif /*!HAVE_DOSISH_SYSTEM*/
157
158 void
159 gnupg_init_signals (int mode, void (*fast_cleanup)(void))
160 {
161   assert (!mode);
162
163   cleanup_fnc = fast_cleanup;
164 #ifndef HAVE_DOSISH_SYSTEM
165   init_one_signal (SIGINT, got_fatal_signal, 1 );
166   init_one_signal (SIGHUP, got_fatal_signal, 1 );
167   init_one_signal (SIGTERM, got_fatal_signal, 1 );
168   init_one_signal (SIGQUIT, got_fatal_signal, 1 );
169   init_one_signal (SIGSEGV, got_fatal_signal, 1 );
170   init_one_signal (SIGUSR1, got_usr_signal, 0 );
171   init_one_signal (SIGPIPE, SIG_IGN, 0 );
172 #endif
173 }
174
175
176 static void
177 do_block (int block)
178 {
179 #ifdef HAVE_DOSISH_SYSTEM
180   (void)block;
181 #else /*!HAVE_DOSISH_SYSTEM*/
182   static int is_blocked;
183 #ifdef HAVE_SIGPROCMASK
184   static sigset_t oldmask;
185
186   if (block)
187     {
188       sigset_t newmask;
189
190       if (is_blocked)
191         log_bug ("signals are already blocked\n");
192       sigfillset( &newmask );
193       sigprocmask( SIG_BLOCK, &newmask, &oldmask );
194       is_blocked = 1;
195     }
196   else
197     {
198       if (!is_blocked)
199         log_bug("signals are not blocked\n");
200       sigprocmask (SIG_SETMASK, &oldmask, NULL);
201       is_blocked = 0;
202     }
203 #else /*!HAVE_SIGPROCMASK*/
204   static void (*disposition[MAXSIG])();
205   int sig;
206
207   if (block)
208     {
209       if (is_blocked)
210         log_bug("signals are already blocked\n");
211       for (sig=1; sig < MAXSIG; sig++)
212         {
213           disposition[sig] = sigset (sig, SIG_HOLD);
214         }
215       is_blocked = 1;
216     }
217   else
218     {
219       if (!is_blocked)
220         log_bug ("signals are not blocked\n");
221       for (sig=1; sig < MAXSIG; sig++) {
222         sigset (sig, disposition[sig]);
223       }
224       is_blocked = 0;
225     }
226 #endif /*!HAVE_SIGPROCMASK*/
227 #endif /*!HAVE_DOSISH_SYSTEM*/
228 }
229
230
231 void
232 gnupg_block_all_signals ()
233 {
234   do_block(1);
235 }
236
237 void
238 gnupg_unblock_all_signals ()
239 {
240   do_block(0);
241 }