155 lines
4.3 KiB
C
155 lines
4.3 KiB
C
/***************************************************************************
|
|
* Copyright (C) 07/2007 by Olaf Rempel *
|
|
* razzor@kopf-tisch.de *
|
|
* *
|
|
* This program is free software; you can redistribute it and/or modify *
|
|
* it under the terms of the GNU General Public License as published by *
|
|
* the Free Software Foundation; version 2 of the License *
|
|
* *
|
|
* This program is distributed in the hope that it will be useful, *
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
|
|
* GNU General Public License for more details. *
|
|
* *
|
|
* You should have received a copy of the GNU General Public License *
|
|
* along with this program; if not, write to the *
|
|
* Free Software Foundation, Inc., *
|
|
* 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. *
|
|
***************************************************************************/
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
|
|
#include <getopt.h>
|
|
#include <pwd.h>
|
|
|
|
#include "configfile.h"
|
|
#include "connection.h"
|
|
#include "event.h"
|
|
#include "httpd.h"
|
|
#include "list.h"
|
|
#include "logging.h"
|
|
#include "sockaddr.h"
|
|
#include "tcpsocket.h"
|
|
|
|
#define DEFAULT_CONFIG "torrent-stats.conf"
|
|
#define DEFAULT_LOGFILE "torrent-stats.log"
|
|
|
|
static struct option opts[] = {
|
|
{"config", 1, 0, 'c'},
|
|
{"user", 1, 0, 'u'},
|
|
{"debug", 0, 0, 'd'},
|
|
{"help", 0, 0, 'h'},
|
|
{0, 0, 0, 0}
|
|
};
|
|
|
|
static int listen_cb(const char *parameter, void *privdata)
|
|
{
|
|
struct sockaddr_in addr;
|
|
|
|
if (parse_sockaddr(parameter, &addr) < 0) {
|
|
log_print(LOG_WARN, "listen_cb(): invalid address");
|
|
return -1;
|
|
}
|
|
|
|
int sockfd = tcp_listen(&addr);
|
|
if (sockfd < 0) {
|
|
log_print(LOG_WARN, "listen_cb(): tcp_listen()");
|
|
return -1;
|
|
}
|
|
|
|
log_print(LOG_INFO, "listen on %s", get_sockaddr_buf(&addr));
|
|
event_add_readfd(NULL, sockfd, privdata, NULL);
|
|
return 0;
|
|
}
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
char *config = DEFAULT_CONFIG;
|
|
char *user = NULL;
|
|
int debug = 0;
|
|
int code, arg = 0;
|
|
|
|
do {
|
|
code = getopt_long(argc, argv, "c:u:dh", opts, &arg);
|
|
|
|
switch (code) {
|
|
case 'c': /* config */
|
|
config = optarg;
|
|
break;
|
|
|
|
case 'u': /* user */
|
|
user = optarg;
|
|
break;
|
|
|
|
case 'd': /* debug */
|
|
debug = 1;
|
|
break;
|
|
|
|
case 'h': /* help */
|
|
printf("Usage: torrent-stats [options]\n"
|
|
"Options: \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"
|
|
"\n");
|
|
exit(0);
|
|
break;
|
|
|
|
case '?': /* error */
|
|
exit(1);
|
|
break;
|
|
|
|
default: /* unknown / all options parsed */
|
|
break;
|
|
}
|
|
} 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)
|
|
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-http", listen_cb, httpd_accept_handler);
|
|
|
|
httpd_add_cb("/quit", 0, ctcs_httpd_quit, NULL);
|
|
httpd_add_cb("/", 1, ctcs_httpd_show, NULL);
|
|
|
|
int interval = config_get_int("global", "status-interval", 10);
|
|
int timeout = config_get_int("global", "seed-timeout", 300);
|
|
|
|
struct timeval tv = { .tv_sec = interval, .tv_usec = 0 };
|
|
event_add_timeout(&tv, ctcs_trigger_status, (void *)timeout);
|
|
|
|
event_loop();
|
|
return 0;
|
|
}
|