daemon code

This commit is contained in:
Olaf Rempel 2009-02-24 14:55:09 +01:00
parent 49550a0afa
commit 8f0d6c1f54
2 changed files with 47 additions and 1 deletions

View File

@ -21,6 +21,7 @@
#include <unistd.h> #include <unistd.h>
#include <getopt.h> #include <getopt.h>
#include <pwd.h>
#include "configfile.h" #include "configfile.h"
#include "connection.h" #include "connection.h"
@ -32,9 +33,12 @@
#include "tcpsocket.h" #include "tcpsocket.h"
#define DEFAULT_CONFIG "torrent-stats.conf" #define DEFAULT_CONFIG "torrent-stats.conf"
#define DEFAULT_LOGFILE "torrent-stats.log"
static struct option opts[] = { static struct option opts[] = {
{"config", 1, 0, 'c'}, {"config", 1, 0, 'c'},
{"user", 1, 0, 'u'},
{"debug", 0, 0, 'd'},
{"help", 0, 0, 'h'}, {"help", 0, 0, 'h'},
{0, 0, 0, 0} {0, 0, 0, 0}
}; };
@ -62,20 +66,32 @@ static int listen_cb(const char *parameter, void *privdata)
int main(int argc, char *argv[]) int main(int argc, char *argv[])
{ {
char *config = DEFAULT_CONFIG; char *config = DEFAULT_CONFIG;
char *user = NULL;
int debug = 0;
int code, arg = 0; int code, arg = 0;
do { do {
code = getopt_long(argc, argv, "c:h", opts, &arg); code = getopt_long(argc, argv, "c:u:dh", opts, &arg);
switch (code) { switch (code) {
case 'c': /* config */ case 'c': /* config */
config = optarg; config = optarg;
break; break;
case 'u': /* user */
user = optarg;
break;
case 'd': /* debug */
debug = 1;
break;
case 'h': /* help */ case 'h': /* help */
printf("Usage: torrent-stats [options]\n" printf("Usage: torrent-stats [options]\n"
"Options: \n" "Options: \n"
" --config -c configfile use this configfile\n" " --config -c configfile use this configfile\n"
" --debug -d do not fork and log to stderr\n"
" --user -u username change uid to username\n"
" --help -h this help\n" " --help -h this help\n"
"\n"); "\n");
exit(0); exit(0);
@ -90,9 +106,37 @@ int main(int argc, char *argv[])
} }
} while (code != -1); } while (code != -1);
/* userwechsel */
if (user) {
struct passwd *pwl;
if (!(pwl = getpwnam(user))) {
log_print(LOG_ERROR, "unknown user: %s", user);
exit(-1);
}
if (setgid(pwl->pw_gid) || setuid(pwl->pw_uid)) {
log_print(LOG_ERROR, "setgid/setuid");
exit(-1);
}
}
/* parse config file */
if (config_parse(config) < 0) if (config_parse(config) < 0)
exit(1); exit(1);
/* check logfile */
const char *logfile = config_get_string("global", "logfile", DEFAULT_LOGFILE);
if (!debug) {
/* start logging */
if (log_init(logfile) < 0)
exit(-1);
/* zum daemon mutieren */
daemon(-1, 0);
}
log_print(LOG_INFO, "torrent-stats started (user: %s, pid: %d)", getpwuid(getuid())->pw_name, getpid());
config_get_strings("global", "listen", listen_cb, ctcs_accept_handler); config_get_strings("global", "listen", listen_cb, ctcs_accept_handler);
config_get_strings("global", "listen-http", listen_cb, httpd_accept_handler); config_get_strings("global", "listen-http", listen_cb, httpd_accept_handler);

View File

@ -4,3 +4,5 @@ listen-http 0.0.0.0:8080
status-interval 10 status-interval 10
seed-timeout 300 seed-timeout 300
logfile torrent-stats.log