diff options
Diffstat (limited to 'src/win32/winService.cpp')
| -rw-r--r-- | src/win32/winService.cpp | 174 |
1 files changed, 174 insertions, 0 deletions
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 <satp@wirdorange.org> + * + * 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 <http://www.gnu.org/licenses/>. + */ + +#include <iostream> + +#include <windows.h> + +#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_);
+} |