Add missing file and other changes.
[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 2 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, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
20  * USA.
21  */
22
23 #include <config.h>
24 #include <stdio.h>
25 #include <stdlib.h>
26 #include <signal.h>
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   const char *s;
93
94   if (caught_fatal_sig)
95         raise (sig);
96   caught_fatal_sig = 1;
97
98   if (cleanup_fnc)
99     cleanup_fnc ();
100   /* Better don't translate these messages. */
101   write (2, "\n", 1 );
102   s = log_get_prefix (NULL);
103   if (s)
104     write(2, s, strlen (s));
105   write (2, ": signal ", 9 );
106   s = get_signal_name(sig);
107   if (s)
108     write (2, s, strlen(s) );
109   else
110     {
111       /* We are in a signal handler so we can't use any kind of printf
112          even not sprintf.  USe a straightforward algorithm. */
113       if (sig < 0 || sig >= 100000)
114         write (2, "?", 1);
115       else 
116         {
117           int i, any=0;
118
119           for (i=10000; i; i /= 10)
120             {
121               if (sig >= i || ((any || i==1) && !(sig/i)))
122                 {
123                   write (2, "0123456789"+(sig/i), 1);
124                   if ((sig/i))
125                     any = 1;
126                   sig %= i;
127                 }
128             }
129         }
130     }
131   write (2, " caught ... exiting\n", 20);
132   
133   /* Reset action to default action and raise signal again */
134   init_one_signal (sig, SIG_DFL, 0);
135   /* Fixme: remove_lockfiles ();*/
136 #ifdef __riscos__
137   close_fds ();
138 #endif /* __riscos__ */
139   raise( sig );
140 }
141 #endif /*!HAVE_DOSISH_SYSTEM*/
142
143 #ifndef HAVE_DOSISH_SYSTEM
144 static RETSIGTYPE
145 got_usr_signal (int sig)
146 {
147   caught_sigusr1 = 1;
148 }
149 #endif /*!HAVE_DOSISH_SYSTEM*/
150
151 void
152 gnupg_init_signals (int mode, void (*fast_cleanup)(void))
153 {
154   assert (!mode);
155
156   cleanup_fnc = fast_cleanup;
157 #ifndef HAVE_DOSISH_SYSTEM
158   init_one_signal (SIGINT, got_fatal_signal, 1 );
159   init_one_signal (SIGHUP, got_fatal_signal, 1 );
160   init_one_signal (SIGTERM, got_fatal_signal, 1 );
161   init_one_signal (SIGQUIT, got_fatal_signal, 1 );
162   init_one_signal (SIGSEGV, got_fatal_signal, 1 );
163   init_one_signal (SIGUSR1, got_usr_signal, 0 );
164   init_one_signal (SIGPIPE, SIG_IGN, 0 );
165 #endif
166 }
167
168 void
169 gnupg_pause_on_sigusr (int which)
170 {
171 #ifndef HAVE_DOSISH_SYSTEM
172 # ifdef HAVE_SIGPROCMASK
173   sigset_t mask, oldmask;
174
175   assert (which == 1);
176   sigemptyset( &mask );
177   sigaddset( &mask, SIGUSR1 );
178   
179   sigprocmask( SIG_BLOCK, &mask, &oldmask );
180   while (!caught_sigusr1)
181     sigsuspend (&oldmask);
182   caught_sigusr1 = 0;
183   sigprocmask (SIG_UNBLOCK, &mask, NULL);
184 # else 
185   assert (which == 1);
186   sighold (SIGUSR1);
187   while (!caught_sigusr1)
188     sigpause(SIGUSR1);
189   caught_sigusr1 = 0;
190   sigrelease(SIGUSR1);
191 # endif /*!HAVE_SIGPROCMASK*/
192 #endif
193 }
194
195
196 static void
197 do_block( int block )
198 {
199 #ifndef HAVE_DOSISH_SYSTEM
200   static int is_blocked;
201 #ifdef HAVE_SIGPROCMASK
202   static sigset_t oldmask;
203
204   if (block)
205     {
206       sigset_t newmask;
207
208       if (is_blocked)
209         log_bug ("signals are already blocked\n");
210       sigfillset( &newmask );
211       sigprocmask( SIG_BLOCK, &newmask, &oldmask );
212       is_blocked = 1;
213     }
214   else
215     {
216       if (!is_blocked)
217         log_bug("signals are not blocked\n");
218       sigprocmask (SIG_SETMASK, &oldmask, NULL);
219       is_blocked = 0;
220     }
221 #else /*!HAVE_SIGPROCMASK*/
222   static void (*disposition[MAXSIG])();
223   int sig;
224
225   if (block)
226     {
227       if (is_blocked)
228         log_bug("signals are already blocked\n");
229       for (sig=1; sig < MAXSIG; sig++)
230         {
231           disposition[sig] = sigset (sig, SIG_HOLD);
232         }
233       is_blocked = 1;
234     }
235   else
236     {
237       if (!is_blocked)
238         log_bug ("signals are not blocked\n");
239       for (sig=1; sig < MAXSIG; sig++) {
240         sigset (sig, disposition[sig]);
241       }
242       is_blocked = 0;
243     }
244 #endif /*!HAVE_SIGPROCMASK*/
245 #endif /*HAVE_DOSISH_SYSTEM*/
246 }
247
248
249 void
250 gnupg_block_all_signals ()
251 {
252   do_block(1);
253 }
254
255 void
256 gnupg_unblock_all_signals ()
257 {
258   do_block(0);
259 }