src/signalController.cpp

   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 #include <map>
  47 #include <iostream>
  48 #include <boost/bind.hpp>
  49
  50 #include "signalController.h"
  51 #include "log.h"
  52 #include "anytunError.h"
  53 #include "threadUtils.hpp"
  54
  55
  56 SignalController& gSignalController = SignalController::instance();
  57
  58 SignalController& SignalController::instance()
  59 {
  60   static SignalController instance;
  61   return instance;
  62 }
  63
  64 int SigErrorHandler(int /*sig*/, const std::string& msg)
  65 {
  66   AnytunError::throwErr() << msg;
  67
  68   return 0;
  69 }
  70
  71 //use system specific signal handler
  72 #if !defined(_MSC_VER) && !defined(MINGW)
  73 #include "signalHandler.hpp"
  74 #else
  75 #ifdef WIN_SERVICE
  76 #include "win32/signalServiceHandler.hpp"
  77 #else
  78 #include "win32/signalHandler.hpp"
  79 #endif
  80 #endif
  81
  82 void SignalController::init(DaemonService& service)
  83 {
  84   registerSignalHandler(*this, service);
  85   handler[SIGERROR] = boost::bind(SigErrorHandler, _1, _2);
  86 }
  87
  88 void SignalController::inject(int sig, const std::string& msg)
  89 {
  90   {
  91     Lock lock(sigQueueMutex);
  92     sigQueue.push(SigPair(sig, msg));
  93   }
  94   sigQueueSem.up();
  95 }
  96
  97 int SignalController::run()
  98 {
  99   for(CallbackMap::iterator it = callbacks.begin(); it != callbacks.end(); ++it)
 100     if(it->first == CALLB_RUNNING) {
 101       it->second();
 102     }
 103
 104   int ret = 0;
 105   for(;;) {
 106     sigQueueSem.down();
 107     SigPair sig;
 108     {
 109       Lock lock(sigQueueMutex);
 110       sig = sigQueue.front();
 111       sigQueue.pop();
 112     }
 113
 114     HandlerMap::iterator it = handler.find(sig.first);
 115     if(it != handler.end()) {
 116       ret = it->second(sig.first, sig.second);
 117
 118       if(ret) {
 119         break;
 120       }
 121     } else {
 122       it = handler.find(SIGUNKNOWN);
 123       if(it != handler.end()) {
 124         it->second(sig.first, sig.second);
 125       } else {
 126         cLog.msg(Log::PRIO_NOTICE) << "SIG " << sig.first << " caught with message '" << sig.second << "' - ignoring";
 127       }
 128     }
 129   }
 130
 131   for(CallbackMap::iterator it = callbacks.begin(); it != callbacks.end(); ++it)
 132     if(it->first == CALLB_STOPPING) {
 133       it->second();
 134     }
 135
 136   return ret;
 137 }
 138