mirror of https://github.com/PurpleI2P/i2pd.git
orignal
9 years ago
committed by
GitHub
2 changed files with 120 additions and 114 deletions
@ -1,113 +1,115 @@
@@ -1,113 +1,115 @@
|
||||
#include <thread> |
||||
#include "Config.h" |
||||
#include "Daemon.h" |
||||
#include "util.h" |
||||
#include "Log.h" |
||||
|
||||
#ifdef _WIN32 |
||||
|
||||
#include "Win32/Win32Service.h" |
||||
#ifdef WIN32_APP |
||||
#include "Win32/Win32App.h" |
||||
#endif |
||||
|
||||
namespace i2p |
||||
{ |
||||
namespace util |
||||
{ |
||||
bool DaemonWin32::init(int argc, char* argv[]) |
||||
{ |
||||
setlocale(LC_CTYPE, ""); |
||||
SetConsoleCP(1251); |
||||
SetConsoleOutputCP(1251); |
||||
|
||||
if (!Daemon_Singleton::init(argc, argv)) |
||||
return false; |
||||
|
||||
std::string serviceControl; i2p::config::GetOption("svcctl", serviceControl); |
||||
if (serviceControl == "install") |
||||
{ |
||||
LogPrint(eLogInfo, "WinSVC: installing ", SERVICE_NAME, " as service"); |
||||
InstallService( |
||||
SERVICE_NAME, // Name of service
|
||||
SERVICE_DISPLAY_NAME, // Name to display
|
||||
SERVICE_START_TYPE, // Service start type
|
||||
SERVICE_DEPENDENCIES, // Dependencies
|
||||
SERVICE_ACCOUNT, // Service running account
|
||||
SERVICE_PASSWORD // Password of the account
|
||||
); |
||||
return false; |
||||
} |
||||
else if (serviceControl == "remove") |
||||
{ |
||||
LogPrint(eLogInfo, "WinSVC: uninstalling ", SERVICE_NAME, " service"); |
||||
UninstallService(SERVICE_NAME); |
||||
return false; |
||||
} |
||||
|
||||
#include <thread> |
||||
#include <clocale> |
||||
#include "Config.h" |
||||
#include "Daemon.h" |
||||
#include "util.h" |
||||
#include "Log.h" |
||||
|
||||
#ifdef _WIN32 |
||||
|
||||
#include "Win32/Win32Service.h" |
||||
#ifdef WIN32_APP |
||||
#include "Win32/Win32App.h" |
||||
#endif |
||||
|
||||
namespace i2p |
||||
{ |
||||
namespace util |
||||
{ |
||||
bool DaemonWin32::init(int argc, char* argv[]) |
||||
{ |
||||
setlocale(LC_CTYPE, ""); |
||||
SetConsoleCP(1251); |
||||
SetConsoleOutputCP(1251); |
||||
setlocale(LC_ALL, "Russian"); |
||||
|
||||
if (!Daemon_Singleton::init(argc, argv)) |
||||
return false; |
||||
|
||||
std::string serviceControl; i2p::config::GetOption("svcctl", serviceControl); |
||||
if (serviceControl == "install") |
||||
{ |
||||
LogPrint(eLogInfo, "WinSVC: installing ", SERVICE_NAME, " as service"); |
||||
InstallService( |
||||
SERVICE_NAME, // Name of service
|
||||
SERVICE_DISPLAY_NAME, // Name to display
|
||||
SERVICE_START_TYPE, // Service start type
|
||||
SERVICE_DEPENDENCIES, // Dependencies
|
||||
SERVICE_ACCOUNT, // Service running account
|
||||
SERVICE_PASSWORD // Password of the account
|
||||
); |
||||
return false; |
||||
} |
||||
else if (serviceControl == "remove") |
||||
{ |
||||
LogPrint(eLogInfo, "WinSVC: uninstalling ", SERVICE_NAME, " service"); |
||||
UninstallService(SERVICE_NAME); |
||||
return false; |
||||
} |
||||
|
||||
if (isDaemon) |
||||
{ |
||||
LogPrint(eLogDebug, "Daemon: running as service"); |
||||
I2PService service(SERVICE_NAME); |
||||
if (!I2PService::Run(service)) |
||||
{ |
||||
LogPrint(eLogError, "Daemon: Service failed to run w/err 0x%08lx\n", GetLastError()); |
||||
return false; |
||||
} |
||||
return false; |
||||
} |
||||
else |
||||
LogPrint(eLogDebug, "Daemon: running as user"); |
||||
|
||||
return true; |
||||
} |
||||
|
||||
bool DaemonWin32::start() |
||||
{ |
||||
setlocale(LC_CTYPE, ""); |
||||
SetConsoleCP(1251); |
||||
SetConsoleOutputCP(1251); |
||||
setlocale(LC_ALL, "Russian"); |
||||
#ifdef WIN32_APP |
||||
if (!i2p::win32::StartWin32App ()) return false; |
||||
|
||||
// override log
|
||||
i2p::config::SetOption("log", std::string ("file")); |
||||
#endif |
||||
bool ret = Daemon_Singleton::start(); |
||||
if (ret && i2p::log::Logger().GetLogType() == eLogFile) |
||||
{ |
||||
// TODO: find out where this garbage to console comes from
|
||||
SetStdHandle(STD_OUTPUT_HANDLE, INVALID_HANDLE_VALUE); |
||||
SetStdHandle(STD_ERROR_HANDLE, INVALID_HANDLE_VALUE); |
||||
} |
||||
bool insomnia; i2p::config::GetOption("insomnia", insomnia); |
||||
if (insomnia) |
||||
SetThreadExecutionState(ES_CONTINUOUS | ES_SYSTEM_REQUIRED); |
||||
return ret; |
||||
} |
||||
|
||||
bool DaemonWin32::stop() |
||||
{ |
||||
#ifdef WIN32_APP |
||||
i2p::win32::StopWin32App (); |
||||
#endif |
||||
return Daemon_Singleton::stop(); |
||||
} |
||||
|
||||
void DaemonWin32::run () |
||||
{ |
||||
#ifdef WIN32_APP |
||||
i2p::win32::RunWin32App (); |
||||
#else |
||||
while (running) |
||||
{ |
||||
std::this_thread::sleep_for (std::chrono::seconds(1)); |
||||
} |
||||
|
||||
#endif |
||||
} |
||||
} |
||||
} |
||||
|
||||
#endif |
||||
{ |
||||
LogPrint(eLogDebug, "Daemon: running as service"); |
||||
I2PService service(SERVICE_NAME); |
||||
if (!I2PService::Run(service)) |
||||
{ |
||||
LogPrint(eLogError, "Daemon: Service failed to run w/err 0x%08lx\n", GetLastError()); |
||||
return false; |
||||
} |
||||
return false; |
||||
} |
||||
else |
||||
LogPrint(eLogDebug, "Daemon: running as user"); |
||||
|
||||
return true; |
||||
} |
||||
|
||||
bool DaemonWin32::start() |
||||
{ |
||||
setlocale(LC_CTYPE, ""); |
||||
SetConsoleCP(1251); |
||||
SetConsoleOutputCP(1251); |
||||
setlocale(LC_ALL, "Russian"); |
||||
#ifdef WIN32_APP |
||||
if (!i2p::win32::StartWin32App ()) return false; |
||||
|
||||
// override log
|
||||
i2p::config::SetOption("log", std::string ("file")); |
||||
#endif |
||||
bool ret = Daemon_Singleton::start(); |
||||
if (ret && i2p::log::Logger().GetLogType() == eLogFile) |
||||
{ |
||||
// TODO: find out where this garbage to console comes from
|
||||
SetStdHandle(STD_OUTPUT_HANDLE, INVALID_HANDLE_VALUE); |
||||
SetStdHandle(STD_ERROR_HANDLE, INVALID_HANDLE_VALUE); |
||||
} |
||||
bool insomnia; i2p::config::GetOption("insomnia", insomnia); |
||||
if (insomnia) |
||||
SetThreadExecutionState(ES_CONTINUOUS | ES_SYSTEM_REQUIRED); |
||||
return ret; |
||||
} |
||||
|
||||
bool DaemonWin32::stop() |
||||
{ |
||||
#ifdef WIN32_APP |
||||
i2p::win32::StopWin32App (); |
||||
#endif |
||||
return Daemon_Singleton::stop(); |
||||
} |
||||
|
||||
void DaemonWin32::run () |
||||
{ |
||||
#ifdef WIN32_APP |
||||
i2p::win32::RunWin32App (); |
||||
#else |
||||
while (running) |
||||
{ |
||||
std::this_thread::sleep_for (std::chrono::seconds(1)); |
||||
} |
||||
|
||||
#endif |
||||
} |
||||
} |
||||
} |
||||
|
||||
#endif |
||||
|
Loading…
Reference in new issue