secur32: Update ntlm_auth version detection to detect new samba4 version numbers.
[wine] / server / signal.c
1 /*
2  * Server signal handling
3  *
4  * Copyright (C) 2003 Alexandre Julliard
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19  */
20
21 #include "config.h"
22
23 #include <signal.h>
24 #include <stdio.h>
25 #ifdef HAVE_POLL_H
26 #include <poll.h>
27 #endif
28 #ifdef HAVE_SYS_POLL_H
29 #include <sys/poll.h>
30 #endif
31 #ifdef HAVE_SYS_RESOURCE_H
32 #include <sys/resource.h>
33 #endif
34 #include <unistd.h>
35
36 #include "file.h"
37 #include "object.h"
38 #include "process.h"
39 #include "thread.h"
40
41 #if defined(linux) && defined(__SIGRTMIN)
42 /* the signal used by linuxthreads as exit signal for clone() threads */
43 # define SIG_PTHREAD_CANCEL (__SIGRTMIN+1)
44 #endif
45
46 typedef void (*signal_callback)(void);
47
48 struct handler
49 {
50     struct object    obj;         /* object header */
51     struct fd       *fd;          /* file descriptor for the pipe side */
52     int              pipe_write;  /* unix fd for the pipe write side */
53     volatile int     pending;     /* is signal pending? */
54     signal_callback  callback;    /* callback function */
55 };
56
57 static void handler_dump( struct object *obj, int verbose );
58 static void handler_destroy( struct object *obj );
59
60 static const struct object_ops handler_ops =
61 {
62     sizeof(struct handler),   /* size */
63     handler_dump,             /* dump */
64     no_add_queue,             /* add_queue */
65     NULL,                     /* remove_queue */
66     NULL,                     /* signaled */
67     NULL,                     /* satisfied */
68     no_signal,                /* signal */
69     no_get_fd,                /* get_fd */
70     no_map_access,            /* map_access */
71     no_lookup_name,           /* lookup_name */
72     no_close_handle,          /* close_handle */
73     handler_destroy           /* destroy */
74 };
75
76 static void handler_poll_event( struct fd *fd, int event );
77
78 static const struct fd_ops handler_fd_ops =
79 {
80     NULL,                     /* get_poll_events */
81     handler_poll_event,       /* poll_event */
82     no_flush,                 /* flush */
83     no_get_file_info,         /* get_file_info */
84     no_queue_async,           /* queue_async */
85     no_cancel_async           /* cancel_async */
86 };
87
88 static struct handler *handler_sighup;
89 static struct handler *handler_sigterm;
90 static struct handler *handler_sigint;
91 static struct handler *handler_sigchld;
92 static struct handler *handler_sigio;
93
94 static sigset_t blocked_sigset;
95
96 static int watchdog;
97
98 /* create a signal handler */
99 static struct handler *create_handler( signal_callback callback )
100 {
101     struct handler *handler;
102     int fd[2];
103
104     if (pipe( fd ) == -1) return NULL;
105     if (!(handler = alloc_object( &handler_ops )))
106     {
107         close( fd[0] );
108         close( fd[1] );
109         return NULL;
110     }
111     handler->pipe_write = fd[1];
112     handler->pending    = 0;
113     handler->callback   = callback;
114
115     if (!(handler->fd = create_anonymous_fd( &handler_fd_ops, fd[0], &handler->obj )))
116     {
117         release_object( handler );
118         return NULL;
119     }
120     set_fd_events( handler->fd, POLLIN );
121     return handler;
122 }
123
124 /* handle a signal received for a given handler */
125 static void do_signal( struct handler *handler )
126 {
127     if (!handler->pending)
128     {
129         char dummy = 0;
130         handler->pending = 1;
131         write( handler->pipe_write, &dummy, 1 );
132     }
133 }
134
135 static void handler_dump( struct object *obj, int verbose )
136 {
137     struct handler *handler = (struct handler *)obj;
138     fprintf( stderr, "Signal handler fd=%p\n", handler->fd );
139 }
140
141 static void handler_destroy( struct object *obj )
142 {
143     struct handler *handler = (struct handler *)obj;
144     if (handler->fd) release_object( handler->fd );
145     close( handler->pipe_write );
146 }
147
148 static void handler_poll_event( struct fd *fd, int event )
149 {
150     struct handler *handler = get_fd_user( fd );
151
152     if (event & (POLLERR | POLLHUP))
153     {
154         /* this is not supposed to happen */
155         fprintf( stderr, "wineserver: Error on signal handler pipe\n" );
156         release_object( handler );
157     }
158     else if (event & POLLIN)
159     {
160         char dummy;
161
162         handler->pending = 0;
163         read( get_unix_fd( handler->fd ), &dummy, 1 );
164         handler->callback();
165     }
166 }
167
168 /* SIGHUP callback */
169 static void sighup_callback(void)
170 {
171 #ifdef DEBUG_OBJECTS
172     dump_objects();
173 #endif
174 }
175
176 /* SIGTERM callback */
177 static void sigterm_callback(void)
178 {
179     flush_registry();
180     exit(1);
181 }
182
183 /* SIGINT callback */
184 static void sigint_callback(void)
185 {
186     kill_all_processes( NULL, 1 );
187     flush_registry();
188     exit(1);
189 }
190
191 /* SIGHUP handler */
192 static void do_sighup( int signum )
193 {
194     do_signal( handler_sighup );
195 }
196
197 /* SIGTERM handler */
198 static void do_sigterm( int signum )
199 {
200     do_signal( handler_sigterm );
201 }
202
203 /* SIGINT handler */
204 static void do_sigint( int signum )
205 {
206     do_signal( handler_sigint );
207 }
208
209 /* SIGALRM handler */
210 static void do_sigalrm( int signum )
211 {
212     watchdog = 1;
213 }
214
215 /* SIGCHLD handler */
216 static void do_sigchld( int signum )
217 {
218     do_signal( handler_sigchld );
219 }
220
221 /* SIGSEGV handler */
222 static void do_sigsegv( int signum )
223 {
224     fprintf( stderr, "wineserver crashed, please report this.\n");
225     abort();
226 }
227
228 /* SIGIO handler */
229 #ifdef HAVE_SIGINFO_T_SI_FD
230 static void do_sigio( int signum, siginfo_t *si, void *x )
231 {
232     do_signal( handler_sigio );
233     do_change_notify( si->si_fd );
234 }
235 #endif
236
237 void start_watchdog(void)
238 {
239     alarm( 3 );
240     watchdog = 0;
241 }
242
243 void stop_watchdog(void)
244 {
245     alarm( 0 );
246     watchdog = 0;
247 }
248
249 int watchdog_triggered(void)
250 {
251     return watchdog != 0;
252 }
253
254 static int core_dump_disabled( void )
255 {
256     int r = 0;
257 #ifdef RLIMIT_CORE
258     struct rlimit lim;
259
260     r = !getrlimit(RLIMIT_CORE, &lim) && (lim.rlim_cur == 0);
261 #endif
262     return r;
263 }
264
265 void init_signals(void)
266 {
267     struct sigaction action;
268
269     if (!(handler_sighup  = create_handler( sighup_callback ))) goto error;
270     if (!(handler_sigterm = create_handler( sigterm_callback ))) goto error;
271     if (!(handler_sigint  = create_handler( sigint_callback ))) goto error;
272     if (!(handler_sigchld = create_handler( sigchld_callback ))) goto error;
273     if (!(handler_sigio   = create_handler( sigio_callback ))) goto error;
274
275     sigemptyset( &blocked_sigset );
276     sigaddset( &blocked_sigset, SIGCHLD );
277     sigaddset( &blocked_sigset, SIGHUP );
278     sigaddset( &blocked_sigset, SIGINT );
279     sigaddset( &blocked_sigset, SIGALRM );
280     sigaddset( &blocked_sigset, SIGIO );
281     sigaddset( &blocked_sigset, SIGQUIT );
282     sigaddset( &blocked_sigset, SIGTERM );
283 #ifdef SIG_PTHREAD_CANCEL
284     sigaddset( &blocked_sigset, SIG_PTHREAD_CANCEL );
285 #endif
286
287     action.sa_mask = blocked_sigset;
288     action.sa_flags = 0;
289     action.sa_handler = do_sigchld;
290     sigaction( SIGCHLD, &action, NULL );
291 #ifdef SIG_PTHREAD_CANCEL
292     sigaction( SIG_PTHREAD_CANCEL, &action, NULL );
293 #endif
294     action.sa_handler = do_sighup;
295     sigaction( SIGHUP, &action, NULL );
296     action.sa_handler = do_sigint;
297     sigaction( SIGINT, &action, NULL );
298     action.sa_handler = do_sigalrm;
299     sigaction( SIGALRM, &action, NULL );
300     action.sa_handler = do_sigterm;
301     sigaction( SIGQUIT, &action, NULL );
302     sigaction( SIGTERM, &action, NULL );
303     if (core_dump_disabled())
304     {
305         action.sa_handler = do_sigsegv;
306         sigaction( SIGSEGV, &action, NULL );
307     }
308     action.sa_handler = SIG_IGN;
309     sigaction( SIGXFSZ, &action, NULL );
310 #ifdef HAVE_SIGINFO_T_SI_FD
311     action.sa_sigaction = do_sigio;
312     action.sa_flags = SA_SIGINFO;
313     sigaction( SIGIO, &action, NULL );
314 #endif
315     return;
316
317 error:
318     fprintf( stderr, "failed to initialize signal handlers\n" );
319     exit(1);
320 }
321
322 void close_signals(void)
323 {
324     sigprocmask( SIG_BLOCK, &blocked_sigset, NULL );
325     release_object( handler_sighup );
326     release_object( handler_sigterm );
327     release_object( handler_sigint );
328     release_object( handler_sigchld );
329     release_object( handler_sigio );
330 }