libtransport.git Changeset - a78794c2e7e1

Changeset - a78794c2e7e1

Parent rev.

Child rev.

[Not reviewed]

4 2 2

Vitaly Takmazov - 13 years ago 2012-11-09 23:59:46
vitalyster@gmail.com

Windows Service Support

8 files changed with 190 insertions and 338 deletions:

spectrum/src/CMakeLists.txt

spectrum/src/main.cpp

spectrum/src/win32/ServiceWrapper.cpp

120

spectrum/src/win32/ServiceWrapper.h

spectrum/src/win32/SpectrumService.cpp

spectrum/src/win32/SpectrumService.h

spectrum/src/win32/WindowsService.cpp

214

spectrum/src/win32/WindowsService.h

0 comments (0 inline, 0 general)

spectrum/src/CMakeLists.txt

➞

Show inline comments

 cmake_minimum_required(VERSION 2.6)
 FILE(GLOB SRC *.cpp)
 # if (WIN32)
 # FILE(GLOB WIN_SRC win32/*.cpp)
 # include_directories(win32)
 # ADD_EXECUTABLE(spectrum2 ${SRC} ${WIN_SRC})
 # else()
 if (WIN32)
 FILE(GLOB WIN_SRC win32/*.cpp)
 include_directories(win32)
 ADD_EXECUTABLE(spectrum2 ${SRC} ${WIN_SRC})
 else()
 ADD_EXECUTABLE(spectrum2 ${SRC})
-# endif()
 endif()
 ADD_DEPENDENCIES(spectrum2 spectrum2_libpurple_backend)
 ADD_DEPENDENCIES(spectrum2 spectrum2_libircclient-qt_backend)

spectrum/src/main.cpp

➞

Show inline comments

@@ @@ -30,24 +30,31 @@ @@
 #ifndef __FreeBSD__
 #include <malloc.h>
 #endif
 #else
 #include <process.h>
 #define getpid _getpid
-// #include "win32/SpectrumService.h"
+#include "win32/ServiceWrapper.h"
 #endif
 #include <sys/stat.h>
 using namespace Transport;
 using namespace Transport::Util;
 DEFINE_LOGGER(logger, "Spectrum");
 Swift::SimpleEventLoop *eventLoop_ = NULL;
 Component *component_ = NULL;
 UserManager *userManager_ = NULL;
 Config *config_ = NULL;
 void stop() {
 	userManager_->removeAllUsers(false);
 	component_->stop();
 	eventLoop_->stop();
+}
 static void stop_spectrum() {
 	userManager_->removeAllUsers(false);
 	component_->stop();
 	eventLoop_->stop();
+}
 #endif
 int main(int argc, char **argv)
+{
 	Config config(argc, argv);
+	config_ = &config;
 	boost::program_options::variables_map vm;
 	bool no_daemon = false;
 	std::string config_file;
 	std::string jid;
 	setlocale(LC_ALL, "");
@@ @@ -157,15 +164,15 @@ int main(int argc, char **argv) @@
 		("no-debug,d", "Create coredumps on crash")
 		("jid,j", boost::program_options::value<std::string>(&jid)->default_value(""), "Specify JID of transport manually")
 		("config", boost::program_options::value<std::string>(&config_file)->default_value(""), "Config file")
 		("version,v", "Shows Spectrum version")
+		;
 #ifdef WIN32
 // 	desc.add_options()
 // 		("install-service,i", "Install spectrum as Windows service")
 // 		("uninstall-service,u", "Uninstall Windows service");
  	desc.add_options()
  		("install-service,i", "Install spectrum as Windows service")
  		("uninstall-service,u", "Uninstall Windows service");
 #endif
 	try
+	{
 		boost::program_options::positional_options_description p;
 		p.add("config", -1);
 		boost::program_options::store(boost::program_options::command_line_parser(argc, argv).
@@ @@ -189,51 +196,49 @@ int main(int argc, char **argv) @@
+		}
 		if(vm.count("no-daemonize")) {
 			no_daemon = true;
+		}
 #ifdef WIN32
 #if 0
 		if (vm.count("install-service")) {
-			SpectrumService ntservice;
+			ServiceWrapper ntservice("Spectrum2");
 			if (!ntservice.IsInstalled()) {
 					// determine the name of the currently executing file
 				char szFilePath[MAX_PATH];
-				GetModuleFileName(NULL, szFilePath, sizeof(szFilePath));
+				GetModuleFileNameA(NULL, szFilePath, sizeof(szFilePath));
 				std::string exe_file(szFilePath);
 				std::string config_file = exe_file.replace(exe_file.end() - 4, exe_file.end(), ".cfg");
 				std::string service_path = std::string(szFilePath) + std::string(" --config ") + config_file;
 				if (ntservice.Install(service_path.c_str())) {
+				if (ntservice.Install((char *)service_path.c_str())) {
 					std::cout << "Successfully installed" << std::endl;
 					return 0;
 				} else {
 					std::cout << "Error installing service, are you an Administrator?" << std::endl;
 					return 1;
+				}
 			} else {
 				std::cout << "Already installed" << std::endl;
 				return 1;
+			}
+		}
 		if (vm.count("uninstall-service")) {
-			SpectrumService ntservice;
+			ServiceWrapper ntservice("Spectrum2");
 			if (ntservice.IsInstalled()) {
-				if (ntservice.Remove()) {
+				if (ntservice.UnInstall()) {
 					std::cout << "Successfully removed" << std::endl;
 					return 0;
 				} else {
 					std::cout << "Error removing service, are you an Administrator?" << std::endl;
 					return 1;
+				}
 			} else {
 				std::cout << "Service not installed" << std::endl;
 				return 1;
+			}
+		}
 #endif
 #endif
+	}
 	catch (std::runtime_error& e)
+	{
 		std::cout << desc << "\n";
 		return 1;
@@ @@ -307,42 +312,55 @@ int main(int argc, char **argv) @@
 	if (!no_daemon) {
 		// daemonize
 		daemonize(CONFIG_STRING(&config, "service.working_dir").c_str(), CONFIG_STRING(&config, "service.pidfile").c_str());
 // 		removeOldIcons(CONFIG_STRING(&config, "service.working_dir") + "/icons");
+    }
 #endif
 #ifdef WIN32
 	ServiceWrapper ntservice("Spectrum2");
 	if (ntservice.IsInstalled()) {
 		ntservice.RunService();
 	} else {
 		mainloop();
+	}
 #else
 	mainloop();
 #endif
+}
 int mainloop() {
-	Logging::initMainLogging(&config);
+	Logging::initMainLogging(config_);
 #ifndef WIN32
-	if (!CONFIG_STRING(&config, "service.group").empty() ||!CONFIG_STRING(&config, "service.user").empty() ) {
+	if (!CONFIG_STRING(config_, "service.group").empty() ||!CONFIG_STRING(config_, "service.user").empty() ) {
 		struct rlimit limit;
 		getrlimit(RLIMIT_CORE, &limit);
-		if (!CONFIG_STRING(&config, "service.group").empty()) {
+		if (!CONFIG_STRING(config_, "service.group").empty()) {
 			struct group *gr;
 			if ((gr = getgrnam(CONFIG_STRING(&config, "service.group").c_str())) == NULL) {
 				std::cerr << "Invalid service.group name " << CONFIG_STRING(&config, "service.group") << "\n";
 			if ((gr = getgrnam(CONFIG_STRING(config_, "service.group").c_str())) == NULL) {
 				std::cerr << "Invalid service.group name " << CONFIG_STRING(config_, "service.group") << "\n";
 				return 1;
+			}
 			if (((setgid(gr->gr_gid)) != 0) || (initgroups(CONFIG_STRING(&config, "service.user").c_str(), gr->gr_gid) != 0)) {
 				std::cerr << "Failed to set service.group name " << CONFIG_STRING(&config, "service.group") << " - " << gr->gr_gid << ":" << strerror(errno) << "\n";
 			if (((setgid(gr->gr_gid)) != 0) || (initgroups(CONFIG_STRING(config_, "service.user").c_str(), gr->gr_gid) != 0)) {
 				std::cerr << "Failed to set service.group name " << CONFIG_STRING(config_, "service.group") << " - " << gr->gr_gid << ":" << strerror(errno) << "\n";
 				return 1;
+			}
+		}
-		if (!CONFIG_STRING(&config, "service.user").empty()) {
+		if (!CONFIG_STRING(config_, "service.user").empty()) {
 			struct passwd *pw;
 			if ((pw = getpwnam(CONFIG_STRING(&config, "service.user").c_str())) == NULL) {
 				std::cerr << "Invalid service.user name " << CONFIG_STRING(&config, "service.user") << "\n";
 			if ((pw = getpwnam(CONFIG_STRING(config_, "service.user").c_str())) == NULL) {
 				std::cerr << "Invalid service.user name " << CONFIG_STRING(config_, "service.user") << "\n";
 				return 1;
+			}
 			if ((setuid(pw->pw_uid)) != 0) {
-				std::cerr << "Failed to set service.user name " << CONFIG_STRING(&config, "service.user") << " - " << pw->pw_uid << ":" << strerror(errno) << "\n";
+				std::cerr << "Failed to set service.user name " << CONFIG_STRING(config_, "service.user") << " - " << pw->pw_uid << ":" << strerror(errno) << "\n";
 				return 1;
+			}
+		}
 		setrlimit(RLIMIT_CORE, &limit);
+	}
@@ @@ -352,20 +370,20 @@ int main(int argc, char **argv) @@
 	setrlimit(RLIMIT_CORE, &limit);
 #endif
 	Swift::SimpleEventLoop eventLoop;
 	Swift::BoostNetworkFactories *factories = new Swift::BoostNetworkFactories(&eventLoop);
-	UserRegistry userRegistry(&config, factories);
+	UserRegistry userRegistry(config_, factories);
-	Component transport(&eventLoop, factories, &config, NULL, &userRegistry);
+	Component transport(&eventLoop, factories, config_, NULL, &userRegistry);
 	component_ = &transport;
 // 	Logger logger(&transport);
 	std::string error;
-	StorageBackend *storageBackend = StorageBackend::createBackend(&config, error);
+	StorageBackend *storageBackend = StorageBackend::createBackend(config_, error);
 	if (storageBackend == NULL) {
 		if (!error.empty()) {
 			std::cerr << error << "\n";
 			return -2;
+		}
+	}
@@ @@ -390,13 +408,13 @@ int main(int argc, char **argv) @@
 		usersReconnecter = new UsersReconnecter(&transport, storageBackend);
+	}
 	FileTransferManager ftManager(&transport, &userManager);
-	NetworkPluginServer plugin(&transport, &config, &userManager, &ftManager, &discoItemsResponder);
+	NetworkPluginServer plugin(&transport, config_, &userManager, &ftManager, &discoItemsResponder);
 	plugin.start();
 	AdminInterface adminInterface(&transport, &userManager, &plugin, storageBackend, userRegistration);
 	plugin.setAdminInterface(&adminInterface);
 	StatsResponder statsResponder(&transport, &userManager, &plugin, storageBackend);

spectrum/src/win32/ServiceWrapper.cpp

➞

Show inline comments

@@ new file 100644 @@
 #include "ServiceWrapper.h"
 LPSTR ServiceName;
 SERVICE_STATUS ServiceStatus;
 SERVICE_STATUS_HANDLE ServiceStatusHandle;
 bool doAction (int action, int desiredAccess, LPSTR path = NULL);
 enum actions {
 	DO_INSTALL, DO_DELETE, DO_CHECK
 };
 ServiceWrapper::ServiceWrapper(LPSTR serviceName)
+{
 	ServiceName = serviceName;
 	ServiceStatusHandle = 0;
+}
 ServiceWrapper::~ServiceWrapper(void)
+{
+}
 bool ServiceWrapper::Install(LPSTR commandLine) {
 	return doAction(DO_INSTALL, SC_MANAGER_ALL_ACCESS, commandLine);
+}
 bool ServiceWrapper::UnInstall() {
 	return doAction(DO_DELETE, SC_MANAGER_ALL_ACCESS);
+}
 bool ServiceWrapper::IsInstalled() {
 	return doAction(DO_CHECK, SC_MANAGER_CONNECT);
+}
 bool doAction(int action, int desiredAccess, LPSTR path) {
 	SC_HANDLE scm = OpenSCManager(NULL, NULL, desiredAccess);
 	SC_HANDLE service = NULL;
 	if (!scm) return FALSE;
 	switch(action) {
 	case DO_INSTALL:
 		service = CreateServiceA(
 			scm,
 			ServiceName,
 			ServiceName,
 			SERVICE_ALL_ACCESS,
 			SERVICE_WIN32_OWN_PROCESS,
 			SERVICE_DEMAND_START,
 			SERVICE_ERROR_NORMAL,
 			path,
 			NULL,
 			NULL,
 			NULL,
 			NULL,
 			NULL
 			);
 		return (service != NULL);
 		break;
 	case DO_DELETE:
 		service = OpenServiceA(scm, ServiceName, DELETE);
 		if (service == NULL)
 			return FALSE;
 		if (DeleteService(service))
 			return TRUE;
 		break;
 	case DO_CHECK:
 		service = OpenServiceA(scm, ServiceName, SERVICE_QUERY_STATUS);
 		return (service != NULL);
 	default:
 		return FALSE;
+	}
 	CloseServiceHandle(service);
 	CloseServiceHandle(scm);
 	return FALSE;
+}
 void WINAPI ServiceControlHandler(DWORD controlCode) {
 	switch (controlCode) {
 	case SERVICE_CONTROL_INTERROGATE:
 		break;
 	case SERVICE_CONTROL_STOP:
 		ServiceStatus.dwCurrentState = SERVICE_STOP_PENDING;
 		break;
 	default:
 		break;
+	}
 	SetServiceStatus(ServiceStatusHandle, &ServiceStatus);
 	stop();
+}
 void WINAPI ServiceMain(DWORD argc, LPSTR *argv) {
 	ServiceStatus.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
 	ServiceStatus.dwServiceSpecificExitCode = NO_ERROR;
 	ServiceStatusHandle = RegisterServiceCtrlHandlerA(ServiceName, ServiceControlHandler);
 	if (ServiceStatusHandle) {
 		ServiceStatus.dwCurrentState = SERVICE_START_PENDING;
 		SetServiceStatus(ServiceStatusHandle, &ServiceStatus);
 		ServiceStatus.dwControlsAccepted |= (SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN);
 		ServiceStatus.dwCurrentState = SERVICE_RUNNING;
 		SetServiceStatus(ServiceStatusHandle, &ServiceStatus);
 		mainloop();
 		ServiceStatus.dwCurrentState = SERVICE_STOP_PENDING;
 		SetServiceStatus(ServiceStatusHandle, &ServiceStatus);
 		ServiceStatus.dwControlsAccepted &= ~(SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN);
 		ServiceStatus.dwCurrentState = SERVICE_STOPPED;
 		SetServiceStatus(ServiceStatusHandle, &ServiceStatus);
+	}
+}
 void ServiceWrapper::RunService() {
 	SERVICE_TABLE_ENTRYA serviceTable[] = {
 		{ ServiceName, ServiceMain },
 		{ NULL, NULL}
 	};
 	StartServiceCtrlDispatcherA(serviceTable);
+}
@@ \ No newline at end of file @@

spectrum/src/win32/ServiceWrapper.h

➞

Show inline comments

@@ new file 100644 @@
 #pragma once
 #include "transport/config.h"
 #include <windows.h>
 #include <tchar.h>
 class ServiceWrapper
+{
 public:
 	ServiceWrapper(LPSTR serviceName);
 	~ServiceWrapper(void);
 	bool Install(LPSTR commandLine);
 	bool UnInstall();
 	bool IsInstalled();
 	void RunService();
 };
 int mainloop();
 void stop();

spectrum/src/win32/SpectrumService.cpp

➞

Show inline comments

deleted file

spectrum/src/win32/SpectrumService.h

➞

Show inline comments

deleted file

spectrum/src/win32/WindowsService.cpp

➞

Show inline comments

deleted file

spectrum/src/win32/WindowsService.h

➞

Show inline comments

deleted file

0 comments (0 inline, 0 general)