mirror of
https://github.com/PurpleI2P/i2pd.git
synced 2024-11-17 21:26:04 +00:00
84 lines
1.8 KiB
C++
84 lines
1.8 KiB
C++
#include "Daemon.h"
|
|
#include "util.h"
|
|
#include "Log.h"
|
|
|
|
#ifdef _WIN32
|
|
|
|
#include "./Win32/Win32Service.h"
|
|
|
|
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;
|
|
if (I2PService::isService())
|
|
isDaemon = 1;
|
|
else
|
|
isDaemon = 0;
|
|
|
|
std::string serviceControl = i2p::util::config::GetArg("-service", "none");
|
|
if (serviceControl == "install")
|
|
{
|
|
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
|
|
);
|
|
exit(0);
|
|
}
|
|
else if (serviceControl == "remove")
|
|
{
|
|
UninstallService(SERVICE_NAME);
|
|
exit(0);
|
|
}
|
|
else if (serviceControl != "none")
|
|
{
|
|
printf(" --service=install to install the service.\n");
|
|
printf(" --service=remove to remove the service.\n");
|
|
}
|
|
|
|
if (isDaemon == 1)
|
|
{
|
|
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());
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
exit(EXIT_SUCCESS);
|
|
}
|
|
else
|
|
LogPrint(eLogDebug, "Daemon: running as user");
|
|
|
|
return true;
|
|
}
|
|
bool DaemonWin32::start()
|
|
{
|
|
setlocale(LC_CTYPE, "");
|
|
SetConsoleCP(1251);
|
|
SetConsoleOutputCP(1251);
|
|
setlocale(LC_ALL, "Russian");
|
|
|
|
return Daemon_Singleton::start();
|
|
}
|
|
|
|
bool DaemonWin32::stop()
|
|
{
|
|
return Daemon_Singleton::stop();
|
|
}
|
|
}
|
|
}
|
|
|
|
#endif
|