mirror of https://github.com/PurpleI2P/i2pd.git
I2P: End-to-End encrypted and anonymous Internet
https://i2pd.website/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
116 lines
2.7 KiB
116 lines
2.7 KiB
9 years ago
|
#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;
|
||
|
|
}
|
||
|
|
|
||
|
9 years ago
|
if (isDaemon)
|
||
|
9 years ago
|
{
|
||
|
|
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
|