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 }