src/win32/winService.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 #ifdef WIN_SERVICE
  47
  48 #include <iostream>
  49
  50 #include <windows.h>
  51
  52 #include "winService.h"
  53 #include "../log.h"
  54 #include "../anytunError.h"
  55 #include "../threadUtils.hpp"
  56
  57 void WinService::install()
  58 {
  59   SC_HANDLE schSCManager;
  60   SC_HANDLE schService;
  61   char szPath[MAX_PATH];
  62
  63   if(!GetModuleFileNameA(NULL, szPath, MAX_PATH)) {
  64     AnytunError::throwErr() << "Error on GetModuleFileName: " << AnytunErrno(GetLastError());
  65   }
  66
  67   schSCManager = OpenSCManagerA(NULL, NULL, SC_MANAGER_ALL_ACCESS);
  68   if(NULL == schSCManager) {
  69     AnytunError::throwErr() << "Error on OpenSCManager: " << AnytunErrno(GetLastError());
  70   }
  71
  72   schService = CreateServiceA(schSCManager, SVC_NAME, SVC_NAME, SERVICE_ALL_ACCESS, SERVICE_WIN32_OWN_PROCESS,
  73                               SERVICE_DEMAND_START, SERVICE_ERROR_NORMAL, szPath, NULL, NULL, NULL, NULL, NULL);
  74   if(schService == NULL) {
  75     CloseServiceHandle(schSCManager);
  76     AnytunError::throwErr() << "Error on CreateService: " << AnytunErrno(GetLastError());
  77   }
  78
  79   std::cout << "Service installed successfully" << std::endl;
  80
  81   CloseServiceHandle(schService);
  82   CloseServiceHandle(schSCManager);
  83 }
  84
  85 void WinService::uninstall()
  86 {
  87   SC_HANDLE schSCManager;
  88   SC_HANDLE schService;
  89
  90   schSCManager = OpenSCManagerA(NULL, NULL, SC_MANAGER_ALL_ACCESS);
  91   if(NULL == schSCManager) {
  92     AnytunError::throwErr() << "Error on OpenSCManager: " << AnytunErrno(GetLastError());
  93   }
  94
  95   schService = OpenServiceA(schSCManager, SVC_NAME, SERVICE_ALL_ACCESS);
  96   if(schService == NULL) {
  97     CloseServiceHandle(schSCManager);
  98     AnytunError::throwErr() << "Error on CreateService: " << AnytunErrno(GetLastError());
  99   }
 100
 101   if(!DeleteService(schService)) {
 102     CloseServiceHandle(schService);
 103     CloseServiceHandle(schSCManager);
 104     AnytunError::throwErr() << "Error on DeleteService: " << AnytunErrno(GetLastError());
 105   }
 106
 107   std::cout << "Service uninstalled successfully" << std::endl;
 108
 109   CloseServiceHandle(schService);
 110   CloseServiceHandle(schSCManager);
 111 }
 112
 113 void WinService::start()
 114 {
 115   SERVICE_TABLE_ENTRY DispatchTable[] = {
 116     {SVC_NAME, (LPSERVICE_MAIN_FUNCTION)WinService::main },
 117     {NULL, NULL}
 118   };
 119
 120   if(!StartServiceCtrlDispatcherA(DispatchTable)) {
 121     AnytunError::throwErr() << "Error on StartServiceCtrlDispatcher: " << AnytunErrno(GetLastError());
 122   }
 123 }
 124
 125 int real_main(int argc, char* argv[], WinService& service);
 126
 127 VOID WINAPI WinService::main(DWORD dwArgc, LPTSTR* lpszArgv)
 128 {
 129   WinService service;
 130
 131   service.status_handle_ = RegisterServiceCtrlHandlerA(SVC_NAME, WinService::ctrlHandler);
 132   if(!service.status_handle_) {
 133     cLog.msg(Log::PRIO_ERROR) << "Error on RegisterServiceCtrlHandler: " << AnytunErrno(GetLastError());
 134     return;
 135   }
 136   service.status_.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
 137   service.status_.dwServiceSpecificExitCode = 0;
 138   service.reportStatus(SERVICE_START_PENDING, NO_ERROR);
 139
 140   real_main(dwArgc, lpszArgv, service);
 141
 142   service.reportStatus(SERVICE_STOPPED, NO_ERROR);
 143 }
 144
 145 VOID WINAPI WinService::ctrlHandler(DWORD dwCtrl)
 146 {
 147   gSignalController.inject(dwCtrl);
 148 }
 149
 150 int WinService::handleCtrlSignal(int sig, const std::string& msg)
 151 {
 152   switch(sig) {
 153   case SERVICE_CONTROL_STOP: {
 154     reportStatus(SERVICE_STOP_PENDING, NO_ERROR);
 155     cLog.msg(Log::PRIO_NOTICE) << "received service stop signal, exitting";
 156     return 1;
 157   }
 158   case SERVICE_CONTROL_INTERROGATE:
 159     break;
 160   default:
 161     break;
 162   }
 163   reportStatus(status_.dwCurrentState, NO_ERROR);
 164
 165   return 0;
 166 }
 167
 168 void WinService::reportStatus(DWORD dwCurrentState, DWORD dwWin32ExitCode)
 169 {
 170   static DWORD dwCheckPoint = 1;
 171
 172   status_.dwCurrentState = dwCurrentState;
 173   status_.dwWin32ExitCode = dwWin32ExitCode;
 174   status_.dwWaitHint = 0;
 175
 176   if((dwCurrentState == SERVICE_START_PENDING) ||
 177       (dwCurrentState == SERVICE_STOP_PENDING)) {
 178     status_.dwControlsAccepted = 0;
 179   } else {
 180     status_.dwControlsAccepted = SERVICE_ACCEPT_STOP;
 181   }
 182
 183   if((dwCurrentState == SERVICE_RUNNING) ||
 184       (dwCurrentState == SERVICE_STOPPED)) {
 185     status_.dwCheckPoint = 0;
 186   } else {
 187     status_.dwCheckPoint = dwCheckPoint++;
 188   }
 189
 190   SetServiceStatus(status_handle_, &status_);
 191 }
 192
 193 void WinService::initPrivs(std::string const& username, std::string const& groupname)
 194 {
 195   // nothing here
 196 }
 197
 198 void WinService::dropPrivs()
 199 {
 200   // nothing here
 201 }
 202
 203 void WinService::chroot(std::string const& dir)
 204 {
 205   // nothing here
 206 }
 207
 208 void WinService::daemonize()
 209 {
 210   // nothing here
 211 }
 212
 213 bool WinService::isDaemonized()
 214 {
 215   return true;
 216 }
 217
 218 #endif