stv0g/spectrum2
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Merge pull request #16 from vitalyster/win_service

Browse source 
Multiple Windows services support
This commit is contained in:
Jan Kaluza 2012-11-21 02:28:47 -08:00
commit ae953ed2e0
1 changed files with 31 additions and 22 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

53
spectrum/src/main.cpp
View file

@ -266,7 +266,13 @@ int main(int argc, char **argv)
bool no_daemon = false; bool no_daemon = false;
std::string config_file; std::string config_file;
std::string jid; std::string jid;
#ifdef WIN32
std::string install_service_name, uninstall_service_name, run_service_name;
// determine the name of the currently executing file
char szFilePath[MAX_PATH];
GetModuleFileNameA(NULL, szFilePath, sizeof(szFilePath));
std::string exe_file(szFilePath);
#endif
setlocale(LC_ALL, ""); setlocale(LC_ALL, "");
#ifndef WIN32 #ifndef WIN32
#ifndef __FreeBSD__ #ifndef __FreeBSD__
@ -297,8 +303,9 @@ int main(int argc, char **argv)
; ;
#ifdef WIN32 #ifdef WIN32
desc.add_options() desc.add_options()
("install-service,i", "Install spectrum as Windows service") ("install-service,i", boost::program_options::value<std::string>(&install_service_name)->default_value(""), "Install spectrum as Windows service")
("uninstall-service,u", "Uninstall Windows service"); ("uninstall-service,u", boost::program_options::value<std::string>(&uninstall_service_name)->default_value(""), "Uninstall Windows service")
("run-as-service,r", boost::program_options::value<std::string>(&run_service_name)->default_value(""), "stub for Windows Service Manager");
#endif #endif
try try
{ {
@ -327,41 +334,38 @@ int main(int argc, char **argv)
if(vm.count("no-daemonize")) { if(vm.count("no-daemonize")) {
no_daemon = true; no_daemon = true;
} }
#ifdef WIN32 #ifdef WIN32
if (vm.count("install-service")) { if (!install_service_name.empty()) {
ServiceWrapper ntservice("Spectrum2"); // build command line for Service Manager
std::string service_path = exe_file + std::string(" --config ") + vm["config"].as<std::string>()
+ std::string(" --run-as-service ") + install_service_name;
ServiceWrapper ntservice((char *)install_service_name.c_str());
if (!ntservice.IsInstalled()) { if (!ntservice.IsInstalled()) {
// determine the name of the currently executing file
char szFilePath[MAX_PATH];
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((char *)service_path.c_str())) { if (ntservice.Install((char *)service_path.c_str())) {
std::cout << "Successfully installed" << std::endl; std::cout << "Successfully installed " << install_service_name << std::endl;
return 0; return 0;
} else { } else {
std::cout << "Error installing service, are you an Administrator?" << std::endl; std::cout << "Error installing service, are you an Administrator?" << std::endl;
return 1; return 1;
} }
} else { } else {
std::cout << "Already installed" << std::endl; std::cout << "Already installed " << install_service_name << std::endl;
return 1; return 1;
} }
} }
if (vm.count("uninstall-service")) { if (!uninstall_service_name.empty()) {
ServiceWrapper ntservice("Spectrum2"); ServiceWrapper ntservice((char *)uninstall_service_name.c_str());
if (ntservice.IsInstalled()) { if (ntservice.IsInstalled()) {
if (ntservice.UnInstall()) { if (ntservice.UnInstall()) {
std::cout << "Successfully removed" << std::endl; std::cout << "Successfully removed " << uninstall_service_name << std::endl;
return 0; return 0;
} else { } else {
std::cout << "Error removing service, are you an Administrator?" << std::endl; std::cout << "Error removing service, are you an Administrator?" << std::endl;
return 1; return 1;
} }
} else { } else {
std::cout << "Service not installed" << std::endl; std::cout << "Service not installed: " << uninstall_service_name << std::endl;
return 1; return 1;
} }
} }
@ -441,9 +445,14 @@ int main(int argc, char **argv)
} }
#endif #endif
#ifdef WIN32 #ifdef WIN32
ServiceWrapper ntservice("Spectrum2"); if (!run_service_name.empty()) {
if (ntservice.IsInstalled()) { ServiceWrapper ntservice((char *)run_service_name.c_str());
ntservice.RunService(); if (ntservice.IsInstalled()) {
ntservice.RunService();
} else {
std::cerr << "Service not installed: " << run_service_name << std::endl;
return 1;
}
} else { } else {
mainloop(); mainloop();
} }