mirror of https://github.com/PurpleI2P/i2pd.git
orignal
4 years ago
committed by
GitHub
39 changed files with 617 additions and 777 deletions
@ -1,414 +0,0 @@
@@ -1,414 +0,0 @@
|
||||
/*
|
||||
* Copyright (c) 2013-2020, The PurpleI2P Project |
||||
* |
||||
* This file is part of Purple i2pd project and licensed under BSD3 |
||||
* |
||||
* See full license text in LICENSE file at top of project tree |
||||
*/ |
||||
|
||||
#ifdef _WIN32 |
||||
#define _CRT_SECURE_NO_WARNINGS // to use freopen
|
||||
#endif |
||||
|
||||
#include "Win32Service.h" |
||||
#include <assert.h> |
||||
//#include <strsafe.h>
|
||||
#include <windows.h> |
||||
|
||||
#include "Daemon.h" |
||||
#include "Log.h" |
||||
|
||||
I2PService *I2PService::s_service = NULL; |
||||
|
||||
BOOL I2PService::isService() |
||||
{ |
||||
BOOL bIsService = FALSE; |
||||
HWINSTA hWinStation = GetProcessWindowStation(); |
||||
if (hWinStation != NULL) |
||||
{ |
||||
USEROBJECTFLAGS uof = { 0 }; |
||||
if (GetUserObjectInformation(hWinStation, UOI_FLAGS, &uof, sizeof(USEROBJECTFLAGS), NULL) && ((uof.dwFlags & WSF_VISIBLE) == 0)) |
||||
{ |
||||
bIsService = TRUE; |
||||
} |
||||
} |
||||
return bIsService; |
||||
} |
||||
|
||||
BOOL I2PService::Run(I2PService &service) |
||||
{ |
||||
s_service = &service; |
||||
SERVICE_TABLE_ENTRY serviceTable[] = |
||||
{ |
||||
{ service.m_name, ServiceMain }, |
||||
{ NULL, NULL } |
||||
}; |
||||
return StartServiceCtrlDispatcher(serviceTable); |
||||
} |
||||
|
||||
void WINAPI I2PService::ServiceMain(DWORD dwArgc, PSTR *pszArgv) |
||||
{ |
||||
assert(s_service != NULL); |
||||
s_service->m_statusHandle = RegisterServiceCtrlHandler( |
||||
s_service->m_name, ServiceCtrlHandler); |
||||
if (s_service->m_statusHandle == NULL) |
||||
{ |
||||
throw GetLastError(); |
||||
} |
||||
s_service->Start(dwArgc, pszArgv); |
||||
} |
||||
|
||||
|
||||
void WINAPI I2PService::ServiceCtrlHandler(DWORD dwCtrl) |
||||
{ |
||||
switch (dwCtrl) |
||||
{ |
||||
case SERVICE_CONTROL_STOP: s_service->Stop(); break; |
||||
case SERVICE_CONTROL_PAUSE: s_service->Pause(); break; |
||||
case SERVICE_CONTROL_CONTINUE: s_service->Continue(); break; |
||||
case SERVICE_CONTROL_SHUTDOWN: s_service->Shutdown(); break; |
||||
case SERVICE_CONTROL_INTERROGATE: break; |
||||
default: break; |
||||
} |
||||
} |
||||
|
||||
I2PService::I2PService(PSTR pszServiceName, |
||||
BOOL fCanStop, |
||||
BOOL fCanShutdown, |
||||
BOOL fCanPauseContinue) |
||||
{ |
||||
m_name = (pszServiceName == NULL) ? (PSTR)"" : pszServiceName; |
||||
m_statusHandle = NULL; |
||||
m_status.dwServiceType = SERVICE_WIN32_OWN_PROCESS; |
||||
m_status.dwCurrentState = SERVICE_START_PENDING; |
||||
|
||||
DWORD dwControlsAccepted = 0; |
||||
if (fCanStop) |
||||
dwControlsAccepted |= SERVICE_ACCEPT_STOP; |
||||
if (fCanShutdown) |
||||
dwControlsAccepted |= SERVICE_ACCEPT_SHUTDOWN; |
||||
if (fCanPauseContinue) |
||||
dwControlsAccepted |= SERVICE_ACCEPT_PAUSE_CONTINUE; |
||||
|
||||
m_status.dwControlsAccepted = dwControlsAccepted; |
||||
m_status.dwWin32ExitCode = NO_ERROR; |
||||
m_status.dwServiceSpecificExitCode = 0; |
||||
m_status.dwCheckPoint = 0; |
||||
m_status.dwWaitHint = 0; |
||||
m_fStopping = FALSE; |
||||
// Create a manual-reset event that is not signaled at first to indicate
|
||||
// the stopped signal of the service.
|
||||
m_hStoppedEvent = CreateEvent(NULL, TRUE, FALSE, NULL); |
||||
if (m_hStoppedEvent == NULL) |
||||
{ |
||||
throw GetLastError(); |
||||
} |
||||
} |
||||
|
||||
I2PService::~I2PService(void) |
||||
{ |
||||
if (m_hStoppedEvent) |
||||
{ |
||||
CloseHandle(m_hStoppedEvent); |
||||
m_hStoppedEvent = NULL; |
||||
} |
||||
} |
||||
|
||||
void I2PService::Start(DWORD dwArgc, PSTR *pszArgv) |
||||
{ |
||||
try |
||||
{ |
||||
SetServiceStatus(SERVICE_START_PENDING); |
||||
OnStart(dwArgc, pszArgv); |
||||
SetServiceStatus(SERVICE_RUNNING); |
||||
} |
||||
catch (DWORD dwError) |
||||
{ |
||||
LogPrint(eLogError, "Win32Service Start", dwError); |
||||
SetServiceStatus(SERVICE_STOPPED, dwError); |
||||
} |
||||
catch (...) |
||||
{ |
||||
LogPrint(eLogError, "Win32Service failed to start.", EVENTLOG_ERROR_TYPE); |
||||
SetServiceStatus(SERVICE_STOPPED); |
||||
} |
||||
} |
||||
|
||||
void I2PService::OnStart(DWORD dwArgc, PSTR *pszArgv) |
||||
{ |
||||
LogPrint(eLogInfo, "Win32Service in OnStart", EVENTLOG_INFORMATION_TYPE); |
||||
Daemon.start(); |
||||
//i2p::util::config::OptionParser(dwArgc, pszArgv);
|
||||
//i2p::util::filesystem::ReadConfigFile(i2p::util::config::mapArgs, i2p::util::config::mapMultiArgs);
|
||||
//i2p::context.OverrideNTCPAddress(i2p::util::config::GetCharArg("-host", "127.0.0.1"),
|
||||
// i2p::util::config::GetArg("-port", 17070));
|
||||
_worker = new std::thread(std::bind(&I2PService::WorkerThread, this)); |
||||
} |
||||
|
||||
void I2PService::WorkerThread() |
||||
{ |
||||
while (!m_fStopping) |
||||
{ |
||||
::Sleep(1000); // Simulate some lengthy operations.
|
||||
} |
||||
// Signal the stopped event.
|
||||
SetEvent(m_hStoppedEvent); |
||||
} |
||||
|
||||
void I2PService::Stop() |
||||
{ |
||||
DWORD dwOriginalState = m_status.dwCurrentState; |
||||
try |
||||
{ |
||||
SetServiceStatus(SERVICE_STOP_PENDING); |
||||
OnStop(); |
||||
SetServiceStatus(SERVICE_STOPPED); |
||||
} |
||||
catch (DWORD dwError) |
||||
{ |
||||
LogPrint(eLogInfo, "Win32Service Stop", dwError); |
||||
SetServiceStatus(dwOriginalState); |
||||
} |
||||
catch (...) |
||||
{ |
||||
LogPrint(eLogError, "Win32Service failed to stop.", EVENTLOG_ERROR_TYPE); |
||||
SetServiceStatus(dwOriginalState); |
||||
} |
||||
} |
||||
|
||||
void I2PService::OnStop() |
||||
{ |
||||
// Log a service stop message to the Application log.
|
||||
LogPrint(eLogInfo, "Win32Service in OnStop", EVENTLOG_INFORMATION_TYPE); |
||||
Daemon.stop(); |
||||
m_fStopping = TRUE; |
||||
if (WaitForSingleObject(m_hStoppedEvent, INFINITE) != WAIT_OBJECT_0) |
||||
{ |
||||
throw GetLastError(); |
||||
} |
||||
_worker->join(); |
||||
delete _worker; |
||||
} |
||||
|
||||
void I2PService::Pause() |
||||
{ |
||||
try |
||||
{ |
||||
SetServiceStatus(SERVICE_PAUSE_PENDING); |
||||
OnPause(); |
||||
SetServiceStatus(SERVICE_PAUSED); |
||||
} |
||||
catch (DWORD dwError) |
||||
{ |
||||
LogPrint(eLogError, "Win32Service Pause", dwError); |
||||
SetServiceStatus(SERVICE_RUNNING); |
||||
} |
||||
catch (...) |
||||
{ |
||||
LogPrint(eLogError, "Win32Service failed to pause.", EVENTLOG_ERROR_TYPE); |
||||
SetServiceStatus(SERVICE_RUNNING); |
||||
} |
||||
} |
||||
|
||||
void I2PService::OnPause() |
||||
{ |
||||
} |
||||
|
||||
void I2PService::Continue() |
||||
{ |
||||
try |
||||
{ |
||||
SetServiceStatus(SERVICE_CONTINUE_PENDING); |
||||
OnContinue(); |
||||
SetServiceStatus(SERVICE_RUNNING); |
||||
} |
||||
catch (DWORD dwError) |
||||
{ |
||||
LogPrint(eLogError, "Win32Service Continue", dwError); |
||||
SetServiceStatus(SERVICE_PAUSED); |
||||
} |
||||
catch (...) |
||||
{ |
||||
LogPrint(eLogError, "Win32Service failed to resume.", EVENTLOG_ERROR_TYPE); |
||||
SetServiceStatus(SERVICE_PAUSED); |
||||
} |
||||
} |
||||
|
||||
void I2PService::OnContinue() |
||||
{ |
||||
} |
||||
|
||||
void I2PService::Shutdown() |
||||
{ |
||||
try |
||||
{ |
||||
OnShutdown(); |
||||
SetServiceStatus(SERVICE_STOPPED); |
||||
} |
||||
catch (DWORD dwError) |
||||
{ |
||||
LogPrint(eLogError, "Win32Service Shutdown", dwError); |
||||
} |
||||
catch (...) |
||||
{ |
||||
LogPrint(eLogError, "Win32Service failed to shut down.", EVENTLOG_ERROR_TYPE); |
||||
} |
||||
} |
||||
|
||||
void I2PService::OnShutdown() |
||||
{ |
||||
} |
||||
|
||||
void I2PService::SetServiceStatus(DWORD dwCurrentState, |
||||
DWORD dwWin32ExitCode, |
||||
DWORD dwWaitHint) |
||||
{ |
||||
static DWORD dwCheckPoint = 1; |
||||
m_status.dwCurrentState = dwCurrentState; |
||||
m_status.dwWin32ExitCode = dwWin32ExitCode; |
||||
m_status.dwWaitHint = dwWaitHint; |
||||
m_status.dwCheckPoint = |
||||
((dwCurrentState == SERVICE_RUNNING) || |
||||
(dwCurrentState == SERVICE_STOPPED)) ? |
||||
0 : dwCheckPoint++; |
||||
|
||||
::SetServiceStatus(m_statusHandle, &m_status); |
||||
} |
||||
|
||||
//*****************************************************************************
|
||||
|
||||
void FreeHandles(SC_HANDLE schSCManager, SC_HANDLE schService) |
||||
{ |
||||
if (schSCManager) |
||||
{ |
||||
CloseServiceHandle(schSCManager); |
||||
schSCManager = NULL; |
||||
} |
||||
if (schService) |
||||
{ |
||||
CloseServiceHandle(schService); |
||||
schService = NULL; |
||||
} |
||||
} |
||||
|
||||
void InstallService(PCSTR pszServiceName, PCSTR pszDisplayName, DWORD dwStartType, PCSTR pszDependencies, PCSTR pszAccount, PCSTR pszPassword) |
||||
{ |
||||
printf("Try to install Win32Service (%s).\n", pszServiceName); |
||||
|
||||
char szPath[MAX_PATH]; |
||||
SC_HANDLE schSCManager = NULL; |
||||
SC_HANDLE schService = NULL; |
||||
|
||||
if (GetModuleFileName(NULL, szPath, ARRAYSIZE(szPath)) == 0) |
||||
{ |
||||
printf("GetModuleFileName failed w/err 0x%08lx\n", GetLastError()); |
||||
FreeHandles(schSCManager, schService); |
||||
return; |
||||
} |
||||
char SvcOpt[] = " --daemon"; |
||||
strncat(szPath, SvcOpt, strlen(SvcOpt)); |
||||
|
||||
// Open the local default service control manager database
|
||||
schSCManager = OpenSCManager(NULL, NULL, SC_MANAGER_CONNECT | SC_MANAGER_CREATE_SERVICE); |
||||
if (schSCManager == NULL) |
||||
{ |
||||
printf("OpenSCManager failed w/err 0x%08lx\n", GetLastError()); |
||||
FreeHandles(schSCManager, schService); |
||||
return; |
||||
} |
||||
|
||||
// Install the service into SCM by calling CreateService
|
||||
schService = CreateService( |
||||
schSCManager, // SCManager database
|
||||
pszServiceName, // Name of service
|
||||
pszDisplayName, // Name to display
|
||||
SERVICE_QUERY_STATUS, // Desired access
|
||||
SERVICE_WIN32_OWN_PROCESS, // Service type
|
||||
dwStartType, // Service start type
|
||||
SERVICE_ERROR_NORMAL, // Error control type
|
||||
szPath, // Service's binary
|
||||
NULL, // No load ordering group
|
||||
NULL, // No tag identifier
|
||||
pszDependencies, // Dependencies
|
||||
pszAccount, // Service running account
|
||||
pszPassword // Password of the account
|
||||
); |
||||
|
||||
if (schService == NULL) |
||||
{ |
||||
printf("CreateService failed w/err 0x%08lx\n", GetLastError()); |
||||
FreeHandles(schSCManager, schService); |
||||
return; |
||||
} |
||||
|
||||
printf("Win32Service is installed as %s.\n", pszServiceName); |
||||
|
||||
// Centralized cleanup for all allocated resources.
|
||||
FreeHandles(schSCManager, schService); |
||||
} |
||||
|
||||
void UninstallService(PCSTR pszServiceName) |
||||
{ |
||||
printf("Try to uninstall Win32Service (%s).\n", pszServiceName); |
||||
|
||||
SC_HANDLE schSCManager = NULL; |
||||
SC_HANDLE schService = NULL; |
||||
SERVICE_STATUS ssSvcStatus = {}; |
||||
|
||||
// Open the local default service control manager database
|
||||
schSCManager = OpenSCManager(NULL, NULL, SC_MANAGER_CONNECT); |
||||
if (schSCManager == NULL) |
||||
{ |
||||
printf("OpenSCManager failed w/err 0x%08lx\n", GetLastError()); |
||||
FreeHandles(schSCManager, schService); |
||||
return; |
||||
} |
||||
|
||||
// Open the service with delete, stop, and query status permissions
|
||||
schService = OpenService(schSCManager, pszServiceName, SERVICE_STOP | SERVICE_QUERY_STATUS | DELETE); |
||||
if (schService == NULL) |
||||
{ |
||||
printf("OpenService failed w/err 0x%08lx\n", GetLastError()); |
||||
FreeHandles(schSCManager, schService); |
||||
return; |
||||
} |
||||
|
||||
// Try to stop the service
|
||||
if (ControlService(schService, SERVICE_CONTROL_STOP, &ssSvcStatus)) |
||||
{ |
||||
printf("Stopping %s.\n", pszServiceName); |
||||
Sleep(1000); |
||||
|
||||
while (QueryServiceStatus(schService, &ssSvcStatus)) |
||||
{ |
||||
if (ssSvcStatus.dwCurrentState == SERVICE_STOP_PENDING) |
||||
{ |
||||
printf("."); |
||||
Sleep(1000); |
||||
} |
||||
else break; |
||||
} |
||||
|
||||
if (ssSvcStatus.dwCurrentState == SERVICE_STOPPED) |
||||
{ |
||||
printf("\n%s is stopped.\n", pszServiceName); |
||||
} |
||||
else |
||||
{ |
||||
printf("\n%s failed to stop.\n", pszServiceName); |
||||
} |
||||
} |
||||
|
||||
// Now remove the service by calling DeleteService.
|
||||
if (!DeleteService(schService)) |
||||
{ |
||||
printf("DeleteService failed w/err 0x%08lx\n", GetLastError()); |
||||
FreeHandles(schSCManager, schService); |
||||
return; |
||||
} |
||||
|
||||
printf("%s is removed.\n", pszServiceName); |
||||
|
||||
// Centralized cleanup for all allocated resources.
|
||||
FreeHandles(schSCManager, schService); |
||||
} |
||||
@ -1,92 +0,0 @@
@@ -1,92 +0,0 @@
|
||||
/*
|
||||
* Copyright (c) 2013-2020, The PurpleI2P Project |
||||
* |
||||
* This file is part of Purple i2pd project and licensed under BSD3 |
||||
* |
||||
* See full license text in LICENSE file at top of project tree |
||||
*/ |
||||
|
||||
#ifndef WIN_32_SERVICE_H__ |
||||
#define WIN_32_SERVICE_H__ |
||||
|
||||
#include <thread> |
||||
#include <windows.h> |
||||
|
||||
#ifdef _WIN32 |
||||
// Internal name of the service
|
||||
#define SERVICE_NAME "i2pdService" |
||||
|
||||
// Displayed name of the service
|
||||
#define SERVICE_DISPLAY_NAME "i2pd router service" |
||||
|
||||
// Service start options.
|
||||
#define SERVICE_START_TYPE SERVICE_DEMAND_START |
||||
|
||||
// List of service dependencies - "dep1\0dep2\0\0"
|
||||
#define SERVICE_DEPENDENCIES "" |
||||
|
||||
// The name of the account under which the service should run
|
||||
#define SERVICE_ACCOUNT "NT AUTHORITY\\LocalService" |
||||
|
||||
// The password to the service account name
|
||||
#define SERVICE_PASSWORD NULL |
||||
#endif |
||||
|
||||
class I2PService |
||||
{ |
||||
public: |
||||
|
||||
I2PService(PSTR pszServiceName, |
||||
BOOL fCanStop = TRUE, |
||||
BOOL fCanShutdown = TRUE, |
||||
BOOL fCanPauseContinue = FALSE); |
||||
|
||||
virtual ~I2PService(void); |
||||
|
||||
static BOOL isService(); |
||||
static BOOL Run(I2PService &service); |
||||
void Stop(); |
||||
|
||||
protected: |
||||
|
||||
virtual void OnStart(DWORD dwArgc, PSTR *pszArgv); |
||||
virtual void OnStop(); |
||||
virtual void OnPause(); |
||||
virtual void OnContinue(); |
||||
virtual void OnShutdown(); |
||||
void SetServiceStatus(DWORD dwCurrentState, |
||||
DWORD dwWin32ExitCode = NO_ERROR, |
||||
DWORD dwWaitHint = 0); |
||||
|
||||
private: |
||||
|
||||
static void WINAPI ServiceMain(DWORD dwArgc, LPSTR *lpszArgv); |
||||
static void WINAPI ServiceCtrlHandler(DWORD dwCtrl); |
||||
void WorkerThread(); |
||||
void Start(DWORD dwArgc, PSTR *pszArgv); |
||||
void Pause(); |
||||
void Continue(); |
||||
void Shutdown(); |
||||
static I2PService* s_service; |
||||
PSTR m_name; |
||||
SERVICE_STATUS m_status; |
||||
SERVICE_STATUS_HANDLE m_statusHandle; |
||||
|
||||
BOOL m_fStopping; |
||||
HANDLE m_hStoppedEvent; |
||||
|
||||
std::thread* _worker; |
||||
}; |
||||
|
||||
void InstallService( |
||||
PCSTR pszServiceName, |
||||
PCSTR pszDisplayName, |
||||
DWORD dwStartType, |
||||
PCSTR pszDependencies, |
||||
PCSTR pszAccount, |
||||
PCSTR pszPassword |
||||
); |
||||
|
||||
void UninstallService(PCSTR pszServiceName); |
||||
|
||||
#endif // WIN_32_SERVICE_H__
|
||||
Loading…
Reference in new issue