random ai stuff
This commit is contained in:
matst80
166
telldus-core/service/main_unix.cpp
Normal file
166
telldus-core/service/main_unix.cpp
Normal file
@@ -0,0 +1,166 @@
|
||||
//
|
||||
// Copyright (C) 2012 Telldus Technologies AB. All rights reserved.
|
||||
//
|
||||
// Copyright: See COPYING file that comes with this distribution
|
||||
//
|
||||
//
|
||||
#include <signal.h>
|
||||
#include <string.h>
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <unistd.h>
|
||||
#include <sys/stat.h>
|
||||
#include <errno.h>
|
||||
#include <pwd.h>
|
||||
#include <grp.h>
|
||||
#include <string>
|
||||
|
||||
#include "service/TelldusMain.h"
|
||||
#include "service/Settings.h"
|
||||
#include "common/Strings.h"
|
||||
#include "service/Log.h"
|
||||
|
||||
#define DAEMON_NAME "telldusd"
|
||||
#define PID_FILE "/var/run/" DAEMON_NAME ".pid"
|
||||
|
||||
TelldusMain tm;
|
||||
|
||||
void signalHandler(int sig) {
|
||||
switch(sig) {
|
||||
case SIGHUP:
|
||||
Log::warning("Received SIGHUP signal.");
|
||||
break;
|
||||
case SIGTERM:
|
||||
case SIGINT:
|
||||
Log::warning("Received SIGTERM or SIGINT signal.");
|
||||
Log::warning("Shutting down");
|
||||
tm.stop();
|
||||
break;
|
||||
case SIGPIPE:
|
||||
break;
|
||||
default:
|
||||
Log::warning("Unhandled signal (%d) %s", sig, strsignal(sig));
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
int main(int argc, char **argv) {
|
||||
pid_t pid, sid;
|
||||
FILE *fd;
|
||||
bool deamonize = true;
|
||||
|
||||
for (int i = 1; i < argc; ++i) {
|
||||
if (strcmp(argv[i], "--nodaemon") == 0) {
|
||||
deamonize = false;
|
||||
Log::setLogOutput(Log::StdOut);
|
||||
} else if (strcmp(argv[i], "--debug") == 0) {
|
||||
Log::setDebug();
|
||||
} else if (strcmp(argv[i], "--help") == 0) {
|
||||
printf("Telldus TellStick background service\n\nStart with --nodaemon to not run as daemon\n\n");
|
||||
printf("Report bugs to <info.tech@telldus.com>\n");
|
||||
exit(EXIT_SUCCESS);
|
||||
} else if (strcmp(argv[i], "--version") == 0) {
|
||||
printf("telldusd " VERSION "\n\n");
|
||||
printf("Copyright (C) 2011 Telldus Technologies AB\n\n");
|
||||
printf("Written by Micke Prag <micke.prag@telldus.se>\n");
|
||||
exit(EXIT_SUCCESS);
|
||||
} else {
|
||||
printf("Unknown option %s\n", argv[i]);
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
}
|
||||
|
||||
if (deamonize) {
|
||||
pid = fork();
|
||||
if (pid < 0) {
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
if (pid > 0) {
|
||||
// We are the parent
|
||||
// Let the parent store the clients pid,
|
||||
// This way anyone starting the daemon can read the pidfile immediately
|
||||
|
||||
// Record the pid
|
||||
fd = fopen(PID_FILE, "w");
|
||||
if (fd) {
|
||||
fprintf(fd, "%d\n", pid);
|
||||
fclose(fd);
|
||||
} else {
|
||||
Log::error("Could not open pid file %s: %s", PID_FILE, strerror(errno));
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
exit(EXIT_SUCCESS);
|
||||
}
|
||||
}
|
||||
|
||||
Log::notice("%s daemon starting up", DAEMON_NAME);
|
||||
|
||||
if (deamonize) {
|
||||
/* Change the file mode mask */
|
||||
umask(0);
|
||||
|
||||
sid = setsid();
|
||||
|
||||
if (sid < 0) {
|
||||
// Something went wrong
|
||||
printf("Could not set sid\n");
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
close(STDIN_FILENO);
|
||||
close(STDOUT_FILENO);
|
||||
close(STDERR_FILENO);
|
||||
}
|
||||
|
||||
/* Reduce our permissions (change user and group) */
|
||||
if (getuid() == 0 || geteuid() == 0) {
|
||||
Settings settings;
|
||||
std::string user = TelldusCore::wideToString(settings.getSetting(L"user"));
|
||||
std::string group = TelldusCore::wideToString(settings.getSetting(L"group"));
|
||||
|
||||
// We use the non threadsafe function getgrnam() here. Since this is startup code
|
||||
// and no other threads have been started yet.
|
||||
struct group *grp = getgrnam(group.c_str()); // NOLINT(runtime/threadsafe_fn)
|
||||
if (grp) {
|
||||
setgid(grp->gr_gid);
|
||||
} else {
|
||||
Log::warning("Group %s could not be found", group.c_str());
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
// We use the non threadsafe function getpwnam() here. Since this is startup code
|
||||
// and no other threads have been started yet.
|
||||
struct passwd *pw = getpwnam(user.c_str()); // NOLINT(runtime/threadsafe_fn)
|
||||
if (pw) {
|
||||
setuid( pw->pw_uid );
|
||||
} else {
|
||||
Log::warning("User %s could not be found", user.c_str());
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
}
|
||||
|
||||
/* Change the current working directory */
|
||||
if ((chdir("/")) < 0) {
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
/* Install signal traps for proper shutdown */
|
||||
signal(SIGTERM, signalHandler);
|
||||
signal(SIGINT, signalHandler);
|
||||
signal(SIGPIPE, signalHandler);
|
||||
|
||||
// Change so we do not need to call wait() for children processes
|
||||
struct sigaction sa;
|
||||
sa.sa_handler = SIG_IGN;
|
||||
sa.sa_flags = SA_NOCLDWAIT;
|
||||
sigemptyset(&sa.sa_mask);
|
||||
if (sigaction(SIGCHLD, &sa, NULL) == -1) {
|
||||
Log::error("Could not set the SA_NOCLDWAIT flag. We will be creating zombie processes!");
|
||||
}
|
||||
|
||||
tm.start();
|
||||
|
||||
Log::notice("%s daemon exited", DAEMON_NAME);
|
||||
// Cleanup
|
||||
Log::destroy();
|
||||
exit(EXIT_SUCCESS);
|
||||
}
|
||||
Reference in New Issue
Block a user