i2pd/DaemonWin32.cpp

#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