Use AC_USE_SYSTEM_EXTENSIONS
[openafs.git] / src / util / softsig.c
1 /*
2  * Copyright 2000, International Business Machines Corporation and others.
3  * All Rights Reserved.
4  *
5  * This software has been released under the terms of the IBM Public
6  * License.  For details, see the LICENSE file in the top-level source
7  * directory or online at http://www.openafs.org/dl/license10.html
8  *
9  * Portions Copyright (c) 2003 Apple Computer, Inc.
10  */
11
12 #include <afsconfig.h>
13 #include <afs/param.h>
14
15 #include <afs/param.h>
16 #include <assert.h>
17 #include <stdlib.h>
18 #include <stdio.h>
19 #ifndef  AFS_NT40_ENV
20 #include <signal.h>
21 #include <unistd.h>
22 #else
23 #include <afs/procmgmt.h>
24 #endif
25 #include <pthread.h>
26
27 #include "pthread_nosigs.h"
28
29 /*------------------------------------------------------------------------
30  * Under Darwin 6.x (including 7.0), sigwait() is broken, so we use
31  * sigsuspend() instead.  We also don't block signals we don't know
32  * about, so they should kill us, rather than us returning zero status.
33  *------------------------------------------------------------------------*/
34
35 static pthread_t softsig_tid;
36 static struct {
37     void (*handler) (int);
38     int pending;
39 #if !(defined(AFS_DARWIN_ENV) || (defined(AFS_NBSD_ENV) && !defined(AFS_NBSD50_ENV)))
40     int fatal;
41 #endif /* !defined(AFS_DARWIN_ENV) || !defined(AFS_NBSD_ENV) */
42     int inited;
43 } softsig_sigs[NSIG];
44
45 static void *
46 softsig_thread(void *arg)
47 {
48     sigset_t ss, os;
49     int i;
50
51     sigemptyset(&ss);
52     /* get the list of signals _not_ blocked by AFS_SIGSET_CLEAR() */
53     pthread_sigmask(SIG_BLOCK, &ss, &os);
54     pthread_sigmask(SIG_SETMASK, &os, NULL);
55     sigaddset(&ss, SIGUSR1);
56 #if defined(AFS_DARWIN_ENV) || (defined(AFS_NBSD_ENV) && !defined(AFS_NBSD50_ENV))
57     pthread_sigmask (SIG_BLOCK, &ss, NULL);
58     sigdelset (&os, SIGUSR1);
59 #elif !defined(AFS_HPUX_ENV)
60     /* On HPUX, don't wait for 'critical' signals, as things such as
61      * SEGV won't cause a core, then. Some non-HPUX platforms may need
62      * this, though, since apparently if we wait on some signals but not
63      * e.g. SEGV, the softsig thread will still wait around when the
64      * other threads were killed by the SEGV. */
65     for (i = 0; i < NSIG; i++) {
66         if (!sigismember(&os, i) && i != SIGSTOP && i != SIGKILL) {
67             sigaddset(&ss, i);
68             softsig_sigs[i].fatal = 1;
69         }
70     }
71 #endif /* defined(AFS_DARWIN_ENV) || defined(AFS_NBSD_ENV) */
72
73     while (1) {
74         void (*h) (int);
75 #if !defined(AFS_DARWIN_ENV) && !defined(AFS_NBSD_ENV)
76         int sigw;
77 #endif
78
79         h = NULL;
80
81         for (i = 0; i < NSIG; i++) {
82             if (softsig_sigs[i].handler && !softsig_sigs[i].inited) {
83                 sigaddset(&ss, i);
84 #if defined(AFS_DARWIN_ENV) || (defined(AFS_NBSD_ENV) && !defined(AFS_NBSD50_ENV))
85                 pthread_sigmask (SIG_BLOCK, &ss, NULL);
86                 sigdelset (&os, i);
87 #endif /* defined(AFS_DARWIN_ENV) || defined(AFS_NBSD_ENV) */
88                 softsig_sigs[i].inited = 1;
89             }
90             if (softsig_sigs[i].pending) {
91                 softsig_sigs[i].pending = 0;
92                 h = softsig_sigs[i].handler;
93                 break;
94             }
95         }
96         if (i == NSIG) {
97 #if defined(AFS_DARWIN_ENV) || (defined(AFS_NBSD_ENV) && !defined(AFS_NBSD50_ENV))
98             sigsuspend (&os);
99 #else /* !defined(AFS_DARWIN_ENV) && !defined(AFS_NBSD_ENV) */
100             sigwait(&ss, &sigw);
101             if (sigw != SIGUSR1) {
102                 if (softsig_sigs[sigw].fatal)
103                     exit(0);
104                 softsig_sigs[sigw].pending = 1;
105             }
106 #endif /* defined(AFS_DARWIN_ENV) || defined(AFS_NBSD_ENV) */
107         } else if (h)
108             h(i);
109     }
110     return NULL;
111 }
112
113 static void
114 softsig_usr1(int signo)
115 {
116     signal (SIGUSR1, softsig_usr1);
117 }
118
119 void
120 softsig_init(void)
121 {
122     int rc;
123     AFS_SIGSET_DECL;
124     AFS_SIGSET_CLEAR();
125     rc = pthread_create(&softsig_tid, NULL, &softsig_thread, NULL);
126     assert(0 == rc);
127     AFS_SIGSET_RESTORE();
128     signal (SIGUSR1, softsig_usr1);
129 }
130
131 static void
132 softsig_handler(int signo)
133 {
134     signal(signo, softsig_handler);
135     softsig_sigs[signo].pending = 1;
136     pthread_kill(softsig_tid, SIGUSR1);
137 }
138
139 void
140 softsig_signal(int signo, void (*handler) (int))
141 {
142     softsig_sigs[signo].handler = handler;
143     softsig_sigs[signo].inited = 0;
144     signal(signo, softsig_handler);
145     pthread_kill(softsig_tid, SIGUSR1);
146 }
147
148 #if defined(TEST)
149 static void
150 print_foo(int signo)
151 {
152     printf("foo, signo = %d, tid = %d\n", signo, pthread_self());
153 }
154
155 int
156 main(int argc, char **argv)
157 {
158     softsig_init();
159     softsig_signal(SIGINT, print_foo);
160     printf("main is tid %d\n", pthread_self());
161     while (1)
162         sleep(60);
163 }
164 #endif