src/gromacs/gmxlib/sighandler.c

   1 /*
   2  *
   3  *                This source code is part of
   4  *
   5  *                 G   R   O   M   A   C   S
   6  *
   7  *          GROningen MAchine for Chemical Simulations
   8  *
   9  *                        VERSION 3.2.0
  10  * Written by David van der Spoel, Erik Lindahl, Berk Hess, and others.
  11  * Copyright (c) 1991-2000, University of Groningen, The Netherlands.
  12  * Copyright (c) 2001-2004, The GROMACS development team,
  13  * check out http://www.gromacs.org for more information.
  14
  15  * This program is free software; you can redistribute it and/or
  16  * modify it under the terms of the GNU General Public License
  17  * as published by the Free Software Foundation; either version 2
  18  * of the License, or (at your option) any later version.
  19  *
  20  * If you want to redistribute modifications, please consider that
  21  * scientific software is very special. Version control is crucial -
  22  * bugs must be traceable. We will be happy to consider code for
  23  * inclusion in the official distribution, but derived work must not
  24  * be called official GROMACS. Details are found in the README & COPYING
  25  * files - if they are missing, get the official version at www.gromacs.org.
  26  *
  27  * To help us fund GROMACS development, we humbly ask that you cite
  28  * the papers on the package - you can find them in the top README file.
  29  *
  30  * For more info, check our website at http://www.gromacs.org
  31  *
  32  * And Hey:
  33  * GROwing Monsters And Cloning Shrimps
  34  */
  35 #ifdef HAVE_CONFIG_H
  36 #include <config.h>
  37 #endif
  38
  39 #include "typedefs.h"
  40 #include "gmx_fatal.h"
  41 #include "sighandler.h"
  42
  43
  44 const char *gmx_stop_cond_name[] =
  45 {
  46     "None",
  47     "Stop at the next neighbor search step",
  48     "Stop at the next step",
  49     "Abort"
  50 };
  51
  52 /* these do not neccesarily match the stop condition, but are
  53    referred to in the signal handler. */
  54 const char *gmx_signal_name[] =
  55 {
  56     "None",
  57     "INT",
  58     "TERM",
  59     "second INT/TERM",
  60     "remote INT/TERM",
  61     "remote second INT/TERM",
  62     "USR1",
  63     "Abort"
  64 };
  65
  66 static volatile sig_atomic_t stop_condition=gmx_stop_cond_none;
  67 static volatile sig_atomic_t last_signal_name=0;
  68
  69 static volatile sig_atomic_t usr_condition=0;
  70
  71 static void signal_handler(int n)
  72 {
  73     switch (n) {
  74 /* windows doesn't do SIGINT correctly according to ANSI (yes, signals are in
  75    ANSI C89, and windows spawns a thread specifically to run the INT signal
  76    handler), but that doesn't matter for a simple signal handler like this. */
  77         case SIGTERM:
  78         case SIGINT:
  79             /* we explicitly set things up to allow this: */
  80             stop_condition++;
  81             if (n==SIGINT)
  82                 last_signal_name=1;
  83             if (n==SIGTERM)
  84                 last_signal_name=2;
  85             if (stop_condition == gmx_stop_cond_next)
  86                 last_signal_name=3;
  87             if (stop_condition >= gmx_stop_cond_abort)
  88                 abort();
  89             break;
  90 #ifdef HAVE_SIGUSR1
  91         case SIGUSR1:
  92             usr_condition=1;
  93             break;
  94 #endif
  95         default:
  96             break;
  97     }
  98 }
  99
 100 static void gmx_signal(int signum)
 101 {
 102 #ifdef HAVE_SIGACTION
 103     struct sigaction act;
 104     act.sa_handler = signal_handler;
 105     act.sa_flags = SA_RESTART;
 106     sigaction(signum,&act,NULL);
 107 #else
 108     signal(signum,signal_handler);
 109 #endif
 110 }
 111
 112 void signal_handler_install(void)
 113 {
 114     if (getenv("GMX_NO_TERM") == NULL)
 115     {
 116         if (debug)
 117         {
 118             fprintf(debug,"Installing signal handler for SIGTERM\n");
 119         }
 120         gmx_signal(SIGTERM);
 121     }
 122     if (getenv("GMX_NO_INT") == NULL)
 123     {
 124         if (debug)
 125         {
 126             fprintf(debug,"Installing signal handler for SIGINT\n");
 127         }
 128         gmx_signal(SIGINT);
 129     }
 130 #ifdef HAVE_SIGUSR1
 131     if (getenv("GMX_NO_USR1") == NULL)
 132     {
 133         if (debug)
 134         {
 135             fprintf(debug,"Installing signal handler for SIGUSR1\n");
 136         }
 137         gmx_signal(SIGUSR1);
 138     }
 139 #endif
 140 }
 141
 142 gmx_stop_cond_t gmx_get_stop_condition(void)
 143 {
 144     return (gmx_stop_cond_t)stop_condition;
 145 }
 146
 147 void gmx_set_stop_condition(gmx_stop_cond_t recvd_stop_cond)
 148 {
 149     if (recvd_stop_cond > stop_condition)
 150     {
 151         stop_condition=recvd_stop_cond;
 152         if (stop_condition == gmx_stop_cond_next_ns)
 153             last_signal_name=4;
 154         if (stop_condition == gmx_stop_cond_next)
 155             last_signal_name=5;
 156     }
 157 }
 158
 159 const char *gmx_get_signal_name(void)
 160 {
 161     return gmx_signal_name[last_signal_name];
 162 }
 163
 164 gmx_bool gmx_got_usr_signal(void)
 165 {
 166 #ifdef HAVE_SIGUSR1
 167     gmx_bool ret=(gmx_bool)usr_condition;
 168     usr_condition=0;
 169     return ret;
 170 #else
 171     return FALSE;
 172 #endif
 173 }
 174
 175