From b75cb8243b4c0f073351d0ee1c82bbbb6963db3d Mon Sep 17 00:00:00 2001 From: Christian Pointner Date: Fri, 20 Feb 2009 04:40:38 +0000 Subject: added experimental win32 service code --- src/win32/winService.cpp | 174 +++++++++++++++++++++++++++++++++++++++++++++++ src/win32/winService.h | 76 +++++++++++++++++++++ 2 files changed, 250 insertions(+) create mode 100644 src/win32/winService.cpp create mode 100644 src/win32/winService.h (limited to 'src/win32') diff --git a/src/win32/winService.cpp b/src/win32/winService.cpp new file mode 100644 index 0000000..5ebade1 --- /dev/null +++ b/src/win32/winService.cpp @@ -0,0 +1,174 @@ +/* + * anytun + * + * The secure anycast tunneling protocol (satp) defines a protocol used + * for communication between any combination of unicast and anycast + * tunnel endpoints. It has less protocol overhead than IPSec in Tunnel + * mode and allows tunneling of every ETHER TYPE protocol (e.g. + * ethernet, ip, arp ...). satp directly includes cryptography and + * message authentication based on the methodes used by SRTP. It is + * intended to deliver a generic, scaleable and secure solution for + * tunneling and relaying of packets of any protocol. + * + * + * Copyright (C) 2007-2008 Othmar Gsenger, Erwin Nindl, + * Christian Pointner + * + * This file is part of Anytun. + * + * Anytun is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License version 3 as + * published by the Free Software Foundation. + * + * Anytun is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with anytun. If not, see . + */ + +#include + +#include + +#include "winService.h" +#include "../log.h" +#include "../threadUtils.hpp" + +WinService* WinService::inst = NULL; +Mutex WinService::instMutex; +WinService& gWinService = WinService::instance(); + +WinService& WinService::instance() +{ + Lock lock(instMutex); + static instanceCleaner c; + if(!inst) + inst = new WinService(); + + return *inst; +} + +WinService::~WinService() +{ + if(started_) + CloseHandle(exit_event_); +} + +void WinService::install() +{ + SC_HANDLE schSCManager; + SC_HANDLE schService; + char szPath[MAX_PATH]; + + if(!GetModuleFileNameA(NULL, szPath, MAX_PATH)) { + std::stringstream msg; + msg << "Error on GetModuleFileName: " << LogErrno(GetLastError()); + throw std::runtime_error(msg.str()); + } + + schSCManager = OpenSCManagerA(NULL, NULL, SC_MANAGER_ALL_ACCESS); + if(NULL == schSCManager) { + std::stringstream msg; + msg << "Error on OpenSCManager: " << LogErrno(GetLastError()); + throw std::runtime_error(msg.str()); + } + + schService = CreateServiceA(schSCManager, name_.c_str(), name_.c_str(), SERVICE_ALL_ACCESS, SERVICE_WIN32_OWN_PROCESS, + SERVICE_DEMAND_START, SERVICE_ERROR_NORMAL, szPath, NULL, NULL, NULL, NULL, NULL); // no password + if(schService == NULL) { + CloseServiceHandle(schSCManager); + std::stringstream msg; + msg << "Error on CreateService: " << LogErrno(GetLastError()); + throw std::runtime_error(msg.str()); + } + + std::cout << "Service installed successfully" << std::endl; + + CloseServiceHandle(schService); + CloseServiceHandle(schSCManager); +} + +void WinService::start() +{ + if(started_) + throw std::runtime_error("Service already started"); + + SERVICE_TABLE_ENTRY DispatchTable[] = { + {(LPSTR)name_.c_str(), (LPSERVICE_MAIN_FUNCTION)WinService::main }, + {NULL, NULL} + }; + + if(!StartServiceCtrlDispatcherA(DispatchTable)) { + std::stringstream msg; + msg << "Error on StartServiceCtrlDispatcher: " << LogErrno(GetLastError()); + throw std::runtime_error(msg.str()); + } +} + +void WinService::waitForExit() +{ + if(started_) + throw std::runtime_error("Service not started correctly"); + + WaitForSingleObject(exit_event_, INFINITE); + reportStatus(SERVICE_STOP_PENDING, NO_ERROR); +} + +void WinService::stop() +{ + if(started_) + throw std::runtime_error("Service not started correctly"); + + reportStatus(SERVICE_STOPPED, NO_ERROR); +} + +VOID WINAPI WinService::main(DWORD dwArgc, LPTSTR *lpszArgv) +{ + gWinService.status_handle_ = RegisterServiceCtrlHandlerA(gWinService.name_.c_str(), WinService::ctrlHandler); + if(!gWinService.status_handle_) { + cLog.msg(Log::PRIO_ERR) << "Error on RegisterServiceCtrlHandler: " << LogErrno(GetLastError()); + return; + } + gWinService.status_.dwServiceType = SERVICE_WIN32_OWN_PROCESS; + gWinService.status_.dwServiceSpecificExitCode = 0; + gWinService.reportStatus(SERVICE_START_PENDING, NO_ERROR); + + gWinService.exit_event_ = CreateEvent(NULL, true, false, NULL); + if(!gWinService.exit_event_) { + cLog.msg(Log::PRIO_ERR) << "Error on CreateEvent: " << LogErrno(GetLastError()); + gWinService.reportStatus(SERVICE_STOPPED, -1); + return; + } + gWinService.started_ = true; + gWinService.reportStatus(SERVICE_RUNNING, NO_ERROR); +} + +VOID WINAPI WinService::ctrlHandler(DWORD dwCtrl) +{ +} + +void WinService::reportStatus(DWORD dwCurrentState, DWORD dwWin32ExitCode, DWORD dwWaitHint) +{ + static DWORD dwCheckPoint = 1; + + status_.dwCurrentState = dwCurrentState; + status_.dwWin32ExitCode = dwWin32ExitCode; + status_.dwWaitHint = dwWaitHint; + + if((dwCurrentState == SERVICE_START_PENDING) || + (dwCurrentState == SERVICE_STOP_PENDING)) + status_.dwControlsAccepted = 0; + else + status_.dwControlsAccepted = SERVICE_ACCEPT_STOP; + + if((dwCurrentState == SERVICE_RUNNING) || + (dwCurrentState == SERVICE_STOPPED)) + status_.dwCheckPoint = 0; + else + status_.dwCheckPoint = dwCheckPoint++; + + SetServiceStatus(status_handle_, &status_); +} diff --git a/src/win32/winService.h b/src/win32/winService.h new file mode 100644 index 0000000..dc648d8 --- /dev/null +++ b/src/win32/winService.h @@ -0,0 +1,76 @@ +/* + * anytun + * + * The secure anycast tunneling protocol (satp) defines a protocol used + * for communication between any combination of unicast and anycast + * tunnel endpoints. It has less protocol overhead than IPSec in Tunnel + * mode and allows tunneling of every ETHER TYPE protocol (e.g. + * ethernet, ip, arp ...). satp directly includes cryptography and + * message authentication based on the methodes used by SRTP. It is + * intended to deliver a generic, scaleable and secure solution for + * tunneling and relaying of packets of any protocol. + * + * + * Copyright (C) 2007-2008 Othmar Gsenger, Erwin Nindl, + * Christian Pointner + * + * This file is part of Anytun. + * + * Anytun is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License version 3 as + * published by the Free Software Foundation. + * + * Anytun is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with anytun. If not, see . + */ + +#ifndef _WIN_SERVICE_H_ +#define _WIN_SERVICE_H_ + +#include "../threadUtils.hpp" + +class WinService +{ +public: + static WinService& instance(); + + void install(); + void start(); + void waitForExit(); + void stop(); + + static VOID WINAPI main(DWORD dwArgc, LPTSTR *lpszArgv); + static VOID WINAPI ctrlHandler(DWORD dwCtrl); +private: + WinService() : name_("anytun"), started_(false) {}; + ~WinService(); + WinService(const WinService &w); + void operator=(const WinService &w); + + void reportStatus(DWORD dwCurrentState, DWORD dwWin32ExitCode, DWORD dwWaitHint=0); + + static WinService* inst; + static Mutex instMutex; + class instanceCleaner { + public: ~instanceCleaner() { + if(WinService::inst != NULL) + delete WinService::inst; + } + }; + friend class instanceCleaner; + + std::string name_; + bool started_; + SERVICE_STATUS status_; + SERVICE_STATUS_HANDLE status_handle_; + HANDLE exit_event_; +}; + +extern WinService& gWinService; + +#endif \ No newline at end of file -- cgit v1.2.3