src/posix/signalHandler.hpp

   1 /*
   2  *  anytun
   3  *
   4  *  The secure anycast tunneling protocol (satp) defines a protocol used
   5  *  for communication between any combination of unicast and anycast
   6  *  tunnel endpoints.  It has less protocol overhead than IPSec in Tunnel
   7  *  mode and allows tunneling of every ETHER TYPE protocol (e.g.
   8  *  ethernet, ip, arp ...). satp directly includes cryptography and
   9  *  message authentication based on the methods used by SRTP.  It is
  10  *  intended to deliver a generic, scaleable and secure solution for
  11  *  tunneling and relaying of packets of any protocol.
  12  *
  13  *
  14  *  Copyright (C) 2007-2014 Markus Grüneis, Othmar Gsenger, Erwin Nindl,
  15  *                          Christian Pointner <satp@wirdorange.org>
  16  *
  17  *  This file is part of Anytun.
  18  *
  19  *  Anytun is free software: you can redistribute it and/or modify
  20  *  it under the terms of the GNU General Public License as published by
  21  *  the Free Software Foundation, either version 3 of the License, or
  22  *  any later version.
  23  *
  24  *  Anytun is distributed in the hope that it will be useful,
  25  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  26  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  27  *  GNU General Public License for more details.
  28  *
  29  *  You should have received a copy of the GNU General Public License
  30  *  along with Anytun.  If not, see <http://www.gnu.org/licenses/>.
  31  *
  32  *  In addition, as a special exception, the copyright holders give
  33  *  permission to link the code of portions of this program with the
  34  *  OpenSSL library under certain conditions as described in each
  35  *  individual source file, and distribute linked combinations
  36  *  including the two.
  37  *  You must obey the GNU General Public License in all respects
  38  *  for all of the code used other than OpenSSL.  If you modify
  39  *  file(s) with this exception, you may extend this exception to your
  40  *  version of the file(s), but you are not obligated to do so.  If you
  41  *  do not wish to do so, delete this exception statement from your
  42  *  version.  If you delete this exception statement from all source
  43  *  files in the program, then also delete it here.
  44  */
  45
  46 #ifndef ANYTUN_signalHandler_hpp_INCLUDED
  47 #define ANYTUN_signalHandler_hpp_INCLUDED
  48
  49 #include <csignal>
  50 #include <boost/thread.hpp>
  51 #include <boost/bind.hpp>
  52
  53 int SigIntHandler(int /*sig*/, const std::string& /*msg*/)
  54 {
  55   cLog.msg(Log::PRIO_NOTICE) << "SIG-Int caught, exiting";
  56   return 1;
  57 }
  58
  59 int SigQuitHandler(int /*sig*/, const std::string& /*msg*/)
  60 {
  61   cLog.msg(Log::PRIO_NOTICE) << "SIG-Quit caught, exiting";
  62   return 1;
  63 }
  64
  65 int SigHupHandler(int /*sig*/, const std::string& /*msg*/)
  66 {
  67   cLog.msg(Log::PRIO_NOTICE) << "SIG-Hup caught";
  68   return 0;
  69 }
  70
  71 int SigTermHandler(int /*sig*/, const std::string& /*msg*/)
  72 {
  73   cLog.msg(Log::PRIO_NOTICE) << "SIG-Term caught, exiting";
  74   return 1;
  75 }
  76
  77 int SigUsr1Handler(int /*sig*/, const std::string& /*msg*/)
  78 {
  79   cLog.msg(Log::PRIO_NOTICE) << "SIG-Usr1 caught";
  80   return 0;
  81 }
  82
  83 int SigUsr2Handler(int /*sig*/, const std::string& /*msg*/)
  84 {
  85   cLog.msg(Log::PRIO_NOTICE) << "SIG-Usr2 caught";
  86   return 0;
  87 }
  88
  89 /// TODO: this outstandignly ugly please and i really can't stress the please fix it asap!!!!!!!
  90 extern std::ofstream pidFile;
  91
  92 void handleSignal()
  93 {
  94   if(pidFile.is_open()) {
  95     pid_t pid = getpid();
  96     pidFile << pid;
  97     pidFile.close();
  98   }
  99
 100   struct timespec timeout;
 101   sigset_t signal_set;
 102   int sigNum;
 103   for(;;) {
 104     sigemptyset(&signal_set);
 105     sigaddset(&signal_set, SIGINT);
 106     sigaddset(&signal_set, SIGQUIT);
 107     sigaddset(&signal_set, SIGHUP);
 108     sigaddset(&signal_set, SIGTERM);
 109     sigaddset(&signal_set, SIGUSR1);
 110     sigaddset(&signal_set, SIGUSR2);
 111     timeout.tv_sec = 1;
 112     timeout.tv_nsec = 0;
 113     sigNum = sigtimedwait(&signal_set, NULL, &timeout);
 114     if(sigNum == -1) {
 115       if(errno != EINTR && errno != EAGAIN) {
 116         cLog.msg(Log::PRIO_ERROR) << "sigwait failed with error: \"" << AnytunErrno(errno) << "\" SignalHandling will be disabled";
 117         break;
 118       }
 119     } else {
 120       gSignalController.inject(sigNum);
 121     }
 122   }
 123 }
 124
 125 void registerSignalHandler(SignalController& ctrl, DaemonService& /*service*/)
 126 {
 127   sigset_t signal_set;
 128
 129   sigemptyset(&signal_set);
 130   sigaddset(&signal_set, SIGINT);
 131   sigaddset(&signal_set, SIGQUIT);
 132   sigaddset(&signal_set, SIGHUP);
 133   sigaddset(&signal_set, SIGTERM);
 134   sigaddset(&signal_set, SIGUSR1);
 135   sigaddset(&signal_set, SIGUSR2);
 136
 137 #if defined(BOOST_HAS_PTHREADS)
 138   pthread_sigmask(SIG_BLOCK, &signal_set, NULL);
 139 #else
 140 #error The signalhandler works only with pthreads
 141 #endif
 142
 143   boost::thread(boost::bind(handleSignal));
 144
 145   ctrl.handler[SIGINT] = boost::bind(SigIntHandler, _1, _2);
 146   ctrl.handler[SIGQUIT] = boost::bind(SigQuitHandler, _1, _2);
 147   ctrl.handler[SIGHUP] = boost::bind(SigHupHandler, _1, _2);
 148   ctrl.handler[SIGTERM] = boost::bind(SigTermHandler, _1, _2);
 149   ctrl.handler[SIGUSR1] = boost::bind(SigUsr1Handler, _1, _2);
 150   ctrl.handler[SIGUSR2] = boost::bind(SigUsr2Handler, _1, _2);
 151
 152   cLog.msg(Log::PRIO_DEBUG) << "signal handlers are now registered";
 153 }
 154
 155 #endif