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

service.user, service.group

Browse source
This commit is contained in:
HanzZ 2011-09-09 10:20:58 +02:00
parent 36056deeb9
commit c9e9772506
3 changed files with 42 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

37
spectrum/src/main.cpp
View file

@ -11,6 +11,8 @@
#include <boost/filesystem.hpp>
#ifndef WIN32
#include "sys/signal.h"
#include <pwd.h>
#include <grp.h>
#else
#include <Windows.h>
#include <tchar.h>
@ -25,6 +27,8 @@ using namespace log4cxx;
using namespace Transport;
static LoggerPtr logger = log4cxx::Logger::getLogger("Spectrum");
Swift::SimpleEventLoop *eventLoop_ = NULL;
static void spectrum_sigint_handler(int sig) {
@ -163,6 +167,7 @@ int main(int argc, char **argv)
#ifndef WIN32
if (!no_daemon) {
// create directories
try {
boost::filesystem::create_directories(CONFIG_STRING(&config, "service.working_dir"));
}
@ -179,8 +184,10 @@ int main(int argc, char **argv)
std::cerr << "Can't create service.pidfile directory " << boost::filesystem::path(CONFIG_STRING(&config, "service.pidfile")).parent_path().string() << ".\n";
return 1;
}
// daemonize
daemonize(CONFIG_STRING(&config, "service.working_dir").c_str(), CONFIG_STRING(&config, "service.pidfile").c_str());
}
}
#endif
if (CONFIG_STRING(&config, "logging.config").empty()) {
@ -195,6 +202,34 @@ int main(int argc, char **argv)
log4cxx::PropertyConfigurator::configure(CONFIG_STRING(&config, "logging.config"));
}
#ifndef WIN32
if (!CONFIG_STRING(&config, "service.group").empty()) {
struct group *gr;
if ((gr = getgrnam(CONFIG_STRING(&config, "service.group").c_str())) == NULL) {
LOG4CXX_ERROR(logger, "Invalid service.group name " << CONFIG_STRING(&config, "service.group"));
return 1;
}
if (((setgid(gr->gr_gid)) != 0) || (initgroups(CONFIG_STRING(&config, "service.user").c_str(), gr->gr_gid) != 0)) {
LOG4CXX_ERROR(logger, "Failed to set service.group name " << CONFIG_STRING(&config, "service.group") << " - " << gr->gr_gid << ":" << strerror(errno));
return 1;
}
}
if (!CONFIG_STRING(&config, "service.user").empty()) {
struct passwd *pw;
if ((pw = getpwnam(CONFIG_STRING(&config, "service.user").c_str())) == NULL) {
LOG4CXX_ERROR(logger, "Invalid service.user name " << CONFIG_STRING(&config, "service.user"));
return 1;
}
if ((setuid(pw->pw_uid)) != 0) {
LOG4CXX_ERROR(logger, "Failed to set service.user name " << CONFIG_STRING(&config, "service.user") << " - " << pw->pw_uid << ":" << strerror(errno));
return 1;
}
}
#endif
Swift::SimpleEventLoop eventLoop;
Swift::BoostNetworkFactories *factories = new Swift::BoostNetworkFactories(&eventLoop);

4
spectrum/src/sample2.cfg
View file

@ -2,6 +2,10 @@
# 1 if Spectrum should run in server mode.
server_mode = 1
# The name of user/group Spectrum runs as.
#user=spectrum
#group=spectrum
# JID of Spectrum instance.
jid = localhost

2
src/config.cpp
View file

@ -50,6 +50,8 @@ bool Config::load(std::istream &ifs, boost::program_options::options_description
("service.server", value<std::string>()->default_value(""), "Server to connect to")
("service.password", value<std::string>()->default_value(""), "Password used to auth the server")
("service.port", value<int>()->default_value(0), "Port the server is listening on")
("service.user", value<std::string>()->default_value(""), "The name of user Spectrum runs as.")
("service.group", value<std::string>()->default_value(""), "The name of group Spectrum runs as.")
("service.backend", value<std::string>()->default_value("libpurple_backend"), "Backend")
("service.protocol", value<std::string>()->default_value(""), "Protocol")
("service.pidfile", value<std::string>()->default_value("/var/run/spectrum2/spectrum2.pid"), "Full path to pid file")