|
|
|
#ifndef WIN_32_SERVICE_H__
|
|
|
|
#define WIN_32_SERVICE_H__
|
|
|
|
|
|
|
|
#include <thread>
|
|
|
|
#define WIN32_LEAN_AND_MEAN
|
|
|
|
#include <windows.h>
|
|
|
|
|
|
|
|
|
|
|
|
#ifdef _WIN32
|
|
|
|
// Internal name of the service
|
|
|
|
#define SERVICE_NAME "i2pService"
|
|
|
|
|
|
|
|
// Displayed name of the service
|
|
|
|
#define SERVICE_DISPLAY_NAME "i2p 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(PSTR pszServiceName,
|
|
|
|
PSTR pszDisplayName,
|
|
|
|
DWORD dwStartType,
|
|
|
|
PSTR pszDependencies,
|
|
|
|
PSTR pszAccount,
|
|
|
|
PSTR pszPassword);
|
|
|
|
|
|
|
|
void UninstallService(PSTR pszServiceName);
|
|
|
|
|
|
|
|
#endif // WIN_32_SERVICE_H__
|