Imported Upstream version 0.3.5
[anytun.git] / 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