* tunneling and relaying of packets of any protocol.
*
*
- * Copyright (C) 2007-2009 Othmar Gsenger, Erwin Nindl,
+ * Copyright (C) 2007-2009 Othmar Gsenger, Erwin Nindl,
* Christian Pointner <satp@wirdorange.org>
*
* This file is part of Anytun.
#include <map>
#include <queue>
+#include <boost/function.hpp>
#include "threadUtils.hpp"
-#ifndef _MSC_VER
-#include <csignal>
+#ifdef WIN_SERVICE
+//#include "win32/winService.h"
+class WinService;
+typedef WinService DaemonService;
+#else
+class DaemonService;
#endif
#define SIGERROR -1
+#define SIGUNKNOWN -2
-class SignalHandler
-{
-public:
- virtual ~SignalHandler() {}
-
- virtual int handle() { return 0; }
- virtual int handle(const std::string& msg) { return 0; }
-
-protected:
- SignalHandler(int s) : sigNum(s) {}
-
-private:
- int sigNum;
- friend class SignalController;
-};
-
-class SigErrorHandler : public SignalHandler
-{
-public:
- SigErrorHandler() : SignalHandler(SIGERROR) {}
- int handle(const std::string& msg);
-};
-
-#ifndef _MSC_VER
-class SigIntHandler : public SignalHandler
-{
-public:
- SigIntHandler() : SignalHandler(SIGINT) {}
- int handle();
-};
-
-class SigQuitHandler : public SignalHandler
-{
-public:
- SigQuitHandler() : SignalHandler(SIGQUIT) {}
- int handle();
-};
-
-class SigHupHandler : public SignalHandler
-{
-public:
- SigHupHandler() : SignalHandler(SIGHUP) {}
- int handle();
-};
-
-class SigUsr1Handler : public SignalHandler
-{
-public:
- SigUsr1Handler() : SignalHandler(SIGUSR1) {}
- int handle();
-};
-
-class SigUsr2Handler : public SignalHandler
-{
-public:
- SigUsr2Handler() : SignalHandler(SIGUSR2) {}
- int handle();
-};
-
-class SigTermHandler : public SignalHandler
-{
-public:
- SigTermHandler() : SignalHandler(SIGTERM) {}
- int handle();
-};
-
-#else
-
-class CtrlCHandler : public SignalHandler
-{
-public:
- CtrlCHandler() : SignalHandler(CTRL_C_EVENT) {}
- int handle();
-};
-
-class CtrlBreakHandler : public SignalHandler
-{
-public:
- CtrlBreakHandler() : SignalHandler(CTRL_BREAK_EVENT) {}
- int handle();
-};
-
-class CtrlCloseHandler : public SignalHandler
-{
-public:
- CtrlCloseHandler() : SignalHandler(CTRL_BREAK_EVENT) {}
- int handle();
-};
-
-class CtrlLogoffHandler : public SignalHandler
-{
-public:
- CtrlLogoffHandler() : SignalHandler(CTRL_BREAK_EVENT) {}
- int handle();
-};
-
-class CtrlShutdownHandler : public SignalHandler
-{
-public:
- CtrlShutdownHandler() : SignalHandler(CTRL_BREAK_EVENT) {}
- int handle();
-};
-#endif
+typedef boost::function<int (int, std::string const&)> SignalHandler;
+typedef enum { CALLB_RUNNING, CALLB_STOPPING } CallbackType;
+typedef boost::function<void ()> ServiceCallback;
class SignalController
{
public:
static SignalController& instance();
-#ifndef _MSC_VER
- void handle();
-#else
- static bool handle(DWORD ctrlType);
-#endif
- void init();
+ void init(DaemonService& service);
int run();
void inject(int sig, const std::string& msg = "");
private:
- typedef std::map<int, SignalHandler*> HandlerMap;
-
SignalController() {};
- ~SignalController();
- SignalController(const SignalController &s);
- void operator=(const SignalController &s);
+ ~SignalController() {};
+ SignalController(const SignalController& s);
+ void operator=(const SignalController& s);
static SignalController* inst;
static Mutex instMutex;
- class instanceCleaner {
- public: ~instanceCleaner() {
- if(SignalController::inst != NULL)
+ class instanceCleaner
+ {
+ public:
+ ~instanceCleaner() {
+ if(SignalController::inst != NULL) {
delete SignalController::inst;
+ }
}
};
friend class instanceCleaner;
Mutex sigQueueMutex;
Semaphore sigQueueSem;
+ typedef std::map<int, SignalHandler> HandlerMap;
HandlerMap handler;
+ typedef std::map<CallbackType, ServiceCallback> CallbackMap;
+ CallbackMap callbacks;
+
+ friend void registerSignalHandler(SignalController& ctrl, DaemonService& service);
};
extern SignalController& gSignalController;